def test_tmgr_mock_tmgr():
# FIXME: what is really tested here?
res_dict = {'resource': 'local.localhost',
'walltime': 40,
'cpus' : 20,
'project' : 'Random'}
rmgr = MockRmgr(resource_desc=res_dict, sid='test.0018')
tmgr = MockTmgr(sid='test.0019',
pending_queue=['pendingq-1'],
completed_queue=['completedq-1'],
rmgr=rmgr,
mq_hostname=hostname,
port=port)
tmgr.start_manager()
proc = mp.Process(target=func_for_mock_tmgr_test,
args=(hostname, port, tmgr._pending_queue[0],
tmgr._completed_queue[0]))
proc.start()
proc.join()
tmgr.terminate_manager()
def test_rmgr_mock_initialization(d):
try:
import glob
import shutil
import os
home = os.environ.get('HOME', '/home')
test_fold = glob.glob('%s/.radical/utils/test.*' % home)
for f in test_fold:
shutil.rmtree(f)
except:
pass
rmgr = MockRmgr(resource_desc=d, sid='test.0000')
assert rmgr._resource_desc == d
assert rmgr._sid == 'test.0000'
assert rmgr._rts == 'mock'
assert rmgr._resource == None
assert rmgr._walltime == None
assert rmgr._cpus == 1
assert rmgr._gpus == 0
assert rmgr._project == None
assert rmgr._access_schema == None
assert rmgr._queue == None
assert rmgr._validated == False
assert rmgr._uid == 'resource_manager.0000'
assert rmgr._logger
assert rmgr._prof
# Shared data list
assert isinstance(rmgr.shared_data, list)
def test_tmgr_mock_initialization(s, l, i):
try:
home = os.environ.get('HOME', '/home')
folder = glob.glob('%s/.radical/utils/test.*' % home)
for f in folder:
shutil.rmtree(f)
except:
pass
sid = 'test.0010'
rmgr = MockRmgr(resource_desc={}, sid=sid)
tmgr = MockTmgr(sid=sid,
pending_queue=['pending'],
completed_queue=['completed'],
rmgr=rmgr,
mq_hostname=hostname,
port=port)
assert tmgr._uid == 'task_manager.0000'
assert tmgr._pending_queue == ['pending']
assert tmgr._completed_queue == ['completed']
assert tmgr._hostname == hostname
assert tmgr._port == port
assert tmgr._rts == 'mock'
assert tmgr._log
assert tmgr._prof
assert tmgr._hb_request_q == '%s-hb-request' % sid
assert tmgr._hb_response_q == '%s-hb-response' % sid
assert tmgr._tmgr_process is None
assert tmgr._hb_thread is None
assert tmgr._rmq_ping_interval == 10
def test_tmgr_mock_tmgr():
res_dict = {
'resource': 'local.localhost',
'walltime': 40,
'cpus': 20,
'project': 'Random'
}
os.environ['RADICAL_PILOT_DBURL'] = 'mlab-url'
os.environ['ENTK_HB_INTERVAL'] = '30'
rmgr = MockRmgr(resource_desc=res_dict, sid='test.0000')
tmgr = MockTmgr(sid='test.0000',
pending_queue=['pendingq-1'],
completed_queue=['completedq-1'],
rmgr=rmgr,
mq_hostname=hostname,
port=port)
tmgr.start_manager()
proc = Process(target=func_for_mock_tmgr_test,
args=(hostname, port, tmgr._pending_queue[0],
tmgr._completed_queue[0]))
proc.start()
proc.join()
tmgr.terminate_manager()
def test_tmgr_mock_tmgr():
# FIXME: what is really tested here?
res_dict = {
'resource': 'local.localhost',
'walltime': 40,
'cpus': 20,
'project': 'Random'
}
credentials = pika.PlainCredentials(username, password)
rmq_conn_params = pika.ConnectionParameters(host=hostname,
port=port,
credentials=credentials)
rmgr = MockRmgr(resource_desc=res_dict, sid='test.0018')
tmgr = MockTmgr(sid='test.0019',
pending_queue=['pendingq-1'],
completed_queue=['completedq-1'],
rmgr=rmgr,
rmq_conn_params=rmq_conn_params)
tmgr.start_manager()
proc = mp.Process(target=func_for_mock_tmgr_test,
args=(hostname, port, username, password,
tmgr._pending_queue[0], tmgr._completed_queue[0]))
proc.start()
proc.join()
tmgr.terminate_manager()
def test_tmgr_mock_initialization():
credentials = pika.PlainCredentials(username, password)
rmq_conn_params = pika.ConnectionParameters(host=hostname,
port=port,
credentials=credentials)
sid = 'test.0010'
rmgr = MockRmgr(resource_desc={}, sid=sid)
tmgr = MockTmgr(sid=sid,
pending_queue=['pending'],
completed_queue=['completed'],
rmgr=rmgr,
rmq_conn_params=rmq_conn_params)
assert tmgr._pending_queue == ['pending']
assert tmgr._completed_queue == ['completed']
assert tmgr._rmq_conn_params == rmq_conn_params
assert tmgr._rts == 'mock'
assert tmgr._log
assert tmgr._prof
assert tmgr._hb_request_q == '%s-hb-request' % sid
assert tmgr._hb_response_q == '%s-hb-response' % sid
assert tmgr._tmgr_process is None
assert tmgr._hb_thread is None
assert tmgr._rmq_ping_interval == 10
def test_rmgr_mock_methods():
rmgr = MockRmgr(resource_desc={}, sid='test.0000')
assert not rmgr.get_resource_allocation_state()
assert not rmgr.get_completed_states()
assert rmgr._validate_resource_desc()
assert not rmgr._populate()
assert not rmgr._submit_resource_request()
assert not rmgr._terminate_resource_request()
def test_rmgr_mock_initialization(d):
rmgr = MockRmgr(resource_desc=d, sid='test.0016')
assert rmgr._resource_desc == d
assert rmgr._sid == 'test.0016'
assert rmgr._rts == 'mock'
assert rmgr._resource is None
assert rmgr._walltime is None
assert rmgr._cpus == 1
assert rmgr._gpus == 0
assert rmgr._project is None
assert rmgr._access_schema is None
assert rmgr._queue is None
assert rmgr._validated is False
assert rmgr._logger
assert rmgr._prof
assert isinstance(rmgr.shared_data, list)
