From 0fef4e802fc37859fc56806a7e3d3965fe8e0d4c Mon Sep 17 00:00:00 2001
From: Crystal Chung <crystal_chung@intuit.com>
Date: Tue, 29 Oct 2019 16:14:17 -0700
Subject: [PATCH] Adding additional unit tests

---
 .../payment/data/QueryResponseTest.java       | 64 +++++++++++++++++++
 .../com/intuit/payment/data/TokenTest.java    | 61 ++++++++++++++++++
 2 files changed, 125 insertions(+)
 create mode 100644 payments-api/src/test/java/com/intuit/payment/data/QueryResponseTest.java
 create mode 100644 payments-api/src/test/java/com/intuit/payment/data/TokenTest.java

diff --git a/payments-api/src/test/java/com/intuit/payment/data/QueryResponseTest.java b/payments-api/src/test/java/com/intuit/payment/data/QueryResponseTest.java
new file mode 100644
index 00000000..4e2c2bbe
--- /dev/null
+++ b/payments-api/src/test/java/com/intuit/payment/data/QueryResponseTest.java
@@ -0,0 +1,64 @@
+package com.intuit.payment.data;
+import org.apache.commons.lang.builder.ReflectionToStringBuilder;
+import org.testng.Assert;
+import org.testng.annotations.BeforeMethod;
+import org.testng.annotations.BeforeTest;
+import org.testng.annotations.Test;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * @author crystal-chung
+ */
+public class QueryResponseTest {
+
+    private List<BankAccount> bankAccounts;
+    private List<Card> cards;
+    private QueryResponse queryResponse;
+
+    @BeforeTest
+    public void init() {
+        bankAccounts = new ArrayList<>();
+        cards = new ArrayList<>();
+    }
+
+    @BeforeMethod
+    public void setUp() {
+        queryResponse = new QueryResponse.Builder()
+                .bankAccounts(bankAccounts)
+                .cards(cards)
+                .build();
+    }
+
+    @Test
+    public void testAllGetters() {
+        Assert.assertEquals(queryResponse.getBankAccounts(), bankAccounts);
+        Assert.assertEquals(queryResponse.getCards(), cards);
+    }
+
+    @Test
+    public void testAllSetters() {
+        List<BankAccount> newBankAccounts = new ArrayList<>();
+        List<Card> newCards = new ArrayList<>();
+
+        BankAccount newBankAccount = new BankAccount();
+        Card newCard = new Card.Builder().expYear("2020").expMonth("02").build();
+
+        newBankAccounts.add(newBankAccount);
+        newCards.add(newCard);
+
+        queryResponse.setBankAccounts(newBankAccounts);
+        queryResponse.setCards(newCards);
+
+        Assert.assertEquals(queryResponse.getBankAccounts(), newBankAccounts);
+        Assert.assertEquals(queryResponse.getCards(), newCards);
+    }
+
+    @Test
+    public void testToString() {
+        String expectedResult = ReflectionToStringBuilder.toString(queryResponse);
+        String actualResult = queryResponse.toString();
+        Assert.assertTrue(actualResult.contains(expectedResult));
+    }
+}
diff --git a/payments-api/src/test/java/com/intuit/payment/data/TokenTest.java b/payments-api/src/test/java/com/intuit/payment/data/TokenTest.java
new file mode 100644
index 00000000..69899d28
--- /dev/null
+++ b/payments-api/src/test/java/com/intuit/payment/data/TokenTest.java
@@ -0,0 +1,61 @@
+package com.intuit.payment.data;
+import org.apache.commons.lang.builder.ReflectionToStringBuilder;
+import org.testng.Assert;
+import org.testng.annotations.BeforeMethod;
+import org.testng.annotations.BeforeTest;
+import org.testng.annotations.Test;
+
+/**
+ * @author crystal-chung
+ */
+public class TokenTest {
+    private Card card;
+    private BankAccount bankAccount;
+    private String value;
+    private Token token;
+
+    @BeforeTest
+    public void init() {
+        card = new Card();
+        bankAccount = new BankAccount();
+        value = "token value";
+    }
+
+    @BeforeMethod
+    public void setUp() {
+        token = new Token.Builder()
+                .card(card)
+                .bankAccount(bankAccount)
+                .value(value)
+                .build();
+    }
+
+    @Test
+    public void testAllGetters() {
+        Assert.assertEquals(token.getCard(), card);
+        Assert.assertEquals(token.getBankAccount(), bankAccount);
+        Assert.assertEquals(token.getValue(), value);
+    }
+
+    @Test
+    public void testAllSetters() {
+        Card newCard = new Card.Builder().expYear("2020").expMonth("02").build();
+        BankAccount newBankAccount = new BankAccount();
+        String newValue = "new token value";
+
+        token.setCard(newCard);
+        token.setBankAccount(newBankAccount);
+        token.setValue(newValue);
+
+        Assert.assertEquals(token.getCard(), newCard);
+        Assert.assertEquals(token.getBankAccount(), newBankAccount);
+        Assert.assertEquals(token.getValue(), newValue);
+    }
+
+    @Test
+    public void testToString() {
+        String expectedResult = ReflectionToStringBuilder.toString(token);
+        String actualResult = token.toString();
+        Assert.assertTrue(actualResult.contains(expectedResult));
+    }
+}