def test_leadership_mock(self):
self.leadership_data["foo"] = "bar"
self.assertEqual(client._get_pgsql_leader_data(), self.leadership_data)
self.assertIsNot(client._get_pgsql_leader_data(), self.leadership_data)
client._set_pgsql_leader_data({"one": "two"})
self.assertEqual(client._get_pgsql_leader_data(), {
"foo": "bar",
"one": "two"
})
client._set_pgsql_leader_data({"foo": "baz"})
self.assertEqual(client._get_pgsql_leader_data(), {
"foo": "baz",
"one": "two"
})
self.assertEqual(self.leadership_data, {"foo": "baz", "one": "two"})
def test_get_pgsql_leader_data_missing(self, leader_get):
leader_get.return_value = None
self.assertEqual(client._get_pgsql_leader_data(), {})
leader_get.assert_called_once_with(client.LEADER_KEY)
def test_get_pgsql_leader_data(self, leader_get):
expected = {2: {"database": "testus"}}
leader_get.return_value = yaml.safe_dump(expected)
self.assertEqual(client._get_pgsql_leader_data(), expected)
leader_get.assert_called_once_with(client.LEADER_KEY)