1 files changed, 115 insertions, 0 deletions
diff --git a/test/lib/test_naming_policy.py b/test/lib/test_naming_policy.py
new file mode 100644
index 0000000..c9b0cd2
--- /dev/null
+++ b/test/lib/test_naming_policy.py
@@ -0,0 +1,115 @@
+
+# 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'))
+        # 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'))
+
+
+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 ##