Making regression test put()'s transactional.

This is to address flaky test failures. See googleapis#535 for more discussion.
dhermes · Jan 22, 2015 · e407a68 · e407a68
1 parent 1bfa469
commit e407a68
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/regression/datastore.py b/regression/datastore.py
@@ -77,7 +77,8 @@ def _get_post(self, id_or_name=None, post_content=None):
 
     def _generic_test_post(self, name=None, key_id=None):
         entity = self._get_post(id_or_name=(name or key_id))
-        datastore.put([entity])
+        with datastore.Transaction():
+            datastore.put([entity])
 
         # Register entity to be deleted.
         self.case_entities_to_delete.append(entity)
@@ -146,7 +147,8 @@ def test_save_key_self_reference(self):
         entity['fullName'] = u'Full name'
         entity['linkedTo'] = key  # Self reference.
 
-        datastore.put([entity])
+        with datastore.Transaction():
+            datastore.put([entity])
         self.case_entities_to_delete.append(entity)
 
         query = datastore.Query(kind='Person')