add tests for JsonPointer.to_last()

stefankoegl · stefankoegl · commit ef7a41cdceca · 2013-07-11T21:46:18.000+02:00
diff --git a/tests.py b/tests.py
@@ -92,10 +92,29 @@ def test_oob(self):
         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']
 
