def test_additional_files_passed_as_extra_params(self, mock_json, mock_call, mock_converters): mock_json.loads = Mock(return_value= [{ "name": "mock_converter", "command": "command --arg {0} --extra_arg {1}", "accepted_content_types": ["text/xml"], "title": "View map as PNG image (thumbnail with background)", "returned_content_type": "image/png", "extraparams": ["country_code"], "additional_files": True }] ) import web import flask from path import path from convert import init_converters import StringIO converters = init_converters() mock_converters.get = Mock(return_value=converters['mock_converter']) files = { 'file': (StringIO.StringIO('shp data'), 'file.shp'), 'shx': (StringIO.StringIO('shx data'), 'file.shx'), 'dbf': (StringIO.StringIO('dbf data'), 'file.dbf') } resp = self.client.post("/convert/mock_converter", data=files) self.assertEqual(2, len(mock_call.mock_calls[0][1][2])) assert path(mock_call.mock_calls[0][1][2][0]).startswith('/tmp/') assert path(mock_call.mock_calls[0][1][2][1]).startswith('/tmp/')
def test_additional_files_param(self, mock_json): mock_json.loads = Mock(return_value= [{ "name": "mock_converter", "command": "command --arg {0} --extra_arg {1}", "accepted_content_types": ["text/xml"], "title": "View map as PNG image (thumbnail with background)", "returned_content_type": "image/png", "extraparams": ["country_code"], "additional_files": True }] ) from convert import init_converters converters = init_converters() conv = converters['mock_converter'] self.assertEqual(conv.additional_files, True)
def test_extraparams_initialisation(self, mock_json): mock_json.loads = Mock(return_value= [{ "name": "mock_converter", "command": "command --arg {0} --extra_arg {1}", "accepted_content_types": ["text/xml"], "title": "View map as PNG image (thumbnail with background)", "returned_content_type": "image/png", "extraparams": ["country_code"] }] ) from convert import init_converters converters = init_converters() self.assertEqual( ['country_code'], converters['mock_converter'].extraparams)