Add unittest

testData/inspection/extra.py

@@ -0,0 +1,15 @@
+from pydantic import BaseModel, Extra
+
+class A(BaseModel):
+    class Config:
+        extra = Extra.ignore
+
+A(a='123')
+
+
+class B(BaseModel):
+    a: str
+    class Config:
+        extra = Extra.forbid
+
+B(a='abc', <error descr="'b' extra fields not permitted">b='123'</error>)

testData/mock/pydanticv1/__init__.py

@@ -1,4 +1,4 @@
-from .main import BaseModel, BaseConfig, create_model
+from .main import BaseModel, BaseConfig, create_model, Extra
 from .class_validators import validator, root_validator
 from .fields import Field, Schema
 from .env_settings import BaseSettings

testData/mock/pydanticv1/main.py

@@ -1,5 +1,8 @@
 from typing import *
 
+from enum import Enum
+
+
 class BaseModel:
     class Config:
         pass
@@ -11,7 +14,7 @@ def from_orm(cls, obj):
         pass
 
 
-class Extra(str):
+class Extra(str, Enum):
     allow = 'allow'
     ignore = 'ignore'
     forbid = 'forbid'

testSrc/com/koxudaxi/pydantic/PydanticInspectionTest.kt

@@ -79,4 +79,8 @@ open class PydanticInspectionTest : PydanticInspectionBase() {
     fun testKwargConfig() {
         doTest()
     }
+
+    fun testExtra() {
+        doTest()
+    }
 }