diff options
-rw-r--r-- | tests/io_mock.c | 3 | ||||
-rw-r--r-- | tests/io_mock.h | 15 | ||||
-rw-r--r-- | tests/lifecycle.c | 14 | ||||
-rw-r--r-- | tests/tests.c | 31 |
4 files changed, 47 insertions, 16 deletions
diff --git a/tests/io_mock.c b/tests/io_mock.c index 51dde9279..9f86fcd67 100644 --- a/tests/io_mock.c +++ b/tests/io_mock.c @@ -14,12 +14,15 @@ * GNU General Public License for more details. */ +#include <include/test.h> #include "io_mock.h" static const struct io_mock *current_io = NULL; void io_mock_register(const struct io_mock *io) { + /* A test can either register its own mock open function or fallback_open_state. */ + assert_true(io == NULL || io->open == NULL || io->fallback_open_state == NULL); current_io = io; } diff --git a/tests/io_mock.h b/tests/io_mock.h index f8f75adcd..a36602a44 100644 --- a/tests/io_mock.h +++ b/tests/io_mock.h @@ -60,6 +60,15 @@ struct pci_dev { /* Always return success for tests. */ #define S_ISREG(x) 0 +/* Maximum number of open calls to mock. This number is arbitrary. */ +#define MAX_MOCK_OPEN 4 + +struct io_mock_fallback_open_state { + unsigned int noc; + const char *paths[MAX_MOCK_OPEN]; + int flags[MAX_MOCK_OPEN]; +}; + struct io_mock { void *state; @@ -105,6 +114,12 @@ struct io_mock { size_t (*fread)(void *state, void *buf, size_t size, size_t len, FILE *fp); int (*fprintf)(void *state, FILE *fp, const char *fmt, va_list args); int (*fclose)(void *state, FILE *fp); + + /* + * An alternative to custom open mock. A test can either register its + * own mock open function or fallback_open_state. + */ + struct io_mock_fallback_open_state *fallback_open_state; }; void io_mock_register(const struct io_mock *io); diff --git a/tests/lifecycle.c b/tests/lifecycle.c index c37f3b99a..5566b3bcf 100644 --- a/tests/lifecycle.c +++ b/tests/lifecycle.c @@ -370,13 +370,6 @@ void linux_spi_probe_lifecycle_test_success(void **state) #endif } -static int realtek_mst_open(void *state, const char *pathname, int flags) -{ - assert_string_equal(pathname, "/dev/i2c-254"); - assert_int_equal(flags & O_RDWR, O_RDWR); - return MOCK_FD; -} - static int realtek_mst_ioctl(void *state, int fd, unsigned long request, va_list args) { assert_int_equal(fd, MOCK_FD); @@ -406,11 +399,16 @@ static int realtek_mst_write(void *state, int fd, const void *buf, size_t sz) void realtek_mst_basic_lifecycle_test_success(void **state) { #if CONFIG_REALTEK_MST_I2C_SPI == 1 + static struct io_mock_fallback_open_state realtek_mst_fallback_open_state = { + .noc = 0, + .paths = { "/dev/i2c-254", NULL }, + .flags = { O_RDWR }, + }; const struct io_mock realtek_mst_io = { - .open = realtek_mst_open, .ioctl = realtek_mst_ioctl, .read = realtek_mst_read, .write = realtek_mst_write, + .fallback_open_state = &realtek_mst_fallback_open_state, }; io_mock_register(&realtek_mst_io); diff --git a/tests/tests.c b/tests/tests.c index f32c14dab..3eef47e4d 100644 --- a/tests/tests.c +++ b/tests/tests.c @@ -74,28 +74,43 @@ uint8_t __wrap_sio_read(uint16_t port, uint8_t reg) return (uint8_t)mock(); } -int __wrap_open(const char *pathname, int flags) +static int mock_open(const char *pathname, int flags) { - LOG_ME; if (get_io() && get_io()->open) return get_io()->open(get_io()->state, pathname, flags); + + if (get_io() && get_io()->fallback_open_state) { + struct io_mock_fallback_open_state *io_state; + unsigned int open_state_flags; + + io_state = get_io()->fallback_open_state; + assert_true(io_state->noc < MAX_MOCK_OPEN); + assert_non_null(io_state->paths[io_state->noc]); + assert_string_equal(pathname, io_state->paths[io_state->noc]); + open_state_flags = io_state->flags[io_state->noc]; + assert_int_equal(flags & open_state_flags, open_state_flags); + io_state->noc++; // proceed to the next path upon next call. + } + return MOCK_FD; } +int __wrap_open(const char *pathname, int flags) +{ + LOG_ME; + return mock_open(pathname, flags); +} + int __wrap_open64(const char *pathname, int flags) { LOG_ME; - if (get_io() && get_io()->open) - return get_io()->open(get_io()->state, pathname, flags); - return MOCK_FD; + return mock_open(pathname, flags); } int __wrap___open64_2(const char *pathname, int flags) { LOG_ME; - if (get_io() && get_io()->open) - return get_io()->open(get_io()->state, pathname, flags); - return MOCK_FD; + return mock_open(pathname, flags); } int __wrap_ioctl(int fd, unsigned long int request, ...) |