NaNtic
/
trytond-party_company
mirror of https://github.com/NaN-tic/trytond-party_company.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

FIX scenario test

This commit is contained in:
Raimon Esteve 2021-04-26 14:59:56 +02:00
parent 9a8805f4ab
commit 1dfa92ff70
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/scenario_party_company.rst
View File

@ -53,8 +53,8 @@ Create companies and currencies::
>>> _ = create_company(party=party3, currency=peso_colombian)
>>> company1, company2, company3 = Company.find([])
>>> admin.main_companies.append(company2)
>>> admin.main_companies.append(company3)
>>> admin.companies.append(company2)
>>> admin.companies.append(company3)
>>> admin.save()
>>> config._context = User.get_preferences(True, config.context)

10
tests/test_party_company.py
View File

@ -17,7 +17,7 @@ def set_company(company):
pool = Pool()
User = pool.get('res.user')
User.write([User(Transaction().user)], {
'main_companies': [('add', [company.id])],
'companies': [('add', [company.id])],
'company': company.id,
})
with Transaction().set_context(User.get_preferences(context_only=True)):
@ -68,14 +68,14 @@ class PartyCompanyTestCase(ModuleTestCase):
self.assertEqual(address2.companies == (company,), True)
user = User(Transaction().user)
self.assertEqual(len(user.main_companies) == 1, True)
self.assertEqual(user.main_companies[0] == company, True)
self.assertEqual(len(user.companies) == 1, True)
self.assertEqual(user.companies[0] == company, True)
company2 = create_company()
with set_company(company2):
user = User(Transaction().user)
self.assertEqual(len(user.main_companies) == 2, True)
self.assertEqual(user.main_companies[1] == company2, True)
self.assertEqual(len(user.companies) == 2, True)
self.assertEqual(user.companies[1] == company2, True)
def suite():