Completed

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

@@ -4,10 +4,12 @@ public class Person {
 
     private String name;
     private String number;
+    private String number2;
 
     public Person(String name, String number) {
         this.name = name;
         this.number = number;
+        this.number2 = number2;
     }
 
     public String getName() {

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

@@ -1,35 +1,56 @@
 package com.zipcodewilmington.phonebook;
+import java.util.*;
 
-import java.util.Set;
-import java.util.TreeMap;
 /**
- * Created by leon on 1/23/18.
+ * Created by Kay.
  */
 public class PhoneBook {
 
-    TreeMap phMap;
+    TreeMap<String, List<String>> phMap;
 
     public PhoneBook() {
-        this.phMap = new TreeMap();
+
+        this.phMap = new TreeMap<String, List<String>>();
     }
 
     public void add(String name, String number) {
-        phMap.put(name, number);
+        if (lookup(name) == null) {
+            List<String> phoneNumbers = new ArrayList<String>();
+            phoneNumbers.add(number);
+            phMap.put(name, phoneNumbers);
+        } else {
+            List<String> phoneNumbers = lookup(name);
+            phoneNumbers.add(number);
+        }
     }
 
     public void remove(String name) {
 
         phMap.remove(name);
     }
-    public String lookup(String name) {
-        return (String)phMap.get(name);
+
+    public List<String> lookup(String name) {
+
+        return phMap.get(name);
 
     }
-    public Set listNames () {
+
+    public Set listNames() {
+
         return phMap.keySet();
     }
+
     public Set listRecords() {
+
         return phMap.entrySet();
     }
-}
 
+    public String reverseLookup(String phoneNumber) {
+        for (Map.Entry<String, List<String>> valuePair : phMap.entrySet()) {
+            if (valuePair.getValue().contains(phoneNumber)){
+                return valuePair.getKey();
+            }
+        }
+        return "Entry not found";
+    }
+}

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

@@ -1,7 +1,7 @@
-package com.zipcodewilmington.phonebook;
-import org.junit.Assert.*;
-import org.junit.Test;
-
+//package com.zipcodewilmington.phonebook;
+//import org.junit.Assert.*;
+//import org.junit.Test;
+//
 //import static org.junit.Assert.assertTrue;
 //
 ///**
@@ -9,19 +9,19 @@ import org.junit.Test;
 // */
 //public class PersonTest {
 //
-//        @Test
-//        public void getNameTest() {
-//            Person person = new Person();
-//            person.setName("Kay");
-//            assertTrue(person.getName().equalsIgnoreCase("Kay"));
-//        }
+//    @Test
+//    public void getNameTest() {
+//        Person person = new Person();
+//        person.setName("Kay");
+//        assertTrue(person.getName().equalsIgnoreCase("Kay"));
+//    }
 //
-//        @Test
-//        public void setNameTest() {
-//            Person person = new Person();
-//            person.getName("Kay");
-//            assertTrue(person.setName().equalsIgnoreCase("Kay"));
-//        }
+//    @Test
+//    public void setNameTest() {
+//        Person person = new Person();
+//        person.getName("Kay");
+//        assertTrue(person.setName().equalsIgnoreCase("Kay"));
+//    }
 //        @Test
 //        public void getNumberTest() {
 //            Person person = new Person();
@@ -34,4 +34,5 @@ import org.junit.Test;
 //            person.setNumber("123");
 //            assertTrue(person.getNumber().equalsIgnoreCase("123"));
 //        }
-//}
+
+

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

@@ -14,7 +14,7 @@ public class PhoneBookTest {
         String name = "Kay";
         String number = "1234567890";
         phTest.add(name, number);
-        Assert.assertTrue(phTest.lookup(name).equals(number));
+        Assert.assertTrue(phTest.lookup(name).contains(number));
     }
 
     @Test
@@ -33,19 +33,20 @@ public class PhoneBookTest {
         String number = "3456784523";
         //adding a test value key pair to test for removal
         phTest.add(name, number);
-        Assert.assertTrue(phTest.lookup(name).equals(number));
+        Assert.assertTrue(phTest.lookup(name).contains(number));
         //after confirmation that it has been added, it needs to be removed
         phTest.remove(name);
         Assert.assertTrue(phTest.lookup(name) == null);
 
     }
+
     @Test
     public void lookupTest() {
         PhoneBook phTest = new PhoneBook();
         String name = "John";
         String number = "8765930";
         phTest.add(name, number);
-        Assert.assertTrue(phTest.lookup(name).equals(number));
+        Assert.assertTrue(phTest.lookup(name).contains(number));
     }
 
 
@@ -81,37 +82,63 @@ public class PhoneBookTest {
     }
 
     @Test
-    public void listRecordsTest(){
+    public void listRecordsTest() {
         PhoneBook phTest = new PhoneBook();
         //added two new entries to test
-        phTest.add("Nico","487542");
-        phTest.add("Laura","1-800-no-sugar");
+        phTest.add("Dumbo", "487542");
+        phTest.add("Blumbo", "1800234583");
 
         //turned into an array to iterate through.
         Object[] array = phTest.listRecords().toArray();
         //to.Array returns Object[] array
         for (int i = 0; i < array.length; i++) {
             //Object that was in array was Map.Entry: a record on map (key value pair)
-            Map.Entry record = (Map.Entry) array[i];
+            Map.Entry<String, List<String>> record = (Map.Entry) array[i];
             //0 index = Laura because treemap is sorted alphabetically
-           if (i == 0 ){
-               Assert.assertTrue(record.getKey().equals("Laura"));
-               Assert.assertTrue(record.getValue().equals("1-800-no-sugar"));
-           }else{
-               Assert.assertTrue(record.getKey().equals("Nico"));
-               Assert.assertTrue(record.getValue().equals("487542"));
-           }
+            if (i == 0) {
+                Assert.assertTrue(record.getKey().equals("Blumbo"));
+                Assert.assertTrue(record.getValue().contains("1800234583"));
+            } else {
+                Assert.assertTrue(record.getKey().equals("Dumbo"));
+                Assert.assertTrue(record.getValue().contains("487542"));
+            }
+        }
+    }
+
+    @Test
+    public void reverselookupTest() {
+        PhoneBook phTest = new PhoneBook();
+        String expectedName = "Blumbo";
+        String expectedNumber = "1234567";
+        phTest.add(expectedName, expectedNumber);
+        //adding entry to test return
+        String actual = phTest.reverseLookup(expectedNumber);
+        Assert.assertEquals(expectedName, actual);
+
+    }
 
+    @Test
+    public void addTwoNumbersTest() {
+        ArrayList<String> phoneNumbers = new ArrayList<String>();
+        phoneNumbers.add("1111");
+        phoneNumbers.add("2222");
+
+        PhoneBook phTest = new PhoneBook();
+        String expectedName = "Blumbo";
+        phTest.add(expectedName, phoneNumbers.get(0));
+        phTest.add(expectedName, phoneNumbers.get(1));
+
+
+
+        //adding entry to test return
+        List<String> actual = phTest.lookup(expectedName);
+
+        for(String phoneNumber : phoneNumbers) {
+            Assert.assertTrue(actual.contains(phoneNumber));
         }
 
+
     }
 
-        //    @Test
-//    public void reverselookupTest() {
-//        String name = "Bob";
-//        String number = "8765930";
-//        phTest.add(name, number);
-//        Assert.assertTrue(phTest.lookup(number).equals(name));
-//    }
 
     }