One last test to pass and then redoing with person class

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

@@ -1,5 +1,7 @@
 package com.zipcodewilmington.phonebook;
 
+import org.omg.CosNaming.NamingContextExtPackage.StringNameHelper;
+
 import java.util.*;
 
 /**
@@ -19,17 +21,19 @@ public class PhoneBook {
 
 
     public void addEntry(String name, String... number) {
-        ArrayList<String> listOfPhoneNumbers = new ArrayList<String>(Arrays.asList(number));
-        myTree.put(name, listOfPhoneNumbers);
+        ArrayList<String> allNumbers = new ArrayList<String>(Arrays.asList(number));
+        myTree.put(name, allNumbers);
 
     }
 
     public void additionalPhoneNumberToKey(String name, String number) {
+
         myTree.get(name).add(number);
     }
 
 
     public void removeNumberFromName(String name, String number) {
+
         myTree.get(name).remove(number);
     }
 
@@ -43,7 +47,7 @@ public class PhoneBook {
         ArrayList<String> matchingNumbers = myTree.get(name);
         String listedNumbers = "";
         for (String phoneNumbers : matchingNumbers) {
-            listedNumbers +=phoneNumbers + "\n";
+            listedNumbers +=phoneNumbers + ", ";
         }
        return listedNumbers.trim();
 
@@ -62,21 +66,22 @@ public class PhoneBook {
         return allNames.trim();
     }
 
+
     public String listPhoneBook() {
 
         Set<String> contacts = myTree.keySet();
-        StringBuilder printBook = new StringBuilder();
+        String phoneBook = "";
 
         for (String name : contacts) {
-            printBook.append(name).append(": ");
-            for (int x = 0; x<myTree.get(name).size(); x++) {
-                printBook.append(x).append("  ");
-            } printBook.append("\n");
+            phoneBook += name + ": ";
+          for (int x = 0; x<myTree.get(name).size(); x++) {
+                phoneBook += myTree.get(name).get(x) + ", ";
+            }
 
         }
-        String result = printBook.toString();
 
-        return result;
+
+        return phoneBook.trim();
     }
 
     public String reverseLookup(String number) {
@@ -104,12 +109,12 @@ public class PhoneBook {
         PhoneBook phonebookEntry = new PhoneBook();
 
         phonebookEntry.addEntry("Brian", "3027619121");
-        phonebookEntry.addEntry("Dan", "4846394190");
+        phonebookEntry.addEntry("Dan", "4846394190", "9897878787");
         phonebookEntry.addEntry("Garret", "3027894586");
         phonebookEntry.addEntry("Luke", "3147619121");
         phonebookEntry.addEntry("Kate", "4899994190");
         phonebookEntry.addEntry("Lenore", "3097894586");
-
+        phonebookEntry.removeEntry("Dan");
 
 
 

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

@@ -0,0 +1,19 @@
+package com.zipcodewilmington.phonebook;
+
+//import org.junit.Assert;
+//import org.junit.Before;
+//import org.junit.Test;
+//import org.junit.experimental.theories.suppliers.TestedOn;
+//
+//public class PersonTest {
+//    @Test
+//    public void testGetName() {
+//
+//        Person p = new Person("Kate");
+//        String name = p.getName();
+//
+//        Assert.assertTrue(name == "Kate");
+//
+//    }
+//
+//}

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

@@ -5,6 +5,8 @@ import org.junit.Before;
 import org.junit.Test;
 import org.junit.experimental.theories.suppliers.TestedOn;
 
+import java.util.ArrayList;
+
 
 /**
  * Created by leon on 1/23/18.
@@ -12,24 +14,26 @@ import org.junit.experimental.theories.suppliers.TestedOn;
 public class PhoneBookTest {
 
 
+
     @Test
     public void addTest() {
         PhoneBook testBook = new PhoneBook();
         testBook.addEntry("Luke", "6109998787");
         String testNumber = testBook.lookupNumber("Luke");
-        Assert.assertTrue(testNumber.equals("6109998787"));
-
-    }
-
-    @Test
-    public void addTest1() {
-        PhoneBook testBook = new PhoneBook();
-        testBook.addEntry("Kate", "4846569090");
-        String testNumber = testBook.lookupNumber("Kate");
-        Assert.assertTrue(testNumber.equals("4846569090"));
+        Assert.assertTrue(testNumber.equals("6109998787,"));
 
     }
 
+//    @Test
+//    public void addTest1() {
+//        PhoneBook testBook = new PhoneBook();
+//        Person expected = new Person("name", "arraylist of phone numbers");
+//        testBook.addEntry("Kate", expected);
+////        Person actual = testBook.lookupNumber("Kate");
+////        Assert.assertTrue(expected.equals(actual));
+//
+//    }
+//
     @Test
     public void removeTest1() {
         PhoneBook testBook = new PhoneBook();
@@ -44,18 +48,9 @@ public class PhoneBookTest {
     @Test
     public void stringLookupTest() {
         PhoneBook testBook = new PhoneBook();
-        testBook.addEntry("Kate", "4846569090");
+        testBook.addEntry("Kate", "4846569090", "9899890099");
         String testNumber = testBook.lookupNumber("Kate");
-        Assert.assertFalse(testNumber.equals("4846587091"));
-
-    }
-
-    @Test
-    public void stringLookupTest1() {
-        PhoneBook testBook = new PhoneBook();
-        testBook.addEntry("Kate", "4846569090");
-        String testNumber = testBook.lookupNumber("Kate");
-        Assert.assertTrue(testNumber.equals("4846569090"));
+        Assert.assertTrue(testNumber.equals("4846569090, 9899890099,"));
 
     }
 
@@ -66,7 +61,7 @@ public class PhoneBookTest {
         testBook.addEntry("Adam", "4864569090");
         testBook.addEntry("Zeke", "4787869090");
         String actual = testBook.listNames();
-        String expected = "Adam\nKate\nZeke\n";
+        String expected = "Adam\nKate\nZeke";
         Assert.assertEquals(actual, expected);
 
 
@@ -76,10 +71,10 @@ public class PhoneBookTest {
     public void listPhonebookTest1() {
         PhoneBook testBook = new PhoneBook();
         testBook.addEntry("Kate", "4846569090");
-        testBook.addEntry("Adam", "4864569090");
+        testBook.addEntry("Adam", "4864569090", "7879891212");
         testBook.addEntry("Zeke", "4787869090");
         String actual = testBook.listPhoneBook();
-        String expected = "Adam: 4864569090\nKate: 4846569090\nZeke: 4787869090\n";
+        String expected = "Adam: 4864569090, 7879891212, Kate: 4846569090, Zeke: 4787869090,";
         Assert.assertEquals(expected, actual);
 
 
@@ -93,4 +88,15 @@ public class PhoneBookTest {
 
     }
 
+
+
+    @Test
+    public void additionalPhoneNumberToKeyTest() {
+        PhoneBook testBook = new PhoneBook();
+        testBook.addEntry("Kate", "5659871212");
+        testBook.additionalPhoneNumberToKey("Kate", "4999999999");
+        String testNumber = testBook.lookupNumber("Kate");
+        Assert.assertTrue(testNumber.equals("5659871212, 4999999999,"));
+
+    }
 }