def test_queued_job_counts(self, mock_pyhorn):
controller = MatterhornController('mh.example.edu')
controller.client = MagicMock(spec_set=MHClient)
controller.client.workflows.return_value = [
Mock(operations=[
Mock(id="foo",
state="RUNNING",
job=Mock(children=[
Mock(status="RUNNING"),
Mock(status="QUEUED"),
])),
Mock(id="foo",
state="RUNNING",
job=Mock(children=[
Mock(status="RUNNING"),
Mock(status="QUEUED")
])),
]),
Mock(operations=[
Mock(id="foo",
state="RUNNING",
job=Mock(children=[
Mock(status="QUEUED"),
Mock(status="QUEUED")
])),
Mock(id="bar",
state="RUNNING",
job=Mock(children=[
Mock(status="QUEUED"),
Mock(status="QUEUED")
])),
Mock(id="foo",
state="INSTANTIATED")
]),
Mock(operations=[
Mock(id="foo",
state="INSTANTIATED"),
Mock(id="foo",
state="RUNNING",
job=Mock(children=[
Mock(status="RUNNING"),
Mock(status="RUNNING")
])),
Mock(id="bar",
state="WAITING",
job=Mock(children=[
Mock(status="QUEUED"),
Mock(status="QUEUED")
])),
Mock(id="baz",
state="RUNNING",
job=Mock(children=[
Mock(status="RUNNING"),
Mock(status="QUEUED")
]))
])
]
self.assertEqual(controller.queued_job_count(), 9)
self.assertEqual(controller.queued_job_count(operation_types=["foo"]), 4)
self.assertEqual(controller.queued_job_count(operation_types=["bar"]), 4)
self.assertEqual(controller.queued_job_count(operation_types=["foo","bar"]), 8)
self.assertEqual(controller.queued_job_count(operation_types=["foo","bar","baz"]), 9)
