def test_existing_reservation_correctly_found(self, mock_reserved_resource, mock_worker_objects): get_collection = mock_reserved_resource.get_collection tasks.get_worker_for_reservation('resource1') get_collection.assert_called_once_with() get_collection.return_value.find_one.assert_called_once_with( {'resource_id': 'resource1'})
def test_no_workers_raised_if_no_reservations(self, mock_reserved_resource): find_one = mock_reserved_resource.get_collection.return_value.find_one find_one.return_value = False try: tasks.get_worker_for_reservation('resource1') except NoWorkers: pass else: self.fail("NoWorkers() Exception should have been raised.")
def test_no_workers_raised_if_no_reservations(self, mock_reserved_resource): find_one = mock_reserved_resource.objects.return_value.first find_one.return_value = False try: tasks.get_worker_for_reservation('resource1') except NoWorkers: pass else: self.fail("NoWorkers() Exception should have been raised.")
def test_correct_worker_returned(self, mock_reserved_resource, mock_worker_objects): find_one = mock_reserved_resource.objects.return_value.first find_one.return_value = {'worker_name': 'worker1'} mock_worker_objects.return_value.first.return_value = find_one.return_value result = tasks.get_worker_for_reservation('resource1') self.assertTrue(result is find_one.return_value)
def test_correct_worker_returned(self, mock_reserved_resource, mock_worker_objects): find_one = mock_reserved_resource.get_collection.return_value.find_one find_one.return_value = {'worker_name': 'worker1'} mock_worker_objects.return_value.first.return_value = find_one.return_value result = tasks.get_worker_for_reservation('resource1') self.assertTrue(result is find_one.return_value)
def test_existing_reservation_correctly_found(self, mock_reserved_resource, mock_worker_objects): get_collection = mock_reserved_resource.get_collection tasks.get_worker_for_reservation('resource1') get_collection.assert_called_once_with() get_collection.return_value.find_one.assert_called_once_with({'resource_id': 'resource1'})
def test_existing_reservation_correctly_found(self, mock_reserved_resource, mock_worker_objects): get_objects = mock_reserved_resource.objects tasks.get_worker_for_reservation('resource1') get_objects.assert_called_once_with(resource_id='resource1') get_objects.return_value.first.assert_called_once_with()