From f53c968d8b17fc526c0883af70dfe99797dbb1ca Mon Sep 17 00:00:00 2001
From: Alexander Kukushkin <alexander.kukushkin@zalando.de>
Date: Tue, 20 Oct 2015 14:36:49 +0200
Subject: [PATCH] Improve tests

---
 tests/test_api.py | 2 ++
 1 file changed, 2 insertions(+)

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)