test: add more context

This commit is contained in:
Frederik Jaeckel 2023-06-07 20:06:36 +02:00
parent ebb6d71be3
commit 765738d9ee
4 changed files with 616 additions and 615 deletions

View file

@ -134,7 +134,9 @@ class AssetTestCase(object):
self.assertEqual( self.assertEqual(
asset.rec_name, asset.rec_name,
'Product 1 | 2.4500 usd/u | 05/15/2022') 'Product 1 | 2.4500 usd/u | 05/15/2022')
self.assertEqual(Asset.search_count([('name', '=', 'Product 1')]), 1) self.assertEqual(
Asset.search_count([('name', '=', 'Product 1')]),
1)
@with_transaction() @with_transaction()
def test_asset_order_and_search_rate_and_date(self): def test_asset_order_and_search_rate_and_date(self):
@ -151,12 +153,8 @@ class AssetTestCase(object):
name='Product 2', name='Product 2',
description='some asset') description='some asset')
asset1 = self.prep_asset_item( asset1 = self.prep_asset_item(company=company, product=product1)
company=company, asset2 = self.prep_asset_item(company=company, product=product2)
product=product1)
asset2 = self.prep_asset_item(
company=company,
product=product2)
Asset.write(*[ Asset.write(*[
[asset1], [asset1],
@ -231,9 +229,7 @@ class AssetTestCase(object):
name='Product 1', name='Product 1',
description='some asset') description='some asset')
asset1 = self.prep_asset_item( asset1 = self.prep_asset_item(company=company, product=product)
company=company,
product=product)
self.assertEqual(asset1.rec_name, 'Product 1 | - usd/u | -') self.assertEqual(asset1.rec_name, 'Product 1 | - usd/u | -')
Asset.write(*[ Asset.write(*[
@ -298,12 +294,8 @@ class AssetTestCase(object):
name='Product 1', name='Product 1',
description='some asset') description='some asset')
asset1 = self.prep_asset_item( asset1 = self.prep_asset_item(company=company, product=product)
company=company, asset2 = self.prep_asset_item(company=company, product=product)
product=product)
asset2 = self.prep_asset_item(
company=company,
product=product)
self.assertEqual(asset1.rec_name, 'Product 1 | - usd/u | -') self.assertEqual(asset1.rec_name, 'Product 1 | - usd/u | -')
self.assertEqual(asset2.rec_name, 'Product 1 | - usd/u | -') self.assertEqual(asset2.rec_name, 'Product 1 | - usd/u | -')
@ -409,9 +401,7 @@ class AssetTestCase(object):
name='Product 1', name='Product 1',
description='some asset') description='some asset')
asset1 = self.prep_asset_item( asset1 = self.prep_asset_item(company=company, product=product)
company=company,
product=product)
self.assertEqual(asset1.rec_name, 'Product 1 | - usd/u | -') self.assertEqual(asset1.rec_name, 'Product 1 | - usd/u | -')
@ -495,9 +485,7 @@ class AssetTestCase(object):
name='Product 1', name='Product 1',
description='some asset') description='some asset')
asset = self.prep_asset_item( asset = self.prep_asset_item(company=company, product=product)
company=company,
product=product)
o_source, = OnlineSource.create([{ o_source, = OnlineSource.create([{
'name': 'Source 1', 'name': 'Source 1',
@ -533,9 +521,7 @@ class AssetTestCase(object):
name='Product 1', name='Product 1',
description='some asset') description='some asset')
asset = self.prep_asset_item( asset = self.prep_asset_item(company=company, product=product)
company=company,
product=product)
o_source, = OnlineSource.create([{ o_source, = OnlineSource.create([{
'name': 'Source 1', 'name': 'Source 1',
@ -547,9 +533,7 @@ class AssetTestCase(object):
'updttime': time(10, 45), 'updttime': time(10, 45),
}]) }])
with Transaction().set_context({ with Transaction().set_context({'qdate': date(2022, 10, 14)}):
'qdate': date(2022, 10, 14)}):
# re-read to make context work # re-read to make context work
asset2, = Asset.browse([asset.id]) asset2, = Asset.browse([asset.id])
@ -560,7 +544,9 @@ class AssetTestCase(object):
# qdate = 2022-10-14 simulates existence of record at this day # qdate = 2022-10-14 simulates existence of record at this day
# next call would be the 15. - but its saturday, # next call would be the 15. - but its saturday,
# next-call-date is moved to 17. # next-call-date is moved to 17.
self.assertEqual(asset2.nextupdate, datetime(2022, 10, 17, 10, 45)) self.assertEqual(
asset2.nextupdate,
datetime(2022, 10, 17, 10, 45))
self.assertEqual( self.assertEqual(
Asset.search_count([ Asset.search_count([
@ -676,9 +662,15 @@ class AssetTestCase(object):
self.assertEqual(asset1.isin, 'XC0009655157') self.assertEqual(asset1.isin, 'XC0009655157')
self.assertEqual(asset1.secsymb, '1472977') self.assertEqual(asset1.secsymb, '1472977')
self.assertEqual(Asset.search_count([('wkn', '=', '965515')]), 1) self.assertEqual(
self.assertEqual(Asset.search_count([('isin', '=', 'XC0009655157')]), 1) Asset.search_count([('wkn', '=', '965515')]),
self.assertEqual(Asset.search_count([('secsymb', '=', '1472977')]), 1) 1)
self.assertEqual(
Asset.search_count([('isin', '=', 'XC0009655157')]),
1)
self.assertEqual(
Asset.search_count([('secsymb', '=', '1472977')]),
1)
self.assertEqual( self.assertEqual(
Asset.search_count([('rec_name', '=', '965515')]), Asset.search_count([('rec_name', '=', '965515')]),
@ -690,11 +682,18 @@ class AssetTestCase(object):
Asset.search_count([('rec_name', '=', '1472977')]), Asset.search_count([('rec_name', '=', '1472977')]),
1) 1)
self.assertEqual(Asset.search_count([('name', '=', '965515')]), 1) self.assertEqual(
self.assertEqual(Asset.search_count([('name', '=', 'XC0009655157')]), 1) Asset.search_count([('name', '=', '965515')]),
self.assertEqual(Asset.search_count([('name', '=', '1472977')]), 1) 1)
self.assertEqual(
self.assertEqual(Asset.search_count([('name', '=', 'Product unit')]), 1) Asset.search_count([('name', '=', 'XC0009655157')]),
1)
self.assertEqual(
Asset.search_count([('name', '=', '1472977')]),
1)
self.assertEqual(
Asset.search_count([('name', '=', 'Product unit')]),
1)
self.assertEqual(Asset.search_count([ self.assertEqual(Asset.search_count([
('wkn', 'ilike', '9655%'), ('wkn', 'ilike', '9655%'),

View file

@ -25,9 +25,7 @@ class RateTestCase(object):
name='Product 1', name='Product 1',
description='some asset') description='some asset')
asset = self.prep_asset_item( asset = self.prep_asset_item(company=company, product=product)
company=company,
product=product)
Asset.write(*[ Asset.write(*[
[asset], [asset],

View file

@ -26,6 +26,7 @@ class SourceTestCase(object):
Product = pool.get('product.product') Product = pool.get('product.product')
company = self.prep_asset_company() company = self.prep_asset_company()
with Transaction().set_context({'company': company.id}):
osource, = OSource.create([{ osource, = OSource.create([{
'name': 'Source 1', 'name': 'Source 1',
'url': 'https://foo.bar/${isin}/${nsin}/${symbol}', 'url': 'https://foo.bar/${isin}/${nsin}/${symbol}',
@ -36,7 +37,6 @@ class SourceTestCase(object):
}]) }])
self.assertEqual(osource.rec_name, 'Source 1') self.assertEqual(osource.rec_name, 'Source 1')
with Transaction().set_context({'company': company.id}):
product = self.prep_asset_product( product = self.prep_asset_product(
name='Product 1', name='Product 1',
description='some asset') description='some asset')
@ -56,9 +56,7 @@ class SourceTestCase(object):
}, ])], }, ])],
}]) }])
asset = self.prep_asset_item( asset = self.prep_asset_item(company=company, product=product)
company=company,
product=product)
Asset.write(*[ Asset.write(*[
[asset], [asset],
@ -77,7 +75,8 @@ class SourceTestCase(object):
self.assertEqual(len(asset2.updtsources), 1) self.assertEqual(len(asset2.updtsources), 1)
self.assertEqual(asset2.updtsources[0].rec_name, 'Source 1') self.assertEqual(asset2.updtsources[0].rec_name, 'Source 1')
self.assertEqual(asset2.updtdays, 'work') self.assertEqual(asset2.updtdays, 'work')
self.assertEqual(asset2.nextupdate, datetime(2022, 10, 3, 14, 0)) self.assertEqual(
asset2.nextupdate, datetime(2022, 10, 3, 14, 0))
self.assertEqual(len(asset.rates), 0) self.assertEqual(len(asset.rates), 0)
# fake server-response # fake server-response

View file

@ -21,6 +21,7 @@ class WizardTestCase(object):
ImportWiz = pool.get('investment.imp_wiz', type='wizard') ImportWiz = pool.get('investment.imp_wiz', type='wizard')
company = self.prep_asset_company() company = self.prep_asset_company()
with Transaction().set_context({'company': company.id}):
product = self.prep_asset_product( product = self.prep_asset_product(
name='Product 1', name='Product 1',
description='some asset') description='some asset')
@ -42,8 +43,12 @@ class WizardTestCase(object):
self.assertEqual(result['view']['defaults']['asset'], asset.id) self.assertEqual(result['view']['defaults']['asset'], asset.id)
self.assertEqual(result['view']['defaults']['dec_divider'], ',') self.assertEqual(result['view']['defaults']['dec_divider'], ',')
self.assertEqual(result['view']['defaults']['date_fmt'], '%d.%m.%Y') self.assertEqual(
self.assertEqual(result['view']['defaults']['field_delimiter'], ';') result['view']['defaults']['date_fmt'],
'%d.%m.%Y')
self.assertEqual(
result['view']['defaults']['field_delimiter'],
';')
w_obj.start.asset = asset w_obj.start.asset = asset
w_obj.start.dec_divider = ',' w_obj.start.dec_divider = ','