karrio-server-graph 2025.5__py3-none-any.whl

karrio/server/graph/tests/base.py

@@ -0,0 +1,153 @@
+import json
+import dataclasses
+from django.urls import reverse
+from rest_framework import status
+from django.contrib.auth import get_user_model
+from rest_framework.test import APITestCase as BaseAPITestCase, APIClient
+
+from karrio.server.core.logging import logger
+from karrio.server.user.models import Token
+import karrio.server.providers.models as providers
+
+
+@dataclasses.dataclass
+class Result:
+    data: dict = None
+    status_code: str = None
+
+
+class GraphTestCase(BaseAPITestCase):
+    def setUp(self) -> None:
+        self.maxDiff = None
+        # Loguru is already configured globally in settings
+
+        # Setup user and API Token.
+        self.user = get_user_model().objects.create_superuser(
+            "admin@example.com", "test"
+        )
+
+        self.token = Token.objects.create(user=self.user, test_mode=False)
+
+        # Create organization for multi-org support (if enabled)
+        from django.conf import settings
+        if settings.MULTI_ORGANIZATIONS:
+            from karrio.server.orgs.models import Organization, TokenLink
+            self.organization = Organization.objects.create(
+                name="Test Organization",
+                slug="test-org"
+            )
+            # Add user as owner (triggers permission sync via signals)
+            owner = self.organization.add_user(self.user, is_admin=True)
+            self.organization.change_owner(owner)
+            self.organization.save()
+
+            # Link token to organization
+            TokenLink.objects.create(
+                item=self.token,
+                org=self.organization
+            )
+
+        # Setup API client.
+        self.client = APIClient()
+        self.client.credentials(HTTP_AUTHORIZATION="Token " + self.token.key)
+
+        # Setup test carrier connections.
+        self.carrier = providers.Carrier.objects.create(
+            carrier_code="canadapost",
+            carrier_id="canadapost",
+            test_mode=False,
+            created_by=self.user,
+            credentials=dict(
+                username="6e93d53968881714",
+                customer_number="2004381",
+                contract_id="42708517",
+                password="0bfa9fcb9853d1f51ee57a",
+            ),
+            capabilities=["pickup", "rating", "tracking", "shipping"],
+        )
+        self.ups_carrier = providers.Carrier.objects.create(
+            carrier_code="ups",
+            carrier_id="ups_package",
+            test_mode=False,
+            created_by=self.user,
+            credentials=dict(
+                client_id="test",
+                client_secret="test",
+                account_number="000000",
+            ),
+            capabilities=["pickup", "rating", "tracking", "shipping"],
+        )
+        self.fedex_carrier = providers.Carrier.objects.create(
+            carrier_code="fedex",
+            carrier_id="fedex_express",
+            test_mode=False,
+            credentials=dict(
+                api_key="test",
+                secret_key="password",
+                account_number="000000",
+                track_api_key="test",
+                track_secret_key="password",
+            ),
+            capabilities=["pickup", "rating", "tracking", "shipping"],
+            is_system=True,
+        )
+        self.dhl_carrier = providers.Carrier.objects.create(
+            carrier_code="dhl_universal",
+            carrier_id="dhl_universal",
+            test_mode=False,
+            is_system=True,
+            credentials=dict(
+                consumer_key="test",
+                consumer_secret="password",
+            ),
+            capabilities=["tracking"],
+        )
+
+    def query(
+        self,
+        query: str,
+        operation_name: str = None,
+        variables: dict = None,
+        org_id: str = None,
+    ) -> Result:
+        url = reverse("karrio.server.graph:graphql")
+        data = dict(
+            query=query,
+            variables=variables,
+            operation_name=operation_name,
+        )
+
+        response = self.client.post(
+            url, data, **({"x-org-id": org_id} if org_id else {})
+        )
+
+        return Result(
+            status_code=response.status_code,
+            data=json.loads(response.content),
+        )
+
+    def getJWTToken(self, email: str, password: str) -> str:
+        url = reverse("jwt-obtain-pair")
+        data = dict(
+            email=email,
+            password=password,
+        )
+        response = self.client.post(url, data)
+
+        return response.data.get("access")
+
+    def assertResponseNoErrors(self, result: Result):
+        if (
+            result.status_code not in [status.HTTP_200_OK, status.HTTP_201_CREATED]
+            or result.data.get("errors") is not None
+        ):
+            logger.error("GraphQL response has errors",
+                        status_code=result.status_code,
+                        response_data=result.data)
+
+        if result.status_code != status.HTTP_201_CREATED:
+            self.assertEqual(result.status_code, status.HTTP_200_OK)
+        else:
+            self.assertEqual(result.status_code, status.HTTP_201_CREATED)
+
+        assert result.data.get("errors") is None

karrio/server/graph/tests/test_carrier_connections.py

