finished assisnment

PhoneBook.java

@@ -1,7 +1,95 @@
- 
+import java.util.*;
+import java.util.List;
+import java.util.Arrays;
+import java.util.Iterator;
 
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
+    String name;
+    String phoneNumber;
+    TreeMap<String,ArrayList<String>> directory = new TreeMap<String,ArrayList<String>>();
+    HashMap<ArrayList<String>,String> reverseDirectory = new HashMap<ArrayList<String>,String>();
+    public PhoneBook(){
+
+    }
+
+    public TreeMap getDirectory(){
+        return directory;
+    }
+
+    public HashMap<ArrayList<String>, String> invertMap() {
+        for(String k: directory.keySet()){
+            reverseDirectory.put(directory.get(k), k);
+        }
+        return reverseDirectory;
+    }
+
+    public TreeMap add(String name, ArrayList<String> phoneNumber){
+        directory.put(name, phoneNumber);
+        return directory;
+    }
+
+    public TreeMap removeRecord(String name){
+        directory.remove(name);
+        return directory;
+    }
+
+    public ArrayList<String> lookUp(String name){
+        return directory.get(name);
+    }
+
+    public String reverseLookUp(String phoneNumber){
+        this.invertMap();
+        String answer = "";
+        for(ArrayList<String> element: reverseDirectory.keySet()){
+            if(arrayContains(element, phoneNumber) == true){
+                answer = reverseDirectory.get(element);
+            }
+        }
+
+        return answer;
+    }
+
+    public String display(){
+        StringBuilder print = new StringBuilder("");
+
+        for(String key : directory.keySet()){
+            print.append(key + " " + printArray(directory.get(key)) + "\n");
+        }
+
+        return print.toString();
+    }
+
+    public String printArray(ArrayList<String> array){
+        String result = String.join(", ", array);
+
+        return result;
+    }
+
+    public boolean arrayContains(ArrayList<String> list, String phoneNumber){
+        boolean correct = false;
+
+        for(String element : list){
+            if(phoneNumber.equals(element)){
+                correct = true;
+                break;
+            }
+        }
+
+        return correct;
+    }
+
+    public TreeMap remove(String name, String phoneNumber){
+
+        Iterator search = directory.get(name).iterator();
+        while(search.hasNext()){
+            if(search.next().equals(phoneNumber)){
+                search.remove();
+                break;
+            }
+        }
+        return directory;
+    }
 }

PhoneBookTest.java

@@ -1,9 +1,9 @@
 
-
 import static org.junit.Assert.*;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
+import java.util.*;
 
 /**
  * The test class PhoneBookTest.
@@ -13,6 +13,7 @@ import org.junit.Test;
  */
 public class PhoneBookTest
 {
+    PhoneBook test = new PhoneBook();
     /**
      * Default constructor for test class PhoneBookTest
      */
@@ -30,6 +31,116 @@ public class PhoneBookTest
     {
     }
 
+    @Test
+    public void addTest(){
+        //Given
+        ArrayList<String> list = new ArrayList<String>();
+        list.add("5704282076");
+        //When
+        int expected = 1;
+        TreeMap actual = test.add("Chad", list);
+
+        //Result
+        assertEquals(expected, actual.size());
+    }
+
+    @Test
+    public void removeRecordTest(){
+        //Given
+        ArrayList<String> list = new ArrayList<String>();
+        list.add("5704282076");
+        list.add("5707429337");
+        test.add("Chad", list);
+        //When
+        int expected = 0;
+        TreeMap actual = test.removeRecord("Chad");
+
+        //Result
+        assertEquals(expected, actual.size());
+    }
+
+    @Test
+    public void lookUpTest(){
+        //Given
+        ArrayList<String> list = new ArrayList<String>();
+        list.add("5704282076");
+        test.add("Chad", list);
+        //When
+        ArrayList<String> expected = list;
+        ArrayList<String> actual = test.lookUp("Chad");
+
+        //Result
+        assertEquals(expected, actual);
+    }
+
+    @Test
+    public void invertMapTest(){
+        //Given
+        ArrayList<String> list = new ArrayList<String>();
+        list.add("5704282076");
+        test.add("Chad", list);
+        test.invertMap();
+        //When
+        String expected = "Chad";
+        String actual = test.reverseLookUp("5704282076");
+
+        //Result
+        assertEquals(expected, actual);
+    }
+
+    @Test
+    public void displayTest(){
+        //Given
+        ArrayList<String> list = new ArrayList<String>();
+        list.add("5704282076");
+        ArrayList<String> list1 = new ArrayList<String>();
+        list1.add("111-222-333");
+        ArrayList<String> list2 = new ArrayList<String>();
+        list2.add("222-444-4444");
+        test.add("Chad", list);
+        test.add("Zebra", list1);
+        test.add("Dog", list2);
+
+        //When
+        String expected = "Chad 5704282076\nDog 222-444-4444\nZebra 111-222-333\n";
+        String actual = test.display();
+
+        //Result
+        assertEquals(expected, actual);
+    }
+    
+    @Test
+    public void arrayContainsTest(){
+        //Given
+        ArrayList<String> list = new ArrayList<String>();
+        list.add("5704282076");
+        //When
+        boolean actual = test.arrayContains(list,"5704282076");
+        //Result
+        assertTrue(actual);
+    }
+
+    @Test
+    public void removeTest(){
+        //Given
+        ArrayList<String> list = new ArrayList<String>();
+        list.add("5704282076");
+        list.add("5707429337");
+        ArrayList<String> list1 = new ArrayList<String>();
+        list1.add("111-222-333");
+        ArrayList<String> list2 = new ArrayList<String>();
+        list2.add("222-444-4444");
+        test.add("Chad", list);
+        test.add("Zebra", list1);
+        test.add("Dog", list2);
+        
+        test.remove("Chad","5704282076");
+        //When
+        boolean actual = test.arrayContains(list,"5704282076");
+        //Result
+        assertFalse(actual);
+    }
+    
     /**
      * Tears down the test fixture.
      *

package.bluej

@@ -1,20 +1,23 @@
 #BlueJ package file
+dependency1.from=PhoneBookTest
+dependency1.to=PhoneBook
+dependency1.type=UsesDependency
 editor.fx.0.height=0
 editor.fx.0.width=0
 editor.fx.0.x=0
 editor.fx.0.y=0
-objectbench.height=101
-objectbench.width=461
+objectbench.height=164
+objectbench.width=776
 package.divider.horizontal=0.6
-package.divider.vertical=0.8007380073800738
-package.editor.height=427
+package.divider.vertical=0.6845018450184502
+package.editor.height=364
 package.editor.width=674
-package.editor.x=427
-package.editor.y=143
+package.editor.x=496
+package.editor.y=94
 package.frame.height=600
 package.frame.width=800
-package.numDependencies=0
-package.numTargets=1
+package.numDependencies=1
+package.numTargets=2
 package.showExtends=true
 package.showUses=true
 project.charset=UTF-8
@@ -23,4 +26,17 @@ readme.name=@README
 readme.width=47
 readme.x=10
 readme.y=10
-
+target1.height=50
+target1.name=PhoneBookTest
+target1.showInterface=false
+target1.type=UnitTestTargetJunit4
+target1.width=120
+target1.x=100
+target1.y=10
+target2.height=50
+target2.name=PhoneBook
+target2.showInterface=false
+target2.type=ClassTarget
+target2.width=100
+target2.x=100
+target2.y=40