6473: some more error tests
authorradhika <radhika@curoverse.com>
Thu, 16 Jul 2015 13:05:54 +0000 (09:05 -0400)
committerradhika <radhika@curoverse.com>
Thu, 16 Jul 2015 13:05:54 +0000 (09:05 -0400)
sdk/python/tests/test_websockets.py

index 6889c6f1fb554d2b3095f29935c8311b770c6bf2..6d0a2f2079b01ed8f8d6a7ccba3a6e251f16df51 100644 (file)
@@ -38,10 +38,10 @@ class WebsocketTest(run_test_server.TestCaseWithServers):
             poll_fallback=False, expect_type=arvados.events.EventClient)
 
     def test_subscribe_websocket_with_start_time_today(self):
-        lastHour = datetime.today() - timedelta(hours = 1)
+        now = datetime.today()
         self._test_subscribe(
             poll_fallback=False, expect_type=arvados.events.EventClient,
-                additional_filters=[['created_at', '>', lastHour.strftime('%Y-%m-%d')]])
+                additional_filters=[['created_at', '>', now.strftime('%Y-%m-%d')]])
 
     def test_subscribe_websocket_with_start_time_last_hour(self):
         lastHour = datetime.today() - timedelta(hours = 1)
@@ -56,6 +56,25 @@ class WebsocketTest(run_test_server.TestCaseWithServers):
                 poll_fallback=False, expect_type=arvados.events.EventClient,
                     additional_filters=[['created_at', '>', nextHour.strftime('%Y-%m-%d %H:%M:%S')]])
 
+    def test_subscribe_websocket_with_start_time_tomorrow(self):
+        tomorrow = datetime.today() + timedelta(hours = 24)
+        with self.assertRaises(Queue.Empty):
+            self._test_subscribe(
+                poll_fallback=False, expect_type=arvados.events.EventClient,
+                    additional_filters=[['created_at', '>', tomorrow.strftime('%Y-%m-%d')]])
+
+    def test_subscribe_websocket_with_start_time_incorrect_date_format(self):
+        now = datetime.today()
+        self._test_subscribe(
+            poll_fallback=False, expect_type=arvados.events.EventClient,
+                additional_filters=[['created_at', '>', now.strftime('%Y-%m')]])
+
+    def test_subscribe_websocket_with_start_time_incorrect_time_format(self):
+        now = datetime.today()
+        self._test_subscribe(
+            poll_fallback=False, expect_type=arvados.events.EventClient,
+                additional_filters=[['created_at', '>', now.strftime('%Y-%m-%d %H:%M')]])
+
     @mock.patch('arvados.events.EventClient.__init__')
     def test_subscribe_poll(self, event_client_constr):
         event_client_constr.side_effect = Exception('All is well')