Ejemplo n.º 1
0
    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)]
Ejemplo n.º 2
0
 def setUp(self):
     self.prefix = 'test'
     self.redis_store = RedisStore(store=FakeStrictRedis, name=self.prefix)
Ejemplo n.º 3
0
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)
Ejemplo n.º 4
0
 def setUp(self):
     self.prefix = 'test'
     self.redis_store = RedisStore(store=FakeStrictRedis, name=self.prefix)
Ejemplo n.º 5
0
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)