phoneBook

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

@@ -1,7 +1,68 @@
 package com.zipcodewilmington.phonebook;
 
+import java.util.ArrayList;
+import java.util.TreeMap;
+
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
+
+    TreeMap<String, String> pb = new TreeMap<String, String>();
+
+    public PhoneBook() {
+    }
+
+    public boolean hasEntry(String name) {
+
+        if(pb.containsKey(name))
+        {
+            return true;
+        }
+
+        return false;
+    }
+
+    public void add(String name, String phoneNumber) {
+
+        pb.put(name, phoneNumber);
+
+    }
+
+
+
+    public String lookUp(String name) {
+
+        String number = pb.get(name);
+
+        return number;
+    }
+
+    public boolean remove(String name) {
+
+        pb.remove(name);
+        if(pb.containsKey(name))
+        {
+            return false;
+        }
+
+        return true;
+    }
+
+    public String reverseLookUp(String phoneN) {
+        String hold = "";
+        for(String a: pb.keySet())
+        {
+            if (pb.get(a).contains(phoneN))
+            {
+                hold+=a;
+            }
+        }
+        return hold;
+    }
+
+    public String namesAndNumbers() {
+
+        return String.valueOf(pb);
+    }
 }

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

@@ -1,7 +1,90 @@
 package com.zipcodewilmington.phonebook;
 
+import org.junit.Assert;
+import org.junit.Test;
+
 /**
  * Created by leon on 1/23/18.
  */
+
 public class PhoneBookTest {
+
+    public PhoneBookTest() {
+    }
+
+
+
+    @Test
+    public void testHasEntry()
+    {
+
+        PhoneBook phone = new PhoneBook();
+        String name = "jose";
+        String phoneNumber = "302-903-9837";
+        phone.add(name,phoneNumber);
+
+        boolean actual = phone.hasEntry(name);
+        boolean expected = true;
+
+        Assert.assertTrue(actual);
+    }
+
+
+    @Test
+    public void testLookUp()
+    {
+        PhoneBook phone = new PhoneBook();
+        String name = "jose";
+        String phoneNumber = "302-903-9837";
+        phone.add(name, phoneNumber);
+
+        String actual = phone.lookUp(name);
+        String exoected =  phoneNumber;
+        Assert.assertEquals(exoected,actual);
+    }
+
+    @Test
+    public void removeTest()
+    {
+        PhoneBook phone = new PhoneBook();
+        String name = "jose ";
+        String phoneNum = "587-242-3413";
+        phone.add(name,phoneNum);
+
+        boolean expected = true;
+        boolean actual = phone.remove(name);
+
+        Assert.assertEquals(expected,actual);
+
+    }
+    @Test
+    public void reverseLookUpTest()
+    {
+        PhoneBook phone = new PhoneBook();
+        String name = "jose";
+        String phoneN = "947-282-8262";
+        phone.add(name,phoneN);
+
+        String expected = name;
+        String actual = phone.reverseLookUp(phoneN);
+
+        Assert.assertEquals(expected,actual);
+    }
+
+    @Test
+    public void namesAndNumbersTest()
+    {
+        PhoneBook phone = new PhoneBook();
+        String name = "jose";
+        String phoneNumber = "938-938-9392";
+        phone.add(name,phoneNumber);
+
+        String expected = "{"+name+"="+phoneNumber+"}";
+
+        String actual = phone.namesAndNumbers();
+
+        Assert.assertEquals(expected,actual);
+
+    }
+    
 }