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_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():
# 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_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_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