/* * Copyright 2014-2015 Groupon, Inc * Copyright 2014-2015 The Billing Project, LLC * * The Billing Project 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 org.killbill.billing.payment.invoice.dao; import java.math.BigDecimal; import java.util.List; import java.util.UUID; import org.joda.time.DateTime; import org.killbill.billing.catalog.api.Currency; import org.killbill.billing.payment.PaymentTestSuiteWithEmbeddedDB; import org.testng.annotations.BeforeClass; import org.testng.annotations.Test; import static org.testng.Assert.assertEquals; public class TestInvoicePaymentControlDao extends PaymentTestSuiteWithEmbeddedDB { private InvoicePaymentControlDao dao; @BeforeClass(groups = "slow") protected void beforeClass() throws Exception { super.beforeClass(); dao = new InvoicePaymentControlDao(dbi); } @Test(groups = "slow") public void testPluginAutoPayOffSimple() { final UUID accountId = UUID.randomUUID(); final UUID attemptId = UUID.randomUUID(); final UUID paymentId = UUID.randomUUID(); final UUID methodId = UUID.randomUUID(); final BigDecimal amount = new BigDecimal("13.33"); final DateTime utcNow = clock.getUTCNow(); final PluginAutoPayOffModelDao entry1 = new PluginAutoPayOffModelDao(attemptId, "key1", "tkey1", accountId, "XXX", paymentId, methodId, amount, Currency.USD, "lulu", utcNow); dao.insertAutoPayOff(entry1); final List<PluginAutoPayOffModelDao> entries = dao.getAutoPayOffEntry(accountId); assertEquals(entries.size(), 1); assertEquals(entries.get(0).getPaymentExternalKey(), "key1"); assertEquals(entries.get(0).getTransactionExternalKey(), "tkey1"); assertEquals(entries.get(0).getAccountId(), accountId); assertEquals(entries.get(0).getPluginName(), "XXX"); assertEquals(entries.get(0).getPaymentId(), paymentId); assertEquals(entries.get(0).getPaymentMethodId(), methodId); assertEquals(entries.get(0).getAmount().compareTo(amount), 0); assertEquals(entries.get(0).getCurrency(), Currency.USD); assertEquals(entries.get(0).getCreatedBy(), "lulu"); assertEquals(entries.get(0).getCreatedDate(), utcNow); } @Test(groups = "slow") public void testPluginAutoPayOffMutlitpleEntries() { final UUID accountId = UUID.randomUUID(); final UUID attemptId = UUID.randomUUID(); final UUID paymentId1 = UUID.randomUUID(); final UUID methodId = UUID.randomUUID(); final BigDecimal amount = new BigDecimal("13.33"); final DateTime utcNow = clock.getUTCNow(); final PluginAutoPayOffModelDao entry1 = new PluginAutoPayOffModelDao(attemptId, "key1", "tkey1", accountId, "XXX", paymentId1, methodId, amount, Currency.USD, "lulu", utcNow); dao.insertAutoPayOff(entry1); final UUID paymentId2 = UUID.randomUUID(); final PluginAutoPayOffModelDao entry2 = new PluginAutoPayOffModelDao(attemptId, "key2", "tkey2", accountId, "XXX", paymentId2, methodId, amount, Currency.USD, "lulu", utcNow); dao.insertAutoPayOff(entry2); final List<PluginAutoPayOffModelDao> entries = dao.getAutoPayOffEntry(accountId); assertEquals(entries.size(), 2); } @Test(groups = "slow") public void testPluginAutoPayOffNoEntries() { final UUID accountId = UUID.randomUUID(); final UUID paymentId1 = UUID.randomUUID(); final UUID attemptId = UUID.randomUUID(); final UUID methodId = UUID.randomUUID(); final BigDecimal amount = new BigDecimal("13.33"); final DateTime utcNow = clock.getUTCNow(); final PluginAutoPayOffModelDao entry1 = new PluginAutoPayOffModelDao(attemptId, "key1", "tkey1", accountId, "XXX", paymentId1, methodId, amount, Currency.USD, "lulu", utcNow); dao.insertAutoPayOff(entry1); final List<PluginAutoPayOffModelDao> entries = dao.getAutoPayOffEntry(UUID.randomUUID()); assertEquals(entries.size(), 0); } }