1 files changed, 93 insertions, 0 deletions
diff --git a/zephyr/test/drivers/default/src/console_cmd/accelinit.c b/zephyr/test/drivers/default/src/console_cmd/accelinit.c
new file mode 100644
index 0000000000..c440faebba
--- /dev/null
+++ b/zephyr/test/drivers/default/src/console_cmd/accelinit.c
@@ -0,0 +1,93 @@
+/* Copyright 2022 The ChromiumOS Authors
+ * Use of this source code is governed by a BSD-style license that can be
+ * found in the LICENSE file.
+ */
+
+#include <zephyr/fff.h>
+#include <zephyr/shell/shell.h>
+#include <zephyr/ztest.h>
+
+#include "accelgyro.h"
+#include "console.h"
+#include "motion_sense.h"
+#include "test/drivers/test_state.h"
+#include "test/drivers/utils.h"
+
+FAKE_VALUE_FUNC(int, mock_init, struct motion_sensor_t *);
+
+struct console_cmd_accelinit_fixture {
+	const struct accelgyro_drv *sensor_0_drv;
+	struct accelgyro_drv mock_drv;
+};
+
+static void *console_cmd_accelinit_setup(void)
+{
+	static struct console_cmd_accelinit_fixture fixture = {
+		.mock_drv = {
+			.init = mock_init,
+		},
+	};
+	fixture.sensor_0_drv = motion_sensors[0].drv;
+
+	return &fixture;
+}
+
+static void console_cmd_accelinit_before(void *fixture)
+{
+	ARG_UNUSED(fixture);
+	RESET_FAKE(mock_init);
+	FFF_RESET_HISTORY();
+}
+static void console_cmd_accelinit_after(void *fixture)
+{
+	struct console_cmd_accelinit_fixture *this = fixture;
+
+	motion_sensors[0].drv = this->sensor_0_drv;
+	motion_sensors[0].drv->init(&motion_sensors[0]);
+}
+
+ZTEST_SUITE(console_cmd_accelinit, drivers_predicate_post_main,
+	    console_cmd_accelinit_setup, console_cmd_accelinit_before,
+	    console_cmd_accelinit_after, NULL);
+
+ZTEST_USER(console_cmd_accelinit, test_invalid_sensor_num)
+{
+	int rv;
+
+	rv = shell_execute_cmd(get_ec_shell(), "accelinit f");
+	zassert_equal(EC_ERROR_PARAM1, rv, "Expected %d, but got %d",
+		      EC_ERROR_PARAM1, rv);
+
+	rv = shell_execute_cmd(get_ec_shell(), "accelinit -1");
+	zassert_equal(EC_ERROR_PARAM1, rv, "Expected %d, but got %d",
+		      EC_ERROR_PARAM1, rv);
+
+	rv = shell_execute_cmd(get_ec_shell(), "accelinit 100");
+	zassert_equal(EC_ERROR_PARAM1, rv, "Expected %d, but got %d",
+		      EC_ERROR_PARAM1, rv);
+}
+
+ZTEST_USER(console_cmd_accelinit, test_state_was_set)
+{
+	motion_sensors[0].state = SENSOR_INIT_ERROR;
+
+	zassert_ok(shell_execute_cmd(get_ec_shell(), "accelinit 0"), NULL);
+	zassert_equal(SENSOR_INITIALIZED, motion_sensors[0].state,
+		      "Expected %d, but got %d", SENSOR_INITIALIZED,
+		      motion_sensors[0].state);
+}
+
+ZTEST_USER_F(console_cmd_accelinit, test_fail_3_times)
+{
+	mock_init_fake.return_val = 1;
+	motion_sensors[0].drv = &fixture->mock_drv;
+	motion_sensors[0].state = SENSOR_INITIALIZED;
+
+	zassert_ok(shell_execute_cmd(get_ec_shell(), "accelinit 0"), NULL);
+	zassert_equal(3, mock_init_fake.call_count,
+		      "Expected 3 calls, but got %d",
+		      mock_init_fake.call_count);
+	zassert_equal(SENSOR_INIT_ERROR, motion_sensors[0].state,
+		      "Expected %d, but got %d", SENSOR_INIT_ERROR,
+		      motion_sensors[0].state);
+}