From aa350b71394fe24ddd8691f16d035f1e4b1ae3e6 Mon Sep 17 00:00:00 2001 From: Oleksii Kliukin Date: Tue, 26 Jan 2016 15:57:23 +0100 Subject: [PATCH] Increase the tests coverage. --- tests/test_patroni.py | 5 +++++ tests/test_wale_restore.py | 6 +++++- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/tests/test_patroni.py b/tests/test_patroni.py index 18f5c14ba..d1db99cb8 100644 --- a/tests/test_patroni.py +++ b/tests/test_patroni.py @@ -80,3 +80,8 @@ def test_nofailover(self): self.assertTrue(self.p.nofailover) self.p.tags['nofailover'] = None self.assertFalse(self.p.nofailover) + + def test_replicatefrom(self): + self.assertIsNone(self.p.replicatefrom) + self.p.tags['replicatefrom'] = 'foo' + self.assertEqual(self.p.replicatefrom, 'foo') diff --git a/tests/test_wale_restore.py b/tests/test_wale_restore.py index ba4c3bea5..f353a4c94 100644 --- a/tests/test_wale_restore.py +++ b/tests/test_wale_restore.py @@ -3,7 +3,7 @@ import os import psycopg2 import subprocess -from patroni.scripts.wale_restore import WALERestore +from patroni.scripts.wale_restore import WALERestore, main def fake_cursor_fetchone(*args, **kwargs): @@ -91,3 +91,7 @@ def test_run(self): with patch.object(self.wale_restore, 'should_use_s3_to_create_replica', MagicMock(return_value=True)): with patch.object(self.wale_restore, 'create_replica_with_s3', MagicMock(return_value=0)): self.assertEqual(self.wale_restore.run(), 0) + + def test_main(self): + with patch('sys.exit', MagicMock(return_value=0)): + main()