Fix example to work with Pydantic V2, add test for examples

Author: dantheman39

.gitignore

@@ -75,3 +75,7 @@ dmypy.json
 
 # PyCharm / IntelliJ
 .idea/
+
+# vim
+*.swp
+*.swo

examples/departments.py

@@ -21,8 +21,8 @@ class SalaryModel(pydantic.BaseModel):
 
 
 class EmployeeModel(PersonModel):
-    hired_on: datetime.datetime = None
-    salary: T.Optional[SalaryModel]
+    hired_on: T.Optional[datetime.datetime] = None
+    salary: T.Optional[SalaryModel] = None
 
 
 class ManagerModel(EmployeeModel):
@@ -96,13 +96,13 @@ def resolve_list_departments(self, info):
                         salary=SalaryModel(rating="GS-9", amount=75000.23),
                         hired_on=datetime.datetime(2019, 1, 1, 15, 26),
                     ),
-                    EmployeeModel(id=uuid.uuid4(), name="Derek"),
+                    EmployeeModel(id=uuid.uuid4(), name="Derek", salary=None),
                 ],
             )
         ]
 
 
-if __name__ == "__main__":
+def main():
     schema = graphene.Schema(query=Query)
     query = """
         query {
@@ -128,7 +128,10 @@ def resolve_list_departments(self, info):
         }
     }
     """
-    result = schema.execute(query)
+    return schema.execute(query)
+
 
+if __name__ == "__main__":
+    result = main()
     print(result.errors)
     print(json.dumps(result.data, indent=2))

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "graphene_pydantic"
-version = "0.6.0"
+version = "0.6.1"
 description = "Graphene Pydantic integration"
 readme = "README.md"
 repository = "https://github.com/graphql-python/graphene-pydantic"

tests/test_examples/__init__.py

@@ -0,0 +1,30 @@
+from examples import departments
+
+
+def test_departments():
+    result = departments.main()
+    assert not result.errors
+
+    deps = result.data["listDepartments"]
+    assert len(deps) == 1
+
+    employees = deps[0]["employees"]
+    assert len(employees) == 3
+
+    def employee_by_name(employees, name):
+        return [e for e in employees if e["name"] == name][0]
+
+    jason = employee_by_name(employees, "Jason")
+    carmen = employee_by_name(employees, "Carmen")
+    derek = employee_by_name(employees, "Derek")
+
+    # Jason is a manager
+    assert jason["teamSize"] == 2
+    assert carmen.get("teamSize") is None
+
+    # some sanity checks on optional fields,
+    # knowing what the test data is
+    assert jason.get("hiredOn") is None
+    assert carmen.get("hiredOn") is not None
+    assert carmen["salary"]["rating"] == "GS-9"
+    assert derek["salary"] is None