1 files changed, 55 insertions, 0 deletions
diff --git a/zephyr/test/skyrim/src/common/alt_charger.c b/zephyr/test/skyrim/src/common/alt_charger.c
new file mode 100644
index 0000000000..c03d31aaeb
--- /dev/null
+++ b/zephyr/test/skyrim/src/common/alt_charger.c
@@ -0,0 +1,55 @@
+/* Copyright 2023 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/devicetree.h>
+#include <zephyr/fff.h>
+#include <zephyr/ztest.h>
+
+#include <charger.h>
+#include <cros_cbi.h>
+#include <hooks.h>
+
+FAKE_VALUE_FUNC(int, cros_cbi_get_fw_config, enum cbi_fw_config_field_id,
+		uint32_t *);
+FAKE_VOID_FUNC(chg_enable_alternate_test, int);
+
+void alt_charger_init(void);
+
+static bool alt_charger;
+static int cros_cbi_get_fw_config_mock(enum cbi_fw_config_field_id field_id,
+				       uint32_t *value)
+{
+	if (field_id != FW_CHARGER)
+		return -EINVAL;
+
+	*value = alt_charger ? FW_CHARGER_ISL9538 : FW_CHARGER_ISL9241;
+	return 0;
+}
+
+static void alt_charger_before(void *fixture)
+{
+	ARG_UNUSED(fixture);
+	RESET_FAKE(cros_cbi_get_fw_config);
+	RESET_FAKE(chg_enable_alternate_test);
+
+	cros_cbi_get_fw_config_fake.custom_fake = cros_cbi_get_fw_config_mock;
+}
+
+ZTEST_SUITE(alt_charger, NULL, NULL, alt_charger_before, NULL, NULL);
+
+ZTEST(alt_charger, normal_charger)
+{
+	alt_charger = false;
+	alt_charger_init();
+	/* Test that the alternative charger wasn't enabled. */
+	zassert_equal(chg_enable_alternate_test_fake.call_count, 0);
+}
+
+ZTEST(alt_charger, alt_charger)
+{
+	alt_charger = true;
+	alt_charger_init();
+	zassert_equal(chg_enable_alternate_test_fake.call_count, 1);
+	zassert_equal(chg_enable_alternate_test_fake.arg0_val, 0);
+}