FIX scenario test
This commit is contained in:
parent
9a8805f4ab
commit
1dfa92ff70
|
@ -53,8 +53,8 @@ Create companies and currencies::
|
||||||
>>> _ = create_company(party=party3, currency=peso_colombian)
|
>>> _ = create_company(party=party3, currency=peso_colombian)
|
||||||
|
|
||||||
>>> company1, company2, company3 = Company.find([])
|
>>> company1, company2, company3 = Company.find([])
|
||||||
>>> admin.main_companies.append(company2)
|
>>> admin.companies.append(company2)
|
||||||
>>> admin.main_companies.append(company3)
|
>>> admin.companies.append(company3)
|
||||||
>>> admin.save()
|
>>> admin.save()
|
||||||
>>> config._context = User.get_preferences(True, config.context)
|
>>> config._context = User.get_preferences(True, config.context)
|
||||||
|
|
||||||
|
|
|
@ -17,7 +17,7 @@ def set_company(company):
|
||||||
pool = Pool()
|
pool = Pool()
|
||||||
User = pool.get('res.user')
|
User = pool.get('res.user')
|
||||||
User.write([User(Transaction().user)], {
|
User.write([User(Transaction().user)], {
|
||||||
'main_companies': [('add', [company.id])],
|
'companies': [('add', [company.id])],
|
||||||
'company': company.id,
|
'company': company.id,
|
||||||
})
|
})
|
||||||
with Transaction().set_context(User.get_preferences(context_only=True)):
|
with Transaction().set_context(User.get_preferences(context_only=True)):
|
||||||
|
@ -68,14 +68,14 @@ class PartyCompanyTestCase(ModuleTestCase):
|
||||||
self.assertEqual(address2.companies == (company,), True)
|
self.assertEqual(address2.companies == (company,), True)
|
||||||
|
|
||||||
user = User(Transaction().user)
|
user = User(Transaction().user)
|
||||||
self.assertEqual(len(user.main_companies) == 1, True)
|
self.assertEqual(len(user.companies) == 1, True)
|
||||||
self.assertEqual(user.main_companies[0] == company, True)
|
self.assertEqual(user.companies[0] == company, True)
|
||||||
|
|
||||||
company2 = create_company()
|
company2 = create_company()
|
||||||
with set_company(company2):
|
with set_company(company2):
|
||||||
user = User(Transaction().user)
|
user = User(Transaction().user)
|
||||||
self.assertEqual(len(user.main_companies) == 2, True)
|
self.assertEqual(len(user.companies) == 2, True)
|
||||||
self.assertEqual(user.main_companies[1] == company2, True)
|
self.assertEqual(user.companies[1] == company2, True)
|
||||||
|
|
||||||
|
|
||||||
def suite():
|
def suite():
|
||||||
|
|
Loading…
Reference in New Issue