Merge b14f2b38a2e028ba9e17f4d1c4e5bd9b47fe45fb into 74a9673e36c7d25c28315cb54247a91792f8fbc0

pom.xml

@@ -5,8 +5,28 @@
     <modelVersion>4.0.0</modelVersion>
 
     <groupId>com.zipcodewilmington</groupId>
-    <artifactId>phonebok</artifactId>
+    <artifactId>phonebook</artifactId>
     <version>1.0-SNAPSHOT</version>
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <configuration>
+                    <source>1.7</source>
+                    <target>1.7</target>
+                </configuration>
+            </plugin>
+        </plugins>
+    </build>
+
+    <dependencies>
+        <dependency>
+            <groupId>junit</groupId>
+            <artifactId>junit</artifactId>
+            <version>4.12</version>
+        </dependency>
+    </dependencies>
 
 
 </project>

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

@@ -1,7 +1,54 @@
 package com.zipcodewilmington.phonebook;
 
-/**
- * Created by leon on 1/23/18.
- */
+import java.util.ArrayList;
+import java.util.Map;
+import java.util.TreeMap;
+
+//Declare Treemap name
 public class PhoneBook {
-}
+    //Declare Treemap name
+    //method 1
+    private TreeMap<String, ArrayList<String>> contacts;
+    private static final String add = "No such entry";
+    public String name;
+    public String phoneNumber;
+
+    public PhoneBook() {
+        contacts = new TreeMap<>();
+    }
+
+    //Instantiate new array list and add the number
+    public void add(String name, String phoneNumber) {
+        this.name = name;
+        this.phoneNumber = phoneNumber;
+        if (contacts.containsKey(name)) {
+            contacts.get(name).add(phoneNumber);
+        }
+
+        else {
+            ArrayList<String> phoneNumbers = new ArrayList<>();
+            phoneNumbers.add(phoneNumber);
+            contacts.put(name, phoneNumbers);
+        }
+    }
+
+    public void remove(String phoneNumber) {
+        for (Map.Entry<String, ArrayList<String>> entry : contacts.entrySet()) {
+            ArrayList<String> phoneNumArrList = entry.getValue();
+            for (int i = 0; i < phoneNumArrList.size(); i++) {
+                if (phoneNumber.equals(phoneNumArrList.get(i))) {
+                    phoneNumArrList.remove(i);
+                }
+            }
+        }
+    }
+
+    public ArrayList<String> lookup(String name) {
+        //not sure of this yet
+        return contacts.get(name);
+    }
+
+    public void removeRecord(String name) {
+        contacts.remove(name);
+    }
+}

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

@@ -0,0 +1,82 @@
+package com.zipcodewilmington.phonebook;
+
+import java.util.Map;
+import java.util.TreeMap;
+import org.junit.Test;
+import org.junit.Assert;
+import java.util.ArrayList;
+
+
+/**
+ * Created by leon on 1/23/18.
+ */
+public class PhoneBook {
+
+    @Test
+    public static void add() {
+        //Given
+        PhoneBook phoneBook = new PhoneBook();
+
+        //When
+        String expected = "1234567890";
+        PhoneBook.add();
+        //Assert
+        String actual = PhoneBook.lookup();
+        Assert.assertEquals(actual, expected);
+    }
+
+    @Test
+    public void add2() {
+        //Given
+        PhoneBook phoneBook = new PhoneBook();
+
+        //When
+        String expected = "1234567890";
+        phoneBook.add("bovice", "1234567890");
+        phoneBook.add("bovice", "1234567890");
+
+        //Assert
+        String actual = phoneBook.lookup("bovice").get(1);
+        Assert.assertEquals(actual, expected);
+    }
+
+    @Test
+    public void remove() {
+        //Given
+        PhoneBook phoneBook = new PhoneBook();
+        //When
+        phoneBook.add("bovice", "7777777777");
+        phoneBook.add("bovice", "1234567890");
+        phoneBook.remove("1234567890");
+
+        //Assert
+        ArrayList<String> actual = phoneBook.lookup("bovice");
+        Assert.assertEquals(1, actual.size());
+    }
+
+    @Test
+    public static String lookup() {
+//Still needs a look up function
+        return null;
+    }
+
+    @Test
+    public void remove2() {
+        //Given
+        PhoneBook phoneBook = new PhoneBook();
+        phoneBook.add("bovice", "7777777777");
+        phoneBook.add("bovice", "1234567890");
+        phoneBook.add("vance", "8888888888");
+        phoneBook.add("vance", "0987654321");
+        phoneBook.add("dr shivago", "3027778899");
+
+        //When
+        phoneBook.remove("7777777777");
+
+        //Assert
+        ArrayList<String> actual1 = phoneBook.lookup("bovice");
+        ArrayList<String> actual2 = phoneBook.lookup("vance");
+        Assert.assertEquals(1, actual1.size());
+        Assert.assertEquals(1, actual2.size());
+    }
+}

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

@@ -1,7 +0,0 @@
-package com.zipcodewilmington.phonebook;
-
-/**
- * Created by leon on 1/23/18.
- */
-public class PhoneBookTest {
-}