build Test for Load File

src/main/java/HamletParser.java

@@ -1,15 +1,15 @@
 import java.io.File;
 import java.io.IOException;
-import java.util.ArrayList;
 import java.util.Scanner;
-
+import java.lang.*;
+import java.util.*;
 /**
  * Created by thook on 10/7/15.
  */
 public class HamletParser {
 
     private String hamletData;
-    String[] wordArr;
+    private Formatter formatter;
 
     public HamletParser(){
         this.hamletData = loadFile();
@@ -19,24 +19,33 @@ public class HamletParser {
         ClassLoader classLoader = getClass().getClassLoader();
         File file = new File(classLoader.getResource("hamlet.txt").getFile());
         StringBuilder result = new StringBuilder("");
+        StringBuilder lineResult;
+        String[] wordArr;
 
         try(Scanner scanner = new Scanner(file)){
             while(scanner.hasNextLine()){
                 String line = scanner.nextLine();
-                if (isHamletOrHoratio(line)){
-                    wordArr = line.split(" ");
+                wordArr = line.split(" ");
+                lineResult = new StringBuilder("");
+                for (String s: wordArr) {
+                    if (findHamlet(s)) {
+                        lineResult.append(changeHamletToLeon()).append(" ");
+                    } else if (findHoratio(s)) {
+                        lineResult.append(changeChangeHoratioToTariq()).append(" ");
+                    } else {
+                        lineResult.append(s).append(" ");
+                    }
                 }
-                //loop through line for next
-                    //if next matches Hamlet, replace with Leon
-                    //if next matches Horatio, replace with Tariq
-                result.append(line).append("\n");
+                result.append(lineResult.toString().trim());
             }
 
             scanner.close();
         }catch(IOException e){
             e.printStackTrace();
         }
-
+        openOutFile();
+        addText(result.toString());
+        closeOutFile();
         return result.toString();
     }
 
@@ -44,12 +53,49 @@ public class HamletParser {
         return hamletData;
     }
 
-    public boolean isHamletOrHoratio(String line){
-        if (line.toLowerCase().matches("hamlet") || line.toLowerCase().matches("horatio")){
+    public boolean findHamlet(String s){
+        if (s.toLowerCase().matches("hamlet")){
+            return true;
+        } else {
+            return false;
+        }
+    }
+
+    public boolean findHoratio(String s){
+        if (s.toLowerCase().matches("horatio")){
             return true;
         } else {
             return false;
         }
     }
 
+    public String changeHamletToLeon(){
+        return "Leon";
+    }
+
+    public String changeChangeHoratioToTariq(){
+        return "Tariq";
+    }
+
+    /**
+     * Methods to print out to a new file
+     */
+    public void openOutFile(){
+        try {
+            formatter = new Formatter("output.txt");
+        }
+        catch (Exception e){
+            System.out.println("Error writing to file.");
+        }
+    }
+
+    public String addText(String s){
+        formatter.format("%s", s);
+        return loadFile();
+    }
+
+    public void closeOutFile(){
+        formatter.close();
+    }
+
 }

src/test/java/HamletParserTest.java

@@ -16,20 +16,67 @@ public class HamletParserTest {
 
     @Test
     public void testChangeHamletToLeon() {
-        //set to true to fail, change to false
-        Assert.assertTrue(hamletText.toLowerCase().contains("horatio"));
-        Assert.assertTrue(hamletText.toLowerCase().contains("hamlet"));
+        //GIVEN
+
+        //WHEN
+        String expected = "Leon";
+        //THEN
+        String actual = hamletParser.changeHamletToLeon();
+        Assert.assertEquals(expected, actual);
     }
 
     @Test
     public void testChangeHoratioToTariq() {
+        //GIVEN
+
+        //WHEN
+        String expected = "Tariq";
+        //THEN
+        String actual = hamletParser.changeChangeHoratioToTariq();
+        Assert.assertEquals(expected, actual);
     }
 
     @Test
     public void testFindHoratio() {
+        //GIVEN
+        String input = "horatio";
+        //WHEN
+        boolean expected = true;
+        //THEN
+        boolean actual = hamletParser.findHoratio(input);
+        Assert.assertEquals(expected, actual);
     }
 
     @Test
     public void testFindHamlet() {
+        //GIVEN
+        String input = "HAMLET";
+        //WHEN
+        boolean expected = true;
+        //THEN
+        boolean actual = hamletParser.findHamlet(input);
+        Assert.assertEquals(expected, actual);
+    }
+
+    @Test
+    public void testFindHoratio2() {
+        //GIVEN
+        String input = "nope";
+        //WHEN
+        boolean expected = false;
+        //THEN
+        boolean actual = hamletParser.findHoratio(input);
+        Assert.assertEquals(expected, actual);
+    }
+
+    @Test
+    public void testFindHamlet2() {
+        //GIVEN
+        String input = "hhamlet";
+        //WHEN
+        boolean expected = false;
+        //THEN
+        boolean actual = hamletParser.findHamlet(input);
+        Assert.assertEquals(expected, actual);
     }
 }