From 685f73f5c27ec7a101bfbc225d9635e269276835 Mon Sep 17 00:00:00 2001
From: "Antonio Melo Jr." <a_junior@apple.com>
Date: Thu, 19 Sep 2024 16:39:35 +0000
Subject: [PATCH] Updating Test Ui Observable Unit Test with the expected
 endpoint information

---
 app/tests/test_engine/test_ui_observer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/app/tests/test_engine/test_ui_observer.py b/app/tests/test_engine/test_ui_observer.py
index 10ce82d8..39ee8345 100644
--- a/app/tests/test_engine/test_ui_observer.py
+++ b/app/tests/test_engine/test_ui_observer.py
@@ -145,7 +145,7 @@ def __expected_test_case_dict(index: int, suite_index: int) -> Dict[str, Any]:
 
 
 def __expected_test_step_dict(
-    index: int, case_index: int, suite_index: int
+    index: int, case_index: int, suite_index: int, endpoint: int | None = None
 ) -> Dict[str, Any]:
     return {
         MessageKeysEnum.TYPE: MessageTypeEnum.TEST_UPDATE,
@@ -155,6 +155,7 @@ def __expected_test_step_dict(
                 "test_suite_execution_index": suite_index,
                 "test_case_execution_index": case_index,
                 "test_step_execution_index": index,
+                "endpoint": endpoint,
                 "state": TestStateEnum.EXECUTING,
                 "errors": [],
                 "failures": [],