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")