class TestUpdate(unittest.TestCase):
def setUp(self):
super(TestUpdate, self).setUp()
self.manager = ConsumerScheduleManager()
self.units = [{'type_id': 'mytype', 'unit_key': {'name': 'foo'}}]
@mock.patch.object(ConsumerScheduleManager, '_validate_consumer')
def test_validation(self, mock_validate):
mock_validate.side_effect = MissingResource
self.assertRaises(MissingResource, self.manager.update_schedule,
'consumer1', 'schedule1', self.units)
mock_validate.assert_called_once_with('consumer1')
@mock.patch('pulp.server.managers.schedule.utils.update')
@mock.patch(
'pulp.server.managers.consumer.cud.ConsumerManager.get_consumer')
def test_units(self, mock_get_consumer, mock_update):
result = self.manager.update_schedule('consumer1', 'schedule1',
self.units)
mock_update.assert_called_once_with('schedule1',
{'kwargs': {
'units': self.units
}})
self.assertEqual(result, mock_update.return_value)
@mock.patch('pulp.server.managers.schedule.utils.update')
@mock.patch(
'pulp.server.managers.consumer.cud.ConsumerManager.get_consumer')
def test_options(self, mock_get_consumer, mock_update):
options = {'foo': 'bar'}
result = self.manager.update_schedule('consumer1',
'schedule1',
options=options)
mock_update.assert_called_once_with('schedule1',
{'kwargs': {
'options': options
}})
self.assertEqual(result, mock_update.return_value)
@mock.patch('pulp.server.managers.schedule.utils.update')
@mock.patch(
'pulp.server.managers.consumer.cud.ConsumerManager.get_consumer')
def test_other_data(self, mock_get_consumer, mock_update):
schedule_data = {'enabled': False}
result = self.manager.update_schedule('consumer1',
'schedule1',
schedule_data=schedule_data)
mock_update.assert_called_once_with('schedule1', {'enabled': False})
self.assertEqual(result, mock_update.return_value)
class TestUpdate(unittest.TestCase):
def setUp(self):
super(TestUpdate, self).setUp()
self.manager = ConsumerScheduleManager()
self.units = [
{'type_id': 'mytype', 'unit_key': {'name': 'foo'}}
]
@mock.patch.object(ConsumerScheduleManager, '_validate_consumer')
def test_validation(self, mock_validate):
mock_validate.side_effect = MissingResource
self.assertRaises(MissingResource, self.manager.update_schedule, 'consumer1', 'schedule1',
self.units)
mock_validate.assert_called_once_with('consumer1')
@mock.patch('pulp.server.managers.schedule.utils.update')
@mock.patch('pulp.server.managers.consumer.cud.ConsumerManager.get_consumer')
def test_units(self, mock_get_consumer, mock_update):
result = self.manager.update_schedule('consumer1', 'schedule1', self.units)
mock_update.assert_called_once_with('schedule1', {'kwargs': {'units': self.units}})
self.assertEqual(result, mock_update.return_value)
@mock.patch('pulp.server.managers.schedule.utils.update')
@mock.patch('pulp.server.managers.consumer.cud.ConsumerManager.get_consumer')
def test_options(self, mock_get_consumer, mock_update):
options = {'foo': 'bar'}
result = self.manager.update_schedule('consumer1', 'schedule1', options=options)
mock_update.assert_called_once_with('schedule1', {'kwargs': {'options': options}})
self.assertEqual(result, mock_update.return_value)
@mock.patch('pulp.server.managers.schedule.utils.update')
@mock.patch('pulp.server.managers.consumer.cud.ConsumerManager.get_consumer')
def test_other_data(self, mock_get_consumer, mock_update):
schedule_data = {'enabled': False}
result = self.manager.update_schedule('consumer1', 'schedule1', schedule_data=schedule_data)
mock_update.assert_called_once_with('schedule1', {'enabled': False})
self.assertEqual(result, mock_update.return_value)
