From dce258d6052992a3868daa5733adf8fe4c13485e Mon Sep 17 00:00:00 2001
From: Lewis Gaul <legaul@cisco.com>
Date: Thu, 14 Dec 2023 02:43:51 -0800
Subject: [PATCH] Handle other connection exception in
 test_go_active_action_fail() testcase

---
 tests/it/test_main_errors.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/it/test_main_errors.py b/tests/it/test_main_errors.py
index f48cc09..6aee020 100644
--- a/tests/it/test_main_errors.py
+++ b/tests/it/test_main_errors.py
@@ -116,7 +116,10 @@ def test_go_active_action_fail(
         "Hit exception when performing aws_activate_vip action on "
         "<xr_interface=HundredGigE0/0/0/1,vrid=1>"
     ) in ctr_logs
-    assert "ConnectTimeoutError: Connect timeout on endpoint URL" in ctr_logs
+    assert (
+        "ConnectTimeoutError: Connect timeout on endpoint URL" in ctr_logs
+        or "EndpointConnectionError: Could not connect to the endpoint URL" in ctr_logs
+    )
 
 
 def test_consistency_check_action_fail(