Add cache test. check_exists checks cache.

common-workflow-language · Nov 22, 2016 · 8e32f71 · 8e32f71
1 parent a747205
commit 8e32f71
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/schema_salad/ref_resolver.py b/schema_salad/ref_resolver.py
@@ -116,6 +116,9 @@ def fetch_text(self, url):
             raise ValueError('Unsupported scheme in url: %s' % url)
 
     def check_exists(self, url):  # type: (unicode) -> bool
+        if url in self.cache:
+            return True
+
         split = urlparse.urlsplit(url)
         scheme, path = split.scheme, split.path
 

diff --git a/schema_salad/tests/test_fetch.py b/schema_salad/tests/test_fetch.py
@@ -35,3 +35,11 @@ def check_exists(self, url):  # type: (unicode) -> bool
         with self.assertRaises(RuntimeError):
             loader.resolve_ref("bar.txt")
         self.assertFalse(loader.check_exists("bar.txt"))
+
+    def test_cache(self):
+        loader = schema_salad.ref_resolver.Loader({})
+        foo = "file://%s/foo.txt" % os.getcwd()
+        loader.cache.update({foo: "hello: foo"})
+        print loader.cache
+        self.assertEqual({"hello": "foo"}, loader.resolve_ref("foo.txt")[0])
+        self.assertTrue(loader.check_exists(foo))