Update codegen tests to source copyright notice from new place.

smithy-lang · Jun 28, 2024 · f444349 · f444349
1 parent cfe3770
commit f444349
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 9 deletions.
diff --git a/smithy-swift-codegen/src/test/kotlin/EnumGeneratorTests.kt b/smithy-swift-codegen/src/test/kotlin/EnumGeneratorTests.kt
@@ -35,7 +35,7 @@ class EnumGeneratorTests {
 
         val contents = writer.toString()
 
-        contents.shouldContain(SwiftWriter.GENERATED_FILE_HEADER)
+        contents.shouldContain(settings.copyrightNotice)
 
         val expectedGeneratedEnum = """
 /// Really long multi-line Documentation for the enum
@@ -94,7 +94,7 @@ public enum MyEnum: Swift.Equatable, Swift.RawRepresentable, Swift.CaseIterable,
 
         val contents = writer.toString()
 
-        contents.shouldContain(SwiftWriter.GENERATED_FILE_HEADER)
+        contents.shouldContain(settings.copyrightNotice)
 
         val expectedGeneratedEnum = """
 /// Really long multi-line Documentation for the enum

diff --git a/smithy-swift-codegen/src/test/kotlin/StructureGeneratorTests.kt b/smithy-swift-codegen/src/test/kotlin/StructureGeneratorTests.kt
@@ -31,7 +31,7 @@ class StructureGeneratorTests {
 
         val contents = writer.toString()
 
-        contents.shouldContain(SwiftWriter.GENERATED_FILE_HEADER)
+        contents.shouldContain(swiftSettings.copyrightNotice)
         val expectedGeneratedStructure =
             """
             /// This is documentation about the shape.
@@ -253,7 +253,7 @@ public struct RecursiveShapesInputOutputLists {
 
         val contents = writer.toString()
 
-        contents.shouldContain(SwiftWriter.GENERATED_FILE_HEADER)
+        contents.shouldContain(swiftSettings.copyrightNotice)
         val expectedGeneratedStructure = """
 public struct MyError: ClientRuntime.ModeledError, ClientRuntime.ServiceError, ClientRuntime.HTTPError, Swift.Error {
 

diff --git a/smithy-swift-codegen/src/test/kotlin/SwiftDelegatorTests.kt b/smithy-swift-codegen/src/test/kotlin/SwiftDelegatorTests.kt
@@ -43,7 +43,7 @@ class SwiftDelegatorTests {
         delegator.useShapeWriter(getFooInputShape, { writer -> writer.write("Hello!") })
         delegator.flushWriters()
         assertEquals(
-            SwiftWriter.GENERATED_FILE_HEADER + "\n\nHello!\n",
+            settings.copyrightNotice + "\n\nHello!\n",
             manifest.getFileString("Sources/example/models/GetFooInput.swift").get()
         )
     }
@@ -62,7 +62,7 @@ class SwiftDelegatorTests {
         delegator.useShapeWriter(getFooInputShape, { writer -> writer.write("Goodbye!") })
         delegator.flushWriters()
         assertEquals(
-            SwiftWriter.GENERATED_FILE_HEADER + "\n\nHello!\n\nGoodbye!\n",
+            settings.copyrightNotice + "\n\nHello!\n\nGoodbye!\n",
             manifest.getFileString("Sources/example/models/GetFooInput.swift").get()
         )
     }

diff --git a/smithy-swift-codegen/src/test/kotlin/UnionGeneratorTests.kt b/smithy-swift-codegen/src/test/kotlin/UnionGeneratorTests.kt
@@ -39,7 +39,7 @@ class UnionGeneratorTests {
 
         val contents = writer.toString()
 
-        contents.shouldContain(SwiftWriter.GENERATED_FILE_HEADER)
+        contents.shouldContain(settings.copyrightNotice)
 
         val expectedGeneratedEnum =
             """
@@ -83,7 +83,7 @@ class UnionGeneratorTests {
 
         val contents = writer.toString()
 
-        contents.shouldContain(SwiftWriter.GENERATED_FILE_HEADER)
+        contents.shouldContain(settings.copyrightNotice)
 
         val expectedGeneratedEnum =
             """
@@ -123,7 +123,7 @@ class UnionGeneratorTests {
 
         val contents = writer.toString()
 
-        contents.shouldContain(SwiftWriter.GENERATED_FILE_HEADER)
+        contents.shouldContain(settings.copyrightNotice)
 
         val expectedGeneratedEnum =
             """