Add tests for api; add checks for ctl and api for the paused state case

tests/test_api.py

@@ -39,6 +39,10 @@ class MockHa(object):
 
     state_handler = MockPostgresql()
 
+    @staticmethod
+    def is_paused():
+        return False
+
     @staticmethod
     def schedule_reinitialize():
         return 'reinitialize'
@@ -221,11 +225,18 @@ class TestRestApiHandler(unittest.TestCase):
                 request = make_request(role='master', postgres_version='9.5.2')
                 MockRestApiServer(RestApiHandler, request)
 
+        with patch.object(MockHa, 'is_paused', Mock(return_value=True)):
+            request = make_request(schedule='2016-08-42 12:45TZ+1', role='master')
+            MockRestApiServer(RestApiHandler, request)
+
     def test_do_DELETE_restart(self):
         for retval in (True, False):
             with patch.object(MockHa, 'delete_future_restart', Mock(return_value=retval)):
                 request = 'DELETE /restart HTTP/1.0' + self._authorization
                 self.assertIsNotNone(MockRestApiServer(RestApiHandler, request))
+                with patch.object(MockHa, 'is_paused', Mock(return_value=True)):
+                    request = 'DELETE /restart HTTP/1.0' + self._authorization
+                    self.assertIsNotNone(MockRestApiServer(RestApiHandler, request))
 
     @patch.object(MockPatroni, 'dcs')
     def test_do_POST_reinitialize(self, dcs):
@@ -239,6 +250,9 @@ class TestRestApiHandler(unittest.TestCase):
         cluster.leader.name = 'test'
         self.assertIsNotNone(MockRestApiServer(RestApiHandler, request))
 
+        with patch.object(MockHa, 'is_paused', Mock(return_value=True)):
+            self.assertIsNotNone(MockRestApiServer(RestApiHandler, request))
+
     @patch('time.sleep', Mock())
     def test_RestApiServer_query(self):
         with patch.object(MockCursor, 'execute', Mock(side_effect=psycopg2.OperationalError)):
@@ -306,6 +320,9 @@ class TestRestApiHandler(unittest.TestCase):
             d.manual_failover.return_value = False
             MockRestApiServer(RestApiHandler, request)
 
+        with patch.object(MockHa, 'is_paused', Mock(return_value=True)):
+            MockRestApiServer(RestApiHandler, request)
+
         # Exception: No timezone specified
         request = post + '97\n\n{"leader": "postgresql1", "member": "postgresql2",' +\
                          ' "scheduled_at": "6016-02-15T18:13:30.568224"}'

tests/test_ctl.py

@@ -82,6 +82,11 @@ class TestCtl(unittest.TestCase):
         result = self.runner.invoke(ctl, ['failover', 'dummy'], input='leader\nother\n2030-01-01T12:23:00\ny')
         assert result.exit_code == 0
 
+        with patch('patroni.ctl.is_paused', Mock(return_value=True)):
+            result = self.runner.invoke(ctl,
+                                        ['failover', 'dummy', '--force', '--scheduled', '2015-01-01T12:00:00+01:00'])
+            assert result.exit_code == 1
+
         # Aborting failover,as we anser NO to the confirmation
         result = self.runner.invoke(ctl, ['failover', 'dummy'], input='leader\nother\n\nN')
         assert result.exit_code == 1
@@ -241,6 +246,12 @@ class TestCtl(unittest.TestCase):
                                               '--scheduled', '2300-10-01T14:30'])
             assert 'Failed: flush scheduled restart' in result.output
 
+        with patch('patroni.ctl.is_paused', Mock(return_value=True)):
+            result = self.runner.invoke(ctl,
+                                        ['restart', 'alpha', 'other', '--force', '--scheduled', '2300-10-01T14:30'])
+            assert result.exit_code == 1
+
+
         with patch('requests.post', Mock(return_value=MockResponse())):
             # normal restart, the schedule is actually parsed, but not validated in patronictl
             result = self.runner.invoke(ctl, ['restart', 'alpha', '--pg-version', '42.0.0',

tests/test_ha.py

@@ -499,3 +499,8 @@ class TestHa(unittest.TestCase):
         self.assertEquals(self.ha.run_cycle(), 'PAUSE: postgres is not running')
         self.ha.has_lock = true
         self.assertEquals(self.ha.run_cycle(), 'PAUSE: removed leader lock because postgres is not running')
+
+    def test_no_etcd_connection_in_pause(self):
+        self.ha.is_paused = true
+        self.ha.load_cluster_from_dcs = Mock(side_effect=DCSError('Etcd is not responding properly'))
+        self.assertEquals(self.ha.run_cycle(), 'PAUSE: DCS is not accessible')