diff --git a/tests/ztest/mock/src/main.c b/tests/ztest/mock/src/main.c index 9e8b72e5770..1192dd9ce67 100644 --- a/tests/ztest/mock/src/main.c +++ b/tests/ztest/mock/src/main.c @@ -52,12 +52,29 @@ static void test_multi_value_tests(void) expect_two_parameters(2, 3); } +static void returns_data(uint8_t *buf, size_t buf_size) +{ + ztest_copy_return_data(buf, buf_size); +} + +static void test_return_data_tests(void) +{ + uint8_t expected_data[] = { 0x5A, 0xA5, 0xDE, 0xAD }; + uint8_t data[] = { 0xFF, 0xFF, 0xFF, 0xFF }; + + ztest_expect_data(returns_data, buf, expected_data); + + returns_data(data, sizeof(data)); + zassert_mem_equal(expected_data, data, sizeof(data), NULL); +} + void test_main(void) { ztest_test_suite(mock_framework_tests, ztest_unit_test(test_parameter_tests), ztest_unit_test(test_return_value_tests), - ztest_unit_test(test_multi_value_tests) + ztest_unit_test(test_multi_value_tests), + ztest_unit_test(test_return_data_tests) ); ztest_run_test_suite(mock_framework_tests);