almost finished

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

@@ -27,18 +27,26 @@ public class PhoneBook {
         }
     }
 
-    //add and remove numbers from existing names
-    //unfinished, no test cases yet
-    public void modifyNumbers(String name, ArrayList<String> number) {
+    //add numbers to existing entries
+    //no test cases yet
+    public void addNumberToEntry(String name, String number) {
         for (Map.Entry<String, ArrayList<String>> entry : treeMap.entrySet()) {
-            ArrayList<String> value = entry.getValue();
-            for (String firstNumber : value) {
-
+            if (name.equals(entry.getKey())) {
+                ArrayList<String> value = entry.getValue();
+                value.add(number);
             }
         }
-
     }
 
+    //remove numbers from existing entries
+    //no test cases yet
+    public void removeNumberFromEntry(String name, String number) {
+        for (Map.Entry<String, ArrayList<String>> entry : treeMap.entrySet()) {
+            if (entry.getValue().contains(number)) {
+                entry.getValue().remove(number);
+            }
+        }
+    }
 
     //remove a name & number entry
     public void removeEntry(String name) {
@@ -77,7 +85,4 @@ public class PhoneBook {
         return printOut.toString();
     }
 
-
-
 }
-

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

@@ -29,23 +29,44 @@ public class PhoneBookTest {
     }
 
     @Test
-    public void testAdd() {
-
+    public void testInputEntry() {
         //Given
         PhoneBook book = new PhoneBook();
 
         ArrayList<String> expectedAddition = new ArrayList<String>();
         expectedAddition.add("302-555-2222");
         expectedAddition.add("302-555-3456");
+
         //When
         book.inputEntry("Bob", expectedAddition);
         ArrayList<String> actualAddition = new ArrayList<String>(book.lookup("Bob"));
 
         //Then
         Assert.assertEquals(expectedAddition, actualAddition);
+    }
+
+    @Test
+    public void addNumberToEntry() {
+        //Given
+        PhoneBook book = new PhoneBook();
+        ArrayList<String> bobNumber = new ArrayList<String>();
+        bobNumber.add("302-555-2222");
+        book.inputEntry("Bob", bobNumber);
+        book.display();
+        //Expected
+        ArrayList<String> expectedAddition = new ArrayList<String>();
+        expectedAddition.add("302-555-2222");
+        expectedAddition.add("302-555-1111");
+        //When
+        book.addNumberToEntry("Bob", "302-555-1111");
+        ArrayList<String> actualAddition = book.lookup("Bob");
+        //Then
+        Assert.assertEquals(expectedAddition, actualAddition);
 
     }
 
+    //add test for removeNumberFromEntry
+
     @Test
     public void testRemoveEntry() {
 
@@ -129,6 +150,7 @@ public class PhoneBookTest {
         String expectedName = "Bob";
         ArrayList<String> bobNumber = new ArrayList<String>();
         bobNumber.add("302-555-2223");
+        bobNumber.add("302-555-5555");
         book.inputEntry(expectedName, bobNumber);
 
         //When;