/*
* 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);
}
}