def test_no_data_dump(self): self.assertFalse(cscollector.resources_producer(testrun=True)) r = cscollector.setup_redis_connection() redis_resources = r.get("condor-resources") self.assertEqual(redis_resources, None)
def test_empty_queue(self): #will return true if a condor command is run, false otherwise. self.assertFalse(cscollector.collector_command_consumer(testrun=True)) #Double check to make sure the queue was emptied r = cscollector.setup_redis_connection() self.assertIsNone(r.lpop("collector_commands"))
def test_no_machine_name(self): r = cscollector.setup_redis_connection() r.rpush("collector_commands", '{"command": "condor_off"}') #will return true if a condor command is run, false otherwise. self.assertFalse(cscollector.collector_command_consumer(testrun=True)) #Double check to make sure the queue was emptied self.assertIsNone(r.lpop("collector_commands"))
def test_data_dump(self): self.assertTrue( cscollector.resources_producer( testrun=True, testfile=cloud_sched_root_dir + "condor_data_collectors/tests/test_condor_resources.txt")) r = cscollector.setup_redis_connection() redis_resources = r.get("condor-resources") redis_resources = json.loads(redis_resources) res_file = open( cloud_sched_root_dir + "condor_data_collectors/tests/test_condor_resources.txt", 'r') condor_resources = res_file.read()[0:-1] #strip newline condor_resources = json.loads(condor_resources) self.assertEqual(redis_resources, condor_resources)
def setUp(self): r = cscollector.setup_redis_connection() r.flushall()