def test_get_jobinfolist(self, locateAll, query):
print ' Now testing with --> test_get_jobinfolist'
test_numjobs = 2
locateAll.return_value = get_schedulers('test-files/schedulers.txt')
query.return_value = get_jobs_scheduled_attributes(test_numjobs)
job_info_list = condor.lrms(MagicMock(condor.lrms)).get_jobinfolist()
lenj = len(job_info_list)
assert lenj == test_numjobs
def test_get_nodeinfolist(self, get_worker_nodes_list, get_schedulers_list, query):
print ' Now testing with --> test_get_nodeinfolist'
get_worker_nodes_list.return_value = get_worker_nodes(
'test-files/workernodes.txt')
get_schedulers_list.return_value = get_schedulers(
'test-files/schedulers.txt')
query.return_value = get_jobs_scheduled_attributes(2)
node_info_list = condor.lrms(MagicMock(condor.lrms)).get_nodeinfolist()
if node_info_list:
result = '[NODE "wn1.condor.vagrant"] state: used, 1/1 (free slots), 457/457 (mem)'
assert str(node_info_list['wn1.condor.vagrant']) == result
else:
print 'No node_info_list'
def test_init_lrms(self):
lrms = condor.lrms('test_ip')
assert lrms._server_ip == 'test_ip'
def test_init_lrms_empty(self):
lrms = condor.lrms()
assert lrms._server_ip == 'htcondoreserver'
