1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
import unittest
import schema_salad.ref_resolver
import schema_salad.main
import schema_salad.schema
from schema_salad.jsonld_context import makerdf
import rdflib
import ruamel.yaml as yaml
import json
import os
import urlparse
class TestFetcher(unittest.TestCase):
def test_fetcher(self):
class TestFetcher(schema_salad.ref_resolver.Fetcher):
def __init__(self, a, b):
pass
def fetch_text(self, url): # type: (unicode) -> unicode
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: (unicode) -> bool
if url.endswith("foo.txt"):
return True
else:
return False
def urljoin(self, base, url):
urlsp = urlparse.urlsplit(url)
if urlsp.scheme:
return url
basesp = urlparse.urlsplit(base)
if basesp.scheme == "keep":
return base + "/" + url
return urlparse.urljoin(base, url)
loader = schema_salad.ref_resolver.Loader({}, fetcher_constructor=TestFetcher)
self.assertEqual({"hello": "foo"}, loader.resolve_ref("foo.txt")[0])
self.assertEqual({"hello": "keepfoo"}, loader.resolve_ref("foo.txt", base_url="keep:abc+123")[0])
self.assertTrue(loader.check_exists("foo.txt"))
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))
|
