phone book

PhoneBook.java

@@ -1,7 +1,61 @@
- 
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.TreeMap;
 
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
+    String name;
+    String phoneNumber = "";
+    Map<String, String> phoneBook = new TreeMap<String, String>();
+
+    public PhoneBook() {
+    }
+
+    public void add(String name, String phoneNumber) {
+        phoneBook.put(name, phoneNumber);
+    }
+
+    public void remove(String name) {
+        phoneBook.remove(name);
+    }
+
+    public String lookup(String name) {
+        String phoneNumber = phoneBook.get(name);
+        return phoneNumber;
+    }
+
+    public String reverseLookup(String value){
+        String key = "";
+
+        for(Map.Entry entry : phoneBook.entrySet())
+            if (value.equals(entry.getValue())) {
+                key = entry.getKey().toString();
+            }
+        return key;
+    }
+
+    public String display() {
+        Set set = phoneBook.entrySet();
+        Iterator iter = set.iterator();
+
+        StringBuilder answer = new StringBuilder();
+
+        while(iter.hasNext()) {
+            Map.Entry i = (Map.Entry)iter.next();
+            answer.append(i.getKey() + " : " + i.getValue());
+        }
+
+        String result = answer.toString();
+        return result;
+    }
+
+    public int getSize() {
+        return phoneBook.size();
+    }
 }
+

PhoneBookTest.java

@@ -16,8 +16,8 @@ public class PhoneBookTest
     /**
      * Default constructor for test class PhoneBookTest
      */
-    public PhoneBookTest()
-    {
+    public PhoneBookTest(){
+        PhoneBook phoneBook = new PhoneBook();
     }
 
     /**
@@ -29,7 +29,67 @@ public class PhoneBookTest
     public void setUp()
     {
     }
-
+    @Test
+    public void assTest() {
+    
+        PhoneBook phoneBook = new PhoneBook();
+        
+        //given
+        phoneBook.add("Ben", "358-1032");
+        
+        //then
+        
+        String actual = phoneBook.lookup("Ben");
+        String expected = "358-1032";
+        
+        //assert
+        
+        assertEquals(expected, actual);
+    
+    }
+    @Test
+    public void removeTest() {
+        PhoneBook phoneBook = new PhoneBook();
+        
+        //given
+        phoneBook.add("Ben", "358-1032");
+        phoneBook.remove("Ben");
+        
+        //then
+        int actual = phoneBook.getSize();
+        int expected = 0;
+        
+        //assert
+        assertEquals(expected, actual);
+    }
+    @Test
+    public void reverseLookupTest() {
+        PhoneBook phoneBook = new PhoneBook();
+        
+        //given
+        phoneBook.add("Ben", "358-1032");
+        
+        //then
+        String actual = phoneBook.reverseLookup("358-1032");
+        String expected = "Ben";
+        
+        //assert
+        assertEquals(expected, actual);
+    }
+    @Test
+    public void displayTest() {
+        PhoneBook phoneBook = new PhoneBook();
+        
+        //given
+        phoneBook.add("Ben", "358-1032");
+        
+        //then
+        String actual = phoneBook.display();
+        String expected = "Ben : 358-1032";
+        
+        //assert
+        assertEquals(expected, actual);
+    }
     /**
      * Tears down the test fixture.
      *

package.bluej

@@ -1,8 +1,8 @@
 #BlueJ package file
-editor.fx.0.height=0
-editor.fx.0.width=0
-editor.fx.0.x=0
-editor.fx.0.y=0
+editor.fx.0.height=722
+editor.fx.0.width=800
+editor.fx.0.x=320
+editor.fx.0.y=53
 objectbench.height=101
 objectbench.width=461
 package.divider.horizontal=0.6
@@ -14,7 +14,7 @@ package.editor.y=143
 package.frame.height=600
 package.frame.width=800
 package.numDependencies=0
-package.numTargets=1
+package.numTargets=2
 package.showExtends=true
 package.showUses=true
 project.charset=UTF-8
@@ -23,4 +23,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