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

@@ -1,7 +1,60 @@
 package com.zipcodewilmington.phonebook;
 
+import apple.laf.JRSUIUtils;
+
+import java.util.ArrayList;
+import java.util.Set;
+import java.util.TreeMap;
+
 /**
- * Created by leon on 1/23/18.
+ * Created by William on 10/28/2018.
  */
 public class PhoneBook {
-}
+
+    //TREEMAP
+    public TreeMap<String, ArrayList<String>> phonebook  = new TreeMap<>();
+
+    //CHECKS IF IT HAS AN ENTRY
+    public boolean hasEntry(String name){
+        if(phonebook.containsKey(name)){
+            return true;
+        }
+        return false;
+    }
+
+    //LOOKUP NUMBER
+    public ArrayList<String> lookUp(String name) {
+        if(hasEntry(name)){
+            return phonebook.get(name);
+        }
+        return null;
+    }
+
+    //PRINTS OUT NAME AND NUMBERS IN TREEMAP
+    public TreeMap<String, ArrayList<String>> listNamesAndNumbers() {
+        return phonebook;
+    }
+
+    //LIST ONLY NAMES IN TREEMAP
+    public Set<String> listNames() {
+        return phonebook.keySet();
+    }
+
+    //ADDS CONTACT TO THE TREEMAP
+    public void addContact(String name, ArrayList<String> number) {
+        phonebook.put(name, number);
+    }
+
+    //REMOVES CONTACT FROM THE TREE
+    public void removeContact(String name, ArrayList<String> number){
+        name = null;
+        number = null;
+    }
+
+    //CHECKS THE SIZE
+    public int size(){
+        return phonebook.size();
+    }
+
+
+}

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

@@ -1,7 +1,123 @@
 package com.zipcodewilmington.phonebook;
 
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.util.ArrayList;
+
 /**
- * Created by leon on 1/23/18.
+ * Created by William on 10/28/2018.
  */
 public class PhoneBookTest {
-}
+
+    @Test
+    public void testHasEntry(){
+        //Given
+        PhoneBook book = new PhoneBook();
+        String name = "AA";
+        ArrayList<String> phoneNumber = new ArrayList<>();
+
+        phoneNumber.add("211-111-2222");
+
+        book.addContact(name, phoneNumber);
+
+        //When
+        boolean actualOutput = book.hasEntry(name);
+
+        //Then
+        Assert.assertTrue(actualOutput);
+    }
+
+    @Test
+    public void testAdd(){
+        //Given
+        PhoneBook book = new PhoneBook();
+        String name = "AA";
+        ArrayList<String> phoneNumber = new ArrayList<>();
+        phoneNumber.add("211-111-2222");
+
+        //When
+        book.addContact(name, phoneNumber);
+
+        //Then
+        Assert.assertEquals(1, book.size());
+    }
+
+    @Test
+    public void testRemove(){
+        //Given
+        PhoneBook book = new PhoneBook();
+        String name = "AA";
+        ArrayList<String> phoneNumber = new ArrayList<>();
+        phoneNumber.add("211-111-2222");
+
+        //When
+        book.removeContact(name, phoneNumber);
+
+        //Then
+        Assert.assertEquals(0, book.size());
+    }
+
+    @Test
+    public void testLookUp(){
+        //Given
+        PhoneBook book = new PhoneBook();
+        String name = "AA";
+        ArrayList<String> phoneNumber = new ArrayList<>();
+        phoneNumber.add("221-111-2222");
+        ArrayList<String> expected = new ArrayList<String>();
+        expected.add("221-111-2222");
+        //When
+        book.addContact("AA", phoneNumber);
+        book.lookUp(name);
+
+        //Then
+        Assert.assertEquals(expected, book.lookUp("AA"));
+    }
+
+    @Test
+    public void testSize(){
+        //Given
+        PhoneBook book = new PhoneBook();
+        ArrayList<String> phoneNumber = new ArrayList<>();
+        String name = "AA";
+        phoneNumber.add("211-111-2222");
+
+        book.addContact(name, phoneNumber);
+
+        //When
+        int actualOutput = book.size();
+        int expectedOutput = 1;
+
+        //Then
+        Assert.assertEquals(expectedOutput, actualOutput);
+    }
+
+    @Test
+    public void testList(){
+
+        PhoneBook phoneBook = new PhoneBook();
+        ArrayList<String> phoneNumber = new ArrayList<>();
+        String name = "AA";
+        phoneNumber.add("221-111-2222");
+
+        //When
+        phoneBook.addContact("AA", phoneNumber);
+
+        phoneBook.listNamesAndNumbers();
+    }
+
+    @Test
+    public void testNameList(){
+
+        PhoneBook phoneBook = new PhoneBook();
+        ArrayList<String> phoneNumber = new ArrayList<>();
+        String name = "AA";
+        phoneNumber.add("221-111-2222");
+
+        //When
+        phoneBook.addContact("AA", phoneNumber);
+
+        phoneBook.listNames();
+    }
+}