Fix testcases for deleting customer

AY2324S2-CS2103T-T08-2 · Apr 13, 2024 · d42dff3 · d42dff3
1 parent 59620de
commit d42dff3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/test/java/seedu/address/logic/commands/DeleteCustomerCommandTest.java b/src/test/java/seedu/address/logic/commands/DeleteCustomerCommandTest.java
@@ -8,7 +8,7 @@
 import static seedu.address.logic.commands.CommandTestUtil.showPersonAtIndex;
 import static seedu.address.testutil.TypicalIndexes.INDEX_FIRST_PERSON;
 import static seedu.address.testutil.TypicalIndexes.INDEX_SECOND_PERSON;
-import static seedu.address.testutil.TypicalPersons.getTypicalAddressBook;
+import static seedu.address.testutil.TypicalPersons.getTypicalAddressBookWithPersonsOnly;
 
 import org.junit.jupiter.api.Test;
 
@@ -25,7 +25,7 @@
  */
 public class DeleteCustomerCommandTest {
 
-    private Model model = new ModelManager(getTypicalAddressBook(), new UserPrefs());
+    private Model model = new ModelManager(getTypicalAddressBookWithPersonsOnly(), new UserPrefs());
 
     @Test
     public void execute_validIndexUnfilteredList_success() {

diff --git a/src/test/java/seedu/address/testutil/TypicalPersons.java b/src/test/java/seedu/address/testutil/TypicalPersons.java
@@ -87,6 +87,17 @@ public static AddressBook getTypicalAddressBook() {
         return ab;
     }
 
+    public static AddressBook getTypicalAddressBookWithPersonsOnly() {
+        AddressBook ab = new AddressBook();
+        ArrayList<Person> typicalPersons = new ArrayList<>();
+        for (Person person : getTypicalPersons()) {
+            Person copy = new PersonBuilder(person).build();
+            ab.addPerson(copy);
+            typicalPersons.add(copy);
+        }
+        return ab;
+    }
+
     public static List<Person> getTypicalPersons() {
         return new ArrayList<>(Arrays.asList(ALICE, BENSON, CARL, DANIEL, ELLE, FIONA, GEORGE));
     }