reverseLookup left

PhoneBook.java

@@ -1,4 +1,58 @@
-
+import java.util.Map;
+import java.util.*;
 public class PhoneBook {
-    
+    Map<String,List<String>> phoneBook = new TreeMap<String,List<String>>();
+
+    public PhoneBook(){
+
+    }
+
+    public void addEntrant(String name, String...phoneNumber){
+        String checkName = lookup(name);
+        if(checkName == null){
+            phoneBook.put(name, Arrays.asList(phoneNumber));
+        } else {
+            System.out.println("Error, user already exists");
+        }
+    }
+
+    public void removeEntrant(String name){
+        phoneBook.remove(name);   
+    }
+
+    public String lookup(String name){
+        if (phoneBook.get(name) == null){
+            return null;
+        }
+        StringBuilder numbersList = new StringBuilder();
+        int counter = 0;
+        for(String number: phoneBook.get(name)){
+            numbersList.append(number);
+            counter++;
+            if(counter != phoneBook.get(name).size()){
+                numbersList.append(", ");
+            }
+        }
+
+        return numbersList.toString();
+    }
+
+    public String reverseLookup(String phoneNumber){
+        for(Map.Entry<String, List<String>> element: phoneBook.entrySet()){
+            String key = element.getKey();
+            List<String> value = element.getValue();
+            if(value.equals(phoneNumber)){
+                return key;
+            }
+        }
+        return "This number does not belong to anyone";
+    }
+
+    public void display(){
+        for(Map.Entry<String, List<String>> element: phoneBook.entrySet()){
+            String key = element.getKey();
+            List<String> value = element.getValue();
+            System.out.println( key + ":" + value);
+        }
+    }
 }

PhoneBookTest.java

@@ -7,12 +7,68 @@ import org.junit.Test;
 
 public class PhoneBookTest
 {
-    /**
-     * Sets up the test fixture.
-     *
-     * Called before every test case method.
-     */
+
     @Before
-    public void setUp(){    
+    public void setUp(){ 
+        PhoneBook pb = new PhoneBook(); 
+    }
+    
+    @Test
+    public void testAdd1(){
+        PhoneBook pb = new PhoneBook();
+        
+        String expected = "123, 456";
+        
+        pb.addEntrant("John" , "123", "456");
+        String actual = pb.lookup("John");
+        
+        assertEquals(expected, actual);
+    }
+    
+    @Test
+    public void testAdd2(){
+        PhoneBook pb = new PhoneBook();
+        
+        String expected = "123";
+        
+        pb.addEntrant("Josh" , "123");
+        String actual = pb.lookup("Josh");
+        
+        assertEquals(expected, actual);
+    }
+    
+    
+    @Test
+    public void testRemove(){
+        PhoneBook pb = new PhoneBook();
+        
+        String expected = null;
+        
+        pb.addEntrant("Josh" , "123");
+        pb.removeEntrant("Josh");
+        String actual = pb.lookup("Josh");
+        
+        assertEquals(expected, actual);
+        
+    }
+    
+    @Test
+    public void testLookup(){
+        PhoneBook pb = new PhoneBook();
+        
+        String expected = "123, 456, 789";
+        
+        pb.addEntrant("Josh" , "123", "456", "789");
+        String actual = pb.lookup("Josh");
+        
+        assertEquals(expected, actual);
+    }
+    
+    @Test
+    public void testReverseLookup(){
+        
+        
     }
+    
+    
 }