add tests for JsonPointer.to_last()

1 files changed, 19 insertions, 0 deletions

diff --git a/tests.py b/tests.py
index 10a56e8..b75723a 100755
--- a/tests.py
+++ b/tests.py
@@ -92,10 +92,29 @@ class WrongInputTests(unittest.TestCase):
         self.assertRaises(JsonPointerException, resolve_pointer, doc, '/10')
 
 
+class ToLastTests(unittest.TestCase):
+
+    def test_empty_path(self):
+        doc = {'a': [1, 2, 3]}
+        ptr = JsonPointer('')
+        last, nxt = ptr.to_last(doc)
+        self.assertEqual(doc, last)
+        self.assertTrue(nxt is None)
+
+
+    def test_path(self):
+        doc = {'a': [{'b': 1, 'c': 2}, 5]}
+        ptr = JsonPointer('/a/0/b')
+        last, nxt = ptr.to_last(doc)
+        self.assertEqual(last, {'b': 1, 'c': 2})
+        self.assertEqual(nxt, 'b')
+
+
 suite = unittest.TestSuite()
 suite.addTest(unittest.makeSuite(SpecificationTests))
 suite.addTest(unittest.makeSuite(ComparisonTests))
 suite.addTest(unittest.makeSuite(WrongInputTests))
+suite.addTest(unittest.makeSuite(ToLastTests))
 
 modules = ['jsonpointer']