|
1 | 1 | package com.introproventures.graphql.jpa.query.autoconfigure;
|
2 | 2 |
|
| 3 | +import static graphql.GraphQLContext.newContext; |
3 | 4 | import static graphql.annotations.AnnotationsSchemaCreator.newAnnotationsSchema;
|
4 | 5 | import static graphql.schema.FieldCoordinates.coordinates;
|
5 | 6 | import static graphql.schema.GraphQLCodeRegistry.newCodeRegistry;
|
|
9 | 10 | import static graphql.schema.idl.RuntimeWiring.newRuntimeWiring;
|
10 | 11 | import static java.util.Collections.emptyMap;
|
11 | 12 | import static org.assertj.core.api.Assertions.assertThat;
|
| 13 | +import static org.dataloader.DataLoaderRegistry.newRegistry; |
12 | 14 | import static org.springframework.transaction.TransactionDefinition.PROPAGATION_REQUIRED;
|
13 | 15 |
|
14 | 16 | import com.introproventures.graphql.jpa.query.autoconfigure.support.AdditionalGraphQLType;
|
|
58 | 60 | import java.util.stream.Collectors;
|
59 | 61 | import java.util.stream.Stream;
|
60 | 62 | import org.assertj.core.api.InstanceOfAssertFactories;
|
| 63 | +import org.dataloader.DataLoaderRegistry; |
61 | 64 | import org.junit.jupiter.api.Test;
|
62 | 65 | import org.reactivestreams.Publisher;
|
63 | 66 | import org.reflections.Reflections;
|
|
69 | 72 | import org.springframework.context.annotation.Bean;
|
70 | 73 | import org.springframework.context.annotation.Configuration;
|
71 | 74 | import org.springframework.core.io.Resource;
|
| 75 | +import org.springframework.graphql.execution.BatchLoaderRegistry; |
72 | 76 | import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
|
73 | 77 | import org.springframework.stereotype.Component;
|
74 | 78 | import org.springframework.transaction.TransactionDefinition;
|
@@ -101,6 +105,9 @@ public class GraphQLSchemaAutoConfigurationTest {
|
101 | 105 | @Autowired
|
102 | 106 | private GraphQLJpaQueryProperties graphQLJpaQueryProperties;
|
103 | 107 |
|
| 108 | + @Autowired |
| 109 | + BatchLoaderRegistry batchLoaderRegistry; |
| 110 | + |
104 | 111 | @SpringBootApplication
|
105 | 112 | @EnableGraphQLJpaQuerySchema(basePackageClasses = TestEntity.class)
|
106 | 113 | static class Application {
|
@@ -520,4 +527,12 @@ void configuresSubscriptionTransactionalExecutionStrategyCustomizer() {
|
520 | 527 | .isEqualTo(true);
|
521 | 528 | });
|
522 | 529 | }
|
| 530 | + |
| 531 | + @Test |
| 532 | + void batchLoaderRegistry() { |
| 533 | + DataLoaderRegistry dataLoaderRegistry = newRegistry().build(); |
| 534 | + batchLoaderRegistry.registerDataLoaders(dataLoaderRegistry, newContext().build()); |
| 535 | + |
| 536 | + assertThat(dataLoaderRegistry.getDataLoadersMap()).isNotEmpty().containsKeys("TestEntity.children"); |
| 537 | + } |
523 | 538 | }
|
0 commit comments