Work in Progress

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

@@ -24,18 +24,23 @@ public class PhoneBook {
 
     }
 
-    public void add(String name,String phoneNumber) {
+    public boolean add(String name,String phoneNumber) {
 
      phoneBook.put(name, phoneNumber);
+     if(phoneBook.get(name).equals(phoneNumber)){
+         return true;
+     }return false;
     }
 
-    public void remove(String name){
+    public boolean remove(String name){
         if(phoneBook.containsKey(name)){
             phoneBook.remove(name);
         }
         else{
             System.out.println(name+" does not exist in the phone book list");
         }
+        return !phoneBook.containsKey(name);
+
 
     }
 

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

@@ -48,12 +48,26 @@ public class PhoneBookTest {
     }
     @Test
     public void testListNumbersAndNumbers(){
-        String expected = "James    "+ "1234554321\n"+
+        String expected = "James   "+ "1234554321\n"+
                 "John   "+"1234567890\n"+
-                "Micheal    "+"5678901234";
+                "Micheal   "+"5678901234";
         String actual =phoneBookRecord.listNamesAndNumbers();
         Assert.assertEquals(expected, actual);
     }
 
+    @Test
+    public void testAdd(){
+        boolean expected = true;
+        boolean actual = phoneBookRecord.add("Peter","7777777777");
+        Assert.assertEquals(expected, actual);
+
+    }
+    @Test
+    public void testRemove(){
+        boolean expected =true;
+        boolean actual = phoneBookRecord.remove("John");
+        Assert.assertEquals(expected, actual);
+    }
+
 
 }