From 4c444c943e60758f0f84500a5fcb88c752bcf2cc Mon Sep 17 00:00:00 2001 From: Alexander Kukushkin Date: Fri, 2 Oct 2015 13:17:58 +0200 Subject: [PATCH] tests for Api.do_GET method --- tests/test_api.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/tests/test_api.py b/tests/test_api.py index 1a8b59e2..e4b8f87e 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -71,10 +71,20 @@ class MockRestApiServer(RestApiServer): class TestRestApiHandler(unittest.TestCase): def test_do_GET(self): - MockRestApiServer(RestApiHandler, b'GET /master') MockRestApiServer(RestApiHandler, b'GET /replica') + with patch.object(RestApiHandler, 'get_postgresql_status', Mock(return_value={})): + MockRestApiServer(RestApiHandler, b'GET /replica') + with patch.object(RestApiHandler, 'get_postgresql_status', Mock(return_value={'role': 'master'})): + MockRestApiServer(RestApiHandler, b'GET /replica') + MockRestApiServer(RestApiHandler, b'GET /master') + MockPatroni.dcs.cluster.leader.name = MockPostgresql.name + MockRestApiServer(RestApiHandler, b'GET /replica') + MockPatroni.dcs.cluster = None + with patch.object(RestApiHandler, 'get_postgresql_status', Mock(return_value={'role': 'master'})): + MockRestApiServer(RestApiHandler, b'GET /master') with patch.object(MockHa, 'restart_scheduled', Mock(return_value=True)): MockRestApiServer(RestApiHandler, b'GET /master') + MockRestApiServer(RestApiHandler, b'GET /master') def test_do_GET_patroni(self): MockRestApiServer(RestApiHandler, b'GET /patroni')