def test_no_workers_raised_when_there_are_no_workers(self, mock_reserved_resource, mock_worker):
mock_worker.objects.return_value = []
find = mock_reserved_resource.objects
find.return_value = [{'worker_name': 'a'}, {'worker_name': 'b'}]
try:
tasks._get_unreserved_worker()
except NoWorkers:
pass
else:
self.fail("NoWorkers() Exception should have been raised.")
def test_no_workers_raised_when_there_are_no_workers(self, mock_reserved_resource, mock_worker):
mock_worker.objects.return_value = []
find = mock_reserved_resource.get_collection.return_value.find
find.return_value = [{'worker_name': 'a'}, {'worker_name': 'b'}]
try:
tasks._get_unreserved_worker()
except NoWorkers:
pass
else:
self.fail("NoWorkers() Exception should have been raised.")
def test_reserved_resources_queried_correctly(self, mock_reserved_resource):
find = mock_reserved_resource.objects.all
find.return_value = [{'worker_name': 'a'}, {'worker_name': 'b'}]
try:
tasks._get_unreserved_worker()
except NoWorkers:
pass
else:
self.fail("NoWorkers() Exception should have been raised.")
find.assert_called_once_with()
def test_worker_returned_when_one_worker_is_not_reserved(self, mock_reserved_resource,
mock_worker_objects):
mock_worker_objects.return_value = [{'name': 'a'}, {'name': 'b'}]
find = mock_reserved_resource.get_collection.return_value.find
find.return_value = [{'worker_name': 'a'}]
result = tasks._get_unreserved_worker()
self.assertEqual(result, {'name': 'b'})
def test_worker_returned_when_one_worker_is_not_reserved(self, mock_reserved_resource,
mock_worker_objects):
get_online = mock_worker_objects.get_online
get_online.return_value = [{'name': 'a'}, {'name': 'b'}]
mock_reserved_resource.objects.all.return_value = [{'worker_name': 'a'}]
result = tasks._get_unreserved_worker()
self.assertEqual(result, {'name': 'b'})
def test_worker_returned_when_one_worker_is_not_reserved(
self, mock_reserved_resource, mock_worker_objects):
mock_worker_objects.return_value = [{'name': 'a'}, {'name': 'b'}]
find = mock_reserved_resource.get_collection.return_value.find
find.return_value = [{'worker_name': 'a'}]
result = tasks._get_unreserved_worker()
self.assertEqual(result, {'name': 'b'})
def test_worker_returned_when_one_worker_is_not_reserved(
self, mock_reserved_resource, mock_worker_objects):
get_online = mock_worker_objects.get_online
get_online.return_value = [{'name': 'a'}, {'name': 'b'}]
mock_reserved_resource.objects.all.return_value = [{
'worker_name': 'a'
}]
result = tasks._get_unreserved_worker()
self.assertEqual(result, {'name': 'b'})
