update

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

@@ -1,6 +1,7 @@
 package com.zipcodewilmington.phonebook;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.Map;
 import java.util.TreeMap;
 
@@ -12,9 +13,11 @@ public class PhoneBook {
     public static void main(String[] args){
 
     }
+
   private Map<String, ArrayList<String>> contacts;
 
     public PhoneBook(){
+
         contacts=new TreeMap<String, ArrayList<String>>();
     }
 
@@ -33,7 +36,14 @@ public class PhoneBook {
     }
 
     public void add(String name, String phonenumber) {
-        ArrayList<String> details = new ArrayList<>();
+    //    contacts.containsKey(name);
+        ArrayList<String>details = new ArrayList<>();
+
+        boolean exists =  contacts.containsKey(name);
+        if(exists){
+
+        details = getNumbersFor(name);
+        }
 
         details.add(phonenumber);
 
@@ -45,7 +55,7 @@ public class PhoneBook {
     public String getContact(String name,String phonenumber) {
         ArrayList<String> phonenumbers = contacts.get(name);
 
-            return (phonenumbers==null? null:name);
+            return (name==null? null:phonenumber);
     }
 
 
@@ -74,6 +84,17 @@ public class PhoneBook {
 
 
     }
+    public String [] lookupNumbersFor(String name){
+        String [] result;
+        ArrayList<String> numbers = contacts.get(name);
+//        result = (String[]d) numbers.toArray();
+                result = numbers.toArray(new String[numbers.size()]);
+
+        return result;
+    }
+    private ArrayList<String> getNumbersFor(String name){
+        return contacts.get(name);
+    }
 
 
 

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

@@ -1,18 +1,33 @@
 package com.zipcodewilmington.phonebook;
 import org.junit.Assert;
+import org.junit.Before;
 import org.junit.Test;
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBookTest {
+    String name;
+    String phonenumber,phonenumber2;
+    PhoneBook book;
 
+@Before
+    public void setup (){
 
+    name = "Steven";
+    phonenumber="122-545-2344";
+    phonenumber2 = "453-343-2322";
+    book=new PhoneBook();
+    book.add(name,phonenumber);
+    book.add(name,phonenumber2);
+
+
+
+}
     @Test
     public void hasEntryphone() {
 
-        PhoneBook book = new PhoneBook();
-        String name = "AA";
-        String phonenumber = "234-232-2311";
+
+//        String name = "AA";
         //when
         book.add(name, phonenumber);
         boolean actualOutput = book.hasEntry(name);
@@ -27,8 +42,8 @@ public class PhoneBookTest {
     public void add() {
 
         PhoneBook book = new PhoneBook();
-        String name = "AA";
-        String phonenumber = "234-232-2311";
+//        String name = "AA";
+//        String phonenumber = "234-232-2311";
         book.add(name, phonenumber);
 
         //then
@@ -39,8 +54,8 @@ public class PhoneBookTest {
     public void getContact_test() {
 
         PhoneBook book = new PhoneBook();
-        String name = "Whitney";
-        String phonenumber = "123-331-3231";
+//        String name = "Whitney";
+//        String phonenumber = "123-331-3231";
         book.add(name, phonenumber);
         String actual = book.getContact(name, phonenumber);
         String expected = name;
@@ -51,8 +66,8 @@ public class PhoneBookTest {
     @Test
     public void reverseLookup_test() {
         PhoneBook book = new PhoneBook();
-        String name = "Whitney";
-        String phonenumber = "231-131-3257";
+//        String name = "Whitney";
+//        String phonenumber = "231-131-3257";
         book.add(name, phonenumber);
         String actual = book.reverselookup(phonenumber);
         String expect = name;
@@ -64,8 +79,8 @@ public class PhoneBookTest {
     @Test
     public void remove_Test() {
         PhoneBook book = new PhoneBook();
-        String name = "AA";
-        String phonenumber = "";
+//        String name = "AA";
+//        String phonenumber = "";
         book.add(name, phonenumber);
         book.remove(name);
         boolean actual = book.hasEntry(name);
@@ -73,15 +88,23 @@ public class PhoneBookTest {
         Assert.assertFalse(actual);
 
     }
+    @Test
+    public void lookupNumbersFor_test() {
+
+            String [] expecting = new String[]{phonenumber,phonenumber2};
+            String [] actual = book.lookupNumbersFor(name);
+
+            Assert.assertEquals(expecting,actual);
+
+
+    }
 
     @Test
     public void ListNamesandNumbers() {
         PhoneBook book = new PhoneBook();
-        String name = "";
-        String Phonenumber = "";
-        book.add(name, Phonenumber);
-        String actual = book.getContact(name, Phonenumber);
-        String expected = Phonenumber;
+        book.add(name, phonenumber);
+        String actual = book.getContact(name, phonenumber);
+        String expected = phonenumber;
         Assert.assertEquals(expected, actual);
 
     }