class TestValidate(unittest.TestCase):
def setUp(self):
super(TestValidate, self).setUp()
self.manager = ConsumerScheduleManager()
@mock.patch('pulp.server.managers.consumer.cud.ConsumerManager.get_consumer')
def test_calls_get_consumer(self, mock_get):
self.manager._validate_consumer('foo')
mock_get.assert_called_once_with('foo')
@mock.patch('pulp.server.db.model.base.Model.get_collection')
def test_raises_missing(self, mock_get_collection):
# mock another layer down to verify manager integration
mock_get_collection.return_value.find_one.side_effect = MissingResource
self.assertRaises(MissingResource, self.manager._validate_consumer, 'foo')
class TestValidate(unittest.TestCase):
def setUp(self):
super(TestValidate, self).setUp()
self.manager = ConsumerScheduleManager()
@mock.patch(
'pulp.server.managers.consumer.cud.ConsumerManager.get_consumer')
def test_calls_get_consumer(self, mock_get):
self.manager._validate_consumer('foo')
mock_get.assert_called_once_with('foo')
@mock.patch('pulp.server.db.model.base.Model.get_collection')
def test_raises_missing(self, mock_get_collection):
# mock another layer down to verify manager integration
mock_get_collection.return_value.find_one.side_effect = MissingResource
self.assertRaises(MissingResource, self.manager._validate_consumer,
'foo')
