killbill-aplcache

Added payment dao tests

2/7/2012 4:39:55 AM

Details

diff --git a/payment/src/test/java/com/ning/billing/payment/dao/TestPaymentDao.java b/payment/src/test/java/com/ning/billing/payment/dao/TestPaymentDao.java
new file mode 100644
index 0000000..6c57c77
--- /dev/null
+++ b/payment/src/test/java/com/ning/billing/payment/dao/TestPaymentDao.java
@@ -0,0 +1,71 @@
+/*
+ * Copyright 2010-2011 Ning, Inc.
+ *
+ * Ning licenses this file to you under the Apache License, version 2.0
+ * (the "License"); you may not use this file except in compliance with the
+ * License.  You may obtain a copy of the License at:
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
+ * License for the specific language governing permissions and limitations
+ * under the License.
+ */
+
+package com.ning.billing.payment.dao;
+
+import java.math.BigDecimal;
+import java.util.UUID;
+
+import org.joda.time.DateTime;
+import org.joda.time.DateTimeZone;
+import org.testng.annotations.Test;
+
+import com.ning.billing.payment.api.PaymentInfo;
+
+public abstract class TestPaymentDao {
+
+    protected PaymentDao dao;
+
+    @Test
+    public void testCreatePayment() {
+        PaymentInfo paymentInfo = new PaymentInfo.Builder().setPaymentId(UUID.randomUUID().toString())
+                                                           .setAmount(BigDecimal.TEN)
+                                                           .setStatus("Processed")
+                                                           .setBankIdentificationNumber("1234")
+                                                           .setPaymentNumber("12345")
+                                                           .setPaymentMethodId("12345")
+                                                           .setReferenceId("12345")
+                                                           .setType("Electronic")
+                                                           .setCreatedDate(new DateTime(DateTimeZone.UTC))
+                                                           .setUpdatedDate(new DateTime(DateTimeZone.UTC))
+                                                           .setEffectiveDate(new DateTime(DateTimeZone.UTC))
+                                                           .build();
+
+        dao.savePaymentInfo(paymentInfo);
+    }
+
+    @Test
+    public void testUpdatePayment() {
+        PaymentInfo paymentInfo = new PaymentInfo.Builder().setPaymentId(UUID.randomUUID().toString())
+                                                           .setAmount(BigDecimal.TEN)
+                                                           .setStatus("Processed")
+                                                           .setBankIdentificationNumber("1234")
+                                                           .setPaymentNumber("12345")
+                                                           .setPaymentMethodId("12345")
+                                                           .setReferenceId("12345")
+                                                           .setType("Electronic")
+                                                           .setCreatedDate(new DateTime(DateTimeZone.UTC))
+                                                           .setUpdatedDate(new DateTime(DateTimeZone.UTC))
+                                                           .setEffectiveDate(new DateTime(DateTimeZone.UTC))
+                                                           .build();
+
+        dao.savePaymentInfo(paymentInfo);
+
+        dao.updatePaymentInfo("CreditCard", paymentInfo.getPaymentId(), "Visa", "US");
+
+    }
+
+}
diff --git a/payment/src/test/java/com/ning/billing/payment/dao/TestPaymentDaoWithEmbeddedDb.java b/payment/src/test/java/com/ning/billing/payment/dao/TestPaymentDaoWithEmbeddedDb.java
new file mode 100644
index 0000000..da48c03
--- /dev/null
+++ b/payment/src/test/java/com/ning/billing/payment/dao/TestPaymentDaoWithEmbeddedDb.java
@@ -0,0 +1,53 @@
+/*
+ * Copyright 2010-2011 Ning, Inc.
+ *
+ * Ning licenses this file to you under the Apache License, version 2.0
+ * (the "License"); you may not use this file except in compliance with the
+ * License.  You may obtain a copy of the License at:
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
+ * License for the specific language governing permissions and limitations
+ * under the License.
+ */
+
+package com.ning.billing.payment.dao;
+
+import java.io.IOException;
+
+import org.apache.commons.io.IOUtils;
+import org.testng.annotations.AfterClass;
+import org.testng.annotations.BeforeClass;
+import org.testng.annotations.BeforeMethod;
+import org.testng.annotations.Test;
+
+import com.ning.billing.dbi.MysqlTestingHelper;
+
+public class TestPaymentDaoWithEmbeddedDb
+{
+    @Test(enabled = true, groups = { "slow", "database" })
+    public class TestPaymentDaoWithEmbeddedDB extends TestPaymentDao {
+        private final MysqlTestingHelper helper = new MysqlTestingHelper();
+
+        @BeforeClass(alwaysRun = true)
+        public void startMysql() throws IOException {
+            final String paymentddl = IOUtils.toString(MysqlTestingHelper.class.getResourceAsStream("/com/ning/billing/payment/ddl.sql"));
+
+            helper.startMysql();
+            helper.initDb(paymentddl);
+        }
+
+        @AfterClass(alwaysRun = true)
+        public void stopMysql() {
+            helper.stopMysql();
+        }
+
+        @BeforeMethod(alwaysRun = true)
+        public void setUp() throws IOException {
+            dao = new DefaultPaymentDao(helper.getDBI());
+        }
+    }
+}
diff --git a/payment/src/test/java/com/ning/billing/payment/dao/TestPaymentDaoWithMock.java b/payment/src/test/java/com/ning/billing/payment/dao/TestPaymentDaoWithMock.java
new file mode 100644
index 0000000..f5af240
--- /dev/null
+++ b/payment/src/test/java/com/ning/billing/payment/dao/TestPaymentDaoWithMock.java
@@ -0,0 +1,30 @@
+/*
+ * Copyright 2010-2011 Ning, Inc.
+ *
+ * Ning licenses this file to you under the Apache License, version 2.0
+ * (the "License"); you may not use this file except in compliance with the
+ * License.  You may obtain a copy of the License at:
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
+ * License for the specific language governing permissions and limitations
+ * under the License.
+ */
+
+package com.ning.billing.payment.dao;
+
+import java.io.IOException;
+
+import org.testng.annotations.BeforeMethod;
+import org.testng.annotations.Test;
+
+@Test(groups = { "fast" })
+public class TestPaymentDaoWithMock extends TestPaymentDao {
+    @BeforeMethod(alwaysRun = true)
+    public void setUp() throws IOException {
+        dao = new MockPaymentDao();
+    }
+}
\ No newline at end of file