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 == []