more additions

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

@@ -26,6 +26,14 @@ public class Person {
     public String getNumbers(){
         return phoneNumber.toString();
     }
+
+    public void removeSingleNumber (String number){
+        phoneNumber.remove(number);
+    }
+
+    public void addAdditionalNumber(String number){
+        phoneNumber.add(number);
+    }
 }
 
 

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

@@ -15,6 +15,7 @@ public class PhoneBook {
         PhoneBook myPhoneBookActualObjectInstance= new PhoneBook();
         myPhoneBookActualObjectInstance.add("vince", "1123456789");
         myPhoneBookActualObjectInstance.add("amy", "29387429");
+        myPhoneBookActualObjectInstance.personTreeMap.get("amy").addAdditionalNumber("2344");
         System.out.println(myPhoneBookActualObjectInstance.displayEntirePhoneBookContents());
     }
 
@@ -34,7 +35,7 @@ public class PhoneBook {
     }
 
 
-    public void remove(String name) {
+    public void removeEntireEntry (String name) {
 
         personTreeMap.remove(name);
     }

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

@@ -12,26 +12,48 @@ import java.util.TreeMap;
  */
 public class PhoneBookTest {
 
-    PhoneBook testPhonebook = new PhoneBook();
+//Start off here. google "how to write a test for a map"
+//    PhoneBook testPhonebook = new PhoneBook(); <=why didn't this work?
+    PhoneBook testPhonebook;
 
     @Before
+    public void setup() {
+            testPhonebook = new PhoneBook();
+    }
 
-    public void setup(){
-
-//Start off here. google "how to write a test for a map"
+    @Test
+    public void addEntryTestCase1(){
+        new Person("Addison", "1234");
+        testPhonebook.add("Addison", "1234");
+        String testNumber = testPhonebook.lookup("Addison");
+        Assert.assertTrue(testNumber.equals("[1234]"));
+    }
 
+    @Test
+    public void addEntryTestCase2(){
+        new Person("AdDy", "123.4");
+        testPhonebook.add("AdDy", "123.4");
+        String testNumber = testPhonebook.lookup("AdDy");
+        Assert.assertTrue(testNumber.equals("[123.4]"));
     }
 
+
+
     @Test
     public void testLookup() {
 
         testPhonebook.add("JohnDoe", "1123");
 
-        String expected = "1123";
+        String expected = "[1123]";
         String actual = testPhonebook.lookup("JohnDoe");
 
         Assert.assertEquals(expected, actual);
 
     }
 
+    @Test
+    public void test (){
+
+    }
+
 }