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()