From c2d89c8679444fa72fa8b6a00729ee5d33149e11 Mon Sep 17 00:00:00 2001
From: Siben Nayak <sibnayak@amazon.com>
Date: Sat, 5 May 2018 18:24:13 +0530
Subject: [PATCH] Fixed unit test cases and formatting

---
 .../main/kotlin/com/baeldung/filesystem/FileWriter.kt  |  6 +++---
 .../kotlin/com/baeldung/filesystem/FileWriterTest.kt   | 10 ++++++++++
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/core-kotlin/src/main/kotlin/com/baeldung/filesystem/FileWriter.kt b/core-kotlin/src/main/kotlin/com/baeldung/filesystem/FileWriter.kt
index 20d9c963c7..6dc9b95f1f 100644
--- a/core-kotlin/src/main/kotlin/com/baeldung/filesystem/FileWriter.kt
+++ b/core-kotlin/src/main/kotlin/com/baeldung/filesystem/FileWriter.kt
@@ -5,13 +5,13 @@ import java.io.File
 class FileWriter {
 
     fun writeFileUsingPrintWriter(fileName: String, fileContent: String) =
-        File(fileName).printWriter().use { out -> out.println(fileContent) }
+            File(fileName).printWriter().use { out -> out.print(fileContent) }
 
     fun writeFileUsingBufferedWriter(fileName: String, fileContent: String) =
-        File(fileName).bufferedWriter().use { out -> out.write(fileContent) }
+            File(fileName).bufferedWriter().use { out -> out.write(fileContent) }
 
     fun writeFileDirectly(fileName: String, fileContent: String) =
-        File(fileName).writeText(fileContent)
+            File(fileName).writeText(fileContent)
 
     fun writeFileDirectlyAsBytes(fileName: String, fileContent: String) =
             File(fileName).writeBytes(fileContent.toByteArray())
diff --git a/core-kotlin/src/test/kotlin/com/baeldung/filesystem/FileWriterTest.kt b/core-kotlin/src/test/kotlin/com/baeldung/filesystem/FileWriterTest.kt
index 43a9957de1..91c66a4fee 100644
--- a/core-kotlin/src/test/kotlin/com/baeldung/filesystem/FileWriterTest.kt
+++ b/core-kotlin/src/test/kotlin/com/baeldung/filesystem/FileWriterTest.kt
@@ -1,6 +1,8 @@
 package com.baeldung.filesystem
 
 import org.junit.jupiter.api.Test
+import java.io.File
+import kotlin.test.assertEquals
 
 internal class FileWriterTest {
 
@@ -13,21 +15,29 @@ internal class FileWriterTest {
     @Test
     fun whenWrittenWithPrintWriter_thenCorrect() {
         fileWriter.writeFileUsingPrintWriter(fileName, fileContent)
+
+        assertEquals(fileContent, File(fileName).readText())
     }
 
     @Test
     fun whenWrittenWithBufferedWriter_thenCorrect() {
         fileWriter.writeFileUsingBufferedWriter(fileName, fileContent)
+
+        assertEquals(fileContent, File(fileName).readText())
     }
 
     @Test
     fun whenWrittenDirectly_thenCorrect() {
         fileWriter.writeFileDirectly(fileName, fileContent)
+
+        assertEquals(fileContent, File(fileName).readText())
     }
 
     @Test
     fun whenWrittenDirectlyAsBytes_thenCorrect() {
         fileWriter.writeFileDirectlyAsBytes(fileName, fileContent)
+
+        assertEquals(fileContent, File(fileName).readText())
     }
 
 }
\ No newline at end of file