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