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