diff --git a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java index 148308b0a7..44f39720f0 100644 --- a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java +++ b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java @@ -185,6 +185,7 @@ import com.linkedin.datahub.graphql.types.test.TestType; import com.linkedin.datahub.graphql.types.usage.UsageType; import com.linkedin.entity.client.EntityClient; import com.linkedin.metadata.config.IngestionConfiguration; +import com.linkedin.metadata.config.TestsConfiguration; import com.linkedin.metadata.entity.EntityService; import com.linkedin.metadata.graph.GraphClient; import com.linkedin.metadata.models.registry.EntityRegistry; @@ -250,6 +251,7 @@ public class GmsGraphQLEngine { private final AuthorizationConfiguration authorizationConfiguration; private final VisualConfiguration visualConfiguration; private final TelemetryConfiguration telemetryConfiguration; + private final TestsConfiguration testsConfiguration; private final DatasetType datasetType; private final CorpUserType corpUserType; @@ -320,7 +322,8 @@ public class GmsGraphQLEngine { final TimelineService timelineService, final boolean supportsImpactAnalysis, final VisualConfiguration visualConfiguration, - final TelemetryConfiguration telemetryConfiguration + final TelemetryConfiguration telemetryConfiguration, + final TestsConfiguration testsConfiguration ) { this.entityClient = entityClient; @@ -343,6 +346,7 @@ public class GmsGraphQLEngine { this.authorizationConfiguration = Objects.requireNonNull(authorizationConfiguration); this.visualConfiguration = visualConfiguration; this.telemetryConfiguration = telemetryConfiguration; + this.testsConfiguration = testsConfiguration; this.datasetType = new DatasetType(entityClient); this.corpUserType = new CorpUserType(entityClient); @@ -538,7 +542,11 @@ public class GmsGraphQLEngine { this.ingestionConfiguration, this.authenticationConfiguration, this.authorizationConfiguration, - supportsImpactAnalysis, this.visualConfiguration, this.telemetryConfiguration)) + this.supportsImpactAnalysis, + this.visualConfiguration, + this.telemetryConfiguration, + this.testsConfiguration + )) .dataFetcher("me", new MeResolver(this.entityClient)) .dataFetcher("search", new SearchResolver(this.entityClient)) .dataFetcher("searchAcrossEntities", new SearchAcrossEntitiesResolver(this.entityClient)) diff --git a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/config/AppConfigResolver.java b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/config/AppConfigResolver.java index e024937b6e..8603ffe47f 100644 --- a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/config/AppConfigResolver.java +++ b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/config/AppConfigResolver.java @@ -14,8 +14,10 @@ import com.linkedin.datahub.graphql.generated.PoliciesConfig; import com.linkedin.datahub.graphql.generated.Privilege; import com.linkedin.datahub.graphql.generated.ResourcePrivileges; import com.linkedin.datahub.graphql.generated.TelemetryConfig; +import com.linkedin.datahub.graphql.generated.TestsConfig; import com.linkedin.datahub.graphql.generated.VisualConfiguration; import com.linkedin.metadata.config.IngestionConfiguration; +import com.linkedin.metadata.config.TestsConfiguration; import com.linkedin.metadata.telemetry.TelemetryConfiguration; import com.linkedin.metadata.version.GitVersion; import graphql.schema.DataFetcher; @@ -37,6 +39,7 @@ public class AppConfigResolver implements DataFetcher