Merge branch '6593-arv-get-stdout' closes #6593

[arvados.git] / sdk / python / tests / test_websockets.py

diff --git a/sdk/python/tests/test_websockets.py b/sdk/python/tests/test_websockets.py
index fc246ac61dd8f30b6ebc6e7ea6cb904c78c85b4a..ffa310baca558ab17cd3c5f245b5010c72d898b9 100644 (file)
--- a/sdk/python/tests/test_websockets.py
+++ b/sdk/python/tests/test_websockets.py
@@ -74,7 +74,8 @@ class WebsocketTest(run_test_server.TestCaseWithServers):
                 additional_filters=[['created_at', '>=', lastHour.strftime('%Y-%m-%d')]])
 
     @mock.patch('arvados.events.EventClient.__init__')
-    def test_poll_websocket_with_start_time_date_only(self, event_client_constr):
+    def test_poll_with_start_time_date_only(self, event_client_constr):
+        event_client_constr.side_effect = Exception('All is well')
         lastHour = datetime.today() - timedelta(hours = 1)
         self._test_subscribe(
             poll_fallback=1, expect_type=arvados.events.PollClient, last_log_id=1,
@@ -88,6 +89,7 @@ class WebsocketTest(run_test_server.TestCaseWithServers):
 
     @mock.patch('arvados.events.EventClient.__init__')
     def test_subscribe_poll_with_start_time_last_hour(self, event_client_constr):
+        event_client_constr.side_effect = Exception('All is well')
         lastHour = datetime.today() - timedelta(hours = 1)
         self._test_subscribe(
             poll_fallback=1, expect_type=arvados.events.PollClient, last_log_id=1,
@@ -102,6 +104,7 @@ class WebsocketTest(run_test_server.TestCaseWithServers):
 
     @mock.patch('arvados.events.EventClient.__init__')
     def test_subscribe_poll_with_start_time_next_hour(self, event_client_constr):
+        event_client_constr.side_effect = Exception('All is well')
         nextHour = datetime.today() + timedelta(hours = 1)
         with self.assertRaises(Queue.Empty):
             self._test_subscribe(
@@ -117,6 +120,7 @@ class WebsocketTest(run_test_server.TestCaseWithServers):
 
     @mock.patch('arvados.events.EventClient.__init__')
     def test_subscribe_poll_with_start_time_tomorrow(self, event_client_constr):
+        event_client_constr.side_effect = Exception('All is well')
         tomorrow = datetime.today() + timedelta(hours = 24)
         with self.assertRaises(Queue.Empty):
             self._test_subscribe(