part2 implemented

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

@@ -10,15 +10,15 @@ import java.util.*;
 
 public class PhoneBook {
 
-    private TreeMap<String, String> treemp;
+    private TreeMap<String, ArrayList<String>> treemp;
 
     public PhoneBook() {
-        treemp = new TreeMap<String,String>() ;
-        addInitialList();
+        treemp = new TreeMap<String, ArrayList<String>>() ;
+        //addInitialList();
     }
 
-    public void setTreemp(String name, String phoneNumber) {
-         this.treemp.put(name, phoneNumber);
+    public void setTreemp(String name, ArrayList phoneNumbers) {
+         this.treemp.put(name, phoneNumbers);
 
     }
 
@@ -31,26 +31,55 @@ public class PhoneBook {
         return false;
     }
 
-    public String add(String name, String phoneNumber)
+    public void add(String name, String phoneNumber)
     {
-        String str = this.treemp.put(name, phoneNumber);
-        return str;
+        boolean keyExistsAlready = hashEntry (name);
+        //Str str = "";
+        //If key already exists then
+        ArrayList<String> alOneTwoMany;
+        if (keyExistsAlready) {
+            // Add new Phonenumber to ArrayList of this Key
+             alOneTwoMany = treemp.get(name);
+            alOneTwoMany.add(phoneNumber);
+            // Add the Arraylist back to the Key
+            treemp.put(name, alOneTwoMany);
+        } else {
+            alOneTwoMany = new ArrayList<String>();
+            alOneTwoMany.add(phoneNumber);
+            treemp.put(name, alOneTwoMany);
+        }
+    }
+
+    public int get (String key) {
+
+        ArrayList al = treemp.get(key);
+        return al.size();
+
     }
 
 
-    public String lmeookup(String name) {
+    /*public String lmeookup(String name) {
 
         String temp = treemp.get(name);
         return temp;
-    }
+    }*/
+
+    public void remove(String key, String number) {
 
-    public String hasRemove(String name) {
+        ArrayList alForRemove = treemp.get(key);
 
-        return name;
+        /* For Loop for iterating ArrayList */
+        for (int counter = 0; counter < alForRemove.size(); counter++) {
+            if (alForRemove.get(counter).toString().equalsIgnoreCase(number)){
+                alForRemove.remove(counter);
+            }
+        }
+        // add the updated Arraylist back to Treemap
+        treemp.put(key, alForRemove);
     }
 
 
-    public void addInitialList () {
+    /* public void addInitialList () {
 
         this.setTreemp("Dog", "222-333-444");
         this.setTreemp("Yashoda","111-222-333");
@@ -83,9 +112,13 @@ public class PhoneBook {
         return sb.toString();
 
     }
+*/
+    public void removeRecord (String key) {
+        treemp.remove(key);
+    }
 
-    public void clearSortedMap () {
-        treemp.clear();
+    public int getNoOfRecords (){
+        return treemp.size();
     }
 
 

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

@@ -4,14 +4,16 @@ package com.zipcodewilmington.phonebook;
 import org.junit.Assert;
 import org.junit.Test;
 
-    /**
+import java.util.ArrayList;
+
+/**
      * Created by leon on 1/23/18.
      */
     public class PhoneBookTest {
 
         private static final int Expected = 0;
 
-        @Test
+       /*** @Test
         public void testHasEntry(){
             //Given
             PhoneBook book = new PhoneBook();
@@ -100,6 +102,95 @@ import org.junit.Test;
             String expected = "Dog 222-444-4444\nZebra 111-222-333\n";
             String actualOutput= book.listNamesAndNumbers();
             Assert.assertEquals(expected, actualOutput);
+        } **/
+
+       @Test
+       //    public void add(String name, String phoneNumber)
+        public void testadd () {
+
+           PhoneBook pb = new PhoneBook();
+           ArrayList <String> al1 = new ArrayList<String>();
+           al1.add("111-111-111");
+           al1.add("222-222-222");
+           al1.add("333-333-333");
+           pb.setTreemp("John", al1);
+           ArrayList <String> al2 = new ArrayList<String>();
+           al2.add("444-444-444");
+           al2.add("222-222-222");
+           al2.add("333-333-333");
+           pb.setTreemp("Mike", al2);
+           ArrayList <String> al3 = new ArrayList<String>();
+           al3.add("111-111-111");
+           al3.add("555-555-555");
+           al3.add("333-333-333");
+           pb.setTreemp("Keenan", al3);
+
+           pb.add("Mike", "666-666-6666");
+
+           int expectedPhoneNumbersCount = 4;
+
+           int actual = pb.get("Mike");
+
+           Assert.assertEquals(expectedPhoneNumbersCount, actual);
+
+       }
+        @Test
+       public void testRemoveRecord () {
+           PhoneBook pb = new PhoneBook();
+           ArrayList <String> al1 = new ArrayList<String>();
+           al1.add("111-111-111");
+           al1.add("222-222-222");
+           al1.add("333-333-333");
+           pb.setTreemp("John", al1);
+           ArrayList <String> al2 = new ArrayList<String>();
+           al2.add("444-444-444");
+           al2.add("222-222-222");
+           al2.add("333-333-333");
+           pb.setTreemp("Mike", al2);
+           ArrayList <String> al3 = new ArrayList<String>();
+           al3.add("111-111-111");
+           al3.add("555-555-555");
+           al3.add("333-333-333");
+           pb.setTreemp("Keenan", al3);
+
+           int before = pb.getNoOfRecords();
+           pb.removeRecord("Mike");
+           int actual = pb.getNoOfRecords();
+           int expectedPhoneNumbersCount = 2;
+           Assert.assertEquals(expectedPhoneNumbersCount, actual);
+
+
+       }
+
+       @Test
+        public void testRemovePhoneNumber ( ) {
+
+            PhoneBook pb = new PhoneBook();
+            ArrayList <String> al1 = new ArrayList<String>();
+            al1.add("111-111-111");
+            al1.add("222-222-222");
+            al1.add("333-333-333");
+            pb.setTreemp("John", al1);
+            ArrayList <String> al2 = new ArrayList<String>();
+            al2.add("444-444-444");
+            al2.add("222-222-222");
+            al2.add("333-333-333");
+            pb.setTreemp("Mike", al2);
+            ArrayList <String> al3 = new ArrayList<String>();
+            al3.add("111-111-111");
+            al3.add("555-555-555");
+            al3.add("333-333-333");
+            pb.setTreemp("Keenan", al3);
+
+            pb.remove("Mike", "222-222-222");
+
+            int expectedPhoneNumbersCount = 2;
+
+            int actual = pb.get("Mike");
+
+            Assert.assertEquals(expectedPhoneNumbersCount, actual);
+
+
         }
 
     }