def test_sanitize_special_character_returns_encoded_character(self):
self.assertEquals("<", sanitize("<"))
def test_sanitize_list_of_xml_returns_list_of_encoded_xml(self):
xml = "<root><element>value</element></root>"
expected = "<root><element>value</element></root>"
xml_list = [xml, xml]
expected_list = [expected, expected]
self.assertEquals(expected_list, sanitize(xml_list))
def test_sanitize_dict_of_xml_returns_dict_of_encoded_xml(self):
xml = "<root><element>value</element></root>"
expected = "<root><element>value</element></root>"
xml_dict = {xml: xml}
expected_dict = {expected: expected}
self.assertEquals(expected_dict, sanitize(xml_dict))
def test_sanitize_bad_xml_returns_encoded_xml(self):
xml = "<root<element>value</element></root>"
expected = "<root<element>value</element></root>"
self.assertEquals(expected, sanitize(xml))
def test_sanitize_float_returns_float(self):
self.assertEquals(1.0, sanitize(1.0))
def test_sanitize_int_returns_int(self):
self.assertEquals(1, sanitize(1))
def test_sanitize_string_returns_string(self):
self.assertEquals("test", sanitize("test"))
def load_resources_view(request):
"""Load resources for a given list of modules
:param request:
:return:
"""
if not request.method == "GET":
return HttpResponseBadRequest({})
if "urlsToLoad" not in request.GET or "urlsLoaded" not in request.GET:
return HttpResponseBadRequest({})
# URLs of the modules to load
mod_urls_qs = sanitize(request.GET["urlsToLoad"])
mod_urls = json.loads(mod_urls_qs)
# URLs of the loaded modules
mod_urls_loaded_qs = sanitize(request.GET["urlsLoaded"])
mod_urls_loaded = json.loads(mod_urls_loaded_qs)
# Request hack to get module resources
request.GET = {"resources": True}
# List of resources
resources = {"scripts": [], "styles": []}
# Add all resources from requested modules
for url in mod_urls:
module = module_api.get_by_url(url)
module_view = AbstractModule.get_view_from_view_path(module.view).as_view()
mod_resources = module_view(request).content.decode("utf-8")
mod_resources = sanitize(mod_resources)
mod_resources = json.loads(mod_resources)
# Append resource to the list
for key in list(resources.keys()):
if mod_resources[key] is None:
continue
for resource in mod_resources[key]:
if resource not in resources[key]:
resources[key].append(resource)
# Remove possible dependencies form already loaded modules
for url in mod_urls_loaded:
module_view = AbstractModule.get_module_view(url)
mod_resources = module_view(request).content.decode("utf-8")
mod_resources = sanitize(mod_resources)
mod_resources = json.loads(mod_resources)
# Remove resources already loaded
for key in list(resources.keys()):
if mod_resources[key] is None:
continue
for resource in mod_resources[key]:
if resource in resources[key]:
i = resources[key].index(resource)
del resources[key][i]
# Build response content
response = {"scripts": "", "styles": ""}
# Aggregate scripts
for script in resources["scripts"]:
if script.startswith("http://") or script.startswith("https://"):
script_tag = '<script class="module" src="' + script + '"></script>'
else:
with open(finders.find(script), "r") as script_file:
script_tag = (
'<script class="module">' + script_file.read() + "</script>"
)
response["scripts"] += script_tag
# Aggregate styles
for style in resources["styles"]:
if style.startswith("http://") or style.startswith("https://"):
script_tag = (
'<link class="module" rel="stylesheet" type="text/css" href="'
+ style
+ '"></link>'
)
else:
with open(finders.find(style), "r") as script_file:
script_tag = '<style class="module">' + script_file.read() + "</style>"
response["styles"] += script_tag
# Send response
return HttpResponse(json.dumps(response))
