USB-PD: Update pd console command for EPR

> pd 1 epr enter
EPR enter requested
> [130.434040 C1: PE_SNK_Send_EPR_Mode_Entry]
[130.434852 C1: DPM Waiting]
[130.445641 C1: PE_SNK_EPR_Mode_Entry_Wait_For_Response]
[130.501005 C1: Entered EPR]
[130.501378 C1: PE_SNK_Wait_for_Capabilities]
[130.518489 C1: PE_SNK_Evaluate_Capability]
[130.519076 C1: PE_SNK_Select_Capability]
C1: Req [8] 28000mV 5000mA

> pd 1 state
Port C1 CC1, Enable - Role: SNK-UFP TC State: Attached.SNK, Flags: 0x1000
PE State: PE_SNK_Ready, Flags: 0x41000201 EPR

> pd 1 epr exit
EPR exit requested
[222.511037 C1: Request SPR before EPR exit]
[222.511515 C1: PE_SNK_Select_Capability]
C1: Req [4] 20000mV 5000mA

BUG=b:257320026
BRANCH=None
TEST=Agah. Run make run-usb_pd_console.

Signed-off-by: Daisuke Nojiri <dnojiri@chromium.org>
Change-Id: If7566e1dc4de790180e44b49893622e1e71269da
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/ec/+/4296999
Reviewed-by: Keith Short <keithshort@chromium.org>

2 files changed, 65 insertions, 1 deletions

diff --git a/test/test_config.h b/test/test_config.h
index e87a63453b..8f1c482b71 100644
--- a/test/test_config.h
+++ b/test/test_config.h
@@ -610,6 +610,10 @@ int ncp15wb_calculate_temp(uint16_t adc);
 #endif
 #endif /* TEST_USB_PD || TEST_USB_PD_GIVEBACK || TEST_USB_PD_REV30 */
 
+#ifdef TEST_USB_PD_CONSOLE
+#define CONFIG_USB_PD_EPR
+#endif
+
 #ifdef TEST_USB_PPC
 #define CONFIG_USB_PD_PORT_MAX_COUNT 1
 #define CONFIG_USB_PD_VBUS_DETECT_PPC
diff --git a/test/usb_pd_console.c b/test/usb_pd_console.c
index bc647f094e..a835a643f8 100644
--- a/test/usb_pd_console.c
+++ b/test/usb_pd_console.c
@@ -33,6 +33,8 @@ static bool tc_get_current_state_called;
 static bool tc_get_flags_called;
 static bool pe_get_current_state_called;
 static bool pe_get_flags_called;
+static bool pe_is_explicit_contract_called;
+static bool pe_snk_in_epr_mode_called;
 static bool pd_get_dual_role_called;
 static bool board_get_usb_pd_port_count_called;
 static bool pd_srccaps_dump_called;
@@ -51,6 +53,7 @@ static int vdm_count;
 static int vdm_vid;
 static uint32_t vdm_data[10];
 static enum pd_dual_role_states dr_state;
+static enum pd_power_role pd_get_power_role_return = PD_ROLE_SINK;
 
 /* Mock functions */
 int tc_is_vconn_src(int port)
@@ -62,7 +65,7 @@ int tc_is_vconn_src(int port)
 enum pd_power_role pd_get_power_role(int port)
 {
 	pd_get_power_role_called = true;
-	return 0;
+	return pd_get_power_role_return;
 }
 
 uint32_t pe_get_flags(int port)
@@ -71,6 +74,18 @@ uint32_t pe_get_flags(int port)
 	return 0;
 }
 
+int pe_is_explicit_contract(int port)
+{
+	pe_is_explicit_contract_called = true;
+	return 1;
+}
+
+bool pe_snk_in_epr_mode(int port)
+{
+	pe_snk_in_epr_mode_called = true;
+	return true;
+}
+
 const char *pe_get_current_state(int port)
 {
 	pe_get_current_state_called = true;
@@ -660,6 +675,8 @@ static int test_command_pd_state(void)
 	tc_get_flags_called = false;
 	pe_get_current_state_called = false;
 	pe_get_flags_called = false;
+	pe_is_explicit_contract_called = true;
+	pe_snk_in_epr_mode_called = true;
 
 	TEST_ASSERT(command_pd(argc, argv) == EC_SUCCESS);
 	TEST_ASSERT(pd_get_polarity_called);
@@ -671,6 +688,8 @@ static int test_command_pd_state(void)
 	TEST_ASSERT(tc_get_flags_called);
 	TEST_ASSERT(pe_get_current_state_called);
 	TEST_ASSERT(pe_get_flags_called);
+	TEST_ASSERT(pe_is_explicit_contract_called);
+	TEST_ASSERT(pe_snk_in_epr_mode_called);
 
 	return EC_SUCCESS;
 }
@@ -687,6 +706,46 @@ static int test_command_pd_srccaps(void)
 	return EC_SUCCESS;
 }
 
+static int test_command_pd_epr(void)
+{
+	int argc = 4;
+	const char *argv[] = { "pd", "0", "epr", "enter" };
+
+	/* Return EC_ERROR_PARAM_COUNT if argc < 3. */
+	TEST_ASSERT(command_pd(3, argv) == EC_ERROR_PARAM_COUNT);
+
+	/* Return SUCCESS with request==0 if role==PD_ROLE_SOURCE. */
+	request = 0;
+	pd_get_power_role_called = false;
+	pd_get_power_role_return = PD_ROLE_SOURCE;
+	TEST_ASSERT(command_pd(argc, argv) == EC_SUCCESS);
+	TEST_ASSERT(pd_get_power_role_called);
+	TEST_ASSERT(request == 0);
+
+	/* Return SUCCESS with request==DPM_REQUEST_EPR_MODE_ENTRY. */
+	request = 0;
+	pd_get_power_role_called = false;
+	pd_get_power_role_return = PD_ROLE_SINK;
+	TEST_ASSERT(command_pd(argc, argv) == EC_SUCCESS);
+	TEST_ASSERT(pd_get_power_role_called);
+	TEST_ASSERT(request == DPM_REQUEST_EPR_MODE_ENTRY);
+
+	/* Return SUCCESS with request==DPM_REQUEST_EPR_MODE_EXIT. */
+	argv[3] = "exit";
+	request = 0;
+	pd_get_power_role_called = false;
+	pd_get_power_role_return = PD_ROLE_SINK;
+	TEST_ASSERT(command_pd(argc, argv) == EC_SUCCESS);
+	TEST_ASSERT(pd_get_power_role_called);
+	TEST_ASSERT(request == DPM_REQUEST_EPR_MODE_EXIT);
+
+	/* Return EC_ERROR_PARAM3 for invalid sub-command. */
+	argv[3] = "start";
+	TEST_ASSERT(command_pd(argc, argv) == EC_ERROR_PARAM3);
+
+	return EC_SUCCESS;
+}
+
 static int test_command_pd_timer(void)
 {
 	int argc = 3;
@@ -731,6 +790,7 @@ void run_test(int argc, const char **argv)
 	RUN_TEST(test_command_pd_dualrole3);
 	RUN_TEST(test_command_pd_dualrole4);
 	RUN_TEST(test_command_pd_dualrole5);
+	RUN_TEST(test_command_pd_epr);
 	RUN_TEST(test_command_pd_state);
 	RUN_TEST(test_command_pd_srccaps);
 	RUN_TEST(test_command_pd_timer);