had to redo everything : (

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

@@ -0,0 +1,54 @@
+package com.zipcodewilmington.phonebook;
+
+import java.util.ArrayList;
+
+public class Contact {
+
+
+    private String name;
+    private ArrayList<String> phoneNumbers;
+
+
+    public Contact(){
+        this.phoneNumbers = new ArrayList<>();
+    }
+
+
+    public Contact(String name, String phoneNumbers){
+        this.name = name;
+        this.phoneNumbers = new ArrayList<>();
+        addNumber(phoneNumbers);
+    }
+
+    public void addNumber(String number) {
+        getPhoneNumbers().add(number);
+    }
+
+    public void removeNumber(String number) {
+        getPhoneNumbers().remove(number);
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public ArrayList<String> getPhoneNumbers() {
+        return phoneNumbers;
+    }
+    @Override
+    public String toString(){
+        return getName() + " " + getListOfPhoneNumbers();
+    }
+    public String getListOfPhoneNumbers(){
+        String listOfNumbers = "";
+        for(String number : getPhoneNumbers()){
+            listOfNumbers += number + " ";
+        }
+        return listOfNumbers.trim();
+    }
+
+}

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

@@ -1,66 +1,67 @@
 package com.zipcodewilmington.phonebook;
+
 import java.util.*;
 import java.util.TreeMap;
-import java.util.Set;
+
 
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
 
-    String name;
-    String number;
-    ArrayList<String> numbersList = new ArrayList<>();
+    private TreeMap<String, Contact> phoneBook;
 
 
-    private TreeMap contactsList;
-
     public PhoneBook() {
-        this.contactsList = new TreeMap<>();
+        this.phoneBook = new TreeMap<>(String::compareToIgnoreCase);
+    }
+    public void addNewEntry(String name, String number){
+        getPhoneBook().put(name, new Contact(name, number));
+    }
+
+    public void removeEntry(String name){
+        getPhoneBook().remove(name);
+    }
+    public void removeNumber(String name, String number){
+        getPhoneBook().get(name).getPhoneNumbers().remove(number);
+    }
 
+    public ArrayList<String> lookUpContactPhoneNumbers(String name){
+        return getPhoneBook().get(name).getPhoneNumbers();
     }
 
-    public Map<String, ArrayList> getContactsList() {
-        return contactsList;
+    public void listAllNamesAndNumbers() {
+        System.out.println(toString());
 
     }
+    public String reverseLookUp(String number) {
+        for (Map.Entry<String, Contact> contactEntry : getPhoneBook().entrySet()) {
+            if (contactEntry.getValue().getPhoneNumbers().contains(number)) {
+                return contactEntry.getKey();
+            }
 
-    public String addContact(String name, String number) {
-        if (!contactsList.containsKey(name)) {
-            contactsList.put(name, number);
         }
-        return "";
+        return "Contact not found";
     }
 
-    public String removeContact(String name) {
-        if (contactsList.containsKey(name)) {
-            contactsList.remove(name);
+    @Override
+    public String toString(){
+        String toBeReturned = "";
+        for(Map.Entry<String, Contact> contactEntry : getPhoneBook().entrySet()){
+           toBeReturned += contactEntry.getValue().toString();
+           toBeReturned += " ";
         }
-        return "";
+        return toBeReturned.trim();
     }
+    public TreeMap<String, Contact> getPhoneBook(){
+        return phoneBook;
+}
 
-
-    public String listAllNamesAndNumbers() {
-        Set listingAll = contactsList.entrySet();
-        return listingAll.toString();
-        /*String getEntries = new String();
-        for (Object allEntries : listingAll) {
-            if(listingAll.equals());*/
         }
 
 
 
-    public String reverseLookUp() {
-        Set<String> listingReverse = contactsList.keySet();
-        for (String allEntries : listingReverse) {
-            if(listingReverse.contains(allEntries.equals(number))) {
-                return allEntries;
-            }
 
-        }
-        return "";
-    }
-}
 
 
 

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

@@ -0,0 +1,66 @@
+package com.zipcodewilmington.phonebook;
+
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+
+import static junit.framework.TestCase.assertTrue;
+import static org.junit.Assert.assertEquals;
+
+public class ContactTest {
+
+    Contact contact;
+
+    @Before
+    public void setUp(){
+        contact = new Contact();
+
+    }
+    @Test
+    public void addNumberTest(){
+        //Given
+        String number = "872-345-2298";
+        contact.addNumber(number);
+        //When
+        boolean numberHasBeenAdded = contact.getPhoneNumbers().contains(number);
+        //Then
+        assertTrue(numberHasBeenAdded);
+
+    }
+    @Test
+    public void removeNumberTest(){
+        //Given
+        String number = "872-345-2298";
+        contact.addNumber(number);
+        //When
+        contact.removeNumber(number);
+        boolean numberHasBeenRemoved = !contact.getPhoneNumbers().contains(number);
+        //Then
+        assertTrue(numberHasBeenRemoved);
+    }
+    @Test
+    public void toStringTest(){
+        //Given
+        String newName = "April";
+        String newNumber = "7584339876";
+        Contact contactUnderTest = new Contact(newName, newNumber);
+
+        String expectedString = "April 7584339876";
+        String actualString = contactUnderTest.toString();
+        assertEquals(expectedString, actualString);
+
+    }
+    @Test
+    public void getListOfPhoneNumbersTest(){
+        String newName = "April";
+        String newNumber = "7584339876";
+        Contact contactUnderTest = new Contact(newName, newNumber);
+        contactUnderTest.addNumber("3457765489");
+        String expectedListOfNumbers = "7584339876 3457765489";
+        String actualListOfNumbers = contactUnderTest.getListOfPhoneNumbers();
+        assertEquals(expectedListOfNumbers, actualListOfNumbers);
+
+    }
+
+
+}

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

@@ -5,54 +5,84 @@ import org.junit.Before;
 import org.junit.Test;
 
 import java.util.ArrayList;
-import java.util.Map;
-import java.util.Set;
+
+import static junit.framework.TestCase.assertTrue;
+import static org.junit.Assert.assertEquals;
 
 
 /**
  * Created by leon on 1/23/18.
  */
-public class PhoneBookTest {
-
+ public class PhoneBookTest {
 
     PhoneBook phoneBook;
 
     @Before
-    public void setUp(){
+    public void setUp() {
         phoneBook = new PhoneBook();
+    }
 
+    @Test
+    public void addNewEntryTest(){
+        String name = "April";
+        String number = "6547783321";
+        phoneBook.addNewEntry(name, number);
+        boolean entryHasBeenCreated = phoneBook.getPhoneBook().containsKey(name);
+        assertTrue(entryHasBeenCreated);
     }
     @Test
-    public void addContactTest(){
-        phoneBook.addContact("Genya", "6578356767");
-        boolean genyaIsInPhoneBook = phoneBook.getContactsList().keySet().contains("Genya");
-        Assert.assertTrue(genyaIsInPhoneBook);
-
+    public void removeEntryTest(){
+        String name = "April";
+        String number = "6547783321";
+        phoneBook.addNewEntry(name, number);
+        phoneBook.removeEntry(name);
+        boolean entryHasBeenRemoved = !phoneBook.getPhoneBook().containsKey(name);
+        assertTrue(entryHasBeenRemoved);
     }
     @Test
-    public void removeContactTest(){
-        phoneBook.removeContact("Pete Jones");
-        boolean peteJonesHasBeenRemoved = !phoneBook.getContactsList().containsKey("Pete Jones");
-        Assert.assertTrue(peteJonesHasBeenRemoved);
+    public void removeNumberTest(){
+        String name = "April";
+        String number = "6547783321";
+        phoneBook.addNewEntry(name, number);
+        phoneBook.removeNumber(name, number);
+        boolean numberHasBeenRemovedFromContact = !phoneBook.getPhoneBook().get(name).getPhoneNumbers().contains(number);
+        assertTrue(numberHasBeenRemovedFromContact);
+    }
+    @Test
+    public void lookUpContactPhoneNumbersTest(){
+        String name = "April";
+        String number = "6547783321";
+        phoneBook.addNewEntry(name, number);
+        ArrayList<String> actualPhoneList = phoneBook.lookUpContactPhoneNumbers(name);
+        ArrayList<String> expectedPhoneList = new ArrayList<>();
+        expectedPhoneList.add(number);
+        assertEquals(expectedPhoneList, actualPhoneList);
     }
-
     @Test
     public void listAllNamesAndNumbersTest(){
+        String name = "April";
+        String number = "6547783321";
+        String secondName = "Genya";
+        String secondNumber = "8753349987";
+        phoneBook.addNewEntry(name, number);
+        phoneBook.addNewEntry(secondName, secondNumber);
+        String actual = phoneBook.toString();
+        String expected = "April 6547783321 Genya 8753349987";
+        assertEquals(expected, actual);
+        //Print all the things
         phoneBook.listAllNamesAndNumbers();
-        String expected = new String();
-        Set<Map.Entry<String, ArrayList>> actual = phoneBook.getContactsList().entrySet();
-        Assert.assertEquals(expected, actual);
     }
-
     @Test
     public void reverseLookUpTest(){
-        phoneBook.reverseLookUp();
-        String expected = new String();
-        Set<Map.Entry<String, ArrayList>> actual = phoneBook.getContactsList().entrySet();
-        Assert.assertEquals(expected, actual);
-
+        String name = "April";
+        String number = "6547783321";
+        String secondName = "Genya";
+        String secondNumber = "8753349987";
+        phoneBook.addNewEntry(name, number);
+        phoneBook.addNewEntry(secondName, secondNumber);
+        String actual = phoneBook.reverseLookUp(number);
+        String expected = "April";
+        assertEquals(expected, actual);
     }
 
 }
-
-