KkS

src/main/java/com/zipcoder/payment/Check.java

@@ -8,7 +8,12 @@ public class Check implements Payment {
 
 
 
-    public Check(long ID, String tia_mowry, String routingNumber, String accountNumber) {
+    public Check(long ID, String payerName, String routingNumber, String accountNumber) {
+
+        this.ID = ID;
+        this.payerName = payerName;
+        this.accountNumber = getlastfour(accountNumber);
+        this.routingNumber = routingNumber;
 
     }
 
@@ -16,6 +21,10 @@ public class Check implements Payment {
 
     }
 
+    public String getlastfour(String accountNumber){
+        return accountNumber.replace(accountNumber.substring(0,accountNumber.length()-4), "");
+    }
+
     public Long getID() {
         return ID;
     }
@@ -25,7 +34,7 @@ public class Check implements Payment {
     }
 
     public String getShortDescription() {
-        return "Check " + payerName + " " + accountNumber;
+        return "Check " + payerName + " " + "***" + accountNumber;
     }
 
     public String getRoutingNumber() {

src/main/java/com/zipcoder/payment/PaymentPresenter.java

@@ -10,11 +10,17 @@ public class PaymentPresenter {
         for (Payment o: payments) {
             o.getID();
             o.getPayerName();
-
             sb.append(o.getShortDescription()+"\n");
         }
         return sb.toString();
     }
 
 
+    public String toStringByPayerName(Payment[] payments) {
+        return null;
+    }
+
+    public String toStringById(Payment[] payments) {
+        return null;
+    }
 }

src/test/java/com/zipcoder/payment/PaymentPresenterTest.java

@@ -30,4 +30,34 @@ public class PaymentPresenterTest {
         String actual = presenter.toString(payments);
         assertEquals(expected, actual);
     }
+    @Test
+    public void toStringByPayerName(){
+        Payment[] payments = new Payment[2];
+        Payment paypal = new Paypal(4L, "Tamara Mowry", "tamara@mowry.com");
+        Payment check = new Check(81L, "Tia Mowry", "11432543", "134344551")
+
+        payments[0] = paypal;
+        payments[1] = check;
+
+        PaymentPresenter presenter = new PaymentPresenter();
+        String expected = "Paypal Tamara Mowry tamara@mowry.com\nCheck Tia Mowry ***4551\n";
+
+        String actual = presenter.toStringByPayerName(payments);
+        assertEquals(expected, actual);
+    }
+    @Test
+    public void toStringByID(){
+        Payment[] payments = new Payment[2];
+        Payment paypal = new Paypal(120L, "Tamara Mowry", "tamara@mowry.com");
+        Payment check = new Check(81L, "Tia Mowry", "11432543", "134344551")
+
+        payments[0] = paypal;
+        payments[1] = check;
+
+        PaymentPresenter presenter = new PaymentPresenter();
+        String expected = "Paypal Tamara Mowry tamara@mowry.com\nCheck Tia Mowry ***4551\n";
+
+        String actual = presenter.toStringById(payments);
+        assertEquals(expected, actual);
+    }
 }