Merge da4a50859783ca132bdbd18fae3208617e019816 into 74a9673e36c7d25c28315cb54247a91792f8fbc0

pom.xml

@@ -8,5 +8,11 @@
     <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

@@ -1,7 +1,90 @@
 package com.zipcodewilmington.phonebook;
 
+import java.util.*;
+
 /**
  * Created by leon on 1/23/18.
  */
 public class PhoneBook {
+    Map<String, ArrayList<String>> phoneBook;
+
+    public PhoneBook() {
+        this.phoneBook = new TreeMap<String, ArrayList<String>>();
+    }
+
+    protected boolean add(String name, String phone) {
+        if (hasEntry(name)) {
+            getArrayListFor(name).add(phone);
+            return true;
+
+        } else {
+            ArrayList<String> contactNumbers = new ArrayList<String>();
+            contactNumbers.add(phone);
+            phoneBook.put(name, contactNumbers);
+        }
+
+        return true;
+    }
+
+    protected void remove(String name) {
+         phoneBook.remove(name);
+    }
+
+    protected String lookup(String name) {
+        String contactInfo = "";
+        ArrayList<String>phoneNumberLists = getArrayListFor(name);
+        for (String phoneNumbers:phoneNumberLists) {
+            contactInfo+=phoneNumbers + " ";
+        }
+        return contactInfo;
+    }
+
+    protected String listNames() {
+        String names = "";
+        for (String a : phoneBook.keySet()) {
+            names += a + "\n";
+        }
+
+        return names;
+    }
+
+    protected String listNamesAndNumbers() {
+        StringBuilder namesAndNumbers = new StringBuilder();
+        for (String keys : phoneBook.keySet()) {
+            namesAndNumbers.append(keys+"==>");
+            ArrayList<String> contact = getArrayListFor(keys);
+            for (String num : contact) {
+                namesAndNumbers.append(num+",");
+            }
+            namesAndNumbers.append("\n");
+        }
+        return namesAndNumbers.toString();
+    }
+
+    protected String reverseLookUp(String number) {
+        for (String keys : phoneBook.keySet()) {
+            for (int i = 0; i < getArrayListFor(keys).size(); i++) {
+                if (getArrayListFor(keys).get(i).equals(number))
+                    return keys;
+            }
+        }
+        return null;
+    }
+    protected boolean hasEntry(String name){
+        if(phoneBook.containsKey(name)){
+            return true;
+        }
+        return false;
+    }
+    protected ArrayList<String> getArrayListFor(String name){
+        return phoneBook.get(name);
+    }
+    protected void removeIndividualNumbers(String name,String number){
+        getArrayListFor(name).remove(number);
+    }
+
+    protected void removeRecord() {
+
+        phoneBook.clear();
+    }
 }

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

@@ -1,7 +1,121 @@
 package com.zipcodewilmington.phonebook;
 
+import org.junit.After;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+
+
 /**
  * Created by leon on 1/23/18.
  */
+
+
 public class PhoneBookTest {
+
+    @Before
+    public void setUp() {
+
+    }
+
+    @After
+    public void tearDown() {
+    }
+
+    @Test
+    public void testLookup() {
+        PhoneBook phoneBookTest = new PhoneBook();
+        phoneBookTest.add("James", "555555555");
+        phoneBookTest.add("Charles", "2026783422");
+
+        String actual = phoneBookTest.lookup("Charles");
+        Assert.assertEquals("2026783422 ", actual);
+    }
+
+    @Test
+    public void testListNames() {
+        PhoneBook phoneBookTest = new PhoneBook();
+        phoneBookTest.add("John", "1234567890");
+        phoneBookTest.add("James", "555555555");
+        String expected = "James\n" +
+                "John\n";
+        String actual = phoneBookTest.listNames();
+        Assert.assertEquals(expected, actual);
+
+    }
+
+    @Test
+    public void testListNumbersAndNumbers() {
+        PhoneBook phoneBookTest = new PhoneBook();
+        phoneBookTest.add("John", "1234567890");
+        phoneBookTest.add("James", "5555555555");
+        phoneBookTest.add("James", "6666666666");
+        phoneBookTest.add("Mike", "5678901234");
+        String expected = "James==>" + "5555555555,6666666666,\n" +
+                "John==>" + "1234567890,\n" +
+                "Mike==>" + "5678901234,\n";
+        String actual = phoneBookTest.listNamesAndNumbers();
+        Assert.assertEquals(expected, actual);
+    }
+
+    @Test
+    public void testAdd() {
+        PhoneBook phoneBookTest = new PhoneBook();
+        boolean expected = true;
+        boolean actual = phoneBookTest.add("Peter", "7777777777");
+        Assert.assertEquals(expected, actual);
+
+    }
+
+    @Test
+    public void testRemove() {
+        PhoneBook phoneBookTest = new PhoneBook();
+        phoneBookTest.add("John", "1234567890");
+        phoneBookTest.add("Peter", "7777777777");
+        String ListPreRemoving = "John==>1234567890,\n" +
+                "Peter==>7777777777,\n";
+        phoneBookTest.remove("John");
+        String postRemoving = "Peter==>7777777777,\n";
+        String expected = postRemoving;
+        String actual = phoneBookTest.listNamesAndNumbers();
+        Assert.assertEquals(expected, actual);
+    }
+
+    @Test
+    public void testReverseLookup() {
+        PhoneBook phoneBookTest = new PhoneBook();
+        phoneBookTest.add("James", "555555555");
+
+        String actual = phoneBookTest.reverseLookUp("555555555");
+        Assert.assertEquals("James", actual);
+    }
+
+    @Test
+    public void testRemoveIndividualNumbers() {
+        PhoneBook phonebookeTest = new PhoneBook();
+        phonebookeTest.add("James", "2222222222");
+        phonebookeTest.add("James", "3333333333");
+
+        String expected = "James==>3333333333,\n";
+        phonebookeTest.removeIndividualNumbers("James", "2222222222");
+
+        String actual = phonebookeTest.listNamesAndNumbers();
+        Assert.assertEquals(expected, actual);
+    }
+
+    @Test
+    public void testRemoveRecord() {
+        PhoneBook phoneBookTest = new PhoneBook();
+        phoneBookTest.add("Efrem", "1234567890");
+        phoneBookTest.add("Brhane", "1010101010");
+
+        phoneBookTest.removeRecord();
+
+        String expected = "";
+        String actual = phoneBookTest.listNamesAndNumbers();
+
+        Assert.assertEquals(expected, actual);
+    }
+
+
 }