class ConsumerContentInstallScheduleStrategyTests(rpm_support_base.PulpClientTests): def setUp(self): super(ConsumerContentInstallScheduleStrategyTests, self).setUp() self.strategy = YumConsumerContentScheduleStrategy(self.context, 'install', TYPE_ID_RPM) @mock.patch('pulp.bindings.consumer.ConsumerContentSchedulesAPI.add_schedule') def test_create_schedule(self, mock_add): # Setup schedule = '2012-09-18' failure_threshold = 3 enabled = True kwargs = {OPTION_CONSUMER_ID.keyword : 'test-consumer', 'name': ['pkg1','pkg2']} # Test self.strategy.create_schedule(schedule, failure_threshold, enabled, kwargs) # Verify self.assertEqual(1, mock_add.call_count) call_args = mock_add.call_args[0] self.assertEqual('install', call_args[0]) self.assertEqual('test-consumer', call_args[1]) self.assertEqual(schedule, call_args[2]) self.assertEqual(failure_threshold, call_args[4]) self.assertEqual(enabled, call_args[5]) self.assertEqual({}, call_args[6]) @mock.patch('pulp.bindings.consumer.ConsumerContentSchedulesAPI.delete_schedule') def test_delete_schedule(self, mock_delete): # Setup schedule_id = 'fake-schedule' kwargs = {OPTION_CONSUMER_ID.keyword : 'fake-consumer'} # Test self.strategy.delete_schedule(schedule_id, kwargs) # Verify self.assertEqual(1, mock_delete.call_count) call_args = mock_delete.call_args[0] self.assertEqual('install', call_args[0]) self.assertEqual('fake-consumer', call_args[1]) self.assertEqual(schedule_id, call_args[2]) @mock.patch('pulp.bindings.consumer.ConsumerContentSchedulesAPI.list_schedules') def test_retrieve_schedules(self, mock_retrieve): # Setup kwargs = {OPTION_CONSUMER_ID.keyword : 'retrieve-consumer'} # Test self.strategy.retrieve_schedules(kwargs) # Verify self.assertEqual(1, mock_retrieve.call_count) call_args = mock_retrieve.call_args[0] self.assertEqual('install', call_args[0]) self.assertEqual('retrieve-consumer', call_args[1]) @mock.patch('pulp.bindings.consumer.ConsumerContentSchedulesAPI.update_schedule') def test_update_schedule(self, mock_update): # Setup kwargs = {OPTION_CONSUMER_ID.keyword : 'fake-consumer', 'a' : 'a'} schedule_id = 'schedule-id' # Test self.strategy.update_schedule(schedule_id, **kwargs) # Verify self.assertEqual(1, mock_update.call_count) call_args = mock_update.call_args[0] self.assertEqual('install', call_args[0]) self.assertEqual('fake-consumer', call_args[1]) self.assertEqual(schedule_id, call_args[2]) call_kwargs = mock_update.call_args[1] self.assertEqual({'a' : 'a'}, call_kwargs)
def setUp(self): super(ConsumerContentInstallScheduleStrategyTests, self).setUp() self.strategy = YumConsumerContentScheduleStrategy(self.context, 'install', TYPE_ID_RPM)
def setUp(self): super(ConsumerContentUpdateScheduleStrategyTests, self).setUp() self.strategy = YumConsumerContentScheduleStrategy(self.context, 'update', TYPE_ID_RPM)