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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
# standard imports
import unittest
# bsie imports
from bsie.utils import ns, errors
from bsie.utils.bsfs import URI
from bsie.utils.node import Node
# objects to test
from bsie.lib.naming_policy import NamingPolicy, NamingPolicyIterator, DefaultNamingPolicy
## code ##
class TestDefaultNamingPolicy(unittest.TestCase):
def test_handle_node(self):
# setup
policy = DefaultNamingPolicy('http://example.com', 'me')
# handle_node doesn't modify existing uris
self.assertEqual(policy.handle_node(
Node(ns.bsn.Invalid, uri='http://example.com/you/foo#bar')).uri,
URI('http://example.com/you/foo#bar'))
# processes bsfs:File
self.assertEqual(policy.handle_node(
Node(ns.bsn.Entity, ucid='abc123cba')).uri,
URI('http://example.com/me/file#abc123cba'))
# processes bsfs:Preview
self.assertEqual(policy.handle_node(
Node(ns.bsn.Preview, ucid='abc123cba', size=123)).uri,
URI('http://example.com/me/preview#abc123cba_s123'))
# processes bsn:Tag
self.assertEqual(policy.handle_node(
Node(ns.bsn.Tag, label='hello')).uri,
URI('http://example.com/me/tag#hello'))
# raises an exception on unknown types
self.assertRaises(errors.ProgrammingError, policy.handle_node,
Node(ns.bsn.Invalid, ucid='abc123cba', size=123))
def test_name_file(self):
# setup
policy = DefaultNamingPolicy('http://example.com', 'me')
# name_file uses ucid
self.assertEqual(policy.name_file(
Node(ns.bsn.Entity, ucid='123abc321')).uri,
URI('http://example.com/me/file#123abc321'))
# name_file falls back to a random guid
self.assertTrue(policy.name_file(
Node(ns.bsn.Entity)).uri.startswith('http://example.com/me/file#'))
def test_name_preview(self):
# setup
policy = DefaultNamingPolicy('http://example.com', 'me')
# name_preview uses ucid
self.assertEqual(policy.name_preview(
Node(ns.bsn.Preview, ucid='123abc321')).uri,
URI('http://example.com/me/preview#123abc321'))
self.assertEqual(policy.name_preview(
Node(ns.bsn.Preview, ucid='123abc321', size=400)).uri,
URI('http://example.com/me/preview#123abc321_s400'))
# name_preview uses source
self.assertEqual(policy.name_preview(
Node(ns.bsn.Preview, source=Node(ns.bsn.Entity, ucid='123file321'))).uri,
URI('http://example.com/me/preview#123file321'))
self.assertEqual(policy.name_preview(
Node(ns.bsn.Preview, source=Node(ns.bsn.Entity, ucid='123file321'), size=300)).uri,
URI('http://example.com/me/preview#123file321_s300'))
# name_preview falls back to a random guid
self.assertTrue(policy.name_preview(
Node(ns.bsn.Preview)).uri.startswith('http://example.com/me/preview#'))
self.assertTrue(policy.name_preview(
Node(ns.bsn.Preview, size=200)).uri.startswith('http://example.com/me/preview#'))
self.assertTrue(policy.name_preview(
Node(ns.bsn.Preview, size=200)).uri.endswith('_s200'))
def test_name_tag(self):
# setup
policy = DefaultNamingPolicy('http://example.com', 'me')
# name_tag uses label
self.assertEqual(policy.name_tag(
Node(ns.bsn.Tag, label='hello')).uri,
URI('http://example.com/me/tag#hello'))
# name_tag matches the label
self.assertEqual(
policy.name_tag(Node(ns.bsn.Tag, label='world')),
policy.name_tag(Node(ns.bsn.Tag, label='world')),
)
self.assertNotEqual(
policy.name_tag(Node(ns.bsn.Tag, label='hello')),
policy.name_tag(Node(ns.bsn.Tag, label='world')),
)
# label can include characters that are not valid for an uri
self.assertEqual(policy.name_tag(
Node(ns.bsn.Preview, label='hello world { foo bar ] ')).uri,
URI('http://example.com/me/tag#hello%20world%20%7B%20foo%20bar%20%5D%20'))
# name_tag falls back to a random guid
self.assertTrue(policy.name_tag(
Node(ns.bsn.Tag,)).uri.startswith('http://example.com/me/tag#'))
class TestNamingPolicyIterator(unittest.TestCase):
def test_call(self): # NOTE: We test NamingPolicy.__call__ here
# setup
policy = DefaultNamingPolicy('http://example.com', 'me')
# call accepts list
triples = [('node', 'pred', 'value'), ('node', 'pred', 'value')]
it = policy(triples)
self.assertIsInstance(it, NamingPolicyIterator)
self.assertEqual(it._iterable, triples)
self.assertEqual(it._policy, policy)
# call accepts iterator
triples = iter([('node', 'pred', 'value'), ('node', 'pred', 'value')])
it = policy(triples)
self.assertIsInstance(it, NamingPolicyIterator)
self.assertEqual(it._iterable, triples)
self.assertEqual(it._policy, policy)
def test_iter(self):
# setup
policy = DefaultNamingPolicy('http://example.com', 'me')
triples = [
(Node(ns.bsn.Entity, ucid='foo'), 'predA', 'hello'),
(Node(ns.bsn.Preview, ucid='bar'), 'predB', 1234),
(Node(ns.bsn.Preview, ucid='hello'), 'predC', Node(ns.bsn.Entity, ucid='world'))
]
# handles nodes, handles values, ignores predicate
self.assertListEqual(list(policy(triples)), [
(Node(ns.bsn.Entity, uri='http://example.com/me/file#foo'), 'predA', 'hello'),
(Node(ns.bsn.Preview, uri='http://example.com/me/preview#bar'), 'predB', 1234),
(Node(ns.bsn.Preview, uri='http://example.com/me/preview#hello'), 'predC',
Node(ns.bsn.Entity, uri='http://example.com/me/file#world')),
])
## main ##
if __name__ == '__main__':
unittest.main()
## EOF ##
|
