Add first unit tests for jinja filter funcs

SUNET · Apr 28, 2021 · aefe3b7 · aefe3b7
1 parent 008ac0d
commit aefe3b7
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/src/cnaas_nms/tools/tests/test_jinja_filters.py b/src/cnaas_nms/tools/tests/test_jinja_filters.py
@@ -0,0 +1,25 @@
+import unittest
+
+from cnaas_nms.tools.jinja_filters import increment_ip
+
+class JinjaFilterTests(unittest.TestCase):
+
+    def test_increment_ipv4_plain(self):
+        self.assertEqual(increment_ip('10.0.0.1'), '10.0.0.2')
+        self.assertEqual(increment_ip(increment_ip('10.0.0.1')), '10.0.0.3')
+        self.assertEqual(increment_ip('10.0.0.3', 4), '10.0.0.7')
+        self.assertEqual(increment_ip('10.0.0.1', 255), '10.0.1.0')
+        self.assertEqual(increment_ip('10.0.0.2', -1), '10.0.0.1')
+
+    def test_increment_ipv4_prefix(self):
+        self.assertEqual(increment_ip('10.0.0.1/24'), '10.0.0.2/24')
+        self.assertNotEqual(increment_ip('10.0.0.1/24', 1), '10.0.0.2/32')
+        self.assertEqual(increment_ip(increment_ip('10.0.0.1/24')), '10.0.0.3/24')
+        self.assertEqual(increment_ip('10.0.0.3/24', 4), '10.0.0.7/24')
+        with self.assertRaises(ValueError):
+            increment_ip('10.0.0.1/24', 255)
+        self.assertEqual(increment_ip('10.0.0.2/24', -1), '10.0.0.1/24')
+        self.assertEqual(increment_ip('10.0.0.1/16', 255), '10.0.1.0/16')
+
+if __name__ == '__main__':
+    unittest.main()