From 2d709a48e6fa0b331c12e42c7e887e58fa63ae51 Mon Sep 17 00:00:00 2001 From: Alexander Kukushkin Date: Mon, 26 Oct 2015 11:30:04 +0100 Subject: [PATCH] fix test_create_replica unit test --- tests/test_postgresql.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_postgresql.py b/tests/test_postgresql.py index 0ed04a15..92f32a02 100644 --- a/tests/test_postgresql.py +++ b/tests/test_postgresql.py @@ -263,9 +263,13 @@ class TestPostgresql(unittest.TestCase): self.assertTrue(self.p.can_rewind) self.p.controldata = tmp + @patch('time.sleep', Mock()) def test_create_replica(self): self.p.delete_trigger_file = Mock(side_effect=OSError()) - self.assertEquals(self.p.create_replica({'host': '', 'port': '', 'user': ''}, ''), 1) + with patch('subprocess.call', Mock(side_effect=[1, 0])): + self.assertEquals(self.p.create_replica({'host': '', 'port': '', 'user': ''}, ''), 0) + with patch('subprocess.call', Mock(side_effect=[Exception(), 0])): + self.assertEquals(self.p.create_replica({'host': '', 'port': '', 'user': ''}, ''), 0) def test_create_connection_users(self): cfg = self.p.config