Python get_worker_for_reservation Examples

Example #1

File: test_tasks.py Project: pombreda/pulp

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

Example #2

File: test_tasks.py Project: hgschmie/pulp

 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.")

Example #3

 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.")

Example #4

 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)

Example #5

File: test_tasks.py Project: hgschmie/pulp

 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)

Example #6

File: test_tasks.py Project: hgschmie/pulp

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

Example #7

 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()

Example #8

File: test_tasks.py Project: zjhuntin/pulp

 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()