Adds Final Tests for Const Type

tests/parser/test_const_type_parser.py

@@ -1,23 +1,49 @@
-from typing_extensions import Annotated, get_args, get_origin
-from webbrowser import get
 from jambo.parser import ConstTypeParser
 
+from typing_extensions import Annotated, get_args, get_origin
+
 from unittest import TestCase
 
 
 class TestConstTypeParser(TestCase):
-    def test_parse_const_type(self):
+    def test_const_type_parser(self):
         parser = ConstTypeParser()
 
         expected_const_value = "United States of America"
-        properties = {
-            "const": expected_const_value
-        }
+        properties = {"const": expected_const_value}
 
-        parsed_type, parsed_properties = parser.from_properties(
+        parsed_type, parsed_properties = parser.from_properties_impl(
             "country", properties
         )
 
         self.assertEqual(get_origin(parsed_type), Annotated)
+        self.assertIn(str, get_args(parsed_type))
 
-        self.assertIn(str, get_args(parsed_type))
+        self.assertEqual(parsed_properties["default"], expected_const_value)
+
+    def test_const_type_parser_invalid_properties(self):
+        parser = ConstTypeParser()
+
+        expected_const_value = "United States of America"
+        properties = {"notConst": expected_const_value}
+
+        with self.assertRaises(ValueError) as context:
+            parser.from_properties_impl("invalid_country", properties)
+
+        self.assertIn(
+            "Const type invalid_country must have 'const' property defined",
+            str(context.exception),
+        )
+
+    def test_const_type_parser_invalid_const_value(self):
+        parser = ConstTypeParser()
+
+        properties = {"const": {}}
+
+        with self.assertRaises(ValueError) as context:
+            parser.from_properties_impl("invalid_country", properties)
+
+        self.assertIn(
+            "Const type invalid_country must have 'const' value of allowed types",
+            str(context.exception),
+        )

tests/test_schema_converter.py

@@ -634,3 +634,26 @@ class TestSchemaConverter(TestCase):
 
         obj = Model()
         self.assertEqual(obj.status.value, "active")
+
+    def test_const_type_parser(self):
+        schema = {
+            "title": "Country",
+            "type": "object",
+            "properties": {
+                "name": {
+                    "const": "United States of America",
+                }
+            },
+            "required": ["name"],
+        }
+
+        Model = SchemaConverter.build(schema)
+
+        obj = Model()
+        self.assertEqual(obj.name, "United States of America")
+
+        with self.assertRaises(ValueError):
+            obj.name = "Canada"
+
+        with self.assertRaises(ValueError):
+            Model(name="Canada")