phonebook

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

@@ -1,7 +1,87 @@
 package com.zipcodewilmington.phonebook;
 
-/**
- * Created by leon on 1/23/18.
- */
+import java.util.Map;
+
+import java.util.TreeMap;
+
+
+
 public class PhoneBook {
-}
+
+    private TreeMap<String, String> treemap;
+
+    public PhoneBook() {
+        treemap = new TreeMap<String, String>();
+        addInitialList();
+    }
+
+    public void setTreemap(String name, String phoneNumber) {
+        this.treemap.put(name, phoneNumber);
+
+    }
+
+    public boolean hasEntry(String name) {
+
+        if (treemap.containsKey(name)) {
+            return true;
+        } else
+            return false;
+    }
+
+    public String add(String name, String phoneNumber) {
+        String str = this.treemap.put(name, phoneNumber);
+        return str;
+    }
+
+
+    public String lookup(String name) {
+
+        String str = treemap.get(name);
+        return str;
+    }
+
+    public String hasRemove(String name) {
+
+        return name;
+    }
+
+
+    public void addInitialList() {
+
+        this.setTreemap("John", "111-111-1111");
+        this.setTreemap("Kayle", "408-111-2222");
+        this.setTreemap("Tyler", "408-999-3333");
+
+    }
+
+    public String reverseLookup(String phoneNumber) {
+        String key = "";
+        for (Map.Entry<String, String> entry : treemap.entrySet()) {
+            if (entry.getValue().equals(phoneNumber)) {
+                key = entry.getKey();
+            }
+        }
+        return key;
+    }
+
+    public String listNamesAndNumbers() {
+
+        StringBuffer buffer = new StringBuffer();
+
+        for (Map.Entry<String, String> entry : treemap.entrySet()) {
+
+
+            buffer.append(entry.getKey());
+            buffer.append(" ");
+            buffer.append(entry.getValue());
+            buffer.append("\n");
+        }
+        return buffer.toString();
+
+    }
+
+    public void clearSortedMap() {
+        treemap.clear();
+    }
+
+}

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

@@ -1,7 +1,103 @@
 package com.zipcodewilmington.phonebook;
 
-/**
- * Created by leon on 1/23/18.
- */
-public class PhoneBookTest {
+import com.zipcodewilmington.phonebook.PhoneBook;
+import org.junit.Assert;
+import org.junit.Test;
+
+public class PhoneBookTest
+{
+    private static final int Expected = 0;
+
+
+@Test
+ public void testHasEntry()
+{
+        //Given
+        PhoneBook book = new PhoneBook();
+        String name = "John";
+        String phoneNumber = "111-111-1111";
+        book.hasEntry(name);
+        //When
+        boolean actualOutput = book.hasEntry(name);
+        //Then
+        Assert.assertEquals(true,actualOutput);
+        }
+
+@Test
+public void testAdd(){
+        //Given
+        PhoneBook book = new PhoneBook();
+        String name = "kayle";
+        String phoneNumber = "408-111-2222";
+
+        //When
+        //book.add(name, phoneNumber);
+        String actualOutput = book.add(name, phoneNumber);
+        //Then
+        Assert.assertEquals(null, actualOutput);
+        }
+
+@Test
+public void testLookup()
+        {
+        //Given
+        PhoneBook book = new PhoneBook();
+        String name = "Tayler";
+        String phoneNumber = "408-999-3333";
+
+        String actualOutput=book.lookup(name);
+
+        Assert.assertEquals(phoneNumber, actualOutput);
+        }
+
+@Test
+public void testLookup_forMultipleEntries()
+        {
+        //Given
+        PhoneBook book = new PhoneBook();
+        book.add("name1", "111");
+        book.add("name2", "408");
+
+        String name = "John";
+        String phoneNumber = "111-111-1111";
+
+        //book.lookup(name);
+        String actualOutput=book.lookup(name);
+
+        Assert.assertEquals(actualOutput ,phoneNumber );
+        }
+@Test
+public void testRemove()
+        {
+        PhoneBook  book= new PhoneBook();
+        String expectedname="Kayle";
+        String name="Kayle";
+
+
+        String actualOutput=book.hasRemove(name);
+        Assert.assertEquals(actualOutput,name);
+
+        }
+@Test
+public void testReserveLookup()
+        {
+        PhoneBook  book= new PhoneBook();
+        String name="John";
+        String phoneNumber = book.lookup(name);
+        String actualOutput = book.reverseLookup(phoneNumber);
+
+        Assert.assertEquals(name,actualOutput);
+        }
+
+@Test
+public void testListNamesandNumbers()
+        {
+        PhoneBook  book= new PhoneBook();
+        book.clearSortedMap();
+        book.add("Kayle", "408-111-2222");
+        book.add("John", "111-111-1111");
+        String expected = "John 111-111-1111\nKayle 408-111-2222\n";
+        String actualOutput= book.listNamesAndNumbers();
+        Assert.assertEquals(expected, actualOutput);
+        }
 }