chore: file edits/ 4 tests still failing

Author: haldanek

lesson_17/bank/bank_app/src/main/java/com/codedifferently/lesson17/bank/BankAtm.java

@@ -12,17 +12,22 @@ public class BankAtm {
   private final Map<UUID, Customer> customerById = new HashMap<>();
   private final Map<String, CheckingAccount> accountByNumber = new HashMap<>();
 
-  public BusinessCheckingAccount createCheckingAccount(
-      String accountNumber, Set<Customer> owners, double initialBalance) {
+  public void addAccount(String accountNumber, Set<Customer> owners, double initialBalance) {
     boolean hasBusinessOwner = owners.stream().anyMatch(Customer::isBusiness);
 
     if (hasBusinessOwner) {
-      return new BusinessCheckingAccount(accountNumber, owners, initialBalance);
+      accountByNumber.put(
+          accountNumber, new BusinessCheckingAccount(accountNumber, owners, initialBalance));
     } else {
-      return new BusinessCheckingAccount(accountNumber, owners, initialBalance);
+      accountByNumber.put(
+          accountNumber, new CheckingAccount(accountNumber, owners, initialBalance));
     }
   }
 
+  public Map<String, CheckingAccount> getAccountByNumber() {
+    return accountByNumber;
+  }
+
   /**
    * Adds a checking account to the bank.
    *

lesson_17/bank/bank_app/src/main/java/com/codedifferently/lesson17/bank/exceptions/IllegalArgumentException.java

@@ -0,0 +1,10 @@
+package com.codedifferently.lesson17.bank.exceptions;
+
+public class IllegalArgumentException extends RuntimeException {
+
+  public IllegalArgumentException() {}
+
+  public IllegalArgumentException(String message) {
+    super(message);
+  }
+}

lesson_17/bank/bank_app/src/test/java/com/codedifferently/lesson17/bank/BankAtmTest.java

@@ -112,20 +112,36 @@ void testWithdrawFunds_AccountNotFound() {
   }
 
   @Test
-  void createCheckingAccount_withNoBusinessOwner_shouldReturnStandardCheckingAccount() {
+  void addAccount_withNoBusinessOwner_shouldReturnStandardCheckingAccount() {
+    String accountNumber = "123456789";
+    double initialBalance = 100.0;
+
     Set<Customer> owners = new HashSet<>();
     owners.add(new Customer(UUID.randomUUID(), "Individual Owner", false));
 
-    CheckingAccount account = bankAtm.createCheckingAccount("123456789", owners, 100.0);
-    assertFalse(account instanceof BusinessCheckingAccount);
+    bankAtm.addAccount(accountNumber, owners, initialBalance);
+    CheckingAccount account = bankAtm.getAccountByNumber().get(accountNumber);
+
+    assertNotNull(account, "Account should be created and added to map.");
+    assertFalse(
+        account instanceof BusinessCheckingAccount,
+        "Account should be a standard CheckingAccount.");
   }
 
   @Test
-  void createCheckingAccount_withBusinessOwner_shouldReturnBusinessCheckingAccount() {
+  void addAccount_withBusinessOwner_shouldReturnBusinessCheckingAccount() {
+    String accountNumber = "987654321";
+    double initialBalance = 200.0;
+
     Set<Customer> owners = new HashSet<>();
     owners.add(new Customer(UUID.randomUUID(), "Business Owner", true));
 
-    BusinessCheckingAccount account = bankAtm.createCheckingAccount("123456789", owners, 100.0);
-    assertTrue(account instanceof BusinessCheckingAccount);
+    bankAtm.addAccount(accountNumber, owners, initialBalance);
+
+    CheckingAccount account = bankAtm.getAccountByNumber().get(accountNumber);
+
+    assertNotNull(account, "Account should be created and added to map.");
+    assertTrue(
+        account instanceof BusinessCheckingAccount, "Account should be a BusinessCheckingAccount.");
   }
 }