1 files changed, 118 insertions, 0 deletions
diff --git a/zephyr/test/drivers/default/src/console_cmd/charge_manager.c b/zephyr/test/drivers/default/src/console_cmd/charge_manager.c
new file mode 100644
index 0000000000..f6ee049ea1
--- /dev/null
+++ b/zephyr/test/drivers/default/src/console_cmd/charge_manager.c
@@ -0,0 +1,118 @@
+/* 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/shell/shell.h>
+#include <zephyr/ztest.h>
+
+#include "charge_manager.h"
+#include "console.h"
+#include "emul/emul_isl923x.h"
+#include "emul/tcpc/emul_tcpci.h"
+#include "emul/tcpc/emul_tcpci_partner_snk.h"
+#include "tcpm/tcpci.h"
+#include "test/drivers/test_state.h"
+#include "test/drivers/utils.h"
+
+struct console_cmd_charge_manager_fixture {
+	struct tcpci_partner_data sink_5v_3a;
+	struct tcpci_snk_emul_data sink_ext;
+	const struct emul *tcpci_emul;
+	const struct emul *charger_emul;
+};
+
+static void *console_cmd_charge_manager_setup(void)
+{
+	static struct console_cmd_charge_manager_fixture test_fixture;
+
+	/* Get references for the emulators */
+	test_fixture.tcpci_emul = EMUL_GET_USBC_BINDING(0, tcpc);
+	test_fixture.charger_emul = EMUL_GET_USBC_BINDING(0, chg);
+
+	/* Initialized the sink to request 5V and 3A */
+	tcpci_partner_init(&test_fixture.sink_5v_3a, PD_REV20);
+	test_fixture.sink_5v_3a.extensions = tcpci_snk_emul_init(
+		&test_fixture.sink_ext, &test_fixture.sink_5v_3a, NULL);
+	test_fixture.sink_ext.pdo[1] =
+		PDO_FIXED(5000, 3000, PDO_FIXED_UNCONSTRAINED);
+
+	return &test_fixture;
+}
+
+static void console_cmd_charge_manager_after(void *state)
+{
+	struct console_cmd_charge_manager_fixture *fixture = state;
+
+	shell_execute_cmd(get_ec_shell(), "chgoverride -1");
+	disconnect_sink_from_port(fixture->tcpci_emul);
+}
+
+ZTEST_SUITE(console_cmd_charge_manager, drivers_predicate_post_main,
+	    console_cmd_charge_manager_setup, NULL,
+	    console_cmd_charge_manager_after, NULL);
+
+/**
+ * Test the chgsup (charge supplier info) command. This command only prints to
+ * console some information which is not yet possible to verify. So just check
+ * that the console command ran successfully.
+ */
+ZTEST_USER(console_cmd_charge_manager, test_chgsup)
+{
+	zassert_ok(shell_execute_cmd(get_ec_shell(), "chgsup"), NULL);
+}
+
+/**
+ * Test chgoverride command with no arguments. This should just print the
+ * current override port.
+ */
+ZTEST_USER(console_cmd_charge_manager, test_chgoverride_missing_port)
+{
+	zassert_ok(shell_execute_cmd(get_ec_shell(), "chgoverride"), NULL);
+}
+
+ZTEST_USER(console_cmd_charge_manager, test_chgoverride_off_from_off)
+{
+	zassert_ok(shell_execute_cmd(get_ec_shell(), "chgoverride -1"), NULL);
+	zassert_equal(charge_manager_get_override(), OVERRIDE_OFF, NULL);
+}
+
+ZTEST_USER(console_cmd_charge_manager, test_chgoverride_disable_from_off)
+{
+	zassert_ok(shell_execute_cmd(get_ec_shell(), "chgoverride -2"), NULL);
+	zassert_equal(charge_manager_get_override(), OVERRIDE_DONT_CHARGE,
+		      NULL);
+}
+
+ZTEST_USER(console_cmd_charge_manager, test_chgoverride_0_from_off)
+{
+	zassert_ok(shell_execute_cmd(get_ec_shell(), "chgoverride 0"), NULL);
+	zassert_equal(charge_manager_get_override(), 0, NULL);
+}
+
+ZTEST_USER_F(console_cmd_charge_manager, test_chgoverride_0_from_sink)
+{
+	test_set_chipset_to_g3();
+	k_sleep(K_SECONDS(1));
+
+	/* Set chipset to ON, this will set TCPM to DRP */
+	test_set_chipset_to_s0();
+
+	/* TODO(b/214401892): Check why need to give time TCPM to spin */
+	k_sleep(K_SECONDS(1));
+
+	connect_sink_to_port(&fixture->sink_5v_3a, fixture->tcpci_emul,
+			     fixture->charger_emul);
+	zassert_equal(shell_execute_cmd(get_ec_shell(), "chgoverride 0"),
+		      EC_ERROR_INVAL, NULL);
+}
+
+ZTEST_USER(console_cmd_charge_manager, test_chgoverride_invalid_port)
+{
+	char cmd[256];
+
+	zassume_true(sprintf(cmd, "chgoverride %d", CHARGE_PORT_COUNT) > 0,
+		     NULL);
+	zassert_equal(shell_execute_cmd(get_ec_shell(), cmd), EC_ERROR_PARAM1,
+		      NULL);
+}