|
@@ -8,6 +8,11 @@ package scot.carricksoftware.grantswriter.writer;
|
|
|
import org.junit.jupiter.api.BeforeEach;
|
|
|
import org.junit.jupiter.api.Test;
|
|
|
|
|
|
+import java.io.File;
|
|
|
+import java.io.FileReader;
|
|
|
+import java.io.IOException;
|
|
|
+import java.util.Scanner;
|
|
|
+
|
|
|
import static org.junit.jupiter.api.Assertions.*;
|
|
|
import static scot.carricksoftware.grantswriter.GenerateRandomNumberValues.GetRandomString;
|
|
|
|
|
@@ -31,4 +36,32 @@ class FileWriterTest {
|
|
|
writer.init(GetRandomString());
|
|
|
assertNotNull(writer.getOs());
|
|
|
}
|
|
|
+
|
|
|
+ @Test
|
|
|
+ void closeTest() throws Exception {
|
|
|
+ writer.init(GetRandomString());
|
|
|
+ writer.close();
|
|
|
+ assertNull(writer.getOs());
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ void writeTest() throws Exception {
|
|
|
+ File file = File.createTempFile("/tmp", ".tmp");
|
|
|
+ String line = GetRandomString();
|
|
|
+ writer.init(file.getAbsolutePath());
|
|
|
+ writer.writeLine(line);
|
|
|
+ writer.close();
|
|
|
+
|
|
|
+ assertEquals(line, getFileContents(file.getAbsolutePath()));
|
|
|
+ }
|
|
|
+
|
|
|
+ private String getFileContents(String fileName) throws IOException {
|
|
|
+ Scanner in = new Scanner(new FileReader(fileName));
|
|
|
+ StringBuilder sb = new StringBuilder();
|
|
|
+ while(in.hasNext()) {
|
|
|
+ sb.append(in.next());
|
|
|
+ }
|
|
|
+ in.close();
|
|
|
+ return sb.toString();
|
|
|
+ }
|
|
|
}
|