def test_store_is_called_with_correct_args(self): store = MagicMock() store.delete = MagicMock() flag = FeatureFlag(self.name, store) self.store.create(self.name) flag.destroy() store.delete.assert_called_once_with(self.name)
def test_client_is_called_with_correct_args(self): client = MagicMock() flag = FeatureFlag(self.name, client) flag.destroy() client.destroy.assert_called_once_with(self.name)