def main():
logging.info("taskflow receiver is running")
# get redis data
redisdb = RedisDB()
while True:
data = redisdb.pop_run_queue()
if data is not None:
message_process(int(data))
time.sleep(1)
class TestRedisDB(unittest.TestCase):
def setUp(self) -> None:
self.db = RedisDB()
def test_run_queue(self):
instance = 1
self.db.push_run_queue(instance)
pop_value = self.db.pop_run_queue()
self.assertEqual(int(pop_value), instance)
def test_check_queue(self):
instance = 1
times = 1
interval = 5
self.db.set_check(instance, times, interval)
data = self.db.get_check_times(instance)
self.assertEqual(int(data), times)
time.sleep(10)
data = self.db.fetch_check_queue()
self.assertTrue(len(data) > 0)
print('test check queue data:', type(data[0]), data[0])
self.db.del_check_queue(instance)
self.db.del_check(instance)