fix test_create_replica unit test

This commit is contained in:
Alexander Kukushkin
2015-10-26 11:30:04 +01:00
parent ab64ae7fe7
commit 2d709a48e6
+5 -1
View File
@@ -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