finished

PhoneBook.java

@@ -1,7 +1,52 @@
- 
+import java.util.*;
 
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
+  String name;
+  String phoneNumber;
+  TreeMap<String, String> phoneBook = new TreeMap<>();
+
+  public PhoneBook(TreeMap<String, String> phoneBook) {
+    this.phoneBook = phoneBook;
+  }
+
+  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){
+    return phoneBook.get(name);
+  }
+
+  public String reverseLookup(String phoneNumber) {
+    for (Map.Entry<String, String> e : phoneBook.entrySet()) {
+      String name = e.getKey();
+      String thisNumber = e.getValue();
+
+      if (thisNumber == phoneNumber) {
+        return name;
+      }
+    }
+    return "Name not found";
+  }
+
+  public String display() {
+    StringBuilder sb = new StringBuilder();
+
+    for (Map.Entry<String, String> e : phoneBook.entrySet()){
+      String name = e.getKey();
+      String thisNumber = e.getValue();
+
+      sb.append(name).append(" ").append(thisNumber).append("\n");
+    }
+    return sb.toString();
+  }
 }

PhoneBookTest.java

@@ -1,10 +1,12 @@
 
 
-import static org.junit.Assert.*;
+import org.junit.Assert;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 
+import java.util.*;
+
 /**
  * The test class PhoneBookTest.
  *
@@ -13,6 +15,8 @@ import org.junit.Test;
  */
 public class PhoneBookTest
 {
+
+    TreeMap<String, String> phoneBook = new TreeMap<>();
     /**
      * Default constructor for test class PhoneBookTest
      */
@@ -28,6 +32,7 @@ public class PhoneBookTest
     @Before
     public void setUp()
     {
+      PhoneBook pb = new PhoneBook(phoneBook);
     }
 
     /**
@@ -38,5 +43,56 @@ public class PhoneBookTest
     @After
     public void tearDown()
     {
+
+    }
+
+    // add entries
+    @Test
+    public void testOne() { 
+      String expected = "Dog 222-444-5555\nZebra 111-222-3333\n";
+
+      PhoneBook pb = new PhoneBook(phoneBook);
+      pb.add("Zebra", "111-222-3333");
+      pb.add("Dog", "222-444-5555");
+
+      String actual = pb.display();
+
+      Assert.assertEquals(expected, actual);
+    }
+    @Test
+    public void testTwo() {
+      String expected = "Dog 222-444-5555\n";
+      PhoneBook pb = new PhoneBook(phoneBook);
+        
+      pb.add("Zebra", "111-222-3333");
+      pb.add("Dog", "222-444-5555");
+      pb.remove("Zebra");
+        
+      String actual = pb.display();
+
+      Assert.assertEquals(expected, actual);  
+    }
+    @Test
+    public void TestThree() {
+      String expected = "222-444-5555";
+      PhoneBook pb = new PhoneBook(phoneBook);
+        
+      pb.add("Zebra", "111-222-3333");
+      pb.add("Dog", "222-444-5555");
+      
+      String actual = pb.lookup("Dog");
+        
+      Assert.assertEquals(expected, actual);
+    }
+    @Test
+    public void TestFour() {
+      String expected = "Dog";
+      PhoneBook pb = new PhoneBook(phoneBook);
+        
+      pb.add("Zebra", "111-222-3333");
+      pb.add("Dog", "222-444-5555");
+      
+      String actual = pb.reverseLookup("222-444-5555");
+      Assert.assertEquals(expected, actual);
     }
 }

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=722
+editor.fx.0.width=800
+editor.fx.0.x=590
+editor.fx.0.y=136
 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=35
+package.editor.y=158
 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