Add federation url check to generator

Author: sjin

generator/graphql-kotlin-federation/src/main/kotlin/com/expediagroup/graphql/generator/federation/FederatedSchemaGeneratorHooks.kt

@@ -100,9 +100,9 @@ open class FederatedSchemaGeneratorHooks(
 
     data class LinkSpec(val namespace: String, val imports: Map<String, String>, val url: String? = FEDERATION_SPEC_LATEST_URL)
 
-    public val linkSpecs: MutableMap<String, LinkSpec> = HashMap()
+    val linkSpecs: MutableMap<String, LinkSpec> = HashMap()
 
-    private val federationUrl: String
+    val federationUrl: String
         get() = linkSpecs[FEDERATION_SPEC]?.url ?: FEDERATION_SPEC_LATEST_URL
 
     // workaround to https://github.com/ExpediaGroup/graphql-kotlin/issues/1815

generator/graphql-kotlin-federation/src/test/kotlin/com/expediagroup/graphql/generator/federation/FederatedSchemaGeneratorTest.kt

@@ -20,6 +20,9 @@ import com.expediagroup.graphql.generator.TopLevelObject
 import com.expediagroup.graphql.generator.extensions.print
 import com.expediagroup.graphql.generator.federation.data.queries.simple.NestedQuery
 import com.expediagroup.graphql.generator.federation.data.queries.simple.SimpleQuery
+import com.expediagroup.graphql.generator.federation.directives.FEDERATION_SPEC
+import com.expediagroup.graphql.generator.federation.directives.FEDERATION_SPEC_LATEST_URL
+import com.expediagroup.graphql.generator.federation.directives.FEDERATION_SPEC_URL_PREFIX
 import com.expediagroup.graphql.generator.federation.directives.KEY_DIRECTIVE_NAME
 import com.expediagroup.graphql.generator.federation.types.ENTITY_UNION_NAME
 import graphql.schema.GraphQLUnionType
@@ -278,4 +281,22 @@ class FederatedSchemaGeneratorTest {
         val schema = toFederatedSchema(config, listOf(TopLevelObject(NestedQuery())))
         assertEquals(expectedSchema, schema.print(includeDirectives = true).trim())
     }
+
+    @Test
+    fun `verify federationUrl property returns correct URL`() {
+        val hooks = FederatedSchemaGeneratorHooks(emptyList()).apply {
+            this.linkSpecs[FEDERATION_SPEC] = FederatedSchemaGeneratorHooks.LinkSpec(
+                namespace = FEDERATION_SPEC,
+                imports = emptyMap(),
+                url = "$FEDERATION_SPEC_URL_PREFIX/v2.5"
+            )
+        }
+        assertEquals("$FEDERATION_SPEC_URL_PREFIX/v2.5", hooks.federationUrl)
+    }
+
+    @Test
+    fun `verify federationUrl property returns default when not specified`() {
+        val hooks = FederatedSchemaGeneratorHooks(emptyList())
+        assertEquals(FEDERATION_SPEC_LATEST_URL, hooks.federationUrl)
+    }
 }