def test_store_context(self): from furious.extras.appengine.ndb_persistence import store_context from furious.extras.appengine.ndb_persistence import load_context from furious.extras.appengine.ndb_persistence import ContextPersist from furious.context import Context ctx = Context() ctx.add('test', args=[1, 2]) ctx_dict = ctx.to_dict() store_context(ctx.id, ctx_dict) ctx_persisted = ContextPersist.get_by_id(ctx.id) self.assertIsNotNone(ctx_persisted) reloaded_ctx = load_context(ctx.id) self.assertEqual(reloaded_ctx, ctx_dict)
def test_save_context(self): """Ensure the passed in context gets serialized and set on the saved FuriousContext entity. """ _id = "contextid" context = Context(id=_id) result = store_context(context) self.assertEqual(result.id(), _id) loaded_context = FuriousContext.from_id(result.id()) self.assertEqual(context.to_dict(), loaded_context.to_dict())