Adding more tests in a second

pom.xml

@@ -7,6 +7,18 @@
     <groupId>com.zipcodewilmington</groupId>
     <artifactId>phonebok</artifactId>
     <version>1.0-SNAPSHOT</version>
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <configuration>
+                    <source>1.8</source>
+                    <target>1.8</target>
+                </configuration>
+            </plugin>
+        </plugins>
+    </build>
     <dependencies>
         <dependency>
             <groupId>junit</groupId>

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

@@ -0,0 +1,28 @@
+/*package com.zipcodewilmington.phonebook;
+
+public class Person {
+    public final String name;
+    public final PhoneNumbers phoneNumbers;
+
+    public Person(String name, PhoneNumbers phoneNumbers) {
+        this.name = name;
+        this.phoneNumbers = phoneNumbers;
+    }
+
+    public void add(String phoneNumber) {
+        phoneNumbers.addToContacts(phoneNumber);
+    }
+
+    public void remove(String phoneNumber) {
+        phoneNumbers.removeFromContacts(phoneNumber);
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public PhoneNumbers getPhoneNumber() {
+        return phoneNumbers;
+    }
+}
+*/

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

@@ -1,25 +1,28 @@
 package com.zipcodewilmington.phonebook;
 
+import java.util.ArrayList;
+import java.util.List;
 import java.util.Map;
 import java.util.TreeMap;
-import java.util.ArrayList;
 
 public class PhoneBook {
 
-    TreeMap<String, String> phoneBookList = new TreeMap<String, String>();
+    TreeMap<String, List<String>> phoneBookList = new TreeMap<String, List<String>>();
 
-    public void addEntry(String name, String number) {
-        phoneBookList.put(name, number);
-        //Look to see if person exists
-        //Get array list value
+    public void addEntry(String name, String ...phoneNumber) {
+        List<String> numberList = new ArrayList<String>();
+        for(String number : phoneNumber) {
+            numberList.add(number);
+        }
+        phoneBookList.put(name, numberList);
     }
 
-    public void removeEntry(String name, String number) {
-        phoneBookList.remove(name, number);
+    public void removeEntry(String name) {
+        phoneBookList.remove(name);
     }
 
     public String lookUp(String name) {
-        return phoneBookList.get(name);
+        return phoneBookList.get(name).toString();
     }
 
     public String listNames() {
@@ -29,28 +32,29 @@ public class PhoneBook {
     public String listAll() {
 
         StringBuilder sb = new StringBuilder();
-        for(Map.Entry<String, String> entry : phoneBookList.entrySet()) {
+        for(Map.Entry<String, List<String>> entry : phoneBookList.entrySet()) {
             sb.append(entry.getKey() + entry.getValue());
             sb.append("\n");
         }
         return sb.toString();
-
     }
 
     public String reverseLookUp(String number) {
-        for(Map.Entry<String, String>entry : phoneBookList.entrySet()) {
-            if(entry.getValue() == number) {
+        for(Map.Entry<String, List<String>> entry : phoneBookList.entrySet()) {
+            if(entry.getValue().contains(number)) {
                 return entry.getKey();
             }
         }
         return "That number is not in the phone book";
     }
 
-    public static void main(String[] args) {
-        PhoneBook pb1 = new PhoneBook();
-        pb1.addEntry("Brian", "12345678");
-        pb1.addEntry("Lawrence", "23458984");
-        pb1.addEntry("Anthony", "12381927");
-
+    public void removeOneNumber(String name, String number) {
+        int i = 0;
+        for (String phoneNumber : phoneBookList.get(name)) {
+            if(phoneNumber == number) {
+                phoneBookList.get(name).remove(i);
+            }
+            i++;
+        }
     }
 }

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

@@ -0,0 +1,17 @@
+/*package com.zipcodewilmington.phonebook;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class PhoneNumbers {
+    List<String> contactList = new ArrayList<String>();
+
+    public void addToContacts(String phoneNumber) {
+        contactList.add(phoneNumber);
+    }
+
+    public void removeFromContacts(String phoneNumber) {
+        contactList.remove(phoneNumber);
+    }
+}
+*/

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

@@ -38,7 +38,7 @@ public class PhoneBookTest {
     public void removeEntryTest() {
         String name = "Person 1";
         String number = "8942418";
-        phoneBookTest.removeEntry(name, number);
+        phoneBookTest.removeEntry(name);
         String expected = null;
         String actual = phoneBookTest.lookUp(name);
         Assert.assertEquals(expected, actual);
@@ -49,7 +49,7 @@ public class PhoneBookTest {
     public void removeEntryTest2() {
         String name = "Person 2";
         String number = "12387874";
-        phoneBookTest.removeEntry(name, number);
+        phoneBookTest.removeEntry(name);
         String expected = null;
         String actual = phoneBookTest.lookUp(name);
         Assert.assertEquals(expected, actual);
@@ -107,4 +107,10 @@ public class PhoneBookTest {
         Assert.assertEquals(expected, actual);
 
     }
+
+    @Test
+    public void removeOneNumberTest() {
+        String name = "Brian";
+
+    }
 }