1 files changed, 90 insertions, 0 deletions
diff --git a/zephyr/test/kingler/src/usb_port_count.c b/zephyr/test/kingler/src/usb_port_count.c
new file mode 100644
index 0000000000..4f55bdc693
--- /dev/null
+++ b/zephyr/test/kingler/src/usb_port_count.c
@@ -0,0 +1,90 @@
+/* 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 "driver/tcpm/tcpm.h"
+#include "ec_app_main.h"
+#include "usb_charge.h"
+#include "usb_pd.h"
+#include "usbc_ppc.h"
+#include "variant_db_detection.h"
+
+#include <zephyr/drivers/gpio/gpio_emul.h>
+#include <zephyr/fff.h>
+#include <zephyr/kernel.h>
+#include <zephyr/logging/log.h>
+#include <zephyr/ztest.h>
+
+#define LOG_LEVEL 0
+LOG_MODULE_REGISTER(usb_port_count);
+
+FAKE_VALUE_FUNC(enum corsola_db_type, corsola_get_db_type);
+FAKE_VALUE_FUNC(bool, in_interrupt_context);
+FAKE_VOID_FUNC(bmi3xx_interrupt);
+FAKE_VOID_FUNC(hdmi_hpd_interrupt);
+FAKE_VOID_FUNC(ps185_hdmi_hpd_mux_set);
+FAKE_VALUE_FUNC(bool, ps8743_field_update, const struct usb_mux *, uint8_t,
+		uint8_t, uint8_t);
+FAKE_VOID_FUNC(pd_set_dual_role, int, enum pd_dual_role_states);
+FAKE_VALUE_FUNC(int, tc_is_attached_src, int);
+
+#define FFF_FAKES_LIST(FAKE)         \
+	FAKE(corsola_get_db_type)    \
+	FAKE(in_interrupt_context)   \
+	FAKE(bmi3xx_interrupt)       \
+	FAKE(hdmi_hpd_interrupt)     \
+	FAKE(ps185_hdmi_hpd_mux_set) \
+	FAKE(ps8743_field_update)    \
+	FAKE(pd_set_dual_role)       \
+	FAKE(tc_is_attached_src)
+
+struct usb_port_count_fixture {
+	int place_holder;
+};
+
+static void *usb_port_count_setup(void)
+{
+	static struct usb_port_count_fixture f;
+
+	return &f;
+}
+
+static void usb_port_count_reset_rule_before(const struct ztest_unit_test *test,
+					     void *data)
+{
+	ARG_UNUSED(test);
+	ARG_UNUSED(data);
+	FFF_FAKES_LIST(RESET_FAKE);
+	FFF_RESET_HISTORY();
+}
+
+ZTEST_RULE(usb_port_count_reset_rule, usb_port_count_reset_rule_before, NULL);
+ZTEST_SUITE(usb_port_count, NULL, usb_port_count_setup, NULL, NULL, NULL);
+
+ZTEST_F(usb_port_count, test_detect_db)
+{
+	struct {
+		enum corsola_db_type test_type;
+		int expected_board_get_port_count;
+		int expected_board_get_adjusted_port_count;
+	} testdata[] = { { CORSOLA_DB_UNINIT, CONFIG_USB_PD_PORT_MAX_COUNT,
+			   CONFIG_USB_PD_PORT_MAX_COUNT - 1 },
+			 { CORSOLA_DB_NO_DETECTION,
+			   CONFIG_USB_PD_PORT_MAX_COUNT,
+			   CONFIG_USB_PD_PORT_MAX_COUNT },
+			 { CORSOLA_DB_NONE, CONFIG_USB_PD_PORT_MAX_COUNT - 1,
+			   CONFIG_USB_PD_PORT_MAX_COUNT - 1 },
+			 { CORSOLA_DB_TYPEC, CONFIG_USB_PD_PORT_MAX_COUNT,
+			   CONFIG_USB_PD_PORT_MAX_COUNT },
+			 { CORSOLA_DB_HDMI, CONFIG_USB_PD_PORT_MAX_COUNT - 1,
+			   CONFIG_USB_PD_PORT_MAX_COUNT - 1 } };
+	for (int i = 0; i < ARRAY_SIZE(testdata); i++) {
+		corsola_get_db_type_fake.return_val = testdata[i].test_type;
+		zassert_equal(board_get_usb_pd_port_count(),
+			      testdata[i].expected_board_get_port_count);
+		zassert_equal(
+			board_get_adjusted_usb_pd_port_count(),
+			testdata[i].expected_board_get_adjusted_port_count);
+	}
+}