Phonebook tests complete

pom.xml

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

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

@@ -0,0 +1,18 @@
+package com.zipcodewilmington.phonebook;
+
+public class Main {
+
+    public static void main(String[] args) {
+
+        PhoneBook tester = new PhoneBook();
+        tester.add("Keith", "555-666-7777");
+        tester.add("John", "555-612-7777");
+        tester.add("Mike", "555-623-7777");
+        tester.add("Steve", "555-645-7777");
+       String s = tester.listNames();
+        System.out.println(s);
+
+        s = tester.listNamesAndNumbers();
+        System.out.println(s);
+    }
+}

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

@@ -1,7 +1,58 @@
 package com.zipcodewilmington.phonebook;
 
+import java.util.TreeMap;
+import java.util.Set;
+
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
+
+        String name;
+        String phonenum;
+        TreeMap<String, String> contacts = new TreeMap<String, String>();
+
+
+
+public PhoneBook() {
+}
+
+ public void add(String name, String phonenum){
+       contacts.put(name, phonenum);
+ }
+
+ public void deleter(String name){
+        contacts.remove(name);
+ }
+
+ public String lookup(String name){
+        return contacts.get(name);
+
+ }
+
+ public String listNames(){
+    StringBuilder namesnumbers = new StringBuilder();
+        Set<String> keys = contacts.keySet();
+        for(String key: keys)
+
+        {
+
+            namesnumbers.append(key+"\n");
+        }
+        return namesnumbers.toString();
+ }
+
+    public String listNamesAndNumbers(){
+        StringBuilder namesnumbers = new StringBuilder();
+        Set<String> keys = contacts.keySet();
+        for(String key: keys)
+
+        {
+
+            namesnumbers.append(key+" "+contacts.get(key)+"\n");
+        }
+        return namesnumbers.toString();
+    }
+
+
 }

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

@@ -1,7 +1,81 @@
 package com.zipcodewilmington.phonebook;
 
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.util.TreeMap;
 /**
  * Created by leon on 1/23/18.
  */
+
 public class PhoneBookTest {
+
+    PhoneBook testbook;
+    @Before
+    public void setUp(){
+        testbook = new PhoneBook();
+    }
+
+    @Test
+    public void testPhoneBookAddName() {
+        // Given
+        PhoneBook testbook = new PhoneBook();
+       testbook.add("Keith", "555-666-7777");
+
+      String testNumber = testbook.lookup("Keith");
+      Assert.assertTrue(testNumber.equals("555-666-7777"));
+
+    }
+@Test
+    public void testPhoneBookDelete(){
+        PhoneBook testbook = new PhoneBook();
+        testbook.add("Keith", "555-666-7777");
+        testbook.deleter("Keith");
+        String testnumber = testbook.lookup("Keith");
+        Assert.assertEquals(null, testnumber);
+    }
+@Test
+    public void testPhoneBookLookUp(){
+        PhoneBook testbook = new PhoneBook();
+        testbook.add("Keith", "555-666-7777");
+        String testnumber = testbook.lookup("Keith");
+        Assert.assertEquals("555-666-7777", testnumber);
+    }
+
+    @Test
+    public void testPhoneBookList(){
+
+        PhoneBook testbook = new PhoneBook();
+        testbook.add("Keith", "555-666-7777");
+        testbook.add("John", "555-612-7777");
+        testbook.add("Mike", "555-623-7777");
+        testbook.add("Steve", "555-645-7777");
+
+        String expected = "John 555-612-7777\n" +
+                "Keith 555-666-7777\n" +
+                "Mike 555-623-7777\n" +
+                "Steve 555-645-7777\n";
+        String actual = testbook.listNamesAndNumbers();
+        Assert.assertEquals(expected, actual);
+    }
+
+    @Test
+    public void testPhoneBookNames(){
+
+        PhoneBook testbook = new PhoneBook();
+        testbook.add("Keith", "555-666-7777");
+        testbook.add("John", "555-612-7777");
+        testbook.add("Mike", "555-623-7777");
+        testbook.add("Steve", "555-645-7777");
+
+        String expected = "John\n" +
+                "Keith\n" +
+                "Mike\n" +
+                "Steve\n";
+        String actual = testbook.listNames();
+        Assert.assertEquals(expected, actual);
+    }
+
+
 }