def test_shard(self):
self._add_cleanup(os.path.join('/tmp', 'tests2.db'))
# make shure we do shard
testsdir = os.path.dirname(__file__)
conf = os.path.join(testsdir, 'tests2.ini')
appdir, config, storage, auth = initenv(conf)
res = storage._engine.execute('select count(*) from wbo1')
self.assertEqual(res.fetchall()[0][0], 0)
# doing a few things on the DB
storage.set_user(_UID, email='*****@*****.**')
storage.set_collection(_UID, 'col1')
id1 = '{ec1b7457-003a-45a9-bf1c-c34e37225ad7}'
id2 = '{339f52e1-deed-497c-837a-1ab25a655e37}'
storage.set_item(_UID, 'col1', id1, payload=_PLD)
storage.set_item(_UID, 'col1', id2, payload=_PLD * 89)
self.assertEquals(len(storage.get_items(_UID, 'col1')), 2)
# now making sure we did that in the right table
table = get_wbo_table_name(_UID)
self.assertEqual(table, 'wbo1')
res = storage._engine.execute('select count(*) from wbo1')
self.assertEqual(res.fetchall()[0][0], 2)
def test_shard(self):
# make shure we do shard
testsdir = os.path.dirname(__file__)
conf = os.path.join(testsdir, 'tests2.ini')
appdir, config, storage, auth = initenv(conf)
self._add_cleanup(cleanupenv, conf)
res = storage._engine.execute('select count(*) from wbo1')
self.assertEqual(res.fetchall()[0][0], 0)
# doing a few things on the DB
storage.set_user(_UID, email='*****@*****.**')
storage.set_collection(_UID, 'col1')
id1 = '{ec1b7457-003a-45a9-bf1c-c34e37225ad7}'
id2 = '{339f52e1-deed-497c-837a-1ab25a655e37}'
storage.set_item(_UID, 'col1', id1, payload=_PLD)
storage.set_item(_UID, 'col1', id2, payload=_PLD * 89)
self.assertEquals(len(storage.get_items(_UID, 'col1')), 2)
# now making sure we did that in the right table
table = get_wbo_table_name(_UID)
self.assertEqual(table, 'wbo1')
res = storage._engine.execute('select count(*) from wbo1')
self.assertEqual(res.fetchall()[0][0], 2)
def _get_wbo_table_name(self, user_id):
if self.shard:
return get_wbo_table_name(user_id)
return 'wbo'
def _get_wbo_table_name(self, user_id):
if self.shard:
return get_wbo_table_name(user_id)
return 'wbo'
