def test_get_and_put_cached(self):
storage = StorageByKeyName(CredentialsModel,
'foo',
'credentials',
cache=memcache)
self.assertEqual(None, storage.get())
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
credmodel = CredentialsModel.get_by_key_name('foo')
self.assertEqual('bar', credmodel.credentials.access_token)
# Now remove the item from the cache.
memcache.delete('foo')
# Check that getting refreshes the cache.
credentials = storage.get()
self.assertEqual('bar', credentials.access_token)
self.assertNotEqual(None, memcache.get('foo'))
# Deleting should clear the cache.
storage.delete()
credentials = storage.get()
self.assertEqual(None, credentials)
self.assertEqual(None, memcache.get('foo'))
def test_get_and_put_simple(self):
storage = StorageByKeyName(CredentialsModel, 'foo', 'credentials')
self.assertEqual(None, storage.get())
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
credmodel = CredentialsModel.get_by_key_name('foo')
self.assertEqual('bar', credmodel.credentials.access_token)
def test_get_and_put_simple(self):
storage = StorageByKeyName(CredentialsModel, "foo", "credentials")
self.assertEqual(None, storage.get())
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
credmodel = CredentialsModel.get_by_key_name("foo")
self.assertEqual("bar", credmodel.credentials.access_token)
def test_get_and_put_mixed_ndb_storage_db_get(self):
# Start empty
storage = StorageByKeyName(CredentialsNDBModel, "foo", "credentials")
self.assertEqual(None, storage.get())
# Set NDB store and refresh to add to storage
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
# Retrieve same key from DB model to confirm mixing works
credmodel = CredentialsModel.get_by_key_name("foo")
self.assertEqual("bar", credmodel.credentials.access_token)
self.assertEqual(self.credentials.to_json(), credmodel.credentials.to_json())
def test_get_and_put_mixed_ndb_storage_db_get(self):
# Start empty
storage = StorageByKeyName(CredentialsNDBModel, 'foo', 'credentials')
self.assertEqual(None, storage.get())
# Set NDB store and refresh to add to storage
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
# Retrieve same key from DB model to confirm mixing works
credmodel = CredentialsModel.get_by_key_name('foo')
self.assertEqual('bar', credmodel.credentials.access_token)
self.assertEqual(self.credentials.to_json(),
credmodel.credentials.to_json())
def test_get_and_put_cached(self):
storage = StorageByKeyName(CredentialsModel, "foo", "credentials", cache=memcache)
self.assertEqual(None, storage.get())
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
credmodel = CredentialsModel.get_by_key_name("foo")
self.assertEqual("bar", credmodel.credentials.access_token)
# Now remove the item from the cache.
memcache.delete("foo")
# Check that getting refreshes the cache.
credentials = storage.get()
self.assertEqual("bar", credentials.access_token)
self.assertNotEqual(None, memcache.get("foo"))
# Deleting should clear the cache.
storage.delete()
credentials = storage.get()
self.assertEqual(None, credentials)
self.assertEqual(None, memcache.get("foo"))
def get(self):
credentials = CredentialsModel.query().fetch
for credential in credentials:
logging.info(credential.key.url_safe())
self.redirect('/')
