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'})