Fix inner entity access control (FragmentTemplate)

apollographql · Apr 1, 2023 · 539512e · 539512e
1 parent f9f69de
commit 539512e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/Sources/ApolloCodegenLib/Templates/FragmentTemplate.swift b/Sources/ApolloCodegenLib/Templates/FragmentTemplate.swift
@@ -11,12 +11,14 @@ struct FragmentTemplate: TemplateRenderer {
   let target: TemplateTarget = .operationFile
 
   var template: TemplateString {
-    TemplateString(
+    let accessControl = embeddedAccessControlModifier(target: target)
+
+    return TemplateString(
     """
-    \(embeddedAccessControlModifier(target: target))\
+    \(accessControl)\
     struct \(fragment.name.firstUppercased): \(config.schemaNamespace.firstUppercased)\
     .\(if: isMutable, "Mutable")SelectionSet, Fragment {
-      static var fragmentDefinition: StaticString { ""\"
+      \(accessControl)static var fragmentDefinition: StaticString { ""\"
         \(fragment.definition.source)
         ""\" }
 

diff --git a/Tests/ApolloCodegenTests/CodeGeneration/Templates/FragmentTemplateTests.swift b/Tests/ApolloCodegenTests/CodeGeneration/Templates/FragmentTemplateTests.swift
@@ -320,6 +320,7 @@ class FragmentTemplateTests: XCTestCase {
 
     let expected = """
     public struct TestFragment: TestSchema.SelectionSet, Fragment {
+      public static var fragmentDefinition: StaticString { ""\"
     """
 
     // when
@@ -335,6 +336,7 @@ class FragmentTemplateTests: XCTestCase {
 
     let expected = """
     public struct TestFragment: TestSchema.SelectionSet, Fragment {
+      public static var fragmentDefinition: StaticString { ""\"
     """
 
     // when
@@ -352,6 +354,7 @@ class FragmentTemplateTests: XCTestCase {
 
     let expected = """
     struct TestFragment: TestSchema.SelectionSet, Fragment {
+      static var fragmentDefinition: StaticString { ""\"
     """
 
     // when
@@ -369,6 +372,7 @@ class FragmentTemplateTests: XCTestCase {
 
     let expected = """
     public struct TestFragment: TestSchema.SelectionSet, Fragment {
+      public static var fragmentDefinition: StaticString { ""\"
     """
 
     // when