Refine execute_find_isCaseSensitive()

LinusMelb · Sep 8, 2016 · 3c31f3e · 3c31f3e
1 parent f2edf7f
commit 3c31f3e
Showing 1 changed file with 9 additions and 18 deletions.
diff --git a/test/java/seedu/addressbook/logic/LogicTest.java b/test/java/seedu/addressbook/logic/LogicTest.java
@@ -457,25 +457,16 @@ public void execute_find_isCaseSensitive() throws Exception {
         Person p1 = helper.generatePersonWithName("key key");
         Person p2 = helper.generatePersonWithName("KEy sduauo");
 
-        AddressBook expectedAB = new AddressBook();
-        expectedAB.addPerson(p1);
-        expectedAB.addPerson(pTarget1);
-        expectedAB.addPerson(p2);
-        expectedAB.addPerson(pTarget2);
-
-        List<Person> expectedList = new ArrayList<>();
-        expectedList.addAll(Arrays.asList(pTarget1, pTarget2));
-
-        addressBook.addPerson(p1);
-        addressBook.addPerson(pTarget1);
-        addressBook.addPerson(p2);
-        addressBook.addPerson(pTarget2);
-        CommandResult r = logic.execute("find KEY");
+        List<Person> fourPersons = helper.generatePersonList(p1, pTarget1, p2, pTarget2);
+        AddressBook expectedAB = helper.generateAddressBook(fourPersons);
+        List<Person> expectedList = helper.generatePersonList(pTarget1, pTarget2);
+        helper.addToAddressBook(addressBook, fourPersons);
 
-        assertEquals(Command.getMessageForPersonListShownSummary(expectedList), r.feedbackToUser);
-        assertTrue(r.getRelevantPersons().isPresent());
-        assertEquals(expectedList, r.getRelevantPersons().get());
-        assertLogicObjectStateEquals(expectedAB, expectedList);
+        assertNonMutatingCommandBehavior("find KEY",
+                Command.getMessageForPersonListShownSummary(expectedList),
+                expectedAB,
+                true,
+                expectedList);
     }
 
     @Test