not completed yet

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>4.12</version>
+        </dependency>
+    </dependencies>
 
 
 </project>

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

@@ -8,6 +8,7 @@ import java.util.Map;
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
+
     //instance variable
     private TreeMap<String, String> treeMap;
 
@@ -22,25 +23,50 @@ public class PhoneBook {
     }
 
     //add a name & number entry
-    public void addNameNumberEntry(String name, String number) {
+    public void add(String name, String number) {
         treeMap.put(name, number);
+        String confirm = "";
+        if(treeMap.put(name, number)!= null) {
+            confirm = name + " and " + number + " are now in the database";
+        } else {
+            confirm = "Error: " + name + " is already in the database";
+        }
     }
 
     //remove a name & number entry
-    public void removeEntry(String name) {
+    public String remove(String name) {
         treeMap.remove(name);
+        String response = "";
+        if(treeMap.remove(name) != null) {
+            response = "Name and number removed";
+        }
+        if(treeMap.remove(name)== null) {
+            response = "Invalid entry";
+        }
+        return response;
     }
 
     //find phone number lookup by name
-    public String lookupByName(String name) {
-         return treeMap.get(name);
+    public String lookup(String name) {
+         String lookup = treeMap.get(name);
+         String notFound = name + " is not in the directory";
+         if(lookup != null) {
+             return lookup;
+         }
+         else {
+             return notFound;
+         }
     }
 
     //print out all of the entries in PhoneBook
-    public Set<Map.Entry<String,String>> printOutPhoneBook() {
-        return treeMap.entrySet();
+    //need to work on code for formatting
+    public String display() {
+        System.out.println(treeMap.entrySet().toString());
+        String printOut = treeMap.entrySet().toString();
+        return printOut;
     }
 
 
 
 }
+

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

@@ -1,8 +1,54 @@
 package com.zipcodewilmington.phonebook;
 
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.util.TreeMap;
+
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBookTest {
 
+    public void testDefaultConstructor() {
+
+        PhoneBook book = new PhoneBook();
+
+
+
+    }
+
+    public void testConstructorWithArgument() {
+
+    }
+
+    public void testAdd() {
+        PhoneBook book = new PhoneBook();
+        book.add("Bob", "302-555-1234");
+        Assert.assertNotNull(book);
+    }
+
+    public void testRemove() {
+
+    }
+
+    public void testLookup(){
+
+        //Given
+        String expectedNumber = "302-555-1234";
+        PhoneBook book = new PhoneBook();
+        book.add("Bob", "302-555-1234");
+
+        //When
+        book.lookup("Bob");
+
+        //Then
+        Assert.assertEquals(expectedNumber, "302-555-1234");
+    }
+
+    public void testDisplay() {
+
+    }
+
+
 }