def test_store_is_called_with_correct_args(self): self.store.create(self.name) store = MagicMock() store.set = MagicMock() flag = FeatureFlag(self.name, store) flag.disable() store.set.assert_called_once_with(self.name, False)
def test_client_is_called_with_correct_args(self): client = MagicMock() flag = FeatureFlag(self.name, client) flag.disable() client.disable.assert_called_once_with(self.name)