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)
Beispiel #2
0
    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)