diff --git a/tests/test_api.py b/tests/test_api.py index 013e5065..4b24704d 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -16,6 +16,8 @@ class MockPostgresql(Mock): name = 'test' state = 'running' role = 'master' + server_version = '999999' + scope = 'dummy' def connection(self): return psycopg2_connect() @@ -51,6 +53,7 @@ class MockPatroni: ha = MockHa() dcs = Mock() tags = {} + version = '0.00' class MockRequest: diff --git a/tests/test_etcd.py b/tests/test_etcd.py index e9a334c8..443dc2d8 100644 --- a/tests/test_etcd.py +++ b/tests/test_etcd.py @@ -40,6 +40,9 @@ class MockResponse: class MockPostgresql(Mock): + server_version = '999999' + scope = 'dummy' + def last_operation(self): return '0' diff --git a/tests/test_ha.py b/tests/test_ha.py index d9a408a4..9ff6547d 100644 --- a/tests/test_ha.py +++ b/tests/test_ha.py @@ -48,6 +48,8 @@ class MockPostgresql(Mock): role = 'replica' state = 'running' connection_string = 'postgres://foo@bar/postgres' + server_version = '999999' + scope = 'dummy' def is_healthy(self): return True