Skip to content

Commit e407a68

Browse files
committed
Making regression test put()'s transactional.
This is to address flaky test failures. See #535 for more discussion.
1 parent 1bfa469 commit e407a68

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

regression/datastore.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,8 @@ def _get_post(self, id_or_name=None, post_content=None):
7777

7878
def _generic_test_post(self, name=None, key_id=None):
7979
entity = self._get_post(id_or_name=(name or key_id))
80-
datastore.put([entity])
80+
with datastore.Transaction():
81+
datastore.put([entity])
8182

8283
# Register entity to be deleted.
8384
self.case_entities_to_delete.append(entity)
@@ -146,7 +147,8 @@ def test_save_key_self_reference(self):
146147
entity['fullName'] = u'Full name'
147148
entity['linkedTo'] = key # Self reference.
148149

149-
datastore.put([entity])
150+
with datastore.Transaction():
151+
datastore.put([entity])
150152
self.case_entities_to_delete.append(entity)
151153

152154
query = datastore.Query(kind='Person')

0 commit comments

Comments
 (0)