Part 3 completed

src/main/java/com/zipcodewilmington/phonebook/PhoneBook.java

@@ -1,7 +1,6 @@
 package com.zipcodewilmington.phonebook;
 
-import java.util.Set;
-import java.util.TreeMap;
+import java.util.*;
 
 /**
  * Created by leon on 1/23/18.
@@ -10,23 +9,26 @@ public class PhoneBook {
 
     private String name;
     private String phoneNumber;
-    private String phoneNumber2;
 
-    private TreeMap<String, String> phoneBookDirectory = new TreeMap<String, String>();
 
-
-    public String lookupNameToGetAccompanyingPhoneNumber(String name) {
-        return phoneBookDirectory.get(name);
+    private TreeMap<String, ArrayList<String>> phoneBookDirectory = new TreeMap<String, ArrayList<String>>();
 
 
+    public ArrayList<String> lookupNameToGetAccompanyingPhoneNumber(String name) {
+       return phoneBookDirectory.get(name);
     }
 
-    public String addEntry(String name, String phoneNumber) {
-        phoneBookDirectory.put(name, phoneNumber);
-        return phoneBookDirectory.get(name);
-
+    public void addEntry(String name, String phoneNumber) {
+        if (phoneBookDirectory.containsKey(name)) {
+            phoneBookDirectory.get(name).add(String.valueOf(phoneNumber));
+        } else {
+            ArrayList<String> nameAndPhoneNumber = new ArrayList<String>();
+            nameAndPhoneNumber.add(0, (String.valueOf(phoneNumber)));
+            phoneBookDirectory.put(name, nameAndPhoneNumber);
+        }
     }
 
+
     public String removeEntry(String name) {
         phoneBookDirectory.remove(name);
         return null;
@@ -44,13 +46,19 @@ public class PhoneBook {
 
 
     public String reverseLookup(String phoneNumber) {
-        Set<String> names = phoneBookDirectory.keySet();
-        for (String key: names) {
-            if (phoneBookDirectory.get(key).contains(phoneNumber)) {
-                return key;
-            }
-        }
-        return null;
 
+       Set<String> keys = phoneBookDirectory.keySet();
+       for (String key: keys) {
+               if (phoneBookDirectory.get(key).contains(phoneNumber)) {
+                   return key;
+           }
+       } return "no phone number";
+
+        /* for (Map.Entry<String, ArrayList<String>> entry : phoneBookDirectory.entrySet()) {
+            if (entry.getValue().equals(phoneNumber)) {
+                return entry.getKey();
+            }
+        } return null;
+        */
     }
 }

src/test/java/com/zipcodewilmington/phonebook/PhoneBookTest.java

@@ -2,6 +2,8 @@ package com.zipcodewilmington.phonebook;
 
 import com.sun.xml.internal.ws.policy.AssertionSet;
 import org.junit.*;
+
+import java.util.ArrayList;
 import java.util.TreeMap;
 
 /**
@@ -17,25 +19,22 @@ public class PhoneBookTest {
     private TreeMap<String, String> phoneBook = new TreeMap<String, String>();
 
 
-
-
     @Test
     public void testlookUpNameToGetAccompanyingPhoneNumber() {
         testPhoneBook.addEntry("Rodrigo", "233-123-3445");
         testPhoneBook.addEntry("Elena", "234-123-3344");
         testPhoneBook.addEntry("Robert", "332-233-7821");
         String expected = "332-233-7821";
-        System.out.print(testPhoneBook.lookupNameToGetAccompanyingPhoneNumber("Robert"));
-        String actual = testPhoneBook.lookupNameToGetAccompanyingPhoneNumber("Robert");
+        String actual = testPhoneBook.lookupNameToGetAccompanyingPhoneNumber("Robert").get(0);
         Assert.assertEquals("should return Robert's phone number", expected, actual);
 
     }
     @Test
-    public void  testAddEntry() {
-        String expected ="233-123-3445";
-        String actual = testPhoneBook.addEntry("Rodrigo", "233-123-3445");
-        System.out.println(actual + " " + expected);
-        Assert.assertEquals(expected, actual);
+    public void testAddEntry() {
+        testPhoneBook.addEntry("Rodrigo", "233-123-3445");
+        String expected = "233-123-3445";
+        String actual = testPhoneBook.lookupNameToGetAccompanyingPhoneNumber("Rodrigo").get(0);
+        Assert.assertEquals(actual, expected);
     }
     @Test
     public void testRemoveEntry(){
@@ -69,9 +68,8 @@ public class PhoneBookTest {
         testPhoneBook.addEntry("Rodrigo", "233-123-3445");
         testPhoneBook.addEntry("Elena", "234-123-3344");
         testPhoneBook.addEntry("Robert", "332-233-7821");
-        String phoneNumber= "234-123-3344";
+        String actual= testPhoneBook.reverseLookup("234-123-3344");
         String expected = "Elena";
-        String actual = testPhoneBook.reverseLookup( "234-123-3344");
         Assert.assertEquals("should return who the phone number is assigned to ", expected, actual);
         }
 }