def setUp(self):
# Setup mock GAE Data Store
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
self.user_datastore = UserAppengineDataStore()
class UserDataStoreFactory():
'''
Factory for getting a UserDataStore implementation
'''
storage_implementations = {'appengine': UserAppengineDataStore()}
def __init__(self, storage_impl='appengine'):
'''
Constructor
'''
self.storage = self.storage_implementations[storage_impl]
def set_storage(self, user_storage):
self.storage = user_storage
def get_storage(self):
return self.storage
