update unittests

willforde · Aug 30, 2018 · 82da1fc · 82da1fc
1 parent e89434d
commit 82da1fc
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 14 deletions.
diff --git a/script.module.codequick/lib/codequick/resolver.py b/script.module.codequick/lib/codequick/resolver.py
@@ -207,7 +207,7 @@ def ytdl_logger(record):
             raise RuntimeError(stored_errors[0])
 
     @staticmethod
-    def extract_youtube(source):
+    def extract_youtube(source):  # pragma: no cover
         # TODO: Remove this method as soon as I found out for sure that youtube.dl works on kodi for Xbox
         import htmlement
         import urlquick

diff --git a/script.module.codequick/lib/codequick/support.py b/script.module.codequick/lib/codequick/support.py
@@ -108,16 +108,15 @@ def _parse_po(self):  # type: () -> dict
         strings_po = strings_po.decode("utf8") if isinstance(strings_po, bytes) else strings_po
 
         # Check if strings.po actrally exists first
-        if not os.path.exists(strings_po):
-            return {}
+        if os.path.exists(strings_po):  # pragma: no branch
+            with open(strings_po, "rb") as fo:
+                raw_strings = fo.read()
 
-        with open(strings_po, "rb") as fo:
-            raw_strings = fo.read()
+            # Parse strings using Regular Expressions
+            res = u"^msgctxt\s+[\"']#(\d+?)[\"']$[\n\r]^msgid\s+[\"'](.+?)[\"']$"
+            data = re.findall(res, raw_strings.decode("utf8"), re.MULTILINE | re.UNICODE)
+            self._strings.update((key, int(value)) for value, key in data)
 
-        # Parse strings using Regular Expressions
-        res = u"^msgctxt\s+[\"']#(\d+?)[\"']$[\n\r]^msgid\s+[\"'](.+?)[\"']$"
-        data = re.findall(res, raw_strings.decode("utf8"), re.MULTILINE | re.UNICODE)
-        self._strings.update((key, int(value)) for value, key in data)
         return self._strings
 
 

diff --git a/tests/test_script.py b/tests/test_script.py
@@ -154,8 +154,15 @@ def test_notify(self):
     def test_localize(self):
         self.assertEqual(self.script.localize(30001), "")
 
-    def test_localize_novideo(self):
-        self.assertEqual(self.script.localize(32401), "Unable to resolve url")
+    def test_localize_allvideos(self):
+        self.assertEqual("All Videos", self.script.localize(32003))
+
+    def test_localize_allvideos_gettext(self):
+        self.assertEqual(self.script.localize("All Videos"), "All Videos")
+
+    def test_localize_allvideos_gettext_non_exist(self):
+        with self.assertRaises(KeyError):
+            self.script.localize("All V")
 
     def test_localize_nodata(self):
         self.assertEqual(self.script.localize(33077), "No data found!")

diff --git a/tests/test_youtube.py b/tests/test_youtube.py
@@ -53,12 +53,12 @@ def test_playlist_playlist_single_page(self):
         self.assertGreaterEqual(len(ret), 10)
 
     def test_playlist_playlist_muilti_page(self):
-        ret = route_caller(youtube.Playlist, "PLGe_S5n7MjT36EUYItEv7Wi3Wkz4pZU7o", loop=False)
+        ret = route_caller(youtube.Playlist, "PL8mG-RkN2uTx1lbFS8z8wRYS3RrHCp8TG", loop=False)
         self.assertGreaterEqual(len(ret), 50)
 
     def test_playlist_playlist_loop(self):
-        ret = route_caller(youtube.Playlist, "PLGe_S5n7MjT36EUYItEv7Wi3Wkz4pZU7o", loop=True)
-        self.assertGreaterEqual(len(ret), 250)
+        ret = route_caller(youtube.Playlist, "PL8mG-RkN2uTx1lbFS8z8wRYS3RrHCp8TG", loop=True)
+        self.assertGreaterEqual(len(ret), 66)
 
     def test_related(self):
         ret = route_caller(youtube.Related, "-QEXPO9zgX8")