close to being finished?

pom.xml

@@ -7,6 +7,18 @@
     <groupId>com.zipcodewilmington</groupId>
     <artifactId>phonebok</artifactId>
     <version>1.0-SNAPSHOT</version>
+    <dependencies>
+        <dependency>
+            <groupId>junit</groupId>
+            <artifactId>junit</artifactId>
+            <version>RELEASE</version>
+        </dependency>
+        <dependency>
+            <groupId>junit</groupId>
+            <artifactId>junit</artifactId>
+            <version>RELEASE</version>
+        </dependency>
+    </dependencies>
 
 
 </project>

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

@@ -1,7 +1,8 @@
 package com.zipcodewilmington.phonebook;
 import java.util.Map;
 import java.util.Set;
-import java.util.HashMap;
+import java.util.SortedMap;
+import java.util.TreeMap;
 
 /**
  * Created by leon on 1/23/18.
@@ -15,7 +16,7 @@ public class PhoneBook {
     private Map<String, String> contactsList;
 
     public PhoneBook(){
-        this.contactsList = new HashMap<String, String>();
+        this.contactsList = new TreeMap<String, String>();
         contactsList.put("Pete Jones", "2678849087");
         contactsList.put("Ron Burgundy", "5604329932");
         contactsList.put("Nala Bits", "2345578122");
@@ -23,6 +24,7 @@ public class PhoneBook {
     }
     public Map<String, String> getContactsList(){
         return contactsList;
+
     }
 
     public void addContact(String name, String number) {
@@ -47,17 +49,21 @@ public class PhoneBook {
         return contactsList.get(name) == null ? "Sorry name not found" : contactsList.get(name);
     }
 
-    public void listAllByName(String name){
-        for(String ContactsByName : contactsList.keySet()){
-            System.out.println(contactsList.get(name));
+    public String listAllByName(String strings) {
+        Set nameSet = contactsList.keySet();
+        for (Object names : nameSet) {
+            return names.toString();
         }
 
-    }
-    public void listAllNamesAndNumbers(){
-        Set<Map.Entry<String, String>>listingAll = contactsList.entrySet();
-        System.out.println(contactsList);
-    }
 
-}
+    public String listAllNamesAndNumbers(){
+            Set listingAll = contactsList.entrySet();
+            for (Object allEntries : listingAll )
+            return allEntries.toString();
 
+        }
+    }
 }
+
+
+

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

@@ -1,5 +1,12 @@
 package com.zipcodewilmington.phonebook;
 
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+
+
+
+
 /**
  * Created by leon on 1/23/18.
  */
@@ -17,14 +24,14 @@ public class PhoneBookTest {
     public void testAddContact(){
         phoneBook.addContact("Genya", "6578356767");
         boolean genyaIsInPhoneBook = phoneBook.getContactsList().keySet().contains("Genya");
-        assertTrue(genyaIsInPhoneBook);
+        Assert.assertTrue(genyaIsInPhoneBook);
 
     }
     @Test
     public void testRemoveContact(){
         phoneBook.removeContact("Pete Jones");
         boolean peteJonesHasBeenRemoved = !phoneBook.getContactsList().containsKey("Pete Jones");
-        assertTrue(peteJonesHasBeenRemoved);
+        Assert.assertTrue(peteJonesHasBeenRemoved);
     }
     @Test
     public void testLookUpContactByName(){
@@ -35,9 +42,18 @@ public class PhoneBookTest {
     }
     @Test
     public void testListAllByName(){
-
+        phoneBook.listAllByName(phoneBook.getContactsList().keySet().toString());
+        String expected = ;
+        String actual = phoneBook.getContactsList().keySet().toString();
+        Assert.assertEquals(expected, actual);
     }
 
+    @Test
+    public void testListAllNamesAndNumbers(){
+        phoneBook.listAllNamesAndNumber();
+        String expected = ;
+        String actual = phoneBook.getContactsList().entrySet();
+        Assert.assertEquals(expected, actual);
+    }
 }
 
-}