add tests

Author: amin-farjadi

tests/client_generators/package_generator/test_generated_files.py

@@ -764,3 +764,48 @@ def test_generate_creates_client_with_custom_scalars_imports(
         f"{generator.client_file_name}.py"
     ).open() as client_file:
         assert "from .abc import ScalarABC" in client_file.read()
+
+
+def test_generate_models_only(tmp_path, schema, async_base_client_import):
+    package_name = "test_graphql_client"
+    generator = PackageGenerator(
+        package_name=package_name,
+        target_path=tmp_path.as_posix(),
+        schema=schema,
+        init_generator=InitFileGenerator(),
+        client_generator=ClientGenerator(
+            base_client_import=async_base_client_import,
+            arguments_generator=ArgumentsGenerator(schema=schema),
+        ),
+        enums_generator=EnumsGenerator(schema=schema),
+        input_types_generator=InputTypesGenerator(schema=schema),
+        fragments_generator=FragmentsGenerator(schema=schema, fragments_definitions={}),
+        models_only=True,
+    )
+    query_str = """
+    query CustomQuery($id: ID!) {
+        query1(id: $id) {
+            field1
+        }
+    }
+    """
+    generator.add_operation(parse(query_str).definitions[0])
+    generated_files = generator.generate()
+
+    package_path = tmp_path / package_name
+    assert (package_path / "__init__.py").exists()
+    assert (package_path / "base_model.py").exists()
+    assert (package_path / f"{generator.enums_module_name}.py").exists()
+    assert (package_path / f"{generator.input_types_module_name}.py").exists()
+    assert (package_path / "custom_query.py").exists()
+    assert "custom_query.py" in generated_files
+    assert not (package_path / "client.py").exists()
+    assert not (package_path / generator.base_client_file_path.name).exists()
+    assert not (package_path / EXCEPTIONS_FILE_PATH.name).exists()
+    assert "client.py" not in generated_files
+    assert EXCEPTIONS_FILE_PATH.name not in generated_files
+    init_content = (package_path / "__init__.py").read_text()
+    assert "from .base_model import BaseModel, Upload" in init_content
+    assert "Client" not in init_content
+    assert "AsyncBaseClient" not in init_content
+    assert "GraphQLClientError" not in init_content

tests/main/models_only/example/expected_client/__init__.py

@@ -0,0 +1,16 @@
+from .base_model import BaseModel, Upload
+from .create_user import CreateUser, CreateUserUserCreate
+from .enums import Color
+from .input_types import UserCreateInput
+from .list_users import ListUsers, ListUsersUsers
+
+__all__ = [
+    "BaseModel",
+    "Color",
+    "CreateUser",
+    "CreateUserUserCreate",
+    "ListUsers",
+    "ListUsersUsers",
+    "Upload",
+    "UserCreateInput",
+]

tests/main/models_only/example/expected_client/base_model.py

@@ -0,0 +1,28 @@
+from io import IOBase
+
+from pydantic import BaseModel as PydanticBaseModel
+from pydantic import ConfigDict
+
+
+class UnsetType:
+    def __bool__(self) -> bool:
+        return False
+
+
+UNSET = UnsetType()
+
+
+class BaseModel(PydanticBaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        arbitrary_types_allowed=True,
+        protected_namespaces=(),
+    )
+
+
+class Upload:
+    def __init__(self, filename: str, content: IOBase, content_type: str):
+        self.filename = filename
+        self.content = content
+        self.content_type = content_type

tests/main/models_only/example/expected_client/create_user.py

@@ -0,0 +1,17 @@
+from typing import Optional
+
+from pydantic import Field
+
+from .base_model import BaseModel
+
+
+class CreateUser(BaseModel):
+    user_create: Optional["CreateUserUserCreate"] = Field(alias="userCreate")
+
+
+class CreateUserUserCreate(BaseModel):
+    id: str
+    email: str
+
+
+CreateUser.model_rebuild()

tests/main/models_only/example/expected_client/enums.py

@@ -0,0 +1,9 @@
+from enum import Enum
+
+
+class Color(str, Enum):
+    BLACK = "BLACK"
+    WHITE = "WHITE"
+    RED = "RED"
+    GREEN = "GREEN"
+    BLUE = "BLUE"

tests/main/models_only/example/expected_client/input_types.py

@@ -0,0 +1,13 @@
+from typing import Optional
+
+from pydantic import Field
+
+from .base_model import BaseModel
+from .enums import Color
+
+
+class UserCreateInput(BaseModel):
+    first_name: Optional[str] = Field(alias="firstName", default=None)
+    last_name: Optional[str] = Field(alias="lastName", default=None)
+    email: str
+    favourite_color: Optional[Color] = Field(alias="favouriteColor", default=None)

tests/main/models_only/example/expected_client/list_users.py

@@ -0,0 +1,20 @@
+from typing import Optional
+
+from pydantic import Field
+
+from .base_model import BaseModel
+from .enums import Color
+
+
+class ListUsers(BaseModel):
+    users: list["ListUsersUsers"]
+
+
+class ListUsersUsers(BaseModel):
+    id: str
+    first_name: Optional[str] = Field(alias="firstName")
+    email: str
+    favourite_color: Optional[Color] = Field(alias="favouriteColor")
+
+
+ListUsers.model_rebuild()

tests/main/models_only/example/pyproject.toml

@@ -0,0 +1,5 @@
+[tool.ariadne-codegen]
+schema_path = "schema.graphql"
+queries_path = "queries.graphql"
+include_comments = "none"
+target_package_name = "example_client"

tests/main/models_only/example/queries.graphql

@@ -0,0 +1,15 @@
+mutation CreateUser($userData: UserCreateInput!) {
+  userCreate(userData: $userData) {
+    id
+    email
+  }
+}
+
+query ListUsers {
+  users {
+    id
+    firstName
+    email
+    favouriteColor
+  }
+}

tests/main/models_only/example/schema.graphql

@@ -0,0 +1,35 @@
+schema {
+  query: Query
+  mutation: Mutation
+}
+
+type Query {
+  users(country: String): [User!]!
+}
+
+type Mutation {
+  userCreate(userData: UserCreateInput!): User
+}
+
+input UserCreateInput {
+  firstName: String
+  lastName: String
+  email: String!
+  favouriteColor: Color
+}
+
+type User {
+  id: ID!
+  firstName: String
+  lastName: String
+  email: String!
+  favouriteColor: Color
+}
+
+enum Color {
+  BLACK
+  WHITE
+  RED
+  GREEN
+  BLUE
+}