@@ -0,0 +1,239 @@
+import karrio.lib as lib
+from unittest.mock import ANY
+from karrio.server.graph.tests.base import GraphTestCase
+
+
+class TestSystemConnections(GraphTestCase):
+    def test_query_system_connections(self):
+        response = self.query(
+            """
+            query get_system_connections {
+              system_connections {
+                edges {
+                  node {
+                    id
+                    carrier_id
+                    carrier_name
+                    active
+                    test_mode
+                  }
+                }
+              }
+            }
+            """,
+            operation_name="get_system_connections",
+        )
+        response_data = response.data
+
+        self.assertResponseNoErrors(response)
+        self.assertDictEqual(
+            lib.to_dict(response_data),
+            SYSTEM_CONNECTIONS,
+        )
+
+
+class TestUserConnections(GraphTestCase):
+    def test_query_user_connections(self):
+        response = self.query(
+            """
+            query get_user_connections {
+                user_connections {
+                edges {
+                  node {
+                    id
+                    carrier_id
+                    carrier_name
+                    active
+                    test_mode
+                    credentials
+                  }
+                }
+              }
+            }
+            """,
+            operation_name="get_user_connections",
+        )
+        response_data = response.data
+
+        self.assertResponseNoErrors(response)
+        self.assertDictEqual(
+            lib.to_dict(response_data),
+            USER_CONNECTIONS,
+        )
+
+    def test_create_user_connection(self):
+        response = self.query(
+            """
+            mutation create_connection($data: CreateCarrierConnectionMutationInput!) {
+              create_carrier_connection(input: $data) {
+                connection {
+                    id
+                    carrier_id
+                    carrier_name
+                    active
+                    test_mode
+                    credentials
+                }
+              }
+            }
+            """,
+            operation_name="create_connection",
+            variables=CONNECTION_DATA,
+        )
+        response_data = response.data
+
+        self.assertResponseNoErrors(response)
+        self.assertDictEqual(response_data, CONNECTION_RESPONSE)
+
+    def test_update_user_connection(self):
+        response = self.query(
+            """
+            mutation update_connection($data: UpdateCarrierConnectionMutationInput!) {
+              update_carrier_connection(input: $data) {
+                connection {
+                    carrier_id
+                    credentials
+                }
+              }
+            }
+            """,
+            operation_name="update_connection",
+            variables={
+                "data": {
+                    "id": self.carrier.id,
+                    **CONNECTION_UPDATE_DATA["data"],
+                },
+            },
+        )
+        response_data = response.data
+
+        self.assertResponseNoErrors(response)
+        self.assertDictEqual(
+            lib.to_dict(response_data),
+            lib.to_dict(CONNECTION_UPDATE_RESPONSE),
+        )
+
+
+SYSTEM_CONNECTIONS = {
+    "data": {
+        "system_connections": {
+            "edges": [
+                {
+                    "node": {
+                        "active": True,
+                        "carrier_id": "dhl_universal",
+                        "carrier_name": "dhl_universal",
+                        "id": ANY,
+                        "test_mode": False,
+                    }
+                },
+                {
+                    "node": {
+                        "active": True,
+                        "carrier_id": "fedex_express",
+                        "carrier_name": "fedex",
+                        "id": ANY,
+                        "test_mode": False,
+                    }
+                },
+            ]
+        }
+    }
+}
+
+USER_CONNECTIONS = {
+    "data": {
+        "user_connections": {
+            "edges": [
+                {
+                    "node": {
+                        "active": True,
+                        "carrier_id": "ups_package",
+                        "carrier_name": "ups",
+                        "credentials": {
+                            "account_number": "000000",
+                            "client_id": "test",
+                            "client_secret": "test",
+                        },
+                        "id": ANY,
+                        "test_mode": False,
+                    }
+                },
+                {
+                    "node": {
+                        "active": True,
+                        "carrier_id": "canadapost",
+                        "carrier_name": "canadapost",
+                        "credentials": {
+                            "contract_id": "42708517",
+                            "customer_number": "2004381",
+                            "password": "0bfa9fcb9853d1f51ee57a",
+                            "username": "6e93d53968881714",
+                        },
+                        "id": ANY,
+                        "test_mode": False,
+                    }
+                },
+            ]
+        }
+    }
+}
+
+CONNECTION_DATA = {
+    "data": {
+        "carrier_name": "sendle",
+        "carrier_id": "sendle",
+        "credentials": {
+            "sendle_id": "test_sendle_id",
+            "api_key": "test_api_key",
+        },
+    }
+}
+
+CONNECTION_RESPONSE = {
+    "data": {
+        "create_carrier_connection": {
+            "connection": {
+                "id": ANY,
+                "active": True,
+                "carrier_id": "sendle",
+                "carrier_name": "sendle",
+                "test_mode": False,
+                "credentials": {
+                    "api_key": "test_api_key",
+                    "sendle_id": "test_sendle_id",
+                    "account_country_code": None,
+                },
+            }
+        }
+    }
+}
+
+CONNECTION_UPDATE_DATA = {
+    "data": {
+        "carrier_id": "canadapost_updated",
+        "credentials": {
+            "username": "6e93d53968881714_updated",
+            "customer_number": "2004381_updated",
+            "contract_id": "42708517_updated",
+            "password": "0bfa9fcb9853d1f51ee57a_updated",
+        },
+    }
+}
+
+CONNECTION_UPDATE_RESPONSE = {
+    "data": {
+        "update_carrier_connection": {
+            "connection": {
+                "carrier_id": "canadapost_updated",
+                "credentials": {
+                    "account_country_code": "CA",
+                    "contract_id": "42708517_updated",
+                    "customer_number": "2004381_updated",
+                    "password": "0bfa9fcb9853d1f51ee57a_updated",
+                    "username": "6e93d53968881714_updated",
+                },
+            }
+        }
+    }
+}