class TestRedisInterface(unittest.TestCase):
def setUp(self):
os.environ["redis_log_keyname_key"] = "briefcam"
os.environ["total_job_enqueued_count_redis_name_key"] = "enqueue"
os.environ["total_job_dequeued_count_redis_name_key"] = "dequeue"
self.dirname = os.path.dirname(os.path.realpath(__file__))
self.create_test_docker_container()
self.redisInterface = RedisInterface()
def test_redis_interface(self):
self.assertEqual(self.redisInterface.get_current_enqueue_count(), -1)
self.assertEqual(self.redisInterface.get_current_dequeue_count(), -1)
self.redisInterface.increment_enqueue_count()
self.assertEqual(
self.redisInterface.get_current_enqueue_count().decode('utf8'),
'1')
self.redisInterface.increment_dequeue_count()
self.assertEqual(
self.redisInterface.get_current_enqueue_count().decode('utf8'),
'1')
def create_test_docker_container(self):
completedProcess = subprocess.run([
"docker-compose", "-f", "{}/docker-compose_redis.yml".format(
self.dirname), "up", "-d"
],
stdout=subprocess.PIPE)
self.assertIsNotNone(completedProcess)
self.assertIsNotNone(completedProcess.stdout)
# time.sleep(120)
def delete_test_docker_container(self):
completedProcess = subprocess.run([
"docker-compose", "-f", "{}/docker-compose_redis.yml".format(
self.dirname), "down"
],
stdout=subprocess.PIPE)
self.assertIsNotNone(completedProcess)
self.assertIsNotNone(completedProcess.stdout)
def tearDown(self):
self.delete_test_docker_container()
