Merge pull request #141 from kmwtechnology/LC-504

LC-504: fixes issue of file creation
kmwtechnology · Jun 19, 2024 · 883d5ab · 883d5ab
2 parents 44dd9bc + f6b903f
commit 883d5ab
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lucille-core/src/test/java/com/kmwllc/lucille/stage/PrintTest.java b/lucille-core/src/test/java/com/kmwllc/lucille/stage/PrintTest.java
@@ -10,6 +10,7 @@
 import org.junit.Test;
 import org.mockito.MockedConstruction;
 import java.io.BufferedWriter;
+import java.io.FileWriter;
 import java.io.IOException;
 import java.util.Set;
 
@@ -40,7 +41,8 @@ public void testGetLegalProperties() throws StageException {
 
   @Test
   public void testStop() throws Exception {
-    try (MockedConstruction<BufferedWriter> mockedConstruction = mockConstruction(BufferedWriter.class)) {
+    try (MockedConstruction<BufferedWriter> mockedConstruction = mockConstruction(BufferedWriter.class);
+        MockedConstruction<FileWriter> mockedWriter = mockConstruction(FileWriter.class)) {
       Stage stage = factory.get("PrintTest/full.conf");
       stage.stop();
       verify(mockedConstruction.constructed().get(0)).close();
@@ -51,7 +53,7 @@ public void testStop() throws Exception {
   public void testStopThrows() throws Exception {
     try (MockedConstruction<BufferedWriter> mockedConstruction = mockConstruction(BufferedWriter.class, (mock, context) -> {
       doThrow(IOException.class).when(mock).close();
-    })) {
+    }); MockedConstruction<FileWriter> mockedWriter = mockConstruction(FileWriter.class)) {
       Stage stage = factory.get("PrintTest/full.conf");
       assertThrows(StageException.class, () -> stage.stop());
     }
@@ -61,7 +63,7 @@ public void testStopThrows() throws Exception {
   public void testAppendThrows() throws StageException {
     try (MockedConstruction<BufferedWriter> mockedConstruction = mockConstruction(BufferedWriter.class, (mock, context) -> {
       doThrow(IOException.class).when(mock).append(any());
-    })) {
+    }); MockedConstruction<FileWriter> mockedWriter = mockConstruction(FileWriter.class)) {
       Stage stage = factory.get("PrintTest/full.conf");
       assertThrows(StageException.class, () -> stage.processDocument(Document.create("foo")));
     }