def test_account_must_be_able_to_return_subaccounts(session):
acc1 = create_account(session)
acc2 = create_account(session)
sacc1 = create_account(session, parent=acc1)
sacc2 = create_account(session, parent=acc1)
ssacc1 = create_account(session, parent=sacc2)
session.commit()
accounts = get_toplevel_accounts(session)
assert acc1 in accounts
assert acc2 in accounts
assert not sacc2 in accounts
accounts = acc1.subaccounts()
assert sacc1 in accounts
assert sacc2 in accounts
assert not acc1 in accounts
accounts = sacc2.subaccounts()
assert accounts == [ssacc1]
accounts = acc2.subaccounts()
assert accounts == []
accounts = get_all_accounts(session)
assert acc1 in accounts
assert acc2 in accounts
assert sacc1 in accounts
assert sacc2 in accounts
assert ssacc1 in accounts
def test_account_plan_creating(session):
create_account(session, 'Бу')
create_account(session, 'Ла')
session.commit()
result = [r.name for r in get_toplevel_accounts(session)]
assert u'Бу' in result
assert u'Ла' in result
assert len(result) == 2
def test_accounts_tree(session):
create_account(session, 'acc1')
create_account(session, 'acc3', create_account(session, 'acc2'))
session.commit()
result = [r.name for r in get_toplevel_accounts(session)]
assert u'acc1' in result
assert u'acc2' in result
assert len(result) == 2
acc = get_account_by_name(session, 'acc2')
result = acc.subaccounts()
assert len(result) == 1
assert result[0].name == 'acc3'
result = result[0].parent_accounts
assert len(result) == 1
assert result[0].name == 'acc2'
result = result[0].parent_accounts
assert result == []
