def setUp(self): super(BaseTestCase, self).setUp() self.app = create_app(config.Testing) self.app_context = self.app.app_context() self.client = self.app.test_client() self.app_context.push() self.db = db self.db.drop_all() self.db.create_all() self.user = dict(username="******", password="******", first_name="Test", last_name="User", _admin=True) self.document = dict(title="This is a Test Title", body="Body Body Body, likeasomebody") self.tag = {"title": "TAGGY"} self.default_user = User.create(self.user) self.default_document = Document.create(self.document) self.default_document.user = self.default_user self.tag = Tag.create(self.tag) self.tag.user = self.default_user self.default_document.tags.append(self.tag) self.db.session.commit() self.redis_store = RedisStore(store=FakeStrictRedis, name='test') token = jwt.create_token_for_user(self.default_user) self.headers = [('Content-Type', 'application/json'), ('Authorization', 'Bearer %s' % token)]
def setUp(self): self.prefix = 'test' self.redis_store = RedisStore(store=FakeStrictRedis, name=self.prefix)
class RedisStoreTestCase(unittest.TestCase): def setUp(self): self.prefix = 'test' self.redis_store = RedisStore(store=FakeStrictRedis, name=self.prefix) def tearDown(self): self.redis_store.store.flushall() def test_set_method(self): self.redis_store.set('a', 'valuevaluevalue') self.assertEqual( self.redis_store.store.get('{}|a'.format(self.prefix)), 'valuevaluevalue') self.redis_store.set('willexpire', 'value', expire=.1) time.sleep(.2) self.assertIsNone(self.redis_store.get('willexpire')) def test_get_method(self): self.redis_store.store.set('{}|key'.format(self.prefix), 'value') self.assertEqual(self.redis_store.get('key'), 'value') def test_name_set(self): self.assertEqual(self.redis_store.name, self.prefix + '|') def test_delete_method(self): self.redis_store.set('key', 'value') self.assertGreater(len(self.redis_store.store.keys()), 0) self.redis_store.delete('key') self.assertEqual(len(self.redis_store.store.keys()), 0) self.redis_store.set('key1', 'value') self.redis_store.set('key2', 'value') self.redis_store.delete('key1', 'key2') self.assertEqual(len(self.redis_store.store.keys()), 0) def test_keys_method(self): self.redis_store.set('key1', 'value') self.redis_store.set('key2', 'value') keys = ['test|key2', 'test|key1'] self.assertEqual(self.redis_store.keys(), keys)
class RedisStoreTestCase(unittest.TestCase): def setUp(self): self.prefix = 'test' self.redis_store = RedisStore(store=FakeStrictRedis, name=self.prefix) def tearDown(self): self.redis_store.store.flushall() def test_set_method(self): self.redis_store.set('a', 'valuevaluevalue') self.assertEqual( self.redis_store.store.get('{}|a'.format(self.prefix)), 'valuevaluevalue' ) self.redis_store.set('willexpire', 'value', expire=.1) time.sleep(.2) self.assertIsNone(self.redis_store.get('willexpire')) def test_get_method(self): self.redis_store.store.set('{}|key'.format(self.prefix), 'value') self.assertEqual( self.redis_store.get('key'), 'value' ) def test_name_set(self): self.assertEqual(self.redis_store.name, self.prefix+'|') def test_delete_method(self): self.redis_store.set('key', 'value') self.assertGreater(len(self.redis_store.store.keys()), 0) self.redis_store.delete('key') self.assertEqual(len(self.redis_store.store.keys()), 0) self.redis_store.set('key1', 'value') self.redis_store.set('key2', 'value') self.redis_store.delete('key1', 'key2') self.assertEqual(len(self.redis_store.store.keys()), 0) def test_keys_method(self): self.redis_store.set('key1', 'value') self.redis_store.set('key2', 'value') keys = ['test|key2', 'test|key1'] self.assertEqual(self.redis_store.keys(), keys)