def test_resource_task(self): _, _ctx = self._prepare_master_node() decorators.resource_task(MagicMock(), MagicMock())(MagicMock())() mock_isfile = MagicMock(return_value=True) _ctx.download_resource = MagicMock(return_value="downloaded_resource") with patch('os.path.isfile', mock_isfile): with self.assertRaises(NonRecoverableError) as error: decorators.resource_task(MagicMock(), MagicMock())( MagicMock(side_effect=KuberentesInvalidApiMethodError( 'error_text')))() self.assertEqual(str(error.exception), "error_text")
def test_resource_task_retrieve_NonRecoverableError(self): _, _ctx = self._prepare_master_node() decorators.resource_task(MagicMock(), MagicMock())(MagicMock())() mock_isfile = MagicMock(return_value=True) _ctx.download_resource = MagicMock(return_value="downloaded_resource") defintion = KubernetesResourceDefinition( **_ctx.node.properties['definition']) with patch('os.path.isfile', mock_isfile): with self.assertRaises(NonRecoverableError) as error: decorators.resource_task( retrieve_resources_definitions=MagicMock( return_value=[defintion]), retrieve_mapping=MagicMock(), use_existing=True)(MagicMock( side_effect=NonRecoverableError('error_text')))() self.assertEqual(str(error.exception), "error_text")