Step two and added tests

src/main/java/com/zipcodewilmington/phonebook/PhoneBook.java

@@ -33,12 +33,8 @@ public class PhoneBook {
     }
 
     //remove a name & number entry
-    public String remove(String name) {
-        String removeName = name + " " + treeMap.get(name) + " has been removed";
-        if(treeMap.containsKey(name)) {
+    public void remove(String name) {
             treeMap.remove(name);
-        }
-        return removeName;
     }
 
     //find phone number lookup by name
@@ -47,9 +43,18 @@ public class PhoneBook {
         return number;
     }
 
-    //print out all of the entries in PhoneBook
-    //need to work on code for formatting
+    //reverse lookup
+    public String reverseLookup (String number) {
+        String name = "";
+        for(Map.Entry<String, String> entry: treeMap.entrySet()) {
+            if(entry.getValue().equals(number)) {
+                name = entry.getKey();
+            }
+        }
+        return name;
+    }
 
+    //print out all of the entries in PhoneBook
     public String display(){
         StringBuilder printOut = new StringBuilder();
         for(Map.Entry<String, String> entry: treeMap.entrySet()) {
@@ -60,7 +65,6 @@ public class PhoneBook {
                     .append("\n");
         }
         String print = printOut.toString();
-        System.out.println(printOut);
         return print;
     }
 

src/test/java/com/zipcodewilmington/phonebook/PhoneBookTest.java

@@ -44,19 +44,36 @@ public class PhoneBookTest {
 
         //Given
         PhoneBook book = new PhoneBook();
-        book.add("Robert", "302-555-1234");
-        String expectedValue = "Robert " + book.lookup("Robert") + " has been removed";
-        book.add("Robert", expectedValue);
+        book.add("Bob", "302-555-1234");
+        String expectedValue = "";
+
+        //When
+        book.remove("Bob");
+        String actualValue = book.display();
+
+        //Then
+        Assert.assertEquals(expectedValue, actualValue);
+    }
+
+    @Test
+    public void testRemove2() {
+
+        //Given
+        PhoneBook book = new PhoneBook();
+        book.add("Bob", "302-555-1234");
+        String expectedValue = "";
 
         //When
-        String actualNumber = book.remove("Robert");
+        book.remove("Robert");
+        String actualValue = book.display();
 
         //Then
-        Assert.assertEquals(expectedValue, actualNumber );
+        Assert.assertNotEquals(expectedValue, actualValue);
 
 
     }
 
+
     @Test
     public void testLookup(){
 
@@ -64,13 +81,53 @@ public class PhoneBookTest {
         PhoneBook book = new PhoneBook();
         String expectedNumber = "302-555-1234";
         book.add("Bob", expectedNumber);
-
         //When
         String actualNumber = book.lookup("Bob");
+        //Then
+        Assert.assertEquals(expectedNumber, actualNumber);
+    }
 
+    @Test
+    public void testLookup2(){
+
+        //Given
+        PhoneBook book = new PhoneBook();
+        book.add("Bob", "302-555-1234");
+        String expectedNumber = null;
+        //When
+        String actualNumber = book.lookup("Frank");
+        //Then
+        Assert.assertEquals(expectedNumber, actualNumber);
+    }
+
+    @Test
+    public void testLookup3(){
+
+        //Given
+        PhoneBook book = new PhoneBook();
+        book.add("Bob", "302-555-1234");
+        String expectedNumber = "302-555-9988";
+        book.add("Frank", expectedNumber);
+        //When
+        String actualNumber = book.lookup("Frank");
         //Then
         Assert.assertEquals(expectedNumber, actualNumber);
     }
+
+    @Test
+    public void testreverseLookup(){
+
+        //Given
+        PhoneBook book = new PhoneBook();
+        String expectedName = "Bob";
+        book.add(expectedName, "302-555-1234");
+        //When
+        String actualName = book.reverseLookup("302-555-1234");
+        //Then
+        Assert.assertEquals(expectedName, actualName);
+    }
+
+
     @Test
     public void testDisplay() {
         //Given: need help from instructor