def test_single_earn_movement(self):
self.assertEqual(0, Movement.objects.count())
data = parse_csv(
cStringIO.StringIO(LLOYDS_SIMPLE_EXAMPLE_EARN_ROW),
parser=LloydsParser,
)
import_movements(data, self.bank_account)
self.assertEqual(1, Movement.objects.count())
movement = Movement.objects.all()[0]
self.assertEqual(data[0]["description"], movement.description)
self.assertEqual(data[0]["amount"], movement.amount)
self.assertEqual(data[0]["date"], movement.date)
self.assertEqual(self.bank_account, movement.bank_account)
def save(self, *args, **kwargs):
if self.is_valid():
parser = ENTITY_TO_PARSER[BankAccount.objects.get(
pk=self.cleaned_data['bank_account']).entity]
data = parse_csv(
self.cleaned_data["estatement"],
parser=parser,
header_lines=1,
reverse_order=True,
)
bank_account = BankAccount.objects.get(
pk=self.cleaned_data["bank_account"])
import_movements(data, bank_account)
def test_avoid_duplicates(self):
self.assertEqual(0, Movement.objects.count())
data = parse_csv(
cStringIO.StringIO(LLOYDS_SIMPLE_EXAMPLE_PAY_ROW),
parser=LloydsParser,
)
imported, rejected = import_movements(data, self.bank_account)
self.assertEqual(1, Movement.objects.count())
self.assertEqual(1, imported)
self.assertEqual(0, len(rejected))
imported, rejected = import_movements(data, self.bank_account)
self.assertEqual(1, Movement.objects.count())
self.assertEqual(0, imported)
self.assertEqual(1, len(rejected))
self.assertEqual(data[0]["description"], rejected[0]["description"])
self.assertEqual(data[0]["amount"], rejected[0]["amount"])
self.assertEqual(data[0]["date"], rejected[0]["date"])
def test_multi_movement(self):
self.assertEqual(0, Movement.objects.count())
data = parse_csv(
cStringIO.StringIO(
LLOYDS_SIMPLE_EXAMPLE_PAY_ROW + LLOYDS_SIMPLE_EXAMPLE_EARN_ROW),
parser=LloydsParser,
)
import_movements(data, self.bank_account)
self.assertEqual(2, Movement.objects.count())
movements = Movement.objects.all()
self.assertEqual(data[0]["description"], movements[0].description)
self.assertEqual(data[0]["amount"], movements[0].amount)
self.assertEqual(data[0]["date"], movements[0].date)
self.assertEqual(self.bank_account, movements[0].bank_account)
self.assertEqual(data[1]["description"], movements[1].description)
self.assertEqual(data[1]["amount"], movements[1].amount)
self.assertEqual(data[1]["date"], movements[1].date)
self.assertEqual(self.bank_account, movements[1].bank_account)
def test_single_earn_row(self):
data = parse_csv(
cStringIO.StringIO(LLOYDS_SIMPLE_EXAMPLE_EARN_ROW),
parser=LloydsParser,
)
self.assertEqual(1, len(data))
row = data[0]
self.assertEqual(6, len(row.keys()))
self.assertTrue('date' in row.keys())
self.assertTrue('account_number' in row.keys())
self.assertTrue('description' in row.keys())
self.assertTrue('amount' in row.keys())
self.assertTrue('balance' in row.keys())
self.assertTrue('category' in row.keys())
self.assertEqual(date(2012, 1, 6), row['date'])
self.assertEqual("00-00-00 0000000", row['account_number'])
self.assertEqual("Tickets", row['description'])
self.assertEqual(134.3, row['amount'])
self.assertEqual(200, row['balance'])
def test_single_pay_row(self):
data = parse_csv(
cStringIO.StringIO(LLOYDS_SIMPLE_EXAMPLE_PAY_ROW),
parser=LloydsParser,
)
self.assertEqual(1, len(data))
row = data[0]
self.assertEqual(6, len(row.keys()))
self.assertTrue('date' in row.keys())
self.assertTrue('account_number' in row.keys())
self.assertTrue('description' in row.keys())
self.assertTrue('amount' in row.keys())
self.assertTrue('balance' in row.keys())
self.assertTrue('category' in row.keys())
self.assertEqual(date(2011, 12, 25), row['date'])
self.assertEqual("00-00-00 0000000", row['account_number'])
self.assertEqual("Christmas presents", row['description'])
self.assertEqual(-134.3, row['amount'])
self.assertEqual(200, row['balance'])