asset: spalten tag/monat/3monate... ok + test
This commit is contained in:
Frederik Jaeckel
parent
64b8383096
commit
6311dce3d1
6 changed files with 377 additions and 177 deletions
|
|
@ -112,8 +112,9 @@ class AssetTestCase(ModuleTestCase):
|
|||
self.assertEqual(asset.rec_name, 'Product 1 - 2.4500 usd/Unit [05/15/2022]')
|
||||
|
||||
@with_transaction()
|
||||
def test_asset_percentges(self):
|
||||
""" create asset, add rates, check percentages
|
||||
def test_asset_percentages_daterange(self):
|
||||
""" create asset, add rates, check selection of
|
||||
value
|
||||
"""
|
||||
Asset = Pool().get('investment.asset')
|
||||
|
||||
|
|
@ -156,8 +157,148 @@ class AssetTestCase(ModuleTestCase):
|
|||
])
|
||||
self.assertEqual(asset1.rec_name, 'Product 1 - 2.6000 usd/Unit [05/16/2022]')
|
||||
self.assertEqual(asset2.rec_name, 'Product 1 - 5.2500 usd/Unit [05/15/2022]')
|
||||
self.assertEqual(asset1.change_day1, Decimal('6.1'))
|
||||
self.assertEqual(asset1.change_day1, Decimal('6.12'))
|
||||
self.assertEqual(asset2.change_day1, Decimal('-8.7'))
|
||||
self.assertEqual(asset1.change_month1, None)
|
||||
self.assertEqual(asset2.change_month1, None)
|
||||
self.assertEqual(asset1.change_month3, None)
|
||||
self.assertEqual(asset2.change_month3, None)
|
||||
self.assertEqual(asset1.change_month6, None)
|
||||
self.assertEqual(asset2.change_month6, None)
|
||||
self.assertEqual(asset1.change_month12, None)
|
||||
self.assertEqual(asset2.change_month12, None)
|
||||
|
||||
# check ordering
|
||||
assets = Asset.search([
|
||||
('change_day1', '!=', Decimal('0.0')),
|
||||
], order=[('change_day1', 'ASC')])
|
||||
self.assertEqual(len(assets), 2)
|
||||
self.assertEqual(assets[0].change_day1, Decimal('-8.7'))
|
||||
self.assertEqual(assets[1].change_day1, Decimal('6.12'))
|
||||
|
||||
assets = Asset.search([
|
||||
('change_day1', '!=', Decimal('0.0')),
|
||||
], order=[('change_day1', 'DESC')])
|
||||
self.assertEqual(len(assets), 2)
|
||||
self.assertEqual(assets[0].change_day1, Decimal('6.12'))
|
||||
self.assertEqual(assets[1].change_day1, Decimal('-8.7'))
|
||||
|
||||
# check 5-day-range
|
||||
# four days
|
||||
Asset.write(*[
|
||||
[asset1],
|
||||
{
|
||||
'rates': [('write', [asset1.rates[1]], {
|
||||
'date': date(2022, 5, 12),
|
||||
})],
|
||||
}])
|
||||
self.assertEqual(asset1.rates[0].date, date(2022, 5, 16))
|
||||
self.assertEqual(asset1.rates[1].date, date(2022, 5, 12))
|
||||
self.assertEqual(asset1.change_day1, Decimal('6.12'))
|
||||
# five days
|
||||
Asset.write(*[
|
||||
[asset1],
|
||||
{
|
||||
'rates': [('write', [asset1.rates[1]], {
|
||||
'date': date(2022, 5, 11),
|
||||
})],
|
||||
}])
|
||||
self.assertEqual(asset1.rates[0].date, date(2022, 5, 16))
|
||||
self.assertEqual(asset1.rates[1].date, date(2022, 5, 11))
|
||||
self.assertEqual(asset1.change_day1, Decimal('6.12'))
|
||||
# six days
|
||||
Asset.write(*[
|
||||
[asset1],
|
||||
{
|
||||
'rates': [('write', [asset1.rates[1]], {
|
||||
'date': date(2022, 5, 10),
|
||||
})],
|
||||
}])
|
||||
self.assertEqual(asset1.rates[0].date, date(2022, 5, 16))
|
||||
self.assertEqual(asset1.rates[1].date, date(2022, 5, 10))
|
||||
self.assertEqual(asset1.change_day1, None)
|
||||
|
||||
@with_transaction()
|
||||
def test_asset_percentges_values(self):
|
||||
""" create asset, add rates, check percentages
|
||||
"""
|
||||
Asset = Pool().get('investment.asset')
|
||||
|
||||
company = self.prep_asset_company()
|
||||
product = self.prep_asset_product(
|
||||
name='Product 1',
|
||||
description='some asset')
|
||||
|
||||
asset1 = self.prep_asset_item(
|
||||
company=company,
|
||||
product = product)
|
||||
|
||||
self.assertEqual(asset1.rec_name, 'Product 1 - - usd/Unit [-]')
|
||||
|
||||
Asset.write(*[
|
||||
[asset1],
|
||||
{
|
||||
'rates': [('create', [{
|
||||
'date': date(2022, 5, 15),
|
||||
'rate': Decimal('2.45'),
|
||||
}, {
|
||||
'date': date(2022, 5, 16),
|
||||
'rate': Decimal('2.6'),
|
||||
}, {
|
||||
'date': date(2022, 4, 14),
|
||||
'rate': Decimal('2.2'),
|
||||
}, {
|
||||
'date': date(2022, 2, 14),
|
||||
'rate': Decimal('2.8'),
|
||||
},])],
|
||||
}])
|
||||
self.assertEqual(asset1.rec_name, 'Product 1 - 2.6000 usd/Unit [05/16/2022]')
|
||||
self.assertEqual(len(asset1.rates), 4)
|
||||
self.assertEqual(asset1.rates[0].date, date(2022, 5, 16))
|
||||
self.assertEqual(asset1.rates[1].date, date(2022, 5, 15))
|
||||
self.assertEqual(asset1.rates[2].date, date(2022, 4, 14))
|
||||
self.assertEqual(asset1.rates[3].date, date(2022, 2, 14))
|
||||
|
||||
self.assertEqual(asset1.change_day1, Decimal('6.12'))
|
||||
self.assertEqual(asset1.change_month1, Decimal('18.18'))
|
||||
self.assertEqual(asset1.change_month3, Decimal('-7.14'))
|
||||
self.assertEqual(asset1.change_month6, None)
|
||||
self.assertEqual(asset1.change_month12, None)
|
||||
|
||||
# call order-functions
|
||||
Asset.search([], order=[('change_day1', 'ASC')])
|
||||
Asset.search([], order=[('change_month1', 'ASC')])
|
||||
Asset.search([], order=[('change_month3', 'ASC')])
|
||||
Asset.search([], order=[('change_month6', 'ASC')])
|
||||
Asset.search([], order=[('change_month12', 'ASC')])
|
||||
|
||||
# searcher
|
||||
self.assertEqual(
|
||||
Asset.search_count([('change_day1', '>', Decimal('6.1'))]),
|
||||
1)
|
||||
self.assertEqual(
|
||||
Asset.search_count([('change_day1', '>', Decimal('6.15'))]),
|
||||
0)
|
||||
self.assertEqual(
|
||||
Asset.search_count([('change_day1', '=', Decimal('6.12'))]),
|
||||
1)
|
||||
|
||||
self.assertEqual(
|
||||
Asset.search_count([('change_month1', '>', Decimal('18.0'))]),
|
||||
1)
|
||||
self.assertEqual(
|
||||
Asset.search_count([('change_month1', '>', Decimal('18.18'))]),
|
||||
0)
|
||||
self.assertEqual(
|
||||
Asset.search_count([('change_month1', '=', Decimal('18.18'))]),
|
||||
1)
|
||||
|
||||
self.assertEqual(
|
||||
Asset.search_count([('change_month3', '=', Decimal('-7.14'))]),
|
||||
1)
|
||||
self.assertEqual(
|
||||
Asset.search_count([('change_month6', '=', None)]),
|
||||
1)
|
||||
|
||||
@with_transaction()
|
||||
def test_asset_check_onlinesource_onoff(self):
|
||||
|
|
@ -181,7 +322,7 @@ class AssetTestCase(ModuleTestCase):
|
|||
}])
|
||||
|
||||
self.assertEqual(asset.updtsource, None)
|
||||
self.assertEqual(asset.updttime, None)
|
||||
self.assertEqual(asset.updttime, time(14,0))
|
||||
|
||||
asset.updtsource = o_source
|
||||
asset.updttime = time(10, 45)
|
||||
|
|
@ -231,13 +372,13 @@ class AssetTestCase(ModuleTestCase):
|
|||
self.assertEqual(asset2.updtsource.rec_name, 'Source 1')
|
||||
self.assertEqual(asset2.updttime, time(10, 45))
|
||||
self.assertEqual(len(asset2.rates), 0)
|
||||
self.assertEqual(asset2.nextupdtate, datetime(2022, 10, 15, 10, 45))
|
||||
self.assertEqual(asset2.nextupdate, datetime(2022, 10, 15, 10, 45))
|
||||
|
||||
self.assertEqual(
|
||||
Asset.search_count([('nextupdtate', '<', datetime(2022, 10, 15, 10, 45))]),
|
||||
Asset.search_count([('nextupdate', '<', datetime(2022, 10, 15, 10, 45))]),
|
||||
0)
|
||||
self.assertEqual(
|
||||
Asset.search_count([('nextupdtate', '>=', datetime(2022, 10, 15, 10, 45))]),
|
||||
Asset.search_count([('nextupdate', '>=', datetime(2022, 10, 15, 10, 45))]),
|
||||
1)
|
||||
|
||||
# add rate at yesterday
|
||||
|
|
@ -256,13 +397,13 @@ class AssetTestCase(ModuleTestCase):
|
|||
self.assertEqual(asset.updttime, time(10, 45))
|
||||
self.assertEqual(len(asset.rates), 1)
|
||||
self.assertEqual(asset.rates[0].date, date(2022, 10, 14))
|
||||
self.assertEqual(asset.nextupdtate, datetime(2022, 10, 15, 10, 45))
|
||||
self.assertEqual(asset.nextupdate, datetime(2022, 10, 15, 10, 45))
|
||||
|
||||
self.assertEqual(
|
||||
Asset.search_count([('nextupdtate', '<', datetime(2022, 10, 15, 10, 45))]),
|
||||
Asset.search_count([('nextupdate', '<', datetime(2022, 10, 15, 10, 45))]),
|
||||
0)
|
||||
self.assertEqual(
|
||||
Asset.search_count([('nextupdtate', '>=', datetime(2022, 10, 15, 10, 45))]),
|
||||
Asset.search_count([('nextupdate', '>=', datetime(2022, 10, 15, 10, 45))]),
|
||||
1)
|
||||
|
||||
# add rate at today
|
||||
|
|
@ -281,13 +422,13 @@ class AssetTestCase(ModuleTestCase):
|
|||
self.assertEqual(asset2.updttime, time(10, 45))
|
||||
self.assertEqual(len(asset2.rates), 2)
|
||||
self.assertEqual(asset2.rates[0].date, date(2022, 10, 15))
|
||||
self.assertEqual(asset2.nextupdtate, datetime(2022, 10, 16, 10, 45))
|
||||
self.assertEqual(asset2.nextupdate, datetime(2022, 10, 16, 10, 45))
|
||||
|
||||
self.assertEqual(
|
||||
Asset.search_count([('nextupdtate', '<', datetime(2022, 10, 15, 10, 45))]),
|
||||
Asset.search_count([('nextupdate', '<', datetime(2022, 10, 15, 10, 45))]),
|
||||
0)
|
||||
self.assertEqual(
|
||||
Asset.search_count([('nextupdtate', '>=', datetime(2022, 10, 15, 10, 45))]),
|
||||
Asset.search_count([('nextupdate', '>=', datetime(2022, 10, 15, 10, 45))]),
|
||||
1)
|
||||
|
||||
@with_transaction()
|
||||
|
|
|
|||
|
|
@ -8,7 +8,7 @@ from trytond.pool import Pool
|
|||
from trytond.modules.company.tests import create_company
|
||||
from trytond.transaction import Transaction
|
||||
from decimal import Decimal
|
||||
from datetime import time, date
|
||||
from datetime import time, date, datetime
|
||||
from unittest.mock import MagicMock
|
||||
from requests import Response
|
||||
import requests
|
||||
|
|
@ -66,26 +66,34 @@ class SourceTestCase(ModuleTestCase):
|
|||
{
|
||||
'updtsource': osource.id,
|
||||
}])
|
||||
self.assertEqual(asset.wkn, '965515')
|
||||
self.assertEqual(asset.isin, 'XC0009655157')
|
||||
self.assertEqual(asset.secsymb, '1472977')
|
||||
self.assertEqual(asset.updtsource.rec_name, 'Source 1')
|
||||
self.assertEqual(len(asset.rates), 0)
|
||||
|
||||
# fake server-response
|
||||
resp1 = Response()
|
||||
resp1._content = """<html><body>Response from finance-server
|
||||
with Transaction().set_context({
|
||||
'qdate': date(2022, 10, 1),
|
||||
'qdatetime': datetime(2022, 10, 2, 10, 0, 0),
|
||||
}):
|
||||
asset2, = Asset.browse([asset])
|
||||
self.assertEqual(asset2.wkn, '965515')
|
||||
self.assertEqual(asset2.isin, 'XC0009655157')
|
||||
self.assertEqual(asset2.secsymb, '1472977')
|
||||
self.assertEqual(asset2.updttime, time(14, 0))
|
||||
self.assertEqual(asset2.updtsource.rec_name, 'Source 1')
|
||||
self.assertEqual(asset2.nextupdate, datetime(2022, 10, 2, 14, 0))
|
||||
self.assertEqual(len(asset.rates), 0)
|
||||
|
||||
# fake server-response
|
||||
resp1 = Response()
|
||||
resp1._content = """<html><body>Response from finance-server
|
||||
Course Date 14.08.2022 Today
|
||||
High 34,87 EUR
|
||||
</body></html>""".encode('utf8')
|
||||
resp1.status_code = 200
|
||||
resp1.reason = 'OK'
|
||||
requests.get = MagicMock(return_value=resp1)
|
||||
resp1.status_code = 200
|
||||
resp1.reason = 'OK'
|
||||
requests.get = MagicMock(return_value=resp1)
|
||||
|
||||
OSource.update_rate(asset)
|
||||
self.assertEqual(len(asset.rates), 1)
|
||||
self.assertEqual(asset.rates[0].date, date(2022, 8, 14))
|
||||
self.assertEqual(asset.rates[0].rate, Decimal('34.87'))
|
||||
OSource.update_rate(asset)
|
||||
self.assertEqual(len(asset.rates), 1)
|
||||
self.assertEqual(asset.rates[0].date, date(2022, 8, 14))
|
||||
self.assertEqual(asset.rates[0].rate, Decimal('34.87'))
|
||||
|
||||
@with_transaction()
|
||||
def test_waitlist_source_check_regex(self):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue