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)
