line: add fields 'Fee' + 'Dividend' - todos
This commit is contained in:
Frederik Jaeckel
parent
438035cca2
commit
982f21f06e
12 changed files with 569 additions and 0 deletions
|
|
@ -6,6 +6,7 @@ import unittest
|
|||
|
||||
from trytond.modules.cashbook_investment.tests.test_book import CbInvTestCase
|
||||
from trytond.modules.cashbook_investment.tests.test_reconciliation import ReconTestCase
|
||||
from trytond.modules.cashbook_investment.tests.test_yield import YieldTestCase
|
||||
|
||||
|
||||
__all__ = ['suite']
|
||||
|
|
@ -14,6 +15,7 @@ __all__ = ['suite']
|
|||
class CashbookInvestmentTestCase(\
|
||||
CbInvTestCase,\
|
||||
ReconTestCase,\
|
||||
YieldTestCase,\
|
||||
):
|
||||
'Test cashbook-investment module'
|
||||
module = 'cashbook_investment'
|
||||
|
|
|
|||
215
tests/test_yield.py
Normal file
215
tests/test_yield.py
Normal file
|
|
@ -0,0 +1,215 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
# This file is part of the cashbook-module from m-ds for Tryton.
|
||||
# The COPYRIGHT file at the top level of this repository contains the
|
||||
# full copyright notices and license terms.
|
||||
|
||||
from decimal import Decimal
|
||||
from datetime import date
|
||||
from trytond.pool import Pool
|
||||
from trytond.transaction import Transaction
|
||||
from trytond.tests.test_tryton import ModuleTestCase, with_transaction
|
||||
|
||||
|
||||
class YieldTestCase(ModuleTestCase):
|
||||
'Test yield calculation module'
|
||||
module = 'cashbook_investment'
|
||||
|
||||
def prep_yield_config(self, fee, dividend, gainloss, company):
|
||||
""" add config for yield-calculation
|
||||
fee: name of fee-category,
|
||||
dividend: name of fee-category,
|
||||
gainloss: name of cashbook for gain/loss booking
|
||||
"""
|
||||
pool = Pool()
|
||||
Category = pool.get('cashbook.category')
|
||||
Cashbook = pool.get('cashbook.book')
|
||||
AssetConf = pool.get('cashbook.assetconf')
|
||||
|
||||
fee_cat = Category.search([('name', '=', fee)])
|
||||
if len(fee_cat) > 0:
|
||||
fee_cat = fee_cat[0]
|
||||
else :
|
||||
fee_cat, = Category.create([{
|
||||
'name': fee,
|
||||
'company': company.id,
|
||||
'cattype': 'out',
|
||||
}])
|
||||
|
||||
dividend_cat = Category.search([('name', '=', dividend)])
|
||||
if len(dividend_cat) > 0:
|
||||
dividend_cat = dividend_cat[0]
|
||||
else :
|
||||
dividend_cat, = Category.create([{
|
||||
'name': dividend,
|
||||
'company': company.id,
|
||||
'cattype': 'in',
|
||||
}])
|
||||
|
||||
gainloss_book = Cashbook.search([('name', '=', gainloss)])
|
||||
if len(gainloss_book) > 0:
|
||||
gainloss_book = gainloss_book[0]
|
||||
else :
|
||||
types = self.prep_type()
|
||||
gainloss_book, = Cashbook.create([{
|
||||
'name': gainloss,
|
||||
'btype': types.id,
|
||||
'company': company.id,
|
||||
'currency': company.currency.id,
|
||||
'number_sequ': self.prep_sequence().id,
|
||||
}])
|
||||
|
||||
as_cfg = None
|
||||
with Transaction().set_context({
|
||||
'company': company.id,
|
||||
}):
|
||||
as_cfg, = AssetConf.create([{
|
||||
'fee_category': fee_cat.id,
|
||||
'dividend_category': dividend_cat.id,
|
||||
'gainloss_book': gainloss_book.id,
|
||||
}])
|
||||
self.assertEqual(as_cfg.fee_category.rec_name, 'Fee')
|
||||
self.assertEqual(as_cfg.fee_category.cattype, 'out')
|
||||
|
||||
self.assertEqual(as_cfg.dividend_category.rec_name, 'Dividend')
|
||||
self.assertEqual(as_cfg.dividend_category.cattype, 'in')
|
||||
|
||||
self.assertEqual(as_cfg.gainloss_book.rec_name,
|
||||
'GainLoss | 0.00 usd | Open')
|
||||
return as_cfg
|
||||
|
||||
@with_transaction()
|
||||
def test_yield_config(self):
|
||||
""" check config
|
||||
"""
|
||||
pool = Pool()
|
||||
AssetConf = pool.get('cashbook.assetconf')
|
||||
company = self.prep_company()
|
||||
|
||||
as_cfg = self.prep_yield_config('Fee', 'Dividend', 'GainLoss', company)
|
||||
self.assertEqual(as_cfg.fee_category.rec_name, 'Fee')
|
||||
self.assertEqual(as_cfg.dividend_category.rec_name, 'Dividend')
|
||||
self.assertEqual(as_cfg.gainloss_book.rec_name, 'GainLoss | 0.00 usd | Open')
|
||||
|
||||
@with_transaction()
|
||||
def test_yield_fee_dividend_gainloss(self):
|
||||
""" add cashbook, categories, bookings for
|
||||
fees, dididend...
|
||||
"""
|
||||
pool = Pool()
|
||||
Cashbook = pool.get('cashbook.book')
|
||||
BType = pool.get('cashbook.type')
|
||||
Category = pool.get('cashbook.category')
|
||||
Line = pool.get('cashbook.line')
|
||||
|
||||
company = self.prep_company()
|
||||
as_cfg = self.prep_yield_config('Fee', 'Dividend', 'GainLoss', company)
|
||||
|
||||
type_depot = self.prep_type('Depot', 'D')
|
||||
type_cash = self.prep_type('Cash', 'C')
|
||||
BType.write(*[
|
||||
[type_depot],
|
||||
{
|
||||
'feature': 'asset',
|
||||
}])
|
||||
|
||||
asset = self.prep_asset_item(
|
||||
company=company,
|
||||
product = self.prep_asset_product(name='Product 1'))
|
||||
self.assertEqual(asset.symbol, 'usd/u')
|
||||
|
||||
category_office, = Category.create([{
|
||||
'name': 'Office',
|
||||
'company': company.id,
|
||||
}])
|
||||
|
||||
book_asset, = Cashbook.create([{
|
||||
'name': 'Depot',
|
||||
'btype': type_depot.id,
|
||||
'company': company.id,
|
||||
'currency': company.currency.id,
|
||||
'number_sequ': self.prep_sequence().id,
|
||||
'asset': asset.id,
|
||||
'quantity_uom': asset.uom.id,
|
||||
'start_date': date(2022, 5, 1),
|
||||
'lines': [('create', [{
|
||||
'bookingtype': 'out',
|
||||
'date': date(2022, 5, 1),
|
||||
'amount': Decimal('2.0'),
|
||||
'quantity': Decimal('0.0'),
|
||||
'category': as_cfg.fee_category.id,
|
||||
'description': 'Fee',
|
||||
}, {
|
||||
'bookingtype': 'in',
|
||||
'date': date(2022, 5, 1),
|
||||
'amount': Decimal('10.0'),
|
||||
'quantity': Decimal('1.0'),
|
||||
'category': as_cfg.dividend_category.id,
|
||||
'description': 'reinvested dividend',
|
||||
}])],
|
||||
}])
|
||||
|
||||
# dividend, incoming + link to depot-account
|
||||
book_cash, = Cashbook.create([{
|
||||
'name': 'Cash',
|
||||
'btype': type_cash.id,
|
||||
'company': company.id,
|
||||
'currency': company.currency.id,
|
||||
'number_sequ': self.prep_sequence().id,
|
||||
'start_date': date(2022, 5, 1),
|
||||
'lines': [('create', [{
|
||||
'bookingtype': 'spin',
|
||||
'date': date(2022, 5, 1),
|
||||
'description': 'Dividend',
|
||||
'splitlines': [('create', [{
|
||||
'description': 'Dividend',
|
||||
'splittype': 'cat',
|
||||
'category': as_cfg.dividend_category.id,
|
||||
'amount': Decimal('5.0'),
|
||||
}, {
|
||||
'description': 'Dividend',
|
||||
'splittype': 'tr',
|
||||
'booktransf': book_asset.id,
|
||||
'amount': Decimal('0.0'),
|
||||
'quantity': Decimal('0.0'),
|
||||
}])],
|
||||
}])],
|
||||
}])
|
||||
self.assertEqual(len(book_cash.lines), 1)
|
||||
Line.wfcheck([book_cash.lines[0]])
|
||||
|
||||
self.assertEqual(book_cash.lines[0].rec_name,
|
||||
'05/01/2022|Rev/Sp|5.00 usd|Dividend [-]')
|
||||
self.assertEqual(book_cash.lines[0].reference, None)
|
||||
self.assertEqual(len(book_cash.lines[0].references), 1)
|
||||
self.assertEqual(book_cash.lines[0].references[0].rec_name,
|
||||
'05/01/2022|to|0.00 usd|Dividend [Cash | 5.00 usd | Open]|0.0000 u')
|
||||
|
||||
self.assertEqual(book_asset.name, 'Depot')
|
||||
self.assertEqual(book_asset.rec_name, 'Depot | 8.00 usd | Open | 1.0000 u')
|
||||
self.assertEqual(book_asset.btype.rec_name, 'D - Depot')
|
||||
self.assertEqual(book_asset.state, 'open')
|
||||
self.assertEqual(book_asset.feature, 'asset')
|
||||
self.assertEqual(len(book_asset.lines), 3)
|
||||
|
||||
# reference to dividend (1)
|
||||
self.assertEqual(book_asset.lines[0].rec_name,
|
||||
'05/01/2022|to|0.00 usd|Dividend [Cash | 5.00 usd | Open]|0.0000 u')
|
||||
self.assertEqual(book_asset.lines[0].trade_fee, Decimal('0.0'))
|
||||
self.assertEqual(book_asset.lines[0].asset_dividend, Decimal('5.0'))
|
||||
self.assertEqual(book_asset.lines[0].asset_gainloss, Decimal('0.0'))
|
||||
|
||||
# fee payed from asset-account
|
||||
self.assertEqual(book_asset.lines[1].rec_name,
|
||||
'05/01/2022|Exp|-2.00 usd|Fee [Fee]|0.0000 u')
|
||||
self.assertEqual(book_asset.lines[1].trade_fee, Decimal('-2.0'))
|
||||
self.assertEqual(book_asset.lines[1].asset_dividend, Decimal('0.0'))
|
||||
self.assertEqual(book_asset.lines[1].asset_gainloss, Decimal('0.0'))
|
||||
|
||||
# dividend (2) added to asset-account
|
||||
self.assertEqual(book_asset.lines[2].rec_name,
|
||||
'05/01/2022|Rev|10.00 usd|reinvested dividend [Dividend]|1.0000 u')
|
||||
self.assertEqual(book_asset.lines[2].trade_fee, Decimal('0.0'))
|
||||
self.assertEqual(book_asset.lines[2].asset_dividend, Decimal('10.0'))
|
||||
self.assertEqual(book_asset.lines[2].asset_gainloss, Decimal('0.0'))
|
||||
|
||||
# end YieldTestCase
|
||||
Loading…
Add table
Add a link
Reference in a new issue