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)