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)