@@ -27,9 +27,12 @@ class GraphqlFieldVisibilityTest extends Specification {
2727 def " visibility is enforced" () {
2828
2929 GraphqlFieldVisibility banNameVisibility = newBlock(). addPattern(" .*\\ .name" ). build()
30+ GraphQLCodeRegistry codeRegistry = GraphQLCodeRegistry . newCodeRegistry()
31+ .fieldVisibility(banNameVisibility)
32+ .build()
3033 def schema = GraphQLSchema . newSchema()
3134 .query(StarWarsSchema . queryType)
32- .fieldVisibility(banNameVisibility )
35+ .codeRegistry(codeRegistry )
3336 .build()
3437
3538 def graphQL = GraphQL . newGraphQL(schema). build()
@@ -56,13 +59,13 @@ class GraphqlFieldVisibilityTest extends Specification {
5659 }
5760
5861 def " introspection visibility is enforced" () {
59-
60-
6162 given :
62-
63+ GraphQLCodeRegistry codeRegistry = GraphQLCodeRegistry . newCodeRegistry()
64+ .fieldVisibility(fieldVisibility)
65+ .build()
6366 def schema = GraphQLSchema . newSchema()
6467 .query(StarWarsSchema . queryType)
65- .fieldVisibility(fieldVisibility )
68+ .codeRegistry(codeRegistry )
6669 .build()
6770
6871 def graphQL = GraphQL . newGraphQL(schema). build()
@@ -93,10 +96,12 @@ class GraphqlFieldVisibilityTest extends Specification {
9396
9497 def " introspection turned off via field visibility" () {
9598 given :
96-
99+ GraphQLCodeRegistry codeRegistry = GraphQLCodeRegistry . newCodeRegistry()
100+ .fieldVisibility(NO_INTROSPECTION_FIELD_VISIBILITY )
101+ .build()
97102 def schema = GraphQLSchema . newSchema()
98103 .query(StarWarsSchema . queryType)
99- .fieldVisibility( NO_INTROSPECTION_FIELD_VISIBILITY )
104+ .codeRegistry(codeRegistry )
100105 .build()
101106
102107 def graphQL = GraphQL . newGraphQL(schema). build()
@@ -278,16 +283,15 @@ enum Episode {
278283 }
279284
280285 def " ensure execution cant get to the field" () {
281-
282-
283286 when :
287+ GraphQLCodeRegistry codeRegistry = GraphQLCodeRegistry . newCodeRegistry()
288+ .fieldVisibility(ban([' Droid.appearsIn' ]))
289+ .build()
284290 def schema = GraphQLSchema . newSchema()
285291 .query(StarWarsSchema . queryType)
286- .fieldVisibility(ban([ ' Droid.appearsIn ' ]) )
292+ .codeRegistry(codeRegistry )
287293 .build()
288294
289-
290-
291295 def executionStrategy = new AsyncExecutionStrategy () {
292296
293297 // gives us access to this unit tested method
@@ -340,9 +344,12 @@ enum Episode {
340344 er. getData() == [" hello" : " world" ]
341345
342346 when :
347+ GraphQLCodeRegistry codeRegistry = GraphQLCodeRegistry . newCodeRegistry()
348+ .fieldVisibility(ban([' InputType.closedField' ]))
349+ .build()
343350 schema = GraphQLSchema . newSchema()
344351 .query(inputQueryType)
345- .fieldVisibility(ban([ ' InputType.closedField ' ]) )
352+ .codeRegistry(codeRegistry )
346353 .build()
347354
348355 graphQL = GraphQL . newGraphQL(schema). build()
@@ -366,9 +373,12 @@ enum Episode {
366373
367374 given :
368375
376+ GraphQLCodeRegistry codeRegistry = GraphQLCodeRegistry . newCodeRegistry()
377+ .fieldVisibility(fieldVisibility)
378+ .build()
369379 def schema = GraphQLSchema . newSchema()
370380 .query(inputQueryType)
371- .fieldVisibility(fieldVisibility )
381+ .codeRegistry(codeRegistry )
372382 .build()
373383
374384 def graphQL = GraphQL . newGraphQL(schema). build()
0 commit comments