done

PhoneBook.java

@@ -16,6 +16,33 @@ public class PhoneBook {
         }
     }
 
+    public void addNumber(String name, String newNumber){
+        String checkName = lookup(name);
+        if(checkName == null){
+            addEntrant(name, newNumber);
+        } else {
+            List<String> newList = new ArrayList<>(phoneBook.get(name));
+            newList.add(newNumber);
+            phoneBook.put(name, newList);
+        }
+    }
+    
+    public void removeNumber(String name, String numberToDelete){
+        String checkName = lookup(name);
+        if(checkName == null){
+            System.out.println("User does not exist");
+        } else {
+            List<String> newList = new ArrayList<>(phoneBook.get(name));
+            ListIterator<String> pbIterator = newList.listIterator();
+            while(pbIterator.hasNext()){
+                if(pbIterator.next().equals(numberToDelete)){
+                    pbIterator.remove();
+                }
+            }
+            phoneBook.put(name, newList);
+        }
+    }
+
     public void removeEntrant(String name){
         phoneBook.remove(name);   
     }
@@ -33,16 +60,16 @@ public class PhoneBook {
                 numbersList.append(", ");
             }
         }
-
         return numbersList.toString();
     }
 
     public String reverseLookup(String phoneNumber){
         for(Map.Entry<String, List<String>> element: phoneBook.entrySet()){
             String key = element.getKey();
-            List<String> value = element.getValue();
-            if(value.equals(phoneNumber)){
-                return key;
+            for(String number: phoneBook.get(key)){
+                if(number.equals(phoneNumber)){
+                    return key;
+                }
             }
         }
         return "This number does not belong to anyone";

PhoneBookTest.java

@@ -65,10 +65,88 @@ public class PhoneBookTest
     }
     
     @Test
-    public void testReverseLookup(){
+    public void testReverseLookup1(){
+        PhoneBook pb = new PhoneBook();
+        
+        String expected = "Joe";
+        
+        pb.addEntrant("Josh" , "123", "456", "789");
+        pb.addEntrant("John" , "321", "654", "987");
+        pb.addEntrant("Joe" , "111", "222", "333");
         
+        String actual = pb.reverseLookup("222");
         
+        assertEquals(expected, actual);
     }
     
+    @Test
+    public void testReverseLookup2(){
+        PhoneBook pb = new PhoneBook();
+        
+        String expected = "John";
+        
+        pb.addEntrant("Josh" , "123", "456", "789");
+        pb.addEntrant("John" , "321", "654", "987");
+        pb.addEntrant("Joe" , "111", "222", "333");
+        
+        String actual = pb.reverseLookup("654");
+        
+        assertEquals(expected, actual);
+    }
+    
+    @Test
+    public void testAddNumber1(){
+        PhoneBook pb = new PhoneBook();
+        
+        String expected = "123, 456";
+        
+        pb.addEntrant("Josh" , "123");
+        pb.addNumber("Josh","456");
+        
+        String actual = pb.lookup("Josh");
+        assertEquals(expected, actual);
+    }
+    
+    @Test
+    public void testAddNumber2(){
+        PhoneBook pb = new PhoneBook();
+        
+        String expected = "123, 456, 789";
+        
+        pb.addEntrant("Jim" , "123");
+        pb.addNumber("Jim","456");
+        pb.addNumber("Jim","789");
+        
+        String actual = pb.lookup("Jim");
+        assertEquals(expected, actual);
+    }
+    
+    @Test
+    public void testRemoveNumber1(){
+        PhoneBook pb = new PhoneBook();
+        
+        String expected = "123, 456";
+        
+        pb.addEntrant("Jim" , "123", "456", "789");
+        pb.removeNumber("Jim", "789");
+        
+        String actual = pb.lookup("Jim");
+        assertEquals(expected, actual);
+    }
+    
+    @Test
+    public void testRemoveNumber2(){
+        PhoneBook pb = new PhoneBook();
+        
+        String expected = "123, 789";
+        
+        pb.addEntrant("Jim" , "123", "456", "789");
+        pb.removeNumber("Jim", "456");
+        
+        String actual = pb.lookup("Jim");
+        assertEquals(expected, actual);
+    }
+    
+    
     
 }