Browse Source

Work in Progress

Kibret Tecle 6 years ago
parent
commit
0cffcded13

+ 7
- 2
src/main/java/com/zipcodewilmington/phonebook/PhoneBook.java View File

24
 
24
 
25
     }
25
     }
26
 
26
 
27
-    public void add(String name,String phoneNumber) {
27
+    public boolean add(String name,String phoneNumber) {
28
 
28
 
29
      phoneBook.put(name, phoneNumber);
29
      phoneBook.put(name, phoneNumber);
30
+     if(phoneBook.get(name).equals(phoneNumber)){
31
+         return true;
32
+     }return false;
30
     }
33
     }
31
 
34
 
32
-    public void remove(String name){
35
+    public boolean remove(String name){
33
         if(phoneBook.containsKey(name)){
36
         if(phoneBook.containsKey(name)){
34
             phoneBook.remove(name);
37
             phoneBook.remove(name);
35
         }
38
         }
36
         else{
39
         else{
37
             System.out.println(name+" does not exist in the phone book list");
40
             System.out.println(name+" does not exist in the phone book list");
38
         }
41
         }
42
+        return !phoneBook.containsKey(name);
43
+
39
 
44
 
40
     }
45
     }
41
 
46
 

+ 16
- 2
src/test/java/com/zipcodewilmington/phonebook/PhoneBookTest.java View File

48
     }
48
     }
49
     @Test
49
     @Test
50
     public void testListNumbersAndNumbers(){
50
     public void testListNumbersAndNumbers(){
51
-        String expected = "James    "+ "1234554321\n"+
51
+        String expected = "James   "+ "1234554321\n"+
52
                 "John   "+"1234567890\n"+
52
                 "John   "+"1234567890\n"+
53
-                "Micheal    "+"5678901234";
53
+                "Micheal   "+"5678901234";
54
         String actual =phoneBookRecord.listNamesAndNumbers();
54
         String actual =phoneBookRecord.listNamesAndNumbers();
55
         Assert.assertEquals(expected, actual);
55
         Assert.assertEquals(expected, actual);
56
     }
56
     }
57
 
57
 
58
+    @Test
59
+    public void testAdd(){
60
+        boolean expected = true;
61
+        boolean actual = phoneBookRecord.add("Peter","7777777777");
62
+        Assert.assertEquals(expected, actual);
63
+
64
+    }
65
+    @Test
66
+    public void testRemove(){
67
+        boolean expected =true;
68
+        boolean actual = phoneBookRecord.remove("John");
69
+        Assert.assertEquals(expected, actual);
70
+    }
71
+
58
 
72
 
59
 }
73
 }