def test_transaction_days(db):
plan = AccountsPlan()
acc1 = plan.add_account()
acc2 = plan.add_account()
plan.create_transaction(acc1, acc2, 200.0, datetime(2010, 5, 20)).save()
plan.create_transaction(acc1, acc2, 300.0, datetime(2010, 5, 31)).save()
plan.create_transaction(acc1, acc2, 100.0, datetime(2010, 6, 01)).save()
result = bank.get_month_transaction_days(acc1, 2010, 5)
assert result == [20, 31]
result = bank.get_month_transaction_days(acc2, 2010, 6)
assert result == [1]
result = bank.get_month_transaction_days(acc2, 2010, 7)
assert result == []
def test_who_choice(db):
plan = AccountsPlan()
acc1 = plan.add_account()
acc2 = plan.add_account()
t = plan.create_transaction(acc1, acc2, 200.0)
t.who = u'Бичиков'
t.save()
t = plan.create_transaction(acc1, acc2, 300.0)
t.who = u'Зубков'
t.save()
result = bank.get_who_choice()
assert result == [u'Бичиков', u'Зубков']
t = plan.create_transaction(acc1, acc2, 300.0)
t.who = u'Зубков'
t.save()
result = bank.get_who_choice()
assert result == [u'Бичиков', u'Зубков']
def test_what_choice(db):
plan = AccountsPlan()
acc1 = plan.add_account()
acc2 = plan.add_account()
t = plan.create_transaction(acc1, acc2, 200.0)
t.what = u'за хлеб'
t.save()
t = plan.create_transaction(acc1, acc2, 300.0)
t.what = u'за воду'
t.save()
result = bank.get_what_choice()
assert result == [u'за воду', u'за хлеб']
t = plan.create_transaction(acc1, acc2, 300.0)
t.what = u'за воду'
t.save()
result = bank.get_what_choice()
assert result == [u'за воду', u'за хлеб']
if header:
header = False
continue
date = datetime.datetime.strptime(r[0], "%m/%d/%y %H:%M:%S")
if date.year < 2010:
continue
inout = int(r[1])
what = r[2].decode('utf-8')
how = float(r[3]) if r[3] else 0.0
num = int(r[4])
kuda = r[5]
who = fam_hash[int(r[6])] if r[6] else u'Неизвестный'
if not kuda:
print 'Error account', r
continue
acc = get_or_create_account(kuda)
if inout == 1:
tran = plan.create_transaction(acc, bank, how, date)
else:
tran = plan.create_transaction(bank, acc, how, date)
tran.what = what
tran.who = who
tran.num = num
tran.save()
