Move test to smithy-utils and add trailing \n

smithy-lang · Oct 31, 2020 · 6eaaa87 · 6eaaa87
1 parent e5a3bee
commit 6eaaa87
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 14 deletions.
diff --git a/...egen-core/src/test/java/software/amazon/smithy/codegen/core/writer/CodegenWriterTest.java b/...egen-core/src/test/java/software/amazon/smithy/codegen/core/writer/CodegenWriterTest.java
@@ -113,18 +113,4 @@ public void importsUseReferencesFromSymbols() {
         assertThat(writer.getImportContainer().imports, hasKey("MyString"));
         assertThat(writer.getImportContainer().imports.get("MyString"), equalTo("java.lang.String"));
     }
-
-    @Test
-    public void canComposeSetWithSection() {
-        String testSection = "testSection";
-        MyWriter writer = new MyWriter("foo");
-
-        writer.onSection(testSection, text -> writer.write(text + "1, "));
-        writer.onSection(testSection, text -> writer.write(text + "2, "));
-        writer.onSection(testSection, text -> writer.write(text + "3"));
-
-        writer.write("[${L@testSection}]", "");
-
-        assertThat(writer.toString(), equalTo("[1, 2, 3]"));
-    }
 }
diff --git a/smithy-utils/src/test/java/software/amazon/smithy/utils/CodeWriterTest.java b/smithy-utils/src/test/java/software/amazon/smithy/utils/CodeWriterTest.java
@@ -682,4 +682,18 @@ public void canFilterSections() {
 
         assertThat(writer.toString(), equalTo("HELLO!\nGOODBYE!\n"));
     }
+
+    @Test
+    public void canComposeSetWithSection() {
+        String testSection = "testSection";
+        CodeWriter writer = new CodeWriter();
+
+        writer.onSection(testSection, text -> writer.write(text + "1, "));
+        writer.onSection(testSection, text -> writer.write(text + "2, "));
+        writer.onSection(testSection, text -> writer.write(text + "3"));
+
+        writer.write("[${L@testSection}]", "");
+
+        assertThat(writer.toString(), equalTo("[1, 2, 3]\n"));
+    }
 }