Esempi in Python per ConsumerScheduleManager.get

Esempio n. 1

File: test_consumer.py Progetto: AndreaGiardini/pulp

class TestGet(unittest.TestCase):
    def setUp(self):
        super(TestGet, self).setUp()
        self.manager = ConsumerScheduleManager()
        self.calls = [
            ScheduledCall('PT1H', ACTIONS_TO_TASKS[UNIT_INSTALL_ACTION]),
            ScheduledCall('PT4H', ACTIONS_TO_TASKS[UNIT_UPDATE_ACTION])
        ]

    @mock.patch('pulp.server.managers.schedule.utils.get_by_resource')
    def test_no_action(self, mock_get_by_resource):
        mock_get_by_resource.return_value = self.calls

        result = self.manager.get('consumer1')

        mock_get_by_resource.assert_called_once_with(Consumer.build_resource_tag('consumer1'))
        self.assertEqual(result, self.calls)

    @mock.patch('pulp.server.managers.schedule.utils.get_by_resource')
    def test_with_action(self, mock_get_by_resource):
        mock_get_by_resource.return_value = self.calls

        result = self.manager.get('consumer1', UNIT_INSTALL_ACTION)

        mock_get_by_resource.assert_called_once_with(Consumer.build_resource_tag('consumer1'))
        self.assertEqual(list(result), self.calls[:1])

Esempio n. 2

class TestGet(unittest.TestCase):
    def setUp(self):
        super(TestGet, self).setUp()
        self.manager = ConsumerScheduleManager()
        self.calls = [
            ScheduledCall('PT1H', ACTIONS_TO_TASKS[UNIT_INSTALL_ACTION]),
            ScheduledCall('PT4H', ACTIONS_TO_TASKS[UNIT_UPDATE_ACTION])
        ]

    @mock.patch('pulp.server.managers.schedule.utils.get_by_resource')
    def test_no_action(self, mock_get_by_resource):
        mock_get_by_resource.return_value = self.calls

        result = self.manager.get('consumer1')

        mock_get_by_resource.assert_called_once_with(
            Consumer.build_resource_tag('consumer1'))
        self.assertEqual(result, self.calls)

    @mock.patch('pulp.server.managers.schedule.utils.get_by_resource')
    def test_with_action(self, mock_get_by_resource):
        mock_get_by_resource.return_value = self.calls

        result = self.manager.get('consumer1', UNIT_INSTALL_ACTION)

        mock_get_by_resource.assert_called_once_with(
            Consumer.build_resource_tag('consumer1'))
        self.assertEqual(list(result), self.calls[:1])