def test_squeue_server_list_instancetype_constraint(self, mock_squeue):
mock_squeue.return_value = """1|1024|0|(Resources)|zzzzz-dz642-zzzzzzzzzzzzzzy|instancetype=z2.test|1234567890\n"""
super(JobQueueMonitorActorTestCase, self).build_monitor(
jobqueue.ServerCalculator([(testutil.MockSize(n), {
'cores': n,
'ram': n * 1024,
'scratch': n
}) for n in range(1, 3)]), True, True)
self.monitor.subscribe(self.subscriber).get(self.TIMEOUT)
self.stop_proxy(self.monitor)
self.subscriber.assert_called_with([testutil.MockSize(2)])
def test_squeue_server_list(self, mock_squeue):
mock_squeue.return_value = """1 0 0 Resources zzzzz-zzzzz-zzzzzzzzzzzzzzy
2 0 0 Resources zzzzz-zzzzz-zzzzzzzzzzzzzzz
"""
super(JobQueueMonitorActorTestCase, self).build_monitor(
jobqueue.ServerCalculator([(testutil.MockSize(n), {
'cores': n,
'ram': n,
'scratch': n
}) for n in range(1, 3)]))
self.monitor.subscribe(self.subscriber).get(self.TIMEOUT)
self.stop_proxy(self.monitor)
self.subscriber.assert_called_with(
[testutil.MockSize(1), testutil.MockSize(2)])
def test_squeue_server_list_suffix(self, mock_squeue):
mock_squeue.return_value = """1|1024M|0|(ReqNodeNotAvail, UnavailableNodes:compute123)|zzzzz-dz642-zzzzzzzzzzzzzzy|(null)|1234567890
1|2G|0|(ReqNodeNotAvail)|zzzzz-dz642-zzzzzzzzzzzzzzz|(null)|1234567890
"""
super(JobQueueMonitorActorTestCase, self).build_monitor(
jobqueue.ServerCalculator([(testutil.MockSize(n), {
'cores': n,
'ram': n * 1024,
'scratch': n
}) for n in range(1, 3)]), True, True)
self.monitor.subscribe(self.subscriber).get(self.TIMEOUT)
self.stop_proxy(self.monitor)
self.subscriber.assert_called_with(
[testutil.MockSize(1), testutil.MockSize(2)])
def make_calculator(self, factors, **kwargs):
return jobqueue.ServerCalculator(
[(testutil.MockSize(n), {'cores': n}) for n in factors], **kwargs)
