@@ -4,10 +4,7 @@ import graphql.GraphQL
44import graphql.execution.AsyncExecutionStrategy
55import graphql.relay.Connection
66import graphql.relay.SimpleListConnection
7- import graphql.schema.DataFetcherFactories
8- import graphql.schema.DataFetchingEnvironment
9- import graphql.schema.GraphQLFieldDefinition
10- import graphql.schema.GraphQLObjectType
7+ import graphql.schema.*
118import graphql.schema.idl.SchemaDirectiveWiring
129import graphql.schema.idl.SchemaDirectiveWiringEnvironment
1310import org.junit.Ignore
@@ -325,7 +322,7 @@ class DirectiveTest {
325322
326323 override fun onField (environment : SchemaDirectiveWiringEnvironment <GraphQLFieldDefinition >): GraphQLFieldDefinition {
327324 val field = environment.element
328- val parentType = environment.fieldsContainer
325+ val parentType = FieldCoordinates .coordinates( environment.fieldsContainer, environment.fieldDefinition)
329326
330327 val originalDataFetcher = environment.codeRegistry.getDataFetcher(parentType, field)
331328 val wrappedDataFetcher = DataFetcherFactories .wrapDataFetcher(originalDataFetcher) { _, value ->
@@ -342,15 +339,15 @@ class DirectiveTest {
342339
343340 override fun onField (environment : SchemaDirectiveWiringEnvironment <GraphQLFieldDefinition >): GraphQLFieldDefinition {
344341 val field = environment.element
345- val parentType = environment.fieldsContainer
342+ val parentType = FieldCoordinates .coordinates( environment.fieldsContainer, environment.fieldDefinition)
346343
347344 val originalDataFetcher = environment.codeRegistry.getDataFetcher(parentType, field)
348345 val wrappedDataFetcher = DataFetcherFactories .wrapDataFetcher(originalDataFetcher) { _, value ->
349346 val string = value as ? String
350347 string + string
351348 }
352349
353- environment.codeRegistry.dataFetcher(parentType, field, wrappedDataFetcher)
350+ environment.codeRegistry.dataFetcher(parentType, wrappedDataFetcher)
354351
355352 return field
356353 }
0 commit comments