diff --git a/metaphor/dbt/cloud/discovery_api/graphql_client/__init__.py b/metaphor/dbt/cloud/discovery_api/graphql_client/__init__.py index 0376334f..3f5290ba 100644 --- a/metaphor/dbt/cloud/discovery_api/graphql_client/__init__.py +++ b/metaphor/dbt/cloud/discovery_api/graphql_client/__init__.py @@ -58,7 +58,6 @@ GetJobRunSourcesJobSourcesColumns, ) from .get_job_run_tests import GetJobRunTests, GetJobRunTestsJob, GetJobRunTestsJobTests -from .get_job_tests import GetJobTests, GetJobTestsJob, GetJobTestsJobTests from .get_macro_arguments import ( GetMacroArguments, GetMacroArgumentsEnvironment, @@ -124,9 +123,6 @@ "GetJobRunTests", "GetJobRunTestsJob", "GetJobRunTestsJobTests", - "GetJobTests", - "GetJobTestsJob", - "GetJobTestsJobTests", "GetMacroArguments", "GetMacroArgumentsEnvironment", "GetMacroArgumentsEnvironmentDefinition", diff --git a/metaphor/dbt/cloud/discovery_api/graphql_client/client.py b/metaphor/dbt/cloud/discovery_api/graphql_client/client.py index 9ce62c18..0959d2b5 100644 --- a/metaphor/dbt/cloud/discovery_api/graphql_client/client.py +++ b/metaphor/dbt/cloud/discovery_api/graphql_client/client.py @@ -12,7 +12,6 @@ from .get_job_run_snapshots import GetJobRunSnapshots from .get_job_run_sources import GetJobRunSources from .get_job_run_tests import GetJobRunTests -from .get_job_tests import GetJobTests from .get_macro_arguments import GetMacroArguments from .input_types import MacroDefinitionFilter @@ -22,30 +21,6 @@ def gql(q: str) -> str: class Client(BaseClient): - def get_job_tests(self, job_id: Any, **kwargs: Any) -> GetJobTests: - query = gql( - """ - query GetJobTests($jobId: BigInt!) { - job(id: $jobId) { - tests { - uniqueId - name - status - columnName - executeCompletedAt - dependsOn - } - } - } - """ - ) - variables: Dict[str, object] = {"jobId": job_id} - response = self.execute( - query=query, operation_name="GetJobTests", variables=variables, **kwargs - ) - data = self.get_data(response) - return GetJobTests.model_validate(data) - def get_job_run_models( self, job_id: Any, run_id: Any, **kwargs: Any ) -> GetJobRunModels: diff --git a/metaphor/dbt/cloud/discovery_api/graphql_client/get_job_tests.py b/metaphor/dbt/cloud/discovery_api/graphql_client/get_job_tests.py deleted file mode 100644 index d86af3d4..00000000 --- a/metaphor/dbt/cloud/discovery_api/graphql_client/get_job_tests.py +++ /dev/null @@ -1,29 +0,0 @@ -# Generated by ariadne-codegen -# Source: queries.graphql - -from typing import Any, List, Optional - -from pydantic import Field - -from .base_model import BaseModel - - -class GetJobTests(BaseModel): - job: Optional["GetJobTestsJob"] - - -class GetJobTestsJob(BaseModel): - tests: List["GetJobTestsJobTests"] - - -class GetJobTestsJobTests(BaseModel): - unique_id: str = Field(alias="uniqueId") - name: Optional[str] - status: Optional[str] - column_name: Optional[str] = Field(alias="columnName") - execute_completed_at: Optional[Any] = Field(alias="executeCompletedAt") - depends_on: List[str] = Field(alias="dependsOn") - - -GetJobTests.model_rebuild() -GetJobTestsJob.model_rebuild() diff --git a/metaphor/dbt/cloud/discovery_api/queries.graphql b/metaphor/dbt/cloud/discovery_api/queries.graphql index 3928f638..5db5ab2d 100644 --- a/metaphor/dbt/cloud/discovery_api/queries.graphql +++ b/metaphor/dbt/cloud/discovery_api/queries.graphql @@ -1,16 +1,3 @@ -query GetJobTests($jobId: BigInt!) { - job(id: $jobId) { - tests { - uniqueId - name - status - columnName - executeCompletedAt - dependsOn - } - } -} - query GetJobRunModels($jobId: BigInt!, $runId: BigInt!) { job(id: $jobId, runId: $runId) { models {