def setUp(self):
self.tmpdir = mkdtemp(prefix="kittystore-testing-")
settings = SettingsModule()
settings.KITTYSTORE_SEARCH_INDEX = self.tmpdir
search_index = _get_search_index(settings)
self.store = get_sa_store(settings, search_index=search_index, auto_create=True)
search_index.upgrade(self.store)
def setUp(self):
self.store = get_sa_store(SettingsModule(), auto_create=True)
self.store.db.cache.get_or_create = Mock()
self.store.db.cache.get_or_create.side_effect = lambda *a, **kw: a[1]()
self.store.db.cache.set = Mock()
# cache.delete() will be called if the cache is invalidated
self.store.db.cache.delete = Mock()
def get_store(settings, debug=None, auto_create=False):
"""Factory for a KittyStore subclass"""
_check_settings(settings)
if debug is None:
debug = getattr(settings, "KITTYSTORE_DEBUG", False)
search_index = _get_search_index(settings)
if getattr(settings, "KITTYSTORE_USE_STORM", False):
from kittystore.storm import get_storm_store
store = get_storm_store(settings, search_index, debug, auto_create)
else:
from kittystore.sa import get_sa_store
store = get_sa_store(settings, search_index, debug, auto_create)
if search_index is not None and search_index.needs_upgrade():
if auto_create:
search_index.upgrade(store)
else:
store.close()
raise SchemaUpgradeNeeded()
register_events()
return store
def create_store(settings, debug=None):
"""Factory for a KittyStore subclass"""
_check_settings(settings)
if debug is None:
debug = getattr(settings, "KITTYSTORE_DEBUG", False)
search_index = _get_search_index(settings)
if getattr(settings, "KITTYSTORE_USE_STORM", False):
from kittystore.storm import get_storm_store, create_storm_db
version = create_storm_db(settings, debug)
store = get_storm_store(settings, search_index, debug)
else:
from kittystore.sa import create_sa_db, get_sa_store
version = create_sa_db(settings, debug)
store = get_sa_store(settings, search_index, debug)
if search_index is not None and search_index.needs_upgrade():
search_index.upgrade(store)
return store, version
def setUp(self):
self.store = get_sa_store(SettingsModule(), auto_create=True)#, debug=True)
def setUp(self):
self.store = get_sa_store(SettingsModule(), auto_create=True, debug=False)
self.user = User(id=uuid.uuid1())
self.store.db.add(self.user)
self.store.db.flush()
