@@ -5,23 +5,22 @@ import graphql.Scalars
55import graphql.execution.instrumentation.Instrumentation
66import graphql.execution.reactive.SingleSubscriberPublisher
77import graphql.schema.*
8- import org.reactivestreams.Publisher
98
109import java.util.concurrent.atomic.AtomicReference
1110
1211class TestUtils {
1312
1413 static def createServlet (DataFetcher queryDataFetcher = { env -> env. arguments. arg },
1514 DataFetcher mutationDataFetcher = { env -> env. arguments. arg },
16- boolean asyncServletModeEnabled = false ) {
17- DataFetcher subscriptionDataFetcher = { env ->
18- AtomicReference<SingleSubscriberPublisher<String > > publisherRef = new AtomicReference<> ();
19- publisherRef. set(new SingleSubscriberPublisher<> ({ subscription ->
20- publisherRef. get(). offer(env. arguments. arg)
21- publisherRef. get(). noMoreData()
22- }))
23- return publisherRef. get()
24- }) {
15+ boolean asyncServletModeEnabled = false ,
16+ DataFetcher subscriptionDataFetcher = { env ->
17+ AtomicReference<SingleSubscriberPublisher<String > > publisherRef = new AtomicReference<> ();
18+ publisherRef. set(new SingleSubscriberPublisher<> ({ subscription ->
19+ publisherRef. get(). offer(env. arguments. arg)
20+ publisherRef. get(). noMoreData()
21+ }))
22+ return publisherRef. get()
23+ }) {
2524 GraphQLHttpServlet servlet = GraphQLHttpServlet . with(GraphQLConfiguration
2625 .with(createGraphQlSchema(queryDataFetcher, mutationDataFetcher, subscriptionDataFetcher))
2726 .with(createInstrumentedQueryInvoker())
@@ -75,23 +74,27 @@ class TestUtils {
7574 }
7675 field. dataFetcher(mutationDataFetcher)
7776 }
78- .field { field ->
77+ .field { field ->
7978 field. name(" echoFile" )
8079 field. type(Scalars.GraphQLString )
8180 field. argument { argument ->
8281 argument. name(" file" )
8382 argument. type(ApolloScalars.Upload )
8483 }
85- field. dataFetcher( { env -> new String (ByteStreams . toByteArray(env. arguments. file. getInputStream())) } )
84+ field. dataFetcher({ env -> new String (ByteStreams . toByteArray(env. arguments. file. getInputStream())) })
8685 }
87- .field { field ->
86+ .field { field ->
8887 field. name(" echoFiles" )
8988 field. type(GraphQLList . list(Scalars.GraphQLString ))
9089 field. argument { argument ->
9190 argument. name(" files" )
9291 argument. type(GraphQLList . list(GraphQLNonNull . nonNull(ApolloScalars.Upload )))
9392 }
94- field. dataFetcher( { env -> env. arguments. files. collect { new String (ByteStreams . toByteArray(it. getInputStream())) } } )
93+ field. dataFetcher({ env ->
94+ env. arguments. files. collect {
95+ new String (ByteStreams . toByteArray(it. getInputStream()))
96+ }
97+ })
9598 }
9699 .build()
97100
@@ -110,11 +113,11 @@ class TestUtils {
110113
111114
112115 return GraphQLSchema . newSchema()
113- .query(query)
114- .mutation(mutation)
115- .subscription(subscription)
116- .additionalType(ApolloScalars.Upload )
117- .build()
116+ .query(query)
117+ .mutation(mutation)
118+ .subscription(subscription)
119+ .additionalType(ApolloScalars.Upload )
120+ .build()
118121 }
119122
120123}
0 commit comments