@@ -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