diff planemo/lib/python3.7/site-packages/schema_salad/tests/test_fetch.py @ 1:56ad4e20f292 draft

"planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/planemo/lib/python3.7/site-packages/schema_salad/tests/test_fetch.py	Fri Jul 31 00:32:28 2020 -0400
@@ -0,0 +1,62 @@
+from __future__ import absolute_import, print_function
+
+import os
+from typing import Text
+
+import pytest
+from six.moves import urllib
+
+import schema_salad.main
+import schema_salad.ref_resolver
+import schema_salad.schema
+
+
+def test_fetcher():
+    class TestFetcher(schema_salad.ref_resolver.Fetcher):
+        def __init__(self, a, b):
+            pass
+
+        def fetch_text(self, url):  # type: (Text) -> Text
+            if url == "keep:abc+123/foo.txt":
+                return "hello: keepfoo"
+            if url.endswith("foo.txt"):
+                return "hello: foo"
+            else:
+                raise RuntimeError("Not foo.txt")
+
+        def check_exists(self, url):  # type: (Text) -> bool
+            if url.endswith("foo.txt"):
+                return True
+            else:
+                return False
+
+        def urljoin(self, base, url):
+            urlsp = urllib.parse.urlsplit(url)
+            if urlsp.scheme:
+                return url
+            basesp = urllib.parse.urlsplit(base)
+
+            if basesp.scheme == "keep":
+                return base + "/" + url
+            return urllib.parse.urljoin(base, url)
+
+    loader = schema_salad.ref_resolver.Loader({}, fetcher_constructor=TestFetcher)
+    assert {"hello": "foo"} == loader.resolve_ref("foo.txt")[0]
+    assert {"hello": "keepfoo"} == loader.resolve_ref(
+        "foo.txt", base_url="keep:abc+123"
+    )[0]
+    assert loader.check_exists("foo.txt")
+
+    with pytest.raises(RuntimeError):
+        loader.resolve_ref("bar.txt")
+    assert not loader.check_exists("bar.txt")
+
+
+def test_cache():
+    loader = schema_salad.ref_resolver.Loader({})
+    foo = os.path.join(os.getcwd(), "foo.txt")
+    foo = schema_salad.ref_resolver.file_uri(foo)
+    loader.cache.update({foo: "hello: foo"})
+    print(loader.cache)
+    assert {"hello": "foo"} == loader.resolve_ref("foo.txt")[0]
+    assert loader.check_exists(foo)