yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-11-03 20:27:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(test): fix test execution (#8889)

Browse Source
This commit is contained in:
david-leifker 2023-09-22 20:43:57 -05:00 committed by GitHub
parent 501522d891
commit b905f26d77
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 21 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
build.gradle
View File

@ -289,6 +289,11 @@ subprojects {
}
// https://docs.gradle.org/current/userguide/performance.html
maxParallelForks = Runtime.runtime.availableProcessors().intdiv(2) ?: 1
if (project.configurations.getByName("testImplementation").getDependencies()
.any{ it.getName() == "testng" }) {
useTestNG()
}
}
afterEvaluate {

24
datahub-graphql-core/src/test/java/com/linkedin/datahub/graphql/resolvers/auth/ListAccessTokensResolverTest.java
View File

@ -1,5 +1,6 @@
package com.linkedin.datahub.graphql.resolvers.auth;
import com.datahub.authentication.Authentication;
import com.google.common.collect.ImmutableList;
import com.linkedin.datahub.graphql.QueryContext;
import com.linkedin.datahub.graphql.TestUtils;
@ -8,6 +9,10 @@ import com.linkedin.datahub.graphql.generated.ListAccessTokenInput;
import com.linkedin.datahub.graphql.generated.ListAccessTokenResult;
import com.linkedin.entity.client.EntityClient;
import com.linkedin.metadata.Constants;
import com.linkedin.metadata.query.SearchFlags;
import com.linkedin.metadata.query.filter.SortCriterion;
import com.linkedin.metadata.search.SearchEntityArray;
import com.linkedin.metadata.search.SearchResult;
import graphql.schema.DataFetchingEnvironment;
import java.util.Collections;
import org.mockito.Mockito;
@ -36,14 +41,17 @@ public class ListAccessTokensResolverTest {
Mockito.when(mockEnv.getArgument(Mockito.eq("input"))).thenReturn(input);
final EntityClient mockClient = Mockito.mock(EntityClient.class);
Mockito.when(Mockito.eq(mockClient.filter(
Mockito.eq(Constants.ACCESS_TOKEN_ENTITY_NAME),
Mockito.eq(buildFilter(filters, Collections.emptyList())),
Mockito.notNull(),
Mockito.eq(input.getStart()),
Mockito.eq(input.getCount()),
Mockito.eq(getAuthentication(mockEnv)))))
.thenReturn(null);
final Authentication testAuth = getAuthentication(mockEnv);
Mockito.when(mockClient.search(
Mockito.eq(Constants.ACCESS_TOKEN_ENTITY_NAME),
Mockito.eq(""),
Mockito.eq(buildFilter(filters, Collections.emptyList())),
Mockito.any(SortCriterion.class),
Mockito.eq(input.getStart()),
Mockito.eq(input.getCount()),
Mockito.eq(testAuth),
Mockito.any(SearchFlags.class)))
.thenReturn(new SearchResult().setFrom(0).setNumEntities(0).setPageSize(0).setEntities(new SearchEntityArray()));
final ListAccessTokensResolver resolver = new ListAccessTokensResolver(mockClient);
final ListAccessTokenResult listAccessTokenResult = resolver.get(mockEnv).get();