test getCompany

1 files changed, 28 insertions, 0 deletions

diff --git a/src/Q3/CreditCardTest.java b/src/Q3/CreditCardTest.java
index 3b8744b..373e79e 100644
--- a/src/Q3/CreditCardTest.java
+++ b/src/Q3/CreditCardTest.java
@@ -6,6 +6,8 @@ import junit.framework.TestCase;
 import junit.framework.TestSuite;
 
 public class CreditCardTest extends TestCase {
+  private CreditCard subject;
+
   public CreditCardTest(String testName) {
     super(testName);
   }
@@ -18,4 +20,30 @@ public class CreditCardTest extends TestCase {
     assertEquals(true, true);
     assertTrue(true);
   }
+
+  // 4 for Visa cards
+  // 5 for Master cards,
+  // 37 for American Express cards,
+  // 6 for Discover cards.
+  public void test_getCompany() {
+    assertEquals(CreateSUT("4111111111111111").getCompany(), "Visa");
+    assertEquals(CreateSUT("5555555555554444").getCompany(), "MasterCard");
+    assertEquals(CreateSUT("378282246310005").getCompany(), "American Express");
+    assertEquals(CreateSUT("6011111111111117").getCompany(), "Discover");
+  }
+
+  // public void test_isValid() {
+  // assertTrue(CreateSUT("49927398716").isValid());
+  // assertFalse(CreateSUT("49927398717").isValid());
+  // assertFalse(CreateSUT("1234567812345678").isValid());
+  // assertTrue(CreateSUT("1234567812345670").isValid());
+  // }
+
+  private CreditCard CreateSUT(String number) {
+    return new CreditCard(number);
+  }
+
+  private CreditCard CreateSUT() {
+    return new CreditCard("1234567812345670");
+  }
 }