def ensure_context():
context = get_context()
context.memcache_enabled = getattr(context, "memcache_enabled", True)
context.context_enabled = getattr(context, "context_enabled", True)
context.stack = context.stack if hasattr(context,
"stack") else ContextStack()
def test_property_deletion(self):
stack = ContextStack()
entity = FakeEntity({"field1": "one", "field2": "two"})
stack.top.cache_entity(["entity"], entity, caching.CachingSituation.DATASTORE_PUT)
stack.push() # Enter transaction
entity["field1"] = "oneone"
del entity["field2"]
stack.top.cache_entity(["entity"], entity, caching.CachingSituation.DATASTORE_PUT)
stack.pop(apply_staged=True, clear_staged=True)
self.assertEqual({"field1": "oneone"}, stack.top.cache["entity"])
def test_push_pop(self):
stack = ContextStack()
self.assertEqual({}, stack.top.cache)
entity = FakeEntity({"bananas": 1})
stack.top.cache_entity(["bananas:1"], entity,
caching.CachingSituation.DATASTORE_PUT)
self.assertEqual({"bananas": 1}, stack.top.cache.values()[0])
stack.push()
self.assertEqual([], stack.top.cache.values())
self.assertEqual(2, stack.size)
stack.push()
stack.top.cache_entity(["apples:2"], entity,
caching.CachingSituation.DATASTORE_PUT)
self.assertItemsEqual(["apples:2"], stack.top.cache.keys())
stack.pop()
self.assertItemsEqual([], stack.top.cache.keys())
self.assertEqual(2, stack.size)
self.assertEqual(1, stack.staged_count)
updated = FakeEntity({"bananas": 3})
stack.top.cache_entity(["bananas:1"], updated,
caching.CachingSituation.DATASTORE_PUT)
stack.pop(apply_staged=True, clear_staged=True)
self.assertEqual(1, stack.size)
self.assertEqual({"bananas": 3}, stack.top.cache["bananas:1"])
self.assertEqual(0, stack.staged_count)
def test_push_pop(self):
stack = ContextStack()
self.assertEqual({}, stack.top.cache)
entity = FakeEntity({"bananas": 1})
stack.top.cache_entity(["bananas:1"], entity, caching.CachingSituation.DATASTORE_PUT)
self.assertEqual({"bananas": 1}, stack.top.cache.values()[0])
stack.push()
self.assertEqual([], stack.top.cache.values())
self.assertEqual(2, stack.size)
stack.push()
stack.top.cache_entity(["apples:2"], entity, caching.CachingSituation.DATASTORE_PUT)
self.assertItemsEqual(["apples:2"], stack.top.cache.keys())
stack.pop()
self.assertItemsEqual([], stack.top.cache.keys())
self.assertEqual(2, stack.size)
self.assertEqual(1, stack.staged_count)
updated = FakeEntity({"bananas": 3})
stack.top.cache_entity(["bananas:1"], updated, caching.CachingSituation.DATASTORE_PUT)
stack.pop(apply_staged=True, clear_staged=True)
self.assertEqual(1, stack.size)
self.assertEqual({"bananas": 3}, stack.top.cache["bananas:1"])
self.assertEqual(0, stack.staged_count)
def reset_context(keep_disabled_flags=False, *args, **kwargs):
"""
Called at the beginning and end of each request, resets the thread local
context. If you pass keep_disabled_flags=True the memcache_enabled and context_enabled
flags will be preserved, this is really only useful for testing.
"""
context = get_context()
memcache_enabled = getattr(context, "memcache_enabled", True)
context_enabled = getattr(context, "context_enabled", True)
context.memcache_enabled = True
context.context_enabled = True
context.stack = ContextStack()
if keep_disabled_flags:
context.memcache_enabled = memcache_enabled
context.context_enabled = context_enabled
def test_property_deletion(self):
stack = ContextStack()
entity = FakeEntity({"field1": "one", "field2": "two"})
stack.top.cache_entity(["entity"], entity,
caching.CachingSituation.DATASTORE_PUT)
stack.push() # Enter transaction
entity["field1"] = "oneone"
del entity["field2"]
stack.top.cache_entity(["entity"], entity,
caching.CachingSituation.DATASTORE_PUT)
stack.pop(apply_staged=True, clear_staged=True)
self.assertEqual({"field1": "oneone"}, stack.top.cache["entity"])
