import unittest from ..navigation import ( get_parent_url, get_root_url, join_url, parse_url, pop_first_segment, remove_dot_segments, remove_last_segment, set_parameter, ) class TestNavigation(unittest.TestCase): def test_parse_url(self): # Basic complete URL. res = parse_url("gemini://netloc/parse-me.gmi") self.assertEqual(res["scheme"], "gemini") self.assertEqual(res["netloc"], "netloc") self.assertEqual(res["path"], "/parse-me.gmi") # No scheme. res_netloc = parse_url("//netloc/parse-me.gmi") self.assertIsNone(res_netloc["scheme"], None) for key in res_netloc: if key == "scheme": continue self.assertEqual(res_netloc[key], res[key]) # No scheme but a default is provided. res_netloc = parse_url("//netloc/parse-me.gmi", default_scheme="gemini") self.assertDictEqual(res_netloc, res) # No scheme nor netloc: only a path should be produced. res = parse_url("dece.space/parse-me.gmi") self.assertIsNone(res["scheme"]) self.assertIsNone(res["netloc"]) self.assertEqual(res["path"], "dece.space/parse-me.gmi") # HTTPS scheme. res = parse_url("https://dece.space/index.html") self.assertEqual(res["scheme"], "https") self.assertEqual(res["netloc"], "dece.space") self.assertEqual(res["path"], "/index.html") # File scheme. res = parse_url("file:///home/dece/gemini/index.gmi") self.assertEqual(res["scheme"], "file") self.assertEqual(res["path"], "/home/dece/gemini/index.gmi") # Bebop scheme. res = parse_url("bebop:welcome") self.assertEqual(res["scheme"], "bebop") self.assertIsNone(res["netloc"]) self.assertEqual(res["path"], "welcome") def test_join_url(self): url = join_url("gemini://dece.space/", "some-file.gmi") self.assertEqual(url, "gemini://dece.space/some-file.gmi") url = join_url("gemini://dece.space/", "./some-file.gmi") self.assertEqual(url, "gemini://dece.space/some-file.gmi") url = join_url("gemini://dece.space/dir1", "/some-file.gmi") self.assertEqual(url, "gemini://dece.space/some-file.gmi") url = join_url("gemini://dece.space/dir1/file.gmi", "other-file.gmi") self.assertEqual(url, "gemini://dece.space/dir1/other-file.gmi") url = join_url("gemini://dece.space/dir1/file.gmi", "../top-level.gmi") self.assertEqual(url, "gemini://dece.space/top-level.gmi") url = join_url("s://hard/dir/a", "./../test/b/c/../d/e/f/../.././a.gmi") self.assertEqual(url, "s://hard/test/b/d/a.gmi") def test_remove_dot_segments(self): paths = [ ("index.gmi", "index.gmi"), ("/index.gmi", "/index.gmi"), ("./index.gmi", "index.gmi"), ("/./index.gmi", "/index.gmi"), ("/../index.gmi", "/index.gmi"), ("/a/b/c/./../../g", "/a/g"), ("mid/content=5/../6", "mid/6"), ("../../../../g", "g"), ] for path, expected in paths: self.assertEqual( remove_dot_segments(path), expected, msg="path was " + path ) def test_remove_last_segment(self): self.assertEqual(remove_last_segment(""), "") self.assertEqual(remove_last_segment("/"), "") self.assertEqual(remove_last_segment("/a"), "") self.assertEqual(remove_last_segment("/a/"), "/a") self.assertEqual(remove_last_segment("/a/b"), "/a") self.assertEqual(remove_last_segment("/a/b/c/d"), "/a/b/c") self.assertEqual(remove_last_segment("///"), "//") def test_pop_first_segment(self): self.assertEqual(pop_first_segment(""), ("", "")) self.assertEqual(pop_first_segment("a"), ("a", "")) self.assertEqual(pop_first_segment("/a"), ("/a", "")) self.assertEqual(pop_first_segment("/a/"), ("/a", "/")) self.assertEqual(pop_first_segment("/a/b"), ("/a", "/b")) self.assertEqual(pop_first_segment("a/b"), ("a", "/b")) def test_set_parameter(self): url = set_parameter("gemini://gus.guru/search", "my search") self.assertEqual(url, "gemini://gus.guru/search?my%20search") url = set_parameter("gemini://gus.guru/search?old%20search", "new") self.assertEqual(url, "gemini://gus.guru/search?new") def test_get_parent_url(self): urls_and_parents = [ ("gemini://host", "gemini://host"), ("gemini://host/", "gemini://host/"), ("gemini://host/a", "gemini://host/"), ("gemini://host/a/", "gemini://host/"), ("gemini://host/a/index.gmi", "gemini://host/a/"), ("gemini://host/a/b/", "gemini://host/a/"), ("gemini://host/a/b/file.flac", "gemini://host/a/b/"), ("//host/a/b", "//host/a/"), ("hey", "hey"), # does not really make sense but whatever ("hey/ho", "hey/"), ("hey/ho/letsgo", "hey/ho/"), ] for url, parent in urls_and_parents: self.assertEqual( get_parent_url(url), parent, msg="URL was " + url) def test_get_root_url(self): urls_and_roots = [ ("gemini://host", "gemini://host/"), ("gemini://host/", "gemini://host/"), ("gemini://host/a", "gemini://host/"), ("gemini://host/a/b/c", "gemini://host/"), ("//host/path", "//host/"), ("//host/path?query", "//host/"), ("dumb", "/"), ("dumb/dumber", "/"), ] for url, root in urls_and_roots: self.assertEqual( get_root_url(url), root, msg="URL was " + url )