Skip to content

Commit 7bcbc82

Browse files
committed
added propety test
1 parent 668deb0 commit 7bcbc82

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

tests/test_mapper.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,28 @@ class SampleEnum(enum.Enum):
172172
)
173173

174174

175+
def test_convert_relationship_to_strawberry_type():
176+
strawberry_sqlalchemy_mapper = StrawberrySQLAlchemyMapper()
177+
_, Department = _create_employee_and_department_tables()
178+
employees_property = Department.employees.property
179+
assert (
180+
strawberry_sqlalchemy_mapper._convert_relationship_to_strawberry_type(
181+
employees_property
182+
).__name__
183+
== "EmployeeConnection"
184+
)
185+
186+
187+
def test_get_relationship_is_optional():
188+
strawberry_sqlalchemy_mapper = StrawberrySQLAlchemyMapper()
189+
_, Department = _create_employee_and_department_tables()
190+
employees_property = Department.employees.property
191+
assert (
192+
strawberry_sqlalchemy_mapper._get_relationship_is_optional(employees_property)
193+
is True
194+
)
195+
196+
175197
def test_add_annotation():
176198
strawberry_sqlalchemy_mapper = StrawberrySQLAlchemyMapper()
177199

0 commit comments

Comments
 (0)