Missing lookup

pom.xml

@@ -5,7 +5,7 @@
     <modelVersion>4.0.0</modelVersion>
 
     <groupId>com.zipcodewilmington</groupId>
-    <artifactId>phonebok</artifactId>
+    <artifactId>phonebook</artifactId>
     <version>1.0-SNAPSHOT</version>
     <build>
         <plugins>
@@ -19,11 +19,12 @@
             </plugin>
         </plugins>
     </build>
+
     <dependencies>
         <dependency>
-            <groupId>org.junit.jupiter</groupId>
-            <artifactId>junit-jupiter-api</artifactId>
-            <version>RELEASE</version>
+            <groupId>junit</groupId>
+            <artifactId>junit</artifactId>
+            <version>4.12</version>
         </dependency>
     </dependencies>
 

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

@@ -1,36 +1,54 @@
 package com.zipcodewilmington.phonebook;
-import org.junit.Assert;
-import org.junit.Test;
-import org.junit.jupiter.api.Test;
-import sun.jvm.hotspot.utilities.Assert;
-
-//added all useful utilities
-Private class Person {
-    private String name;
-    private String phNumber;
-//initialize strings
-    public Person (String name, String phNumber) {
-        this.name = name;
-        this.phNumber = phNumber;
-    }
-//get entry (name and number)
-    public void removeEntry(String name) {
 
+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<>();
     }
-//remove entry from list
-    public String[] getEntry() {
-        return getEntry();
+
+    //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);
+        }
     }
-//Look up person
-    public void listAll(){
-        String[] listAll = new String[];
-        return;
+
+    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);
+                }
+            }
+        }
     }
-//List all names in registry
-    public String getPhNumber() {
-        return getPhNumber+= listAll();
+
+    public ArrayList<String> lookup(String name) {
+        //not sure of this yet
+        return contacts.get(name);
     }
-//List all entries
-}
 
-//this is just a skeleton, gonna work on syntax
+    public void removeRecord(String name) {
+        contacts.remove(name);
+    }
+}

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

@@ -1,37 +0,0 @@
-package com.zipcodewilmington.phonebook;
-
-
-import org.junit.jupiter.api.Test;
-import sun.jvm.hotspot.utilities.Assert;
-
-
-class PersonTest {
-//need to generate test to add an entry to phbook
-        PhBook = new PhBook(addEntry);
-        name = new Person("name", "number");}
-
-    }
-
-    @Test
-    void removeEntry1() {
-            Person.add("name", "number");
-            Person.remove("name");
-            Assert.assertFalse(Person.entryCheck("name"));
-}
-
-    @Test
-    void getEntry() {
-            Person.add("name", "number");
-            Assert.assertEquals("name".getPhNumber(), Person.get("name"));
-}
-
-    @Test
-    void listAll() {
-//need to set parameters for test
-}
-
-    @Test
-    void getPhNumber1() {
-//not sure of the syntax how to set keys and values in array list and return
-}
-}

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());
+    }
+}