|
8 | 8 | from sqlalchemy.dialects.postgresql.array import ARRAY |
9 | 9 | from sqlalchemy.orm import relationship |
10 | 10 | from strawberry.scalars import JSON as StrawberryJSON |
11 | | -from strawberry.type import StrawberryOptional |
| 11 | +from strawberry.types.base import StrawberryOptional |
12 | 12 | from strawberry_sqlalchemy_mapper import StrawberrySQLAlchemyMapper |
13 | 13 |
|
14 | 14 |
|
@@ -228,8 +228,8 @@ class Employee: |
228 | 228 | assert len(additional_types) == 1 |
229 | 229 | mapped_employee_type = additional_types[0] |
230 | 230 | assert mapped_employee_type.__name__ == "Employee" |
231 | | - assert len(mapped_employee_type.__strawberry_definition__._fields) == 2 |
232 | | - employee_type_fields = mapped_employee_type.__strawberry_definition__._fields |
| 231 | + assert len(mapped_employee_type.__strawberry_definition__.fields) == 2 |
| 232 | + employee_type_fields = mapped_employee_type.__strawberry_definition__.fields |
233 | 233 | name = next(iter(filter(lambda f: f.name == "name", employee_type_fields))) |
234 | 234 | assert name.type == str |
235 | 235 | id = next(iter(filter(lambda f: f.name == "id", employee_type_fields))) |
@@ -275,8 +275,8 @@ class Employee: |
275 | 275 | assert len(additional_types) == 2 |
276 | 276 | mapped_employee_type = additional_types[0] |
277 | 277 | assert mapped_employee_type.__name__ == "Employee" |
278 | | - assert len(mapped_employee_type.__strawberry_definition__._fields) == 4 |
279 | | - employee_type_fields = mapped_employee_type.__strawberry_definition__._fields |
| 278 | + assert len(mapped_employee_type.__strawberry_definition__.fields) == 4 |
| 279 | + employee_type_fields = mapped_employee_type.__strawberry_definition__.fields |
280 | 280 | name = next(iter(filter(lambda f: f.name == "department_id", employee_type_fields))) |
281 | 281 | assert type(name.type) == StrawberryOptional |
282 | 282 | id = next(iter(filter(lambda f: f.name == "department", employee_type_fields))) |
|
0 commit comments