finished lab

PhoneBook.java

@@ -1,7 +1,43 @@
- 
+ import java.util.*;
+
 
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
+    TreeMap<String, String> phoneListName = new TreeMap();
+    TreeMap<String, String> phoneListNumber = new TreeMap();
+
+    public void add(String name, String number){
+        phoneListName.put(name, number);
+        phoneListNumber.put(number, name);
+    }
+ 
+    public void remove(String name){
+        String number = lookUp(name);
+        phoneListName.remove(name);
+        phoneListNumber.remove(number);
+    }
+    
+    public String lookUp(String name){
+        return phoneListName.get(name);
+    }
+    
+    public String reverseLookUp(String number){
+        return phoneListNumber.get(number);
+        
+    }
+    
+    public void display(){
+        Set set = phoneListName.entrySet();
+        Iterator i = set.iterator();
+        
+        while(i.hasNext()){
+            Map.Entry me = (Map.Entry)i.next();
+            System.out.println(me.getKey() + ": " + me.getValue());
+        }
+        
+    }
+    
+    
 }

PhoneBookTest.java

@@ -1,42 +1,37 @@
 
-
 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
 {
-    /**
-     * Default constructor for test class PhoneBookTest
-     */
-    public PhoneBookTest()
-    {
+    @Test
+    public void testAdd(){
+        PhoneBook phonebook = new PhoneBook();
+        String expected = "1234567890"; 
+        phonebook.add("tyler",expected);
+        String actual = phonebook.lookUp("tyler");
+        assertEquals(actual, expected);
+
     }
 
-    /**
-     * Sets up the test fixture.
-     *
-     * Called before every test case method.
-     */
-    @Before
-    public void setUp()
-    {
+    @Test
+    public void testLookUp(){
+        PhoneBook phonebook = new PhoneBook();
+        String expected = "1234567890";
+        phonebook.add("tyler", expected);
+        String actual = phonebook.lookUp("tyler");
+        assertEquals(expected, actual);
     }
 
-    /**
-     * Tears down the test fixture.
-     *
-     * Called after every test case method.
-     */
-    @After
-    public void tearDown()
-    {
+    @Test
+    public void testReverse(){
+        PhoneBook phonebook = new PhoneBook();
+        String expected = "tyler";
+        phonebook.add("tyler", "1234567890");
+        String actual = phonebook.reverseLookUp("1234567890");
+        assertEquals(expected, actual);
     }
+
 }