From 94aa6873f4b4e5eae460f6897d335559c07dc0f8 Mon Sep 17 00:00:00 2001 From: Oleksii Kliukin Date: Tue, 13 Oct 2015 08:19:44 +0200 Subject: [PATCH] Add more tests for the new postgresql methods. --- tests/test_postgresql.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/test_postgresql.py b/tests/test_postgresql.py index 5677c638..2f1c75ac 100644 --- a/tests/test_postgresql.py +++ b/tests/test_postgresql.py @@ -237,6 +237,20 @@ class TestPostgresql(unittest.TestCase): self.p.rewind.return_value = False self.p.follow_the_leader(self.leader, recovery=True) + def test_can_rewind(self): + tmp = self.p.pg_rewind + self.p.pg_rewind = None + self.assertFalse(self.p.can_rewind) + self.p.pg_rewind = tmp + with mock.patch('subprocess.call', MagicMock(return_value=1)): + self.assertFalse(self.p.can_rewind) + with mock.patch('subprocess.call', side_effect=OSError("foo")): + self.assertFalse(self.p.can_rewind) + tmp = self.p.controldata() + self.p.controldata = lambda: {'wal_log_hints setting': 'on'} + self.assertTrue(self.p.can_rewind) + self.p.controldata = tmp + def test_create_replica(self): self.p.delete_trigger_file = Mock(side_effect=OSError()) self.assertEquals(self.p.create_replica({'host': '', 'port': '', 'user': ''}, ''), 1)