add update of cashbooks on update of asset-rates
This commit is contained in:
Frederik Jaeckel
parent
8c2fb8ed7c
commit
8ec7026259
3 changed files with 401 additions and 4 deletions
38
asset.py
38
asset.py
|
|
@ -3,7 +3,7 @@
|
|||
# The COPYRIGHT file at the top level of this repository contains the
|
||||
# full copyright notices and license terms.
|
||||
|
||||
from trytond.pool import PoolMeta
|
||||
from trytond.pool import PoolMeta, Pool
|
||||
|
||||
|
||||
class AssetRate(metaclass=PoolMeta):
|
||||
|
|
@ -13,22 +13,52 @@ class AssetRate(metaclass=PoolMeta):
|
|||
def create(cls, vlist):
|
||||
""" update cache-value
|
||||
"""
|
||||
pool = Pool()
|
||||
Cashbook = pool.get('cashbook.book')
|
||||
ValueStore = pool.get('cashbook.values')
|
||||
|
||||
records = super(AssetRate, cls).create(vlist)
|
||||
# TODO: update cashbooks using this asset
|
||||
|
||||
ValueStore.update_books(
|
||||
ValueStore.get_book_by_books(
|
||||
Cashbook.search([
|
||||
('asset', 'in', [
|
||||
x.asset.id for x in records])])))
|
||||
return records
|
||||
|
||||
@classmethod
|
||||
def write(cls, *args):
|
||||
""" update cache-value
|
||||
"""
|
||||
pool = Pool()
|
||||
Cashbook = pool.get('cashbook.book')
|
||||
ValueStore = pool.get('cashbook.values')
|
||||
|
||||
actions = iter(args)
|
||||
all_rates = []
|
||||
for rates, values in zip(actions, actions):
|
||||
all_rates.extend(rates)
|
||||
|
||||
super(AssetRate, cls).write(*args)
|
||||
# TODO: update cashbooks using this asset
|
||||
|
||||
ValueStore.update_books(
|
||||
ValueStore.get_book_by_books(
|
||||
Cashbook.search([
|
||||
('asset', 'in', [
|
||||
x.asset.id for x in all_rates])])))
|
||||
|
||||
@classmethod
|
||||
def delete(cls, records):
|
||||
""" set cache to None
|
||||
"""
|
||||
pool = Pool()
|
||||
Cashbook = pool.get('cashbook.book')
|
||||
ValueStore = pool.get('cashbook.values')
|
||||
|
||||
books = ValueStore.get_book_by_books(Cashbook.search([
|
||||
('asset', 'in', [x.asset.id for x in records])]))
|
||||
|
||||
super(AssetRate, cls).delete(records)
|
||||
# TODO: update cashbooks using this asset
|
||||
ValueStore.update_books(books)
|
||||
|
||||
# end
|
||||
|
|
|
|||
|
|
@ -10,9 +10,11 @@ from trytond.modules.investment.tests.test_module import InvestmentTestCase
|
|||
from .yieldtest import YieldTestCase
|
||||
from .book import CbInvTestCase
|
||||
from .reconciliation import ReconTestCase
|
||||
from .valuestore import ValueStoreTestCase
|
||||
|
||||
|
||||
class CashbookInvestmentTestCase(
|
||||
ValueStoreTestCase,
|
||||
CashbookTestCase,
|
||||
InvestmentTestCase,
|
||||
CbInvTestCase,
|
||||
|
|
|
|||
365
tests/valuestore.py
Normal file
365
tests/valuestore.py
Normal file
|
|
@ -0,0 +1,365 @@
|
|||
# -*- 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 trytond.tests.test_tryton import with_transaction
|
||||
from trytond.pool import Pool
|
||||
from trytond.transaction import Transaction
|
||||
from datetime import date
|
||||
from decimal import Decimal
|
||||
|
||||
|
||||
class ValueStoreTestCase(object):
|
||||
""" test update of cashbooks on update of asset
|
||||
"""
|
||||
@with_transaction()
|
||||
def test_valstore_update_asset_rate(self):
|
||||
""" update rate of asset, should update cashbook
|
||||
"""
|
||||
pool = Pool()
|
||||
Book = pool.get('cashbook.book')
|
||||
BType = pool.get('cashbook.type')
|
||||
Asset = pool.get('investment.asset')
|
||||
Queue = pool.get('ir.queue')
|
||||
ValStore = pool.get('cashbook.values')
|
||||
|
||||
company = self.prep_company()
|
||||
with Transaction().set_context({'company': company.id}):
|
||||
types = self.prep_type()
|
||||
BType.write(*[
|
||||
[types],
|
||||
{
|
||||
'feature': 'asset',
|
||||
}])
|
||||
category = self.prep_category(cattype='in')
|
||||
|
||||
party = self.prep_party()
|
||||
asset = self.prep_asset_item(
|
||||
company=company,
|
||||
product=self.prep_asset_product(name='Product 1'))
|
||||
|
||||
Asset.write(*[
|
||||
[asset],
|
||||
{
|
||||
'rates': [('create', [{
|
||||
'date': date(2022, 5, 1),
|
||||
'rate': Decimal('2.5'),
|
||||
}, {
|
||||
'date': date(2022, 5, 2),
|
||||
'rate': Decimal('2.8'),
|
||||
}])],
|
||||
}])
|
||||
self.assertEqual(
|
||||
asset.rec_name, 'Product 1 | 2.8000 usd/u | 05/02/2022')
|
||||
|
||||
(usd, euro) = self.prep_2nd_currency(company)
|
||||
self.assertEqual(company.currency.rec_name, 'Euro')
|
||||
self.assertEqual(asset.symbol, 'usd/u')
|
||||
|
||||
self.assertEqual(Queue.search_count([]), 0)
|
||||
|
||||
book, = Book.create([{
|
||||
'start_date': date(2022, 4, 1),
|
||||
'name': 'Book 1',
|
||||
'btype': types.id,
|
||||
'company': company.id,
|
||||
'currency': euro.id,
|
||||
'number_sequ': self.prep_sequence().id,
|
||||
'asset': asset.id,
|
||||
'quantity_uom': asset.uom.id,
|
||||
'quantity_digits': 3,
|
||||
'lines': [('create', [{
|
||||
'date': date(2022, 5, 1),
|
||||
'description': 'Text 1',
|
||||
'category': category.id,
|
||||
'bookingtype': 'in',
|
||||
'amount': Decimal('2.5'),
|
||||
'party': party.id,
|
||||
'quantity': Decimal('1.5'),
|
||||
}])],
|
||||
}])
|
||||
|
||||
# run worker
|
||||
self.assertEqual(
|
||||
ValStore.search_count([]),
|
||||
len(Book.valuestore_fields()))
|
||||
self.prep_valstore_run_worker()
|
||||
|
||||
self.assertEqual(book.name, 'Book 1')
|
||||
self.assertEqual(book.rec_name, 'Book 1 | 2.50 € | Open | 1.500 u')
|
||||
self.assertEqual(len(book.lines), 1)
|
||||
self.assertEqual(
|
||||
book.lines[0].rec_name,
|
||||
'05/01/2022|Rev|2.50 €|Text 1 [Cat1]|1.500 u')
|
||||
|
||||
values = ValStore.search([
|
||||
('cashbook', '=', book.id)], order=[('field_name', 'ASC')])
|
||||
self.assertEqual(len(values), len(Book.valuestore_fields()))
|
||||
|
||||
self.assertEqual(
|
||||
values[0].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance|2.50|2')
|
||||
self.assertEqual(
|
||||
values[1].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance_all|2.50|2')
|
||||
self.assertEqual(
|
||||
values[2].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance_ref|2.50|2')
|
||||
self.assertEqual(
|
||||
values[3].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_rate|2.67|2')
|
||||
self.assertEqual(
|
||||
values[4].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_value|4.00|2')
|
||||
self.assertEqual(
|
||||
values[5].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_value_ref|4.00|2')
|
||||
self.assertEqual(
|
||||
values[6].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|diff_amount|1.50|2')
|
||||
self.assertEqual(
|
||||
values[7].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|diff_percent|60.00|2')
|
||||
self.assertEqual(
|
||||
values[8].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|purchase_amount|2.50|2')
|
||||
self.assertEqual(
|
||||
values[9].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|quantity|1.500|3')
|
||||
self.assertEqual(
|
||||
values[10].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|quantity_all|1.500|3')
|
||||
self.assertEqual(
|
||||
values[11].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_dividend_12m|0.00|2')
|
||||
self.assertEqual(
|
||||
values[12].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_dividend_total|0.00|2')
|
||||
self.assertEqual(
|
||||
values[13].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_fee_12m|0.00|2')
|
||||
self.assertEqual(
|
||||
values[14].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_fee_total|0.00|2')
|
||||
self.assertEqual(
|
||||
values[15].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_sales|2.50|2')
|
||||
self.assertEqual(
|
||||
values[16].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_sales_12m|0.00|2')
|
||||
|
||||
# add rate
|
||||
Asset.write(*[
|
||||
[asset],
|
||||
{
|
||||
'rates': [('create', [{
|
||||
'date': date(2022, 5, 10),
|
||||
'rate': Decimal('3.0'),
|
||||
}])],
|
||||
}])
|
||||
self.assertEqual(
|
||||
asset.rec_name, 'Product 1 | 3.0000 usd/u | 05/10/2022')
|
||||
|
||||
# run worker
|
||||
self.prep_valstore_run_worker()
|
||||
|
||||
values = ValStore.search([
|
||||
('cashbook', '=', book.id)], order=[('field_name', 'ASC')])
|
||||
self.assertEqual(len(values), len(Book.valuestore_fields()))
|
||||
|
||||
self.assertEqual(
|
||||
values[0].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance|2.50|2')
|
||||
self.assertEqual(
|
||||
values[1].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance_all|2.50|2')
|
||||
self.assertEqual(
|
||||
values[2].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance_ref|2.50|2')
|
||||
self.assertEqual(
|
||||
values[3].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_rate|2.86|2')
|
||||
self.assertEqual(
|
||||
values[4].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_value|4.29|2')
|
||||
self.assertEqual(
|
||||
values[5].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_value_ref|4.29|2')
|
||||
self.assertEqual(
|
||||
values[6].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|diff_amount|1.79|2')
|
||||
self.assertEqual(
|
||||
values[7].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|diff_percent|71.60|2')
|
||||
self.assertEqual(
|
||||
values[8].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|purchase_amount|2.50|2')
|
||||
self.assertEqual(
|
||||
values[9].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|quantity|1.500|3')
|
||||
self.assertEqual(
|
||||
values[10].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|quantity_all|1.500|3')
|
||||
self.assertEqual(
|
||||
values[11].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_dividend_12m|0.00|2')
|
||||
self.assertEqual(
|
||||
values[12].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_dividend_total|0.00|2')
|
||||
self.assertEqual(
|
||||
values[13].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_fee_12m|0.00|2')
|
||||
self.assertEqual(
|
||||
values[14].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_fee_total|0.00|2')
|
||||
self.assertEqual(
|
||||
values[15].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_sales|2.50|2')
|
||||
self.assertEqual(
|
||||
values[16].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_sales_12m|0.00|2')
|
||||
|
||||
# update rate
|
||||
self.assertEqual(asset.rates[0].rate, Decimal('3.0'))
|
||||
self.assertEqual(asset.rates[0].date, date(2022, 5, 10))
|
||||
Asset.write(*[
|
||||
[asset],
|
||||
{
|
||||
'rates': [('write', [asset.rates[0]], {
|
||||
'rate': Decimal('3.5'),
|
||||
})],
|
||||
}])
|
||||
self.assertEqual(
|
||||
asset.rec_name, 'Product 1 | 3.5000 usd/u | 05/10/2022')
|
||||
|
||||
# run worker
|
||||
self.prep_valstore_run_worker()
|
||||
|
||||
values = ValStore.search([
|
||||
('cashbook', '=', book.id)], order=[('field_name', 'ASC')])
|
||||
self.assertEqual(len(values), len(Book.valuestore_fields()))
|
||||
|
||||
self.assertEqual(
|
||||
values[0].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance|2.50|2')
|
||||
self.assertEqual(
|
||||
values[1].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance_all|2.50|2')
|
||||
self.assertEqual(
|
||||
values[2].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance_ref|2.50|2')
|
||||
self.assertEqual(
|
||||
values[3].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_rate|3.33|2')
|
||||
self.assertEqual(
|
||||
values[4].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_value|5.00|2')
|
||||
self.assertEqual(
|
||||
values[5].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_value_ref|5.00|2')
|
||||
self.assertEqual(
|
||||
values[6].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|diff_amount|2.50|2')
|
||||
self.assertEqual(
|
||||
values[7].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|diff_percent|100.00|2')
|
||||
self.assertEqual(
|
||||
values[8].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|purchase_amount|2.50|2')
|
||||
self.assertEqual(
|
||||
values[9].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|quantity|1.500|3')
|
||||
self.assertEqual(
|
||||
values[10].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|quantity_all|1.500|3')
|
||||
self.assertEqual(
|
||||
values[11].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_dividend_12m|0.00|2')
|
||||
self.assertEqual(
|
||||
values[12].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_dividend_total|0.00|2')
|
||||
self.assertEqual(
|
||||
values[13].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_fee_12m|0.00|2')
|
||||
self.assertEqual(
|
||||
values[14].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_fee_total|0.00|2')
|
||||
self.assertEqual(
|
||||
values[15].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_sales|2.50|2')
|
||||
self.assertEqual(
|
||||
values[16].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_sales_12m|0.00|2')
|
||||
|
||||
# delete rate
|
||||
self.assertEqual(asset.rates[0].rate, Decimal('3.5'))
|
||||
self.assertEqual(asset.rates[0].date, date(2022, 5, 10))
|
||||
Asset.write(*[
|
||||
[asset],
|
||||
{
|
||||
'rates': [('delete', [asset.rates[0].id])],
|
||||
}])
|
||||
self.assertEqual(
|
||||
asset.rec_name, 'Product 1 | 2.8000 usd/u | 05/02/2022')
|
||||
|
||||
# run worker
|
||||
self.prep_valstore_run_worker()
|
||||
|
||||
values = ValStore.search([
|
||||
('cashbook', '=', book.id)], order=[('field_name', 'ASC')])
|
||||
self.assertEqual(len(values), len(Book.valuestore_fields()))
|
||||
|
||||
self.assertEqual(
|
||||
values[0].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance|2.50|2')
|
||||
self.assertEqual(
|
||||
values[1].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance_all|2.50|2')
|
||||
self.assertEqual(
|
||||
values[2].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|balance_ref|2.50|2')
|
||||
self.assertEqual(
|
||||
values[3].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_rate|2.67|2')
|
||||
self.assertEqual(
|
||||
values[4].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_value|4.00|2')
|
||||
self.assertEqual(
|
||||
values[5].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|current_value_ref|4.00|2')
|
||||
self.assertEqual(
|
||||
values[6].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|diff_amount|1.50|2')
|
||||
self.assertEqual(
|
||||
values[7].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|diff_percent|60.00|2')
|
||||
self.assertEqual(
|
||||
values[8].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|purchase_amount|2.50|2')
|
||||
self.assertEqual(
|
||||
values[9].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|quantity|1.500|3')
|
||||
self.assertEqual(
|
||||
values[10].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|quantity_all|1.500|3')
|
||||
self.assertEqual(
|
||||
values[11].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_dividend_12m|0.00|2')
|
||||
self.assertEqual(
|
||||
values[12].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_dividend_total|0.00|2')
|
||||
self.assertEqual(
|
||||
values[13].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_fee_12m|0.00|2')
|
||||
self.assertEqual(
|
||||
values[14].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_fee_total|0.00|2')
|
||||
self.assertEqual(
|
||||
values[15].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_sales|2.50|2')
|
||||
self.assertEqual(
|
||||
values[16].rec_name,
|
||||
'[Book 1 | 2.50 € | Open | 1.500 u]|yield_sales_12m|0.00|2')
|
||||
|
||||
# end ValueStoreTestCase
|
||||
Loading…
Reference in a new issue