summaryrefslogtreecommitdiff
path: root/tests/vboot_api_kernel_tests.c
diff options
context:
space:
mode:
Diffstat (limited to 'tests/vboot_api_kernel_tests.c')
-rw-r--r--tests/vboot_api_kernel_tests.c8
1 files changed, 3 insertions, 5 deletions
diff --git a/tests/vboot_api_kernel_tests.c b/tests/vboot_api_kernel_tests.c
index b2d4236e..f885cd86 100644
--- a/tests/vboot_api_kernel_tests.c
+++ b/tests/vboot_api_kernel_tests.c
@@ -12,7 +12,6 @@
#include "load_kernel_fw.h"
#include "test_common.h"
#include "vboot_api.h"
-#include "vboot_test.h"
#define MAX_TEST_DISKS 10
#define DEFAULT_COUNT -1
@@ -452,8 +451,7 @@ static void ResetMocks(test_case_t *test_case)
TEST_SUCC(vb2api_init(workbuf, sizeof(workbuf), &ctx),
"vb2api_init failed");
- memset(&kparams, 0, sizeof(VbSelectAndLoadKernelParams));
- *VbApiKernelGetParamsPtr() = &kparams;
+ memset(&kparams, 0, sizeof(kparams));
memset(&mock_disks, 0, sizeof(mock_disks));
@@ -591,7 +589,7 @@ static void VbTryLoadKernelTest(void)
printf("Test case: %s ...\n", normal_tests[i].name);
ResetMocks(&normal_tests[i]);
ctx->flags = t->ctx_flags;
- TEST_EQ(VbTryLoadKernel(ctx, t->want_flags),
+ TEST_EQ(VbTryLoadKernel(ctx, t->want_flags, &kparams),
t->expected_return_val, " return value");
TEST_EQ(got_recovery_request_val,
t->expected_recovery_request_val, " recovery_request");
@@ -618,7 +616,7 @@ static void VbTryLoadMiniOsKernelTest(void)
printf("Test case: %s ...\n", minios_tests[i].name);
ResetMocks(&minios_tests[i]);
ctx->flags = t->ctx_flags;
- TEST_EQ(VbTryLoadMiniOsKernel(ctx, 0),
+ TEST_EQ(VbTryLoadMiniOsKernel(ctx, 0, &kparams),
t->expected_return_val, " return value");
TEST_EQ(got_recovery_request_val,
t->expected_recovery_request_val, " recovery_request");