diff --git a/tests/test_api.py b/tests/test_api.py index ae25964d..72d5c2b0 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -135,6 +135,8 @@ class TestRestApiHandler(unittest.TestCase): MockRestApiServer(RestApiHandler, request) with patch.object(MockPatroni, 'dcs') as d: cluster = d.get_cluster.return_value + cluster.leader.name = 'postgresql0' + MockRestApiServer(RestApiHandler, request) cluster.leader.name = 'postgresql1' cluster.failover = None MockRestApiServer(RestApiHandler, request)