diff tests/test_schema.py @ 16:2a9e7c4b717e

Problem severity now is also an enum
author Franz Glasner <fzglas.hg@dom66.de>
date Fri, 07 Jul 2023 02:33:28 +0200
parents 696b83f29363
children c3a0fe8d4587
line wrap: on
line diff
--- a/tests/test_schema.py	Fri Jul 07 02:17:45 2023 +0200
+++ b/tests/test_schema.py	Fri Jul 07 02:33:28 2023 +0200
@@ -10,7 +10,7 @@
 
 import data_schema
 import data_schema.util
-from data_schema import ERRORS, WARNINGS
+from data_schema import SEVERITY, ERRORS, WARNINGS
 
 
 TYPE_RE = type(re.compile(r"\A.+\Z"))
@@ -1334,6 +1334,7 @@
         code = ERRORS.E10001
         self.assertEqual("ERRORS.E10001", str(code))
         self.assertEqual("E10001", code.name)
+        self.assertTrue(isinstance(code, ERRORS))
 
     def test_schema_must_be_a_dict_alike(self):
         try:
@@ -1349,11 +1350,11 @@
 
     def test_error_ctor(self):
         v = data_schema.ValidationProblem(code=ERRORS.E10000)
-        self.assertEqual(data_schema.ERROR, v.severity)
+        self.assertEqual(SEVERITY.ERROR, v.severity)
 
     def test_warning_ctor(self):
         v = data_schema.ValidationProblem(code=WARNINGS.W80000)
-        self.assertEqual(data_schema.WARNING, v.severity)
+        self.assertEqual(SEVERITY.WARNING, v.severity)
 
     def test_d1(self):
         x = list(data_schema.validate({}, {"type": "dict"}))
@@ -1372,7 +1373,7 @@
     def test_d2(self):
         x = list(data_schema.validate([], {"type": "map"}))
         self.assertEqual(1, len(x))
-        self.assertEqual(data_schema.ERROR, x[0].severity)
+        self.assertEqual(SEVERITY.ERROR, x[0].severity)
         self.assertEqual(ERRORS.E10000, x[0].code)
 
     def test_d3(self):
@@ -1381,10 +1382,10 @@
             {"type": "dict",
              "required": ["key2"]}))
         self.assertEqual(2, len(x))
-        self.assertEqual(data_schema.ERROR, x[0].severity)
+        self.assertEqual(SEVERITY.ERROR, x[0].severity)
         self.assertEqual(ERRORS.E10004, x[0].code)
         self.assertEqual("key", x[0].hint)
-        self.assertEqual(data_schema.ERROR, x[1].severity)
+        self.assertEqual(SEVERITY.ERROR, x[1].severity)
         self.assertEqual(ERRORS.E10005, x[1].code)
         self.assertEqual(["key2"], x[1].hint)