def test_workflow_upload_update(self, *args): module_mock = MagicMock() with patch.dict( 'sys.modules', **{ 'vdirect_client': module_mock, 'vdirect_client.rest_client': module_mock, }): from ansible_collections.notmintest.not_a_real_collection.plugins.modules import vdirect_file vdirect_file.rest_client.RESP_STATUS = 0 vdirect_file.rest_client.WorkflowTemplate = WorkflowTemplate file = vdirect_file.VdirectFile(NONE_PARAMS) path = os.path.dirname(os.path.abspath(__file__)) WorkflowTemplate.set_create_template_from_archive_result([409]) WorkflowTemplate.set_update_archive_result([201]) result = file.upload(os.path.join(path, "wt.zip")) self.assertEqual(result, vdirect_file.WORKFLOW_TEMPLATE_UPDATED_SUCCESS, 'Unexpected result received:' + repr(result)) WorkflowTemplate.set_update_archive_result( [400, "", "Parsing error", ""]) try: result = file.upload(os.path.join(path, "wt.zip")) self.fail("InvalidSourceException was not thrown") except vdirect_file.InvalidSourceException: assert True
def test_wrong_file_extension(self, *args): module_mock = MagicMock() with patch.dict( 'sys.modules', **{ 'vdirect_client': module_mock, 'vdirect_client.rest_client': module_mock, }): from ansible_collections.notmintest.not_a_real_collection.plugins.modules import vdirect_file module_mock.RESP_STATUS = 0 file = vdirect_file.VdirectFile(NONE_PARAMS) result = file.upload("file.??") assert result == vdirect_file.WRONG_EXTENSION_ERROR
def test_missing_file(self, *args): module_mock = MagicMock() with patch.dict( 'sys.modules', **{ 'vdirect_client': module_mock, 'vdirect_client.rest_client': module_mock, }): from ansible_collections.notmintest.not_a_real_collection.plugins.modules import vdirect_file file = vdirect_file.VdirectFile(NONE_PARAMS) try: file.upload("missing_file.vm") self.fail("IOException was not thrown for missing file") except IOError: assert True
def test_missing_parameter(self, *args): module_mock = MagicMock() with patch.dict( 'sys.modules', **{ 'vdirect_client': module_mock, 'vdirect_client.rest_client': module_mock, }): from ansible_collections.notmintest.not_a_real_collection.plugins.modules import vdirect_file try: params = NONE_PARAMS.copy() del params['vdirect_ip'] vdirect_file.VdirectFile(params) self.fail("KeyError was not thrown for missing parameter") except KeyError: assert True