test: Add test_send_caps_error to usb_pe_drp

As requested in CL:2321869 review, make a new version of
test_send_caps_error that uses the external interface of the PE
layer.

BUG=b:161835483
BRANCH=none
TEST=make run-usb_pe_drp

Signed-off-by: Edward Hill <ecgh@chromium.org>
Change-Id: I983b145cd1e731e844363955896898a2374a0a30
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/ec/+/2419834
Reviewed-by: Paul Fagerburg <pfagerburg@chromium.org>
Reviewed-by: Jett Rink <jettrink@chromium.org>
Commit-Queue: Paul Fagerburg <pfagerburg@chromium.org>

1 files changed, 90 insertions, 0 deletions

diff --git a/test/usb_pe_drp.c b/test/usb_pe_drp.c
index b210db4ae9..8c59dcf180 100644
--- a/test/usb_pe_drp.c
+++ b/test/usb_pe_drp.c
@@ -43,12 +43,102 @@ void before_test(void)
 	mock_dpm_reset();
 	mock_dp_alt_mode_reset();
 	mock_prl_reset();
+
+	/* Restart the PD task and let it settle */
+	task_set_event(TASK_ID_PD_C0, TASK_EVENT_RESET_DONE, 0);
+	task_wait_event(SECOND);
+}
+
+test_static int test_send_caps_error(void)
+{
+	/* Enable PE as source, expect SOURCE_CAP. */
+	mock_pd_port[PORT0].power_role = PD_ROLE_SOURCE;
+	mock_tc_port[PORT0].pd_enable = 1;
+	task_wait_event(10 * MSEC);
+	TEST_EQ(fake_prl_get_last_sent_data_msg_type(PORT0),
+		PD_DATA_SOURCE_CAP, "%d");
+	fake_prl_message_sent(PORT0);
+	task_wait_event(10 * MSEC);
+
+	/* REQUEST 5V, expect ACCEPT, PS_RDY. */
+	rx_emsg[PORT0].header = PD_HEADER(PD_DATA_REQUEST, PD_ROLE_SINK,
+			PD_ROLE_UFP, 0,
+			1, PD_REV30, 0);
+	rx_emsg[PORT0].len = 4;
+	*(uint32_t *)rx_emsg[PORT0].buf = RDO_FIXED(1, 500, 500, 0);
+	fake_prl_message_received(PORT0);
+	task_wait_event(10 * MSEC);
+	TEST_EQ(fake_prl_get_last_sent_ctrl_msg(PORT0),
+		PD_CTRL_ACCEPT, "%d");
+	fake_prl_message_sent(PORT0);
+	task_wait_event(10 * MSEC);
+	TEST_EQ(fake_prl_get_last_sent_ctrl_msg(PORT0),
+		PD_CTRL_PS_RDY, "%d");
+	fake_prl_message_sent(PORT0);
+	task_wait_event(30 * MSEC);
+
+	/* Expect VENDOR_DEF, reply NOT_SUPPORTED. */
+	TEST_EQ(fake_prl_get_last_sent_data_msg_type(PORT0),
+		PD_DATA_VENDOR_DEF, "%d");
+	fake_prl_message_sent(PORT0);
+	task_wait_event(10 * MSEC);
+	rx_emsg[PORT0].header = PD_HEADER(PD_CTRL_NOT_SUPPORTED, PD_ROLE_SINK,
+			PD_ROLE_UFP, 2,
+			0, PD_REV30, 0);
+	rx_emsg[PORT0].len = 0;
+	fake_prl_message_received(PORT0);
+	task_wait_event(30 * MSEC);
+
+	/* Expect GET_SOURCE_CAP, reply NOT_SUPPORTED. */
+	TEST_EQ(fake_prl_get_last_sent_ctrl_msg(PORT0),
+		PD_CTRL_GET_SOURCE_CAP, "%d");
+	fake_prl_message_sent(PORT0);
+	task_wait_event(10 * MSEC);
+	rx_emsg[PORT0].header = PD_HEADER(PD_CTRL_NOT_SUPPORTED, PD_ROLE_SINK,
+			PD_ROLE_UFP, 2,
+			0, PD_REV30, 0);
+	rx_emsg[PORT0].len = 0;
+	fake_prl_message_received(PORT0);
+	task_wait_event(200 * MSEC);
+
+	/*
+	 * Now connected. Send GET_SOURCE_CAP, to check how error sending
+	 * SOURCE_CAP is handled.
+	 */
+	rx_emsg[PORT0].header = PD_HEADER(PD_CTRL_GET_SOURCE_CAP, PD_ROLE_SINK,
+			PD_ROLE_UFP, 3,
+			0, PD_REV30, 0);
+	rx_emsg[PORT0].len = 0;
+	fake_prl_message_received(PORT0);
+	task_wait_event(10 * MSEC);
+	TEST_EQ(fake_prl_get_last_sent_data_msg_type(PORT0),
+		PD_DATA_SOURCE_CAP, "%d");
+
+	/* Simulate error sending SOURCE_CAP. */
+	fake_prl_report_error(PORT0, ERR_TCH_XMIT);
+	task_wait_event(20 * MSEC);
+
+	/*
+	 * Expect SOFT_RESET.
+	 * See section 8.3.3.4.1.1 PE_SRC_Send_Soft_Reset State and section
+	 * 8.3.3.2.3 PE_SRC_Send_Capabilities State.
+	 * "The PE_SRC_Send_Soft_Reset state Shall be entered from any state
+	 * when ... A Message has not been sent after retries to the Sink"
+	 */
+	TEST_EQ(fake_prl_get_last_sent_ctrl_msg(PORT0),
+		PD_CTRL_SOFT_RESET, "%d");
+	fake_prl_message_sent(PORT0);
+	task_wait_event(5 * SECOND);
+
+	return EC_SUCCESS;
 }
 
 void run_test(int argc, char **argv)
 {
 	test_reset();
 
+	RUN_TEST(test_send_caps_error);
+
 	/* Do basic state machine validity checks last. */
 	RUN_TEST(test_pe_no_parent_cycles);