class ApiTestCase(unittest.TestCase):
def setUp(self):
self._app = app.test_client()
self._conn = RedisClient()
def tearDown(self):
self._conn.flush()
def test_get(self):
self._conn.put('aaa')
self._conn.put('bbb')
r = self._app.get('/get')
assert 'aaa' in str(r.data)
r = self._app.get('/get')
assert 'bbb' in str(r.data)
def test_count(self):
self._conn.put('aaa')
self._conn.put('bbb')
r = self._app.get('/count')
assert '2' in str(r.data)
self._conn.put('ccc')
self._conn.put('ddd')
r = self._app.get('/count')
assert '4' in str(r.data)
proxy = self._conn.pop()
r = self._app.get('/count')
assert '3' in str(r.data)
def main():
# db = redis.Redis(host=HOST, port=PORT)
# db.flushall()
conn = RedisClient()
conn.flush()
s = Schedule()
s.run()
app.run(host=FLASK_HOST, port=FLASK_PORT, debug=True)
class RedisClientTestCase(unittest.TestCase):
def setUp(self):
self._conn = RedisClient()
def tearDown(self):
self._conn.flush()
def test_put_and_pop(self):
self._conn.put("label")
assert self._conn.pop() == "label"
def test_put_many(self):
self._conn.put_many(['a', 'b'])
assert self._conn.pop() == "a"
assert self._conn.pop() == "b"
def test_len(self):
self._conn.put_many(['a', 'b', 'c'])
assert self._conn.queue_len == 3
def test_get(self):
self._conn.put_many(['a', 'b', 'c', 'd'])
_ = self._conn.get(2)
assert self._conn.queue_len == 2
