Increase the test coverage.

This commit is contained in:
Oleksii Kliukin
2015-10-19 10:13:14 +02:00
parent 11073ac644
commit 4e448015f3
2 changed files with 11 additions and 1 deletions
+7 -1
View File
@@ -1,7 +1,7 @@
import etcd
import unittest
from mock import Mock, patch
from mock import Mock, MagicMock, patch
from patroni.dcs import Cluster, Failover, Leader, Member
from patroni.etcd import Client, Etcd
from patroni.exceptions import DCSError, PostgresException
@@ -130,6 +130,12 @@ class TestHa(unittest.TestCase):
self.ha.has_lock = true
self.assertEquals(self.ha.run_cycle(), 'removed leader key after trying and failing to start postgres')
@patch('sys.exit', return_value=1)
@patch('patroni.ha.Ha.sysid_valid', MagicMock(return_value=True))
def test_sysid_no_match(self, exit_mock):
self.ha.run_cycle()
exit_mock.assert_called_once_with(1)
@patch.object(Cluster, 'is_unlocked', Mock(return_value=False))
def test_start_as_readonly(self):
self.p.is_leader = self.p.is_healthy = false
+4
View File
@@ -429,3 +429,7 @@ class TestPostgresql(unittest.TestCase):
self.p.cleanup_archive_status()
mock_unlink.assert_not_called()
mock_remove.assert_not_called()
@patch('subprocess.check_output', MagicMock(return_value=0, side_effect=pg_controldata_string))
def test_sysid(self):
self.assertEqual(self.p.sysid, "6200971513092291716")