test: add tests for printing QueryBuilder in GraphQL syntax

Author: ericbstie

gqlrequests/builder.py

@@ -43,7 +43,7 @@ def __str__(cls):
         output = f"type {cls.__name__} {{\n"
         for key, value in cls._resolved_fields.items():
             output += f"    {key}: {value.__name__}\n"
-        output += "}"
+        output += "}\n"
         return output
 
 class QueryBuilder(metaclass=QueryBuilderMeta):

tests/test_creating_builder.py

@@ -84,4 +84,38 @@ class InvalidType(gqlrequests.QueryBuilder):
 
 def test_invalid_type_throws_value_error():
     with pytest.raises(ValueError) as e:
-        InvalidType().build()
+        InvalidType().build()
+
+        
+def test_string_representation_of_class_shows_graphql_syntax():
+    class EveryType(gqlrequests.QueryBuilder):
+        id: int
+        age: int
+        money: float
+        name: str
+        company: bool
+
+    correct_string = """
+type EveryType {
+    id: int
+    age: int
+    money: float
+    name: str
+    company: bool
+}
+"""[1:]
+    assert str(EveryType) == correct_string
+
+def test_string_representation_of_nested_class_shows_graphql_syntax():
+    correct_string = """
+type NestedType {
+    something: SomethingType
+}
+"""[1:]
+    class SomethingType(gqlrequests.QueryBuilder):
+        id: int
+
+    class NestedType(gqlrequests.QueryBuilder):
+        something: SomethingType
+    
+    assert str(NestedType) == correct_string