def test_configlet_update_task_no_task(self, mock_info, mock_sleep): ''' Test configlet_update_task does not get task after three tries. ''' module = Mock() mock_info.side_effect = [dict(), dict(), dict()] result = cv_server_provision.configlet_update_task(module) self.assertIsNone(result) self.assertEqual(mock_sleep.call_count, 3) self.assertEqual(mock_info.call_count, 3)
def test_configlet_update_task_no_task(self, mock_info, mock_sleep): ''' Test configlet_update_task does not get task after three tries. ''' module = Mock() mock_info.side_effect = [dict(), dict(), dict()] result = cv_server_provision.configlet_update_task(module) self.assertIsNone(result) self.assertEqual(mock_sleep.call_count, 3) self.assertEqual(mock_info.call_count, 3)
def test_configlet_update_task_good_one_try(self, mock_info, mock_sleep): ''' Test configlet_update_task gets task after one try. ''' module = Mock() task = dict(data=dict(WORKFLOW_ACTION='Configlet Push'), description='Configlet Assign', workOrderId='7') device_info = dict(taskIdList=[task]) mock_info.return_value = device_info result = cv_server_provision.configlet_update_task(module) self.assertEqual(result, '7') mock_sleep.assert_not_called() self.assertEqual(mock_info.call_count, 1)
def test_configlet_update_task_good_one_try(self, mock_info, mock_sleep): ''' Test configlet_update_task gets task after one try. ''' module = Mock() task = dict(data=dict(WORKFLOW_ACTION='Configlet Push'), description='Configlet Assign', workOrderId='7') device_info = dict(taskIdList=[task]) mock_info.return_value = device_info result = cv_server_provision.configlet_update_task(module) self.assertEqual(result, '7') mock_sleep.assert_not_called() self.assertEqual(mock_info.call_count, 1)
def test_configlet_update_task_good_three_tries(self, mock_info, mock_sleep): ''' Test configlet_update_task gets task on third try. ''' module = Mock() task1 = dict(data=dict(WORKFLOW_ACTION='Configlet Push'), description='Configlet Assign', workOrderId='7') task2 = dict(data=dict(WORKFLOW_ACTION='Nonsense'), description='Configlet Assign', workOrderId='700') device_info = dict(taskIdList=[task1, task2]) mock_info.side_effect = [dict(), dict(), device_info] result = cv_server_provision.configlet_update_task(module) self.assertEqual(result, '7') self.assertEqual(mock_sleep.call_count, 2) self.assertEqual(mock_info.call_count, 3)
def test_configlet_update_task_good_three_tries(self, mock_info, mock_sleep): ''' Test configlet_update_task gets task on third try. ''' module = Mock() task1 = dict(data=dict(WORKFLOW_ACTION='Configlet Push'), description='Configlet Assign', workOrderId='7') task2 = dict(data=dict(WORKFLOW_ACTION='Nonsense'), description='Configlet Assign', workOrderId='700') device_info = dict(taskIdList=[task1, task2]) mock_info.side_effect = [dict(), dict(), device_info] result = cv_server_provision.configlet_update_task(module) self.assertEqual(result, '7') self.assertEqual(mock_sleep.call_count, 2) self.assertEqual(mock_info.call_count, 3)