TreeMap version, basic functionality

PhoneBook.java

@@ -1,7 +1,47 @@
  
-
+import java.util.TreeMap;
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
+    private TreeMap<String, String> phoneBook;
+    private TreeMap<String, String> reversePhoneBook;
+    
+    public PhoneBook() {
+        phoneBook = new TreeMap<String, String>();
+        reversePhoneBook = new TreeMap<String, String>();
+    }
+    
+    public void add(String name, String phoneNumber) {
+        phoneBook.put(name, phoneNumber);
+        reversePhoneBook.put(phoneNumber, name);
+    }
+    
+    public void remove(String name) {
+        String phoneNumber = phoneBook.get(name);
+        phoneBook.remove(name);
+        reversePhoneBook.remove(phoneNumber);
+    }
+    
+    public String lookup(String name) {
+        return phoneBook.get(name) != null ? 
+                                   phoneBook.get(name)
+                                   : 
+                                   "N/A";
+    }
+    
+    public String reverseLookup(String phoneNumber) {
+        return reversePhoneBook.get(phoneNumber) != null ? 
+                                                 reversePhoneBook.get(phoneNumber)
+                                                 :
+                                                 "N/A";
+    }
+    
+    public TreeMap getPhoneBook() {
+        return phoneBook; 
+    }
+    
+    public TreeMap getReversePhoneBook() {
+        return reversePhoneBook;
+    }
 }

PhoneBookTest.java

@@ -5,38 +5,57 @@ 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
 {
-    /**
-     * Default constructor for test class PhoneBookTest
-     */
-    public PhoneBookTest()
+    @Test
+    public void testAdd() 
     {
+       PhoneBook test = new PhoneBook();
+       boolean expected = true;
+       String name = "Jenny";
+       String number = "(302)-867-5309";
+       test.add(name, number);
+       
+       boolean actual = test.getPhoneBook().containsKey(name);
+       assertEquals(expected, actual);  
     }
-
-    /**
-     * Sets up the test fixture.
-     *
-     * Called before every test case method.
-     */
-    @Before
-    public void setUp()
+    // vvvv I hope this is okay since I know the add method works now vvvv
+    @Test
+    public void testAddPhoneBookEntry_And_Lookup()
     {
+       PhoneBook test = new PhoneBook();
+       String name = "Jenny";
+       String expectedNumber = "(302)-867-5309";
+       test.add(name, expectedNumber);
+       
+       String actual = test.lookup(name); 
+       assertEquals(expectedNumber, actual);
     }
-
-    /**
-     * Tears down the test fixture.
-     *
-     * Called after every test case method.
-     */
-    @After
-    public void tearDown()
+    
+    @Test
+    public void testRemovePhoneBookEntry_And_Lookup()
     {
+       PhoneBook test = new PhoneBook();
+       String name = "Jenny";
+       String wrongName = "Tommy Tutone";
+       String number = "(302)-867-5309";
+       test.add(name, number);
+       
+       String actual = test.lookup(wrongName); 
+       assertEquals("N/A", actual);
     }
+    
+    @Test
+    public void testReverseLookup()
+    {
+       PhoneBook test = new PhoneBook();
+       String phoneNumber = "(302)-867-5309";
+       String expectedName = "Jenny";
+       test.add(expectedName, phoneNumber);
+       
+       String actual = test.reverseLookup(phoneNumber);
+       assertEquals(expectedName, actual);
+     
+    }   
 }