Skip to content

Commit 2908d6f

Browse files
committed
Updating api.delete() test to rely on new Batch behavior.
Behavior of test_batch._Connection changed in #524.
1 parent ac3f9b1 commit 2908d6f

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

gcloud/datastore/test_api.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -318,9 +318,11 @@ def test_delete_existing_batch(self):
318318
result = self._callFUT([key], connection=connection)
319319

320320
self.assertEqual(result, None)
321-
self.assertEqual(
322-
connection._deleted,
323-
[(_DATASET, [key._key], CURR_BATCH.mutation)])
321+
self.assertEqual(len(CURR_BATCH.mutation.insert_auto_id), 0)
322+
self.assertEqual(len(CURR_BATCH.mutation.upsert), 0)
323+
deletes = list(CURR_BATCH.mutation.delete)
324+
self.assertEqual(len(deletes), 1)
325+
self.assertEqual(deletes[0], key._key)
324326

325327
def test_delete_implicit_connection(self):
326328
from gcloud._testing import _Monkey
@@ -346,9 +348,11 @@ def test_delete_implicit_connection(self):
346348
result = self._callFUT([key])
347349

348350
self.assertEqual(result, None)
349-
self.assertEqual(
350-
connection._deleted,
351-
[(_DATASET, [key._key], CURR_BATCH.mutation)])
351+
self.assertEqual(len(CURR_BATCH.mutation.insert_auto_id), 0)
352+
self.assertEqual(len(CURR_BATCH.mutation.upsert), 0)
353+
deletes = list(CURR_BATCH.mutation.delete)
354+
self.assertEqual(len(deletes), 1)
355+
self.assertEqual(deletes[0], key._key)
352356

353357
def test_delete_no_keys(self):
354358
from gcloud.datastore import _implicit_environ

0 commit comments

Comments
 (0)