Minor Fixes in EnumTypeParser and Adds Better UnitTests

tests/parser/test_enum_type_parser.py

@@ -66,7 +66,7 @@ class TestEnumTypeParser(TestCase):
         self.assertEqual(
             set(parsed_type.__members__.keys()), {"VALUE1", "VALUE2", "VALUE3"}
         )
-        self.assertEqual(parsed_properties["default"], "value2")
+        self.assertEqual(parsed_properties["default"].value, "value2")
 
     def test_enum_type_parser_throws_invalid_default(self):
         parser = EnumTypeParser()

tests/test_schema_converter.py

@@ -597,3 +597,40 @@ class TestSchemaConverter(TestCase):
         self.assertEqual(obj.age, 30)
         self.assertEqual(obj.address.street, "123 Main St")
         self.assertEqual(obj.address.city, "Springfield")
+
+    def test_enum_type_parser(self):
+        schema = {
+            "title": "Person",
+            "type": "object",
+            "properties": {
+                "status": {
+                    "type": "string",
+                    "enum": ["active", "inactive", "pending"],
+                }
+            },
+            "required": ["status"],
+        }
+
+        Model = SchemaConverter.build(schema)
+
+        obj = Model(status="active")
+        self.assertEqual(obj.status.value, "active")
+
+    def test_enum_type_parser_with_default(self):
+        schema = {
+            "title": "Person",
+            "type": "object",
+            "properties": {
+                "status": {
+                    "type": "string",
+                    "enum": ["active", "inactive", "pending"],
+                    "default": "active",
+                }
+            },
+            "required": ["status"],
+        }
+
+        Model = SchemaConverter.build(schema)
+
+        obj = Model()
+        self.assertEqual(obj.status.value, "active")