Raymond

PhoneBook.java

@@ -1,7 +1,55 @@
- 
 
+import java.util.TreeMap;
+import java.util.*;
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
+    TreeMap<String, ArrayList<String>> pb;
+
+    public PhoneBook(){
+        pb = new TreeMap<>();   
+
+    }
+
+    public void addElement(String name, String phoneNumber){
+        ArrayList<String> aList  =  pb.containsKey(name) ? pb.get(name): new ArrayList<String>();
+        aList.add(phoneNumber + "\n");
+        pb.put(name, aList);
+    }
+
+    public void remove(String name){
+        pb.remove(name); 
+    }
+
+    public String lookUp(String name){
+        StringBuilder look = new StringBuilder();
+
+        for(String element: this.pb.get(name)){
+            
+                look.append(element);
+                
+            
+        }
+        return look.toString();
+    }
+
+    public String reverseLookup(String phoneNumber){
+        String gettingThatValue = null;
+        for(Map.Entry<String,ArrayList<String>> element: pb.entrySet()){
+            if(element.getValue().contains(phoneNumber)){
+                gettingThatValue = element.getKey();
+                break;
+            }
+        }
+
+        return gettingThatValue;
+    }
+
+    public void display(){
+        for(Map.Entry<String,ArrayList<String>> element:pb.entrySet()){
+            System.out.print(element.getKey() + ":" + element.getValue().toString());
+        }
+    }
+
 }

PhoneBookTest.java

@@ -1,5 +1,4 @@
 
-
 import static org.junit.Assert.*;
 import org.junit.After;
 import org.junit.Before;
@@ -12,31 +11,98 @@ import org.junit.Test;
  * @version (a version number or a date)
  */
 public class PhoneBookTest
-{
-    /**
-     * Default constructor for test class PhoneBookTest
-     */
-    public PhoneBookTest()
-    {
+{   private PhoneBook phonebook;
+    private String name;
+    private String phoneNumber;
+    
+    @Before
+    public void setUp(){
+        phonebook = new PhoneBook();
+    
+    } 
+    
+    
+
+    @Test
+    public void addTestFirstNumberForName(){
+        String name = "Raymond";
+        String expected = "302-444-0094\n";
+        
+        phonebook.addElement(name,"302-444-0094");
+        String actual = phonebook.lookUp(name);
+        
+        assertEquals(expected,actual);
+        
+    }
+    
+    @Test
+    public void testAddKeyAlreadyExists() {
+        String name = "Raymond";
+        String phoneNumber1 = "302-450-8066";
+        String phoneNumber2 = "302-465-1770";
+        
+        String expected = phoneNumber1 + "\n" + phoneNumber2 + "\n";
+        phonebook.addElement(name,phoneNumber1);
+        phonebook.addElement(name,phoneNumber2);
+        String actual = phonebook.lookUp(name);
+        
+        assertEquals(expected,actual); 
     }
 
-    /**
-     * Sets up the test fixture.
-     *
-     * Called before every test case method.
-     */
-    @Before
-    public void setUp()
-    {
+    @Test
+    public void removeTest(){
+        String name1 = "Raymond";
+        String name2 = "Eric";
+        String phoneNumber1 = "302-450-8099";
+        String phoneNumber2 = "302-333-0003";
+        phonebook.addElement(name1, phoneNumber1);
+        phonebook.addElement(name2, phoneNumber2);
+        
+        String expected = "Eric";
+        phonebook.remove(name1);
+        String actual = name2;
+        
+        assertEquals(expected,actual);
     }
 
-    /**
-     * Tears down the test fixture.
-     *
-     * Called after every test case method.
-     */
-    @After
-    public void tearDown()
-    {
+    @Test
+    public void lookUpTest(){
+        String name = "Raymond";
+        String phoneNumber = "302-333-3333";
+        phonebook.addElement(name,phoneNumber);
+        
+        String expected = "302-333-3333";
+        phonebook.lookUp(name);
+        String actual = phoneNumber;
+        
+        assertEquals(expected,actual);
     }
+
+    @Test
+    public void reverseLookup(){
+        String name = "Raymond";
+        String phoneNumber = "302-333-3333";
+        phonebook.addElement(name,phoneNumber);
+        
+        String expected = "Raymond";
+        phonebook.reverseLookup(phoneNumber);
+        String actual = name;
+        
+        assertEquals(expected,actual);
+    }
+
+    @Test
+    public void displayTest(){
+        String name = "Raymond";
+        String phoneNumber = "302-450-2222";
+        phonebook.addElement(name,phoneNumber);
+        
+        String expected = "Raymond:302-450-2222";
+        phonebook.display();
+        String actual = name + ":" + phoneNumber;
+
+        assertEquals(expected,actual);
+    }
+
+    
 }