display done

PhoneBook.java

@@ -1,7 +1,57 @@
- 
+import java.util.*;
+import java.util.TreeMap;
+import java.util.Set;
+import java.util.Map;
+import java.util.Iterator;
 
-/**
- * Created by leon on 1/23/18.
- */
 public class PhoneBook {
+    String phoneNumber;
+    Map<String, String> pb = new TreeMap <String, String>();
+
+    public PhoneBook(){
+    }
+
+    public void add(String name, String phoneNumber){
+
+        pb.put(name, phoneNumber);
+
+    }
+
+    public void remove(String name){
+        pb.remove(name);
+    }
+
+    public String display(){
+        Set set = pb.entrySet();
+        Iterator it = set.iterator();
+        StringBuilder display = new StringBuilder();
+
+        while(it.hasNext()) {
+            Map.Entry me = (Map.Entry)it.next();
+            display.append(me.getKey() + " : " + me.getValue() + "\n");
+            
+        }
+        return display.toString();
+    }
+
+    public String lookup(String name){
+        String phone = pb.get(name);
+        return phone;
+    }
+
+    public String reverseLookup(String phoneNumber){
+        String key= null;
+        for(Map.Entry entry: pb.entrySet()){
+            if(phoneNumber.equals(entry.getValue())){
+                key = entry.getKey().toString();
+                break; 
+            }
+        }
+        return key;
+
+    }
+
+    public int size() {
+        return pb.size();
+    }
 }

PhoneBookTest.java

@@ -1,42 +1,55 @@
-
-
+import java.util.*;
 import static org.junit.Assert.*;
-import org.junit.After;
-import org.junit.Before;
 import org.junit.Test;
 
-/**
- * The test class PhoneBookTest.
- *
- * @author  (your name)
- * @version (a version number or a date)
- */
-public class PhoneBookTest
+
+public class PhoneBookTest 
 {
-    /**
-     * Default constructor for test class PhoneBookTest
-     */
-    public PhoneBookTest()
-    {
+    @Test
+    public void addTest(){
+    PhoneBook phoneBook = new PhoneBook();
+    phoneBook.add("vince","333-333-3333");
+    int actual = phoneBook.size();
+    int expected = 1;
+    assertEquals(actual,expected);
     }
 
-    /**
-     * Sets up the test fixture.
-     *
-     * Called before every test case method.
-     */
-    @Before
-    public void setUp()
-    {
+    @Test
+    public void removeTest(){
+    PhoneBook phoneBook = new PhoneBook();
+    phoneBook.add("vince","333-333-3333");
+    phoneBook.remove("vince");
+    int actual = phoneBook.size();
+    int expected = 0;
+    assertEquals(actual,expected);
     }
-
-    /**
-     * Tears down the test fixture.
-     *
-     * Called after every test case method.
-     */
-    @After
-    public void tearDown()
-    {
+    
+    @Test
+    public void lookupTest(){
+    PhoneBook phoneBook = new PhoneBook();
+    phoneBook.add("vince","333-333-3333");
+    String actual =  phoneBook.lookup("vince");
+    String expected = "333-333-3333";
+    assertEquals(actual,expected);
+    }
+        @Test
+    public void lookupReverseTest(){
+    PhoneBook phoneBook = new PhoneBook();
+    phoneBook.add("vince","333-333-3333");
+    String actual =  phoneBook.reverseLookup("333-333-3333");
+    String expected = "vince";
+    assertEquals(actual,expected);
+    }
+    
+            @Test
+    public void displayTest(){
+    PhoneBook phoneBook = new PhoneBook();
+    phoneBook.add("vince","333-333-3333");
+    phoneBook.add("bill","444-444-4444");
+    phoneBook.add("john","555-555-5555");
+    String actual =  phoneBook.display();
+    String expected = "bill : 444-444-4444\njohn : 555-555-5555\nvince : 333-333-3333\n";
+    assertEquals(actual,expected);
     }
+    
 }

package.bluej

@@ -1,20 +1,23 @@
 #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=705
+editor.fx.0.width=800
+editor.fx.0.x=123
+editor.fx.0.y=23
 objectbench.height=101
 objectbench.width=461
 package.divider.horizontal=0.6
 package.divider.vertical=0.8007380073800738
 package.editor.height=427
 package.editor.width=674
-package.editor.x=427
-package.editor.y=143
+package.editor.x=33
+package.editor.y=99
 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