1 files changed, 62 insertions, 0 deletions
diff --git a/tests/vboot_api_kernel6_tests.c b/tests/vboot_api_kernel6_tests.c
new file mode 100644
index 00000000..42bd279c
--- /dev/null
+++ b/tests/vboot_api_kernel6_tests.c
@@ -0,0 +1,62 @@
+/* Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
+ * Use of this source code is governed by a BSD-style license that can be
+ * found in the LICENSE file.
+ *
+ * Tests for vboot_api_kernel.c
+ */
+
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "test_common.h"
+#include "vboot_api.h"
+#include "vboot_nvstorage.h"
+
+/* Mock data */
+static uint32_t virtual_dev_mode_fail;
+
+/**
+ * Reset mock data (for use before each test)
+ */
+static void ResetMocks(void)
+{
+	virtual_dev_mode_fail = 0;
+}
+
+/* Mocks */
+uint32_t SetVirtualDevMode(int val)
+{
+	if (virtual_dev_mode_fail)
+		return VBERROR_SIMULATED;
+	return VBERROR_SUCCESS;
+}
+
+static void VbUnlockDeviceTest(void)
+{
+	ResetMocks();
+	TEST_EQ(VbUnlockDevice(), 0, "unlock success");
+
+	ResetMocks();
+	virtual_dev_mode_fail = 1;
+	TEST_EQ(VbUnlockDevice(), VBERROR_TPM_SET_BOOT_MODE_STATE,
+		"set dev fail");
+}
+
+static void VbLockDeviceTest(void)
+{
+	ResetMocks();
+	TEST_EQ(VbLockDevice(), 0, "lock success");
+}
+
+int main(void)
+{
+	VbUnlockDeviceTest();
+	VbLockDeviceTest();
+
+	if (vboot_api_stub_check_memory())
+		return 255;
+
+	return gTestSuccess ? 0 : 255;
+}