1 files changed, 74 insertions, 0 deletions
diff --git a/zephyr/test/rex/src/usb_mux_config.c b/zephyr/test/rex/src/usb_mux_config.c
new file mode 100644
index 0000000000..fe4e69feb8
--- /dev/null
+++ b/zephyr/test/rex/src/usb_mux_config.c
@@ -0,0 +1,74 @@
+/* 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 "cros_cbi.h"
+#include "hooks.h"
+#include "usb_mux.h"
+#include "usbc/usb_muxes.h"
+
+#include <zephyr/devicetree.h>
+#include <zephyr/fff.h>
+#include <zephyr/ztest.h>
+
+FAKE_VALUE_FUNC(int, cros_cbi_get_fw_config, enum cbi_fw_config_field_id,
+		uint32_t *);
+
+int mock_cros_cbi_get_fw_config(enum cbi_fw_config_field_id field_id,
+				uint32_t *value)
+{
+	*value = FW_USB_DB_USB3;
+	return 0;
+}
+
+int mock_cros_cbi_get_fw_config_no_usb_db(enum cbi_fw_config_field_id field_id,
+					  uint32_t *value)
+{
+	*value = FW_USB_DB_NOT_CONNECTED;
+	return 0;
+}
+
+int mock_cros_cbi_get_fw_config_error(enum cbi_fw_config_field_id field_id,
+				      uint32_t *value)
+{
+	return -1;
+}
+
+static void usb_mux_config_before(void *fixture)
+{
+	ARG_UNUSED(fixture);
+
+	RESET_FAKE(cros_cbi_get_fw_config);
+}
+
+ZTEST_USER(usb_mux_config, test_setup_mux)
+{
+	cros_cbi_get_fw_config_fake.custom_fake = mock_cros_cbi_get_fw_config;
+
+	hook_notify(HOOK_INIT);
+
+	zassert_equal(1, cros_cbi_get_fw_config_fake.call_count);
+}
+
+ZTEST_USER(usb_mux_config, test_setup_mux_no_usb_db)
+{
+	cros_cbi_get_fw_config_fake.custom_fake =
+		mock_cros_cbi_get_fw_config_no_usb_db;
+
+	hook_notify(HOOK_INIT);
+
+	zassert_equal(1, cros_cbi_get_fw_config_fake.call_count);
+}
+
+ZTEST_USER(usb_mux_config, test_setup_mux_error_reading_cbi)
+{
+	cros_cbi_get_fw_config_fake.custom_fake =
+		mock_cros_cbi_get_fw_config_error;
+
+	hook_notify(HOOK_INIT);
+
+	zassert_equal(1, cros_cbi_get_fw_config_fake.call_count);
+}
+
+ZTEST_SUITE(usb_mux_config, NULL, NULL, usb_mux_config_before, NULL, NULL);