update

PhoneBook.java

@@ -1,7 +1,22 @@
- 
+import java.util.TreeMap; 
+import java.util.Map;
 
 /**
- * Created by leon on 1/23/18.
+ * Created by Mexi Liang on 06/11/18.
  */
 public class PhoneBook {
+    String name = "";
+    String phoneNumber= "";
+    Map<String,String> phoneBook = new TreeMap<>();    
+    public void addEntry(String name,String phoneNumber){
+        phoneBook.put(name, phoneNumber);
+    }    
+    
+    public void removeEntry(String name){
+        phoneBook.remove(name);
+    }
+    
+    public int size(){
+    return phoneBook.size(); //return size of the phonebook
+    }
 }

PhoneBookTest.java

@@ -1,8 +1,4 @@
-
-
-import static org.junit.Assert.*;
-import org.junit.After;
-import org.junit.Before;
+import org.junit.Assert;
 import org.junit.Test;
 
 /**
@@ -13,30 +9,35 @@ import org.junit.Test;
  */
 public class PhoneBookTest
 {
-    /**
-     * Default constructor for test class PhoneBookTest
-     */
-    public PhoneBookTest()
-    {
+    @Test
+    public void addEntryTest(){
+    //Given
+    PhoneBook phoneBook = new PhoneBook();
+    String name = "Zebra";
+    String phoneNumber = "111-222-333";
+    String expected = "Zebra , 111-222-333";
+    //When
+    phoneBook.addEntry(name,phoneNumber);
+    
+    //Then
+    Assert.assertEquals(1,phoneBook.size());
+    
     }
-
-    /**
-     * Sets up the test fixture.
-     *
-     * Called before every test case method.
-     */
-    @Before
-    public void setUp()
-    {
-    }
-
-    /**
-     * Tears down the test fixture.
-     *
-     * Called after every test case method.
-     */
-    @After
-    public void tearDown()
-    {
+    @Test
+    public void removeEntry(){
+    //Given
+    PhoneBook phoneBook = new PhoneBook();
+    String name = "Zebra";
+    String expected = "";
+    //when
+    phoneBook.removeEntry(name);
+    //Then
+    Assert.assertEquals(0, phoneBook.size());
     }
+    
+    public String lookUp(){
+    return null;
+    } 
+    
+    
 }