From 7de2556d6b35381b087744e849e8e455038b0237 Mon Sep 17 00:00:00 2001
From: Hai Nguyen <nn.hai@samsung.com>
Date: Tue, 3 Jul 2018 17:33:01 +0800
Subject: [PATCH] Move from using main method to Junit test

---
 .../FileToBase64StringConversion.java         | 32 -----------
 .../FileToBase64StringConversionTest.java     | 53 +++++++++++++++++++
 2 files changed, 53 insertions(+), 32 deletions(-)
 delete mode 100644 core-java-8/src/main/java/com/baeldung/fileToBase64StringConversion/FileToBase64StringConversion.java
 create mode 100644 core-java-8/src/test/java/com/baeldung/fileToBase64StringConversion/FileToBase64StringConversionTest.java

diff --git a/core-java-8/src/main/java/com/baeldung/fileToBase64StringConversion/FileToBase64StringConversion.java b/core-java-8/src/main/java/com/baeldung/fileToBase64StringConversion/FileToBase64StringConversion.java
deleted file mode 100644
index aa3bc9adee..0000000000
--- a/core-java-8/src/main/java/com/baeldung/fileToBase64StringConversion/FileToBase64StringConversion.java
+++ /dev/null
@@ -1,32 +0,0 @@
-package com.baeldung.fileToBase64StringConversion;
-
-import java.io.File;
-import java.io.FileNotFoundException;
-import java.io.IOException;
-import java.util.Base64;
-
-import org.apache.commons.io.FileUtils;
-import org.apache.commons.io.FilenameUtils;
-
-public class FileToBase64StringConversion {
-	public static void main(String[] args) throws FileNotFoundException, IOException {
-		//read file path from first argument
-		String filePath = args[0];
-		
-		byte[] fileContent = FileUtils.readFileToByteArray(new File(filePath));
-		String encodedString = Base64.getEncoder().encodeToString(fileContent);
-
-		//print encoded base64 String
-		System.out.println(encodedString);
-		
-		//construct output file name
-		String extension = FilenameUtils.getExtension(filePath);
-		String baseFileName = FilenameUtils.getBaseName(filePath);
-		String directory = new File(filePath).getParentFile().getAbsolutePath();
-		String outputFileName = directory+File.separator+baseFileName+"_copy."+extension;
-		
-		//decode the string and write to file
-		byte[] decodedBytes = Base64.getDecoder().decode(encodedString);
-		FileUtils.writeByteArrayToFile(new File(outputFileName), decodedBytes);
-	}
-}
diff --git a/core-java-8/src/test/java/com/baeldung/fileToBase64StringConversion/FileToBase64StringConversionTest.java b/core-java-8/src/test/java/com/baeldung/fileToBase64StringConversion/FileToBase64StringConversionTest.java
new file mode 100644
index 0000000000..4e023fbe64
--- /dev/null
+++ b/core-java-8/src/test/java/com/baeldung/fileToBase64StringConversion/FileToBase64StringConversionTest.java
@@ -0,0 +1,53 @@
+package com.baeldung.fileToBase64StringConversion;
+
+import static org.junit.Assert.assertTrue;
+
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Base64;
+import java.util.Collection;
+
+import org.apache.commons.io.FileUtils;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.Parameterized;
+
+@RunWith(Parameterized.class)
+public class FileToBase64StringConversionTest {
+
+	private String inputFilePath;
+	private String outputFilePath;
+
+	public FileToBase64StringConversionTest(String inputFilePath, String outputFilePath) {
+		this.inputFilePath = inputFilePath;
+		this.outputFilePath = outputFilePath;
+	}
+
+	@Parameterized.Parameters
+	public static Collection<String[]> parameters() {
+		return Arrays.asList(new String[][] { { "", "" } });
+	}
+
+	@Test
+	public void fileToBase64StringConversion() throws FileNotFoundException, IOException {
+		File outputFile = new File(outputFilePath);
+		File inputFile = new File(inputFilePath);
+
+		if (!inputFile.exists())
+			return;
+
+		byte[] fileContent = FileUtils.readFileToByteArray(inputFile);
+		String encodedString = Base64.getEncoder().encodeToString(fileContent);
+
+		// print encoded base64 String
+		System.out.println(encodedString);
+
+		// decode the string and write to file
+		byte[] decodedBytes = Base64.getDecoder().decode(encodedString);
+		FileUtils.writeByteArrayToFile(outputFile, decodedBytes);
+
+		assertTrue(outputFile.length() == fileContent.length);
+	}
+}