all test pass still might need to look over before I submit a PRclear

PhoneBook.java

@@ -7,13 +7,13 @@ public class PhoneBook {
     
     Map<String, ArrayList<String> > entries = new HashMap<String, ArrayList<String>>();
     
-    public void add (String name, String phoneNumber){
+    public void addEntry(String name, String phoneNumber){
         ArrayList<String> pn = new ArrayList<String>();
         pn.add(phoneNumber);
         this.entries.put(name, pn);
     }
     
-    public void addNumber(String name, String phoneNumber){
+    public void addNumberTo(String name, String phoneNumber){
         for(Map.Entry<String, ArrayList<String>> entry: this.entries.entrySet()){
             if(name.equals(entry.getKey())){
                 entry.getValue().add(phoneNumber);
@@ -22,46 +22,63 @@ public class PhoneBook {
         }    
     }
     
-    public void remove(String name, String phoneNumber){
+    public void removePhoneNumber(String name, String phoneNumber){
+        for(Map.Entry<String, ArrayList<String>> entry: this.entries.entrySet()){
+            if(name.equals(entry.getKey())){
+                int indexOf = entry.getValue().indexOf(phoneNumber);
+                if(indexOf >= 0){
+                    entry.getValue().remove(indexOf);
+                    break;
+                }
+            }
+        }        
     }
     
     public void removeRecord(String name){
         this.entries.remove(name);
     }
     
-    
-    
-    public void lookUp(String name){
-        System.out.print(this.entries.get(name));
+    public String lookUp(String name){
+        StringBuilder sb = new StringBuilder();
+        for(Map.Entry<String, ArrayList<String>> entry: this.entries.entrySet()){
+            if(entry.getKey().equals(name)){
+                for(String pn : entry.getValue()){
+                    sb.append(pn);
+                }   
+            }
+        }
+        return sb.toString();        
+        
     }
     
     public String reverseLoopup(String phoneNumber){
-        String name = "No Records...";
+        String name = null;
         for(Map.Entry<String, ArrayList<String>> entry: this.entries.entrySet()){
-            
             for(String pn : entry.getValue()){
                 if(pn.equals(phoneNumber)){
                     name = entry.getKey();
                 }
             }
-            
         }
         return name;
-        
     }
     
     
-    public void display() {
+    public String display() {
         StringBuilder sb = new StringBuilder();
         
         for(Map.Entry<String, ArrayList<String>> entry :this.entries.entrySet()){
             sb.append(entry.getKey());
-            sb.append(" - ");
-            sb.append(entry.getValue());
-            sb.append("\n");
+            sb.append(": ");
+            
+            for(String pn : entry.getValue()){
+                sb.append(pn);
+                //sb.append(" ");
+            }
+
         }
         
-        System.out.print(sb.toString());
+        return sb.toString();
     }
    
     

PhoneBookTest.java

@@ -39,4 +39,63 @@ public class PhoneBookTest
     public void tearDown()
     {
     }
+
+    @Test
+    public void addEntry()
+    {
+        PhoneBook phoneBoo1 = new PhoneBook();
+        phoneBoo1.addEntry("nicholas", "3027387978");
+        assertEquals("nicholas: 3027387978", phoneBoo1.display());
+    }
+
+    @Test
+    public void addTwoNumberForOnePersonTest()
+    {
+        PhoneBook phoneBoo1 = new PhoneBook();
+        phoneBoo1.addEntry("nick", "302");
+        phoneBoo1.addNumberTo("nick", "103");
+        assertEquals("nick: 302103", phoneBoo1.display());
+    }
+
+    @Test
+    public void reverseLoopupTest()
+    {
+        PhoneBook phoneBoo1 = new PhoneBook();
+        phoneBoo1.addEntry("nick", "302");
+        assertEquals("nick", phoneBoo1.reverseLoopup("302"));
+    }
+
+    @Test
+    public void lookupTest()
+    {
+        PhoneBook phoneBoo1 = new PhoneBook();
+        phoneBoo1.addEntry("nick", "302");
+        assertEquals("302", phoneBoo1.lookUp("nick"));
+    }
+
+    @Test
+    public void RemoveRecordTest()
+    {
+        PhoneBook phoneBoo2 = new PhoneBook();
+        phoneBoo2.addEntry("m", "302");
+        phoneBoo2.removeRecord("m");
+        assertEquals("", phoneBoo2.lookUp("m"));
+    }
+
+    @Test
+    public void removePhoneNumberTest()
+    {
+        PhoneBook phoneBoo1 = new PhoneBook();
+        phoneBoo1.addEntry("m", "123");
+        phoneBoo1.addNumberTo("m", "456");
+        phoneBoo1.removePhoneNumber("m", "456");
+        assertEquals("123", phoneBoo1.lookUp("m"));
+    }
 }
+
+
+
+
+
+
+