cpplearner
diff --git a/‎commands/preprocess.py‎
Lines changed: 4 additions & 2 deletions b/‎commands/preprocess.py‎
Lines changed: 4 additions & 2 deletions
diff --git a/‎tests/__init__.py‎ b/‎tests/__init__.py‎
diff --git a/‎tests/test_preprocess.py‎
Lines changed: 31 additions & 0 deletions b/‎tests/test_preprocess.py‎
Lines changed: 31 additions & 0 deletions
@@ -87,6 +87,8 @@ def add_file_to_rename_map(rename_map, dir, fn, new_fn):
         return
     rename_map.append((dir, fn, new_fn))
 
+# Converts complex URL to resources supplied by MediaWiki loader to a simplified
+# name
 def convert_loader_name(fn):
     if re.search("modules=site&only=scripts", fn):
         return "site_scripts.js"
@@ -99,8 +101,8 @@ def convert_loader_name(fn):
     elif re.search("modules=.*ext.*&only=styles", fn):
         return "ext.css"
     else:
-        print("Loader file " + fn + " does not match any known files")
-        sys.exit(1)
+        raise Exception('Loader file {0} does not match any known files'\
+                            .format(fn))
 
 def find_files_to_be_renamed(root):
     # Returns a rename map: array of tuples each of which contain three strings:
 
@@ -0,0 +1,31 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+from commands.preprocess import *
+import unittest
+
+class TestConvertLoaderName(unittest.TestCase):
+    def test_convert_loader_name(self):
+        url = 'http://en.cppreference.com/mwiki/load.php?debug=false&lang=en&\
+modules=site&only=scripts&skin=cppreference2&*'
+        self.assertEqual('site_scripts.js', convert_loader_name(url))
+
+        url = 'http://en.cppreference.com/mwiki/load.php?debug=false&lang=en&\
+modules=site&only=styles&skin=cppreference2&*'
+        self.assertEqual('site_modules.css', convert_loader_name(url))
+
+        url = 'http://en.cppreference.com/mwiki/load.php?debug=false&lang=en&\
+modules=skins.cppreference2&only=scripts&skin=cppreference2&*'
+        self.assertEqual('skin_scripts.js', convert_loader_name(url))
+
+        url = 'http://en.cppreference.com/mwiki/load.php?debug=false&lang=en&\
+modules=startup&only=scripts&skin=cppreference2&*'
+        self.assertEqual('startup_scripts.js', convert_loader_name(url))
+
+        url = 'http://en.cppreference.com/mwiki/load.php?debug=false&lang=en&\
+modules=ext.gadget.ColiruCompiler%2CMathJax%7Cext.rtlcite%7Cmediawiki.\
+legacy.commonPrint%2Cshared%7Cskins.cppreference2&only=styles&skin=\
+cppreference2&*'
+        self.assertEqual('ext.css', convert_loader_name(url))
+
+        with self.assertRaises(Exception):
+            convert_loader_name('')