def test_stop_worker(self): flexmock(os).should_receive("system").and_return(None) flexmock(monit_interface).should_receive('stop').and_return(False) flexmock(file_io).should_receive("delete").and_return(None) flexmock(file_io).should_receive("mkdir").and_return(None) flexmock(file_io).should_receive("read")\ .and_return("192.168.0.1\n129.168.0.2\n184.48.65.89") dtq = DistributedTaskQueue() json_request = {'app_id': 'test_app'} response = dtq.stop_worker(json.dumps(json_request)) self.assertTrue(json.loads(response)['error']) flexmock(monit_interface).should_receive('stop').and_return(True) response = dtq.stop_worker(json.dumps(json_request)) self.assertFalse(json.loads(response)['error'])
def test_query_tasks(self): mock_file_io() flexmock(DatastoreProxy).should_receive('__init__') db_access = flexmock() zk_client = flexmock() flexmock(GlobalQueueManager).should_receive('__new__').\ and_return(flexmock()) dtq = DistributedTaskQueue(db_access, zk_client)
def test_stop_worker(self): flexmock(os).should_receive("system").and_return(None) flexmock(monit_interface).should_receive('stop').and_return(False) flexmock(file_io).should_receive("delete").and_return(None) flexmock(file_io).should_receive("mkdir").and_return(None) flexmock(file_io).should_receive("read")\ .and_return("192.168.0.1\n129.168.0.2\n184.48.65.89") db_access = flexmock() dtq = DistributedTaskQueue(db_access) json_request = {'app_id': 'test_app'} response = dtq.stop_worker(json.dumps(json_request)) self.assertTrue(json.loads(response)['error']) flexmock(monit_interface).should_receive('stop').and_return(True) response = dtq.stop_worker(json.dumps(json_request)) self.assertFalse(json.loads(response)['error'])
def test_start_worker(self): flexmock(file_io).should_receive("mkdir").and_return(None) flexmock(file_io).should_receive("read")\ .and_return("192.168.0.1\n129.168.0.2\n184.48.65.89") flexmock(monit_app_configuration).should_receive('create_config_file')\ .and_return('') flexmock(monit_interface).should_receive('start').and_return(False) flexmock(TaskQueueConfig).should_receive("load_queues_from_file") flexmock(TaskQueueConfig).should_receive("create_celery_worker_scripts") flexmock(TaskQueueConfig).should_receive("create_celery_file") flexmock(DatastoreProxy).should_receive('__init__') dtq = DistributedTaskQueue() json_request = {} json_request = json.dumps(json_request) self.assertEquals( dtq.start_worker(json_request), json.dumps({'error': True, 'reason': 'Missing app_id tag'})) json_request = "fefwoinfwef=fwf23onr2or3" json_response = dtq.start_worker(json_request) self.assertEquals( json_response, json.dumps({'error': True, 'reason': 'Badly formed JSON'})) json_request = {'app_id':'my-app'} json_request = json.dumps(json_request) assert 'true' in dtq.start_worker(json_request) flexmock(monit_interface).should_receive('start').and_return(True) json_request = {'app_id':'my-app'} json_request = json.dumps(json_request) assert 'false' in dtq.start_worker(json_request)
def test_pause_queue(self): mock_file_io() flexmock(DatastoreProxy).should_receive('__init__') db_access = flexmock() dtq = DistributedTaskQueue(db_access)
def test_update_storage_limit(self): mock_file_io() flexmock(DatastoreProxy).should_receive('__init__') db_access = flexmock() dtq = DistributedTaskQueue(db_access)
def test_delete_group(self): mock_file_io() flexmock(DatastoreProxy).should_receive('__init__') dtq = DistributedTaskQueue()
def test_modify_task_lease(self): mock_file_io() flexmock(DatastoreProxy).should_receive('__init__') dtq = DistributedTaskQueue()
def test_query_and_own_tasks(self): mock_file_io() flexmock(DatastoreProxy).should_receive('__init__') dtq = DistributedTaskQueue()
def test_fetch_queue_stats(self): mock_file_io() flexmock(DatastoreProxy).should_receive('__init__') dtq = DistributedTaskQueue()