add middle name to address book.

2 files changed, 16 insertions, 13 deletions

diff --git a/assignments/assignment1/src/main/java/ca/mokhan/assignment1/AddressBook.java b/assignments/assignment1/src/main/java/ca/mokhan/assignment1/AddressBook.java
index 4a71498..a411442 100644
--- a/assignments/assignment1/src/main/java/ca/mokhan/assignment1/AddressBook.java
+++ b/assignments/assignment1/src/main/java/ca/mokhan/assignment1/AddressBook.java
@@ -3,14 +3,10 @@ package ca.mokhan.assignment1;
 public class AddressBook
 {
   private String firstName;
+  private String middleName;
 
-  public void setFirstName(String name)
-  {
-    this.firstName = name;
-  }
-
-  public String getFirstName()
-  {
-    return this.firstName;
-  }
+  public String getFirstName() { return this.firstName; }
+  public String getMiddleName() { return this.middleName; }
+  public void setFirstName(String name) { this.firstName = name; }
+  public void setMiddleName(String name) { this.middleName = name; }
 }
diff --git a/assignments/assignment1/src/test/java/ca/mokhan/assignment1/AddressBookTest.java b/assignments/assignment1/src/test/java/ca/mokhan/assignment1/AddressBookTest.java
index b68eac2..0a479ae 100644
--- a/assignments/assignment1/src/test/java/ca/mokhan/assignment1/AddressBookTest.java
+++ b/assignments/assignment1/src/test/java/ca/mokhan/assignment1/AddressBookTest.java
@@ -6,9 +6,12 @@ import junit.framework.TestSuite;
 
 public class AddressBookTest extends TestCase
 {
+  private AddressBook subject;
+
   public AddressBookTest(String testName)
   {
     super(testName);
+    this.subject = new AddressBook();
   }
 
   public static Test suite()
@@ -18,9 +21,13 @@ public class AddressBookTest extends TestCase
 
   public void testFirstName()
   {
-    AddressBook addressBook = new AddressBook();
-    addressBook.setFirstName("mo");
-    String actual = addressBook.getFirstName();
-    assertEquals(actual, "mo");
+    subject.setFirstName("mo");
+    assertEquals(subject.getFirstName(), "mo");
+  }
+
+  public void testMiddleName()
+  {
+    subject.setMiddleName("tsuyoshi");
+    assertEquals(subject.getMiddleName(), "tsuyoshi");
   }
 }