From 835042309e8d0a8499546dc8a32965a2b85757ac Mon Sep 17 00:00:00 2001 From: jdungan Date: Sat, 13 Aug 2016 07:36:12 -0500 Subject: [PATCH] add fixed values test, flake8 fix --- payments/management/commands/charge.py | 4 ++-- payments/tests/utils_test.py | 24 ++++++++++++++++++++++-- 2 files changed, 24 insertions(+), 4 deletions(-) diff --git a/payments/management/commands/charge.py b/payments/management/commands/charge.py index b47c309f..733e10ae 100644 --- a/payments/management/commands/charge.py +++ b/payments/management/commands/charge.py @@ -1,11 +1,11 @@ from django.core.management.base import BaseCommand from django.conf import settings +from payments import utils + import stripe stripe.api_key = settings.STRIPE_SECRET_KEY -from payments import utils - class Command(BaseCommand): diff --git a/payments/tests/utils_test.py b/payments/tests/utils_test.py index fdf868c3..cb4ea147 100644 --- a/payments/tests/utils_test.py +++ b/payments/tests/utils_test.py @@ -1,3 +1,5 @@ +from decimal import Decimal + from django.test import TestCase from .. import utils @@ -6,8 +8,7 @@ class PaymentAmountTest(TestCase): def test_calculate_amounts(self): - amounts = [20, 1, 2, 3, 5, 7, 9, 200, 333, 999] - # amounts = range(1, 1000) + amounts = range(1, 1000) for amount in amounts: print "---" offer_values = utils.transaction_amounts(amount) @@ -42,3 +43,22 @@ def test_calculate_amounts(self): - offer_values['application_fee'] ) ) + + def test_fixed_amounts(self): + values = utils.transaction_amounts(10) + self.assertEqual(values['total_stripe_fee'], Decimal('0.61')) + self.assertEqual(values['application_fee'], Decimal('1.11')) + self.assertEqual(values['codesy_fee'], Decimal('0.25')) + self.assertEqual(values['charge_amount'], Decimal('10.55')) + self.assertEqual(values['offer_stripe_fee'], Decimal('0.30')) + self.assertEqual(values['payout_amount'], Decimal('9.44')) + self.assertEqual(values['payout_stripe_fee'], Decimal('0.31')) + + values = utils.transaction_amounts(50) + self.assertEqual(values['total_stripe_fee'], Decimal('1.81')) + self.assertEqual(values['application_fee'], Decimal('4.31')) + self.assertEqual(values['codesy_fee'], Decimal('1.25')) + self.assertEqual(values['charge_amount'], Decimal('52.16')) + self.assertEqual(values['offer_stripe_fee'], Decimal('0.91')) + self.assertEqual(values['payout_amount'], Decimal('47.85')) + self.assertEqual(values['payout_stripe_fee'], Decimal('0.90'))