Next commit

PhoneBook.java

@@ -11,12 +11,13 @@ public class PhoneBook{
     List<String> multipleNumbers = new LinkedList<>();
  //   public static boolean electronicPhonebook = true;
  //   Scanner input = new Scanner(System.in);
- //   Collection<String> c;
- //   Iterator<String> iter = c.iterator();
+ // HashMap<List, String> c;
+ // Iterator<String> iter = c.iterator();
     public PhoneBook(){
     }
-    public void add(List multipleNumbers, String name){
-        phonebook.put(multipleNumbers, name);
+    
+    public void add(List <String> number, String name){
+        phonebook.put(number, name);
     }
     
     public void remove(String name){
@@ -31,6 +32,10 @@ public class PhoneBook{
         return phonebook.get(number);
     }
     
+  //  public void removeRecord(){
+  //      iter.forEachRemaining(Map<List, String>, iter.remove(List));
+  //  }
+    
     public void display(){
         System.out.println("                         ***** WLECOME TO THE WORLD BOOK *****           ");
         for (Map.Entry<List,String> entry : phonebook.entrySet()){

PhoneBookTest.java

@@ -6,6 +6,8 @@ import org.junit.Before;
 import org.junit.Test;
 public class PhoneBookTest{
     Map<String, String> testPhonebook = new TreeMap<>();
+    Collection<String> c;
+    Iterator<String> iter = c.iterator();
     public PhoneBookTest(){
     }
     
@@ -42,12 +44,22 @@ public class PhoneBookTest{
     @Test
     public void reverseLookupTest(){
         testPhonebook.put("9393939", "Hirself");
-        testPhonebook.get("9393939");
         
-        String actual = testPhonebook.get("Hirself");
-        String expected = "9393939";
+        String actual = testPhonebook.get("9393939");
+        String expected = "Hirself";
         assertEquals(actual, expected);
     }
     
+    @Test
+    public void removeRecordTest(){
+        testPhonebook.put("9393939", "Hirself");
+        testPhonebook.put("8484848", "Moped");
+        
+        for(TreeMap<String, String> entry : c){
+            entry.remove();
+        }
+        String actual = null;
+    }
+    
 
 }