tests: Add tests to write on chip

This patch adds two tests and initialises page_size in mock chip
chip_W25Q128_V. page_size was not needed for previous tests
(erase and read). page_size only needed to execute writing on chip
with dummyflasher, so it is added here.

BUG=b:181803212
TEST=ninja test

Change-Id: I6f0336613ab16a7e59857006496e3590ddb14d00
Signed-off-by: Anastasia Klimchuk <aklm@chromium.org>
Reviewed-on: https://review.coreboot.org/c/flashrom/+/58357
Tested-by: build bot (Jenkins) <no-reply@coreboot.org>
Reviewed-by: Edward O'Callaghan <quasisec@chromium.org>

1 files changed, 61 insertions, 0 deletions

diff --git a/tests/chip.c b/tests/chip.c
index 54e14d22f..4181aed5c 100644
--- a/tests/chip.c
+++ b/tests/chip.c
@@ -148,6 +148,7 @@ static const struct flashchip chip_W25Q128_V = {
 	.read		= spi_chip_read,
 	.write		= spi_chip_write_256,
 	.unlock         = spi_disable_blockprotect,
+	.page_size	= 256,
 	.block_erasers  =
 	{
 		{
@@ -256,3 +257,63 @@ void read_chip_with_dummyflasher_test_success(void **state)
 
 	free(param_dup);
 }
+
+void write_chip_test_success(void **state)
+{
+	(void) state; /* unused */
+
+	struct flashrom_flashctx flash = { 0 };
+	struct flashrom_layout *layout;
+	struct flashchip mock_chip = chip_8MiB;
+	const char *param = ""; /* Default values for all params. */
+
+	setup_chip(&flash, &layout, &mock_chip, param);
+
+	/*
+	 * Providing filename "-" means content is taken from standard input.
+	 * This doesn't change much because all file operations are mocked.
+	 * However filename "-" makes a difference for
+	 * flashrom.c#read_buf_from_file and allows to avoid mocking
+	 * image_stat.st_size.
+	 *
+	 * Now this does mean test covers successful path only, but this test
+	 * is designed to cover only successful write operation anyway.
+	 *
+	 * To cover error path of image_stat.st_size != flash size, filename
+	 * needs to be provided and image_stat.st_size needs to be mocked.
+	 */
+	const char *const filename = "-";
+
+	printf("Write chip operation started.\n");
+	assert_int_equal(0, do_write(&flash, filename, NULL));
+	printf("Write chip operation done.\n");
+
+	teardown(&layout);
+}
+
+void write_chip_with_dummyflasher_test_success(void **state)
+{
+	(void) state; /* unused */
+
+	struct flashrom_flashctx flash = { 0 };
+	struct flashrom_layout *layout;
+	struct flashchip mock_chip = chip_W25Q128_V;
+	/*
+	 * Dummyflasher is capable to emulate W25Q128.V, so we ask it to do this.
+	 * Nothing to mock, dummy is taking care of this already.
+	 */
+	char *param_dup = strdup("bus=spi,emulate=W25Q128FV");
+
+	setup_chip(&flash, &layout, &mock_chip, param_dup);
+
+	/* See comment in write_chip_test_success */
+	const char *const filename = "-";
+
+	printf("Write chip operation started.\n");
+	assert_int_equal(0, do_write(&flash, filename, NULL));
+	printf("Write chip operation done.\n");
+
+	teardown(&layout);
+
+	free(param_dup);
+}