committing Phone-book

PhoneBook.java

@@ -1,7 +1,40 @@
  
-
+import java.util.*;
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
+    //private String name;
+    //private String phoneNum;
+    TreeMap<String, String> phoneListByName = new TreeMap();
+    TreeMap<String, String> phoneListByNum = new TreeMap();
+    
+    public void add(String name, String phoneNumber){
+        phoneListByName.put(name, phoneNumber);
+        phoneListByNum.put(phoneNumber, name);
+        
+    }
+    public void remove(String name){
+        String phoneNumber = lookUp(name);
+        phoneListByName.remove(name);
+        phoneListByNum.remove(phoneNumber);
+    }
+    public String lookUp(String name){
+       return phoneListByName.get(name); 
+       
+    }
+    public String reverseLookUp(String phoneNumber){
+       return phoneListByNum.get(phoneNumber); 
+       
+    }
+    public void display(){
+        Set set = phoneListByName.entrySet();
+        //Set set2 = phoneListByNum.entrySet();
+        Iterator i = set.iterator();
+        
+        while(i.hasNext()){
+            Map.Entry name = (Map.Entry)i.next();
+            System.out.println(name.getKey() + ": " + name.getValue()) ;
+        }
+    }
 }

PhoneBookTest.java

@@ -13,6 +13,8 @@ import org.junit.Test;
  */
 public class PhoneBookTest
 {
+    
+    PhoneBook testObj;
     /**
      * Default constructor for test class PhoneBookTest
      */
@@ -28,6 +30,7 @@ public class PhoneBookTest
     @Before
     public void setUp()
     {
+        testObj = new PhoneBook();
     }
 
     /**
@@ -39,4 +42,34 @@ public class PhoneBookTest
     public void tearDown()
     {
     }
-}
+    @Test
+    public void testadd(){
+     String expected = "4545666667"; 
+     //PhoneBook phoneBook = new PhoneBook();
+     testObj.add("john",expected);
+     String actual = testObj.lookUp("john");
+     assertEquals(actual, expected);
+    }
+    @Test
+    public void testremoveAndLookUp(){
+    String expected = "5454";
+    testObj.add("gei", "5454");
+    assertEquals(expected,testObj.lookUp("gei"));
+    
+    String expectedAfterDelete = null;
+    testObj.remove("gei");
+    assertEquals(expectedAfterDelete, testObj.lookUp("gei"));
+      
+    }
+    @Test
+    public void testreverseLookUp(){
+    String expected = "John";
+    testObj.add("John", "4343");
+    assertEquals(expected, testObj.reverseLookUp("4343"));
+    
+    
+    
+    }
+     
+    }
+

package.bluej

@@ -1,8 +1,11 @@
 #BlueJ package file
-editor.fx.0.height=0
-editor.fx.0.width=0
-editor.fx.0.x=0
-editor.fx.0.y=0
+dependency1.from=PhoneBookTest
+dependency1.to=PhoneBook
+dependency1.type=UsesDependency
+editor.fx.0.height=722
+editor.fx.0.width=800
+editor.fx.0.x=240
+editor.fx.0.y=25
 objectbench.height=101
 objectbench.width=461
 package.divider.horizontal=0.6
@@ -13,8 +16,8 @@ package.editor.x=427
 package.editor.y=143
 package.frame.height=600
 package.frame.width=800
-package.numDependencies=0
-package.numTargets=1
+package.numDependencies=1
+package.numTargets=2
 package.showExtends=true
 package.showUses=true
 project.charset=UTF-8
@@ -23,4 +26,17 @@ readme.name=@README
 readme.width=47
 readme.x=10
 readme.y=10
-
+target1.height=50
+target1.name=PhoneBookTest
+target1.showInterface=false
+target1.type=UnitTestTargetJunit4
+target1.width=120
+target1.x=100
+target1.y=10
+target2.height=50
+target2.name=PhoneBook
+target2.showInterface=false
+target2.type=ClassTarget
+target2.width=100
+target2.x=70
+target2.y=70