493 lines
16 KiB
Python
493 lines
16 KiB
Python
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
import unittest
|
|
import doctest
|
|
try:
|
|
import phonenumbers
|
|
except ImportError:
|
|
phonenumbers = None
|
|
|
|
import trytond.tests.test_tryton
|
|
from trytond.tests.test_tryton import ModuleTestCase, with_transaction
|
|
from trytond.tests.test_tryton import doctest_teardown
|
|
from trytond.tests.test_tryton import doctest_checker
|
|
from trytond.pool import Pool
|
|
from trytond.exceptions import UserError
|
|
from trytond.transaction import Transaction
|
|
|
|
from ..party import IDENTIFIER_TYPES
|
|
|
|
|
|
class PartyCheckEraseMixin:
|
|
|
|
@with_transaction()
|
|
def test_check_erase_party(self):
|
|
"Test check erase of party"
|
|
pool = Pool()
|
|
Erase = pool.get('party.erase', type='wizard')
|
|
Session = pool.get('ir.session.wizard')
|
|
party = self.setup_check_erase_party()
|
|
|
|
session = Session()
|
|
session.save()
|
|
|
|
Erase(session.id).check_erase(party)
|
|
|
|
def setup_check_erase_party(self):
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
|
|
party = Party(active=False)
|
|
party.save()
|
|
return party
|
|
|
|
|
|
class PartyTestCase(PartyCheckEraseMixin, ModuleTestCase):
|
|
'Test Party module'
|
|
module = 'party'
|
|
|
|
@with_transaction()
|
|
def test_category(self):
|
|
'Create category'
|
|
pool = Pool()
|
|
Category = pool.get('party.category')
|
|
category1, = Category.create([{
|
|
'name': 'Category 1',
|
|
}])
|
|
self.assertTrue(category1.id)
|
|
|
|
@with_transaction()
|
|
def test_category_recursion(self):
|
|
'Test category recursion'
|
|
pool = Pool()
|
|
Category = pool.get('party.category')
|
|
category1, = Category.create([{
|
|
'name': 'Category 1',
|
|
}])
|
|
category2, = Category.create([{
|
|
'name': 'Category 2',
|
|
'parent': category1.id,
|
|
}])
|
|
self.assertTrue(category2.id)
|
|
|
|
self.assertRaises(Exception, Category.write, [category1], {
|
|
'parent': category2.id,
|
|
})
|
|
|
|
@with_transaction()
|
|
def test_party(self):
|
|
'Create party'
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
party1, = Party.create([{
|
|
'name': 'Party 1',
|
|
}])
|
|
self.assertTrue(party1.id)
|
|
|
|
@with_transaction()
|
|
def test_party_code(self):
|
|
'Test party code constraint'
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
party1, = Party.create([{
|
|
'name': 'Party 1',
|
|
}])
|
|
|
|
code = party1.code
|
|
|
|
party2, = Party.create([{
|
|
'name': 'Party 2',
|
|
}])
|
|
|
|
self.assertRaises(Exception, Party.write, [party2], {
|
|
'code': code,
|
|
})
|
|
|
|
@with_transaction()
|
|
def test_address(self):
|
|
'Create address'
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
Address = pool.get('party.address')
|
|
party1, = Party.create([{
|
|
'name': 'Party 1',
|
|
}])
|
|
|
|
address, = Address.create([{
|
|
'party': party1.id,
|
|
'street': 'St sample, 15',
|
|
'city': 'City',
|
|
}])
|
|
self.assertTrue(address.id)
|
|
self.assertMultiLineEqual(address.full_address,
|
|
"St sample, 15\n"
|
|
"City")
|
|
with Transaction().set_context(address_with_party=True):
|
|
address = Address(address.id)
|
|
self.assertMultiLineEqual(address.full_address,
|
|
"Party 1\n"
|
|
"St sample, 15\n"
|
|
"City")
|
|
|
|
@with_transaction()
|
|
def test_full_address_country_subdivision(self):
|
|
'Test full address with country and subdivision'
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
Country = pool.get('country.country')
|
|
Subdivision = pool.get('country.subdivision')
|
|
Address = pool.get('party.address')
|
|
party, = Party.create([{
|
|
'name': 'Party',
|
|
}])
|
|
country = Country(name='Country')
|
|
country.save()
|
|
subdivision = Subdivision(
|
|
name='Subdivision', country=country, code='SUB', type='area')
|
|
subdivision.save()
|
|
address, = Address.create([{
|
|
'party': party.id,
|
|
'subdivision': subdivision.id,
|
|
'country': country.id,
|
|
}])
|
|
self.assertMultiLineEqual(address.full_address,
|
|
"Subdivision\n"
|
|
"COUNTRY")
|
|
|
|
@with_transaction()
|
|
def test_address_get_no_type(self):
|
|
"Test address_get with no type"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
Address = pool.get('party.address')
|
|
party, = Party.create([{}])
|
|
address1, address2 = Address.create([{
|
|
'party': party.id,
|
|
'sequence': 1,
|
|
}, {
|
|
'party': party.id,
|
|
'sequence': 2,
|
|
}])
|
|
|
|
address = party.address_get()
|
|
|
|
self.assertEqual(address, address1)
|
|
|
|
@with_transaction()
|
|
def test_address_get_no_address(self):
|
|
"Test address_get with no address"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
party, = Party.create([{}])
|
|
|
|
address = party.address_get()
|
|
|
|
self.assertEqual(address, None)
|
|
|
|
@with_transaction()
|
|
def test_address_get_inactive(self):
|
|
"Test address_get with inactive"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
Address = pool.get('party.address')
|
|
party, = Party.create([{}])
|
|
address1, address2 = Address.create([{
|
|
'party': party.id,
|
|
'sequence': 1,
|
|
'active': False,
|
|
}, {
|
|
'party': party.id,
|
|
'sequence': 2,
|
|
'active': True,
|
|
}])
|
|
|
|
address = party.address_get()
|
|
|
|
self.assertEqual(address, address2)
|
|
|
|
@with_transaction()
|
|
def test_address_get_type(self):
|
|
"Test address_get with type"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
Address = pool.get('party.address')
|
|
party, = Party.create([{}])
|
|
address1, address2 = Address.create([{
|
|
'party': party.id,
|
|
'sequence': 1,
|
|
'postal_code': None,
|
|
}, {
|
|
'party': party.id,
|
|
'sequence': 2,
|
|
'postal_code': '1000',
|
|
}])
|
|
|
|
address = party.address_get(type='postal_code')
|
|
|
|
self.assertEqual(address, address2)
|
|
|
|
@with_transaction()
|
|
def test_party_label_report(self):
|
|
'Test party label report'
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
Label = pool.get('party.label', type='report')
|
|
party1, = Party.create([{
|
|
'name': 'Party 1',
|
|
}])
|
|
oext, content, _, _ = Label.execute([party1.id], {})
|
|
self.assertEqual(oext, 'odt')
|
|
self.assertTrue(content)
|
|
|
|
@with_transaction()
|
|
def test_party_without_name(self):
|
|
'Create party without name'
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
party2, = Party.create([{}])
|
|
self.assertTrue(party2.id)
|
|
code = party2.code
|
|
self.assertEqual(party2.rec_name, '[' + code + ']')
|
|
|
|
@unittest.skipIf(phonenumbers is None, 'requires phonenumbers')
|
|
@with_transaction()
|
|
def test_phone_number_format(self):
|
|
'Test phone number format'
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
ContactMechanism = pool.get('party.contact_mechanism')
|
|
transaction = Transaction()
|
|
|
|
def create(mtype, mvalue):
|
|
party1, = Party.create([{
|
|
'name': 'Party 1',
|
|
}])
|
|
return ContactMechanism.create([{
|
|
'party': party1.id,
|
|
'type': mtype,
|
|
'value': mvalue,
|
|
}])[0]
|
|
|
|
# Test format on create
|
|
mechanism = create('phone', '+442083661177')
|
|
self.assertEqual(mechanism.value, '+44 20 8366 1177')
|
|
self.assertEqual(mechanism.value_compact, '+442083661177')
|
|
|
|
# Test format on write
|
|
mechanism.value = '+442083661178'
|
|
mechanism.save()
|
|
self.assertEqual(mechanism.value, '+44 20 8366 1178')
|
|
self.assertEqual(mechanism.value_compact, '+442083661178')
|
|
|
|
ContactMechanism.write([mechanism], {
|
|
'value': '+442083661179',
|
|
})
|
|
self.assertEqual(mechanism.value, '+44 20 8366 1179')
|
|
self.assertEqual(mechanism.value_compact, '+442083661179')
|
|
|
|
# Test rejection of a phone type mechanism to non-phone value
|
|
with self.assertRaises(UserError):
|
|
mechanism.value = 'notaphone@example.com'
|
|
mechanism.save()
|
|
transaction.rollback()
|
|
|
|
# Test rejection of invalid phone number creation
|
|
with self.assertRaises(UserError):
|
|
mechanism = create('phone', 'alsonotaphone@example.com')
|
|
transaction.rollback()
|
|
|
|
# Test acceptance of a non-phone value when type is non-phone
|
|
mechanism = create('email', 'name@example.com')
|
|
|
|
@with_transaction()
|
|
def test_contact_mechanism_get_no_usage(self):
|
|
"Test contact_mechanism_get with no usage"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
ContactMechanism = pool.get('party.contact_mechanism')
|
|
party, = Party.create([{}])
|
|
contact1, contact2 = ContactMechanism.create([{
|
|
'party': party.id,
|
|
'sequence': 1,
|
|
'type': 'email',
|
|
'value': 'test1@example.com',
|
|
}, {
|
|
'party': party.id,
|
|
'sequence': 2,
|
|
'type': 'email',
|
|
'value': 'test2@example.com',
|
|
}])
|
|
|
|
contact = party.contact_mechanism_get('email')
|
|
|
|
self.assertEqual(contact, contact1)
|
|
|
|
@with_transaction()
|
|
def test_contact_mechanism_get_many_types(self):
|
|
"Test contact_mechanism_get with many types"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
ContactMechanism = pool.get('party.contact_mechanism')
|
|
party, = Party.create([{}])
|
|
contact1, contact2 = ContactMechanism.create([{
|
|
'party': party.id,
|
|
'sequence': 1,
|
|
'type': 'other',
|
|
'value': 'test',
|
|
}, {
|
|
'party': party.id,
|
|
'sequence': 2,
|
|
'type': 'email',
|
|
'value': 'test2@example.com',
|
|
}])
|
|
|
|
contact = party.contact_mechanism_get({'email', 'phone'})
|
|
|
|
self.assertEqual(contact, contact2)
|
|
|
|
@with_transaction()
|
|
def test_contact_mechanism_get_no_contact_mechanism(self):
|
|
"Test contact_mechanism_get with no contact mechanism"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
party, = Party.create([{}])
|
|
|
|
contact = party.contact_mechanism_get()
|
|
|
|
self.assertEqual(contact, None)
|
|
|
|
@with_transaction()
|
|
def test_contact_mechanism_get_no_type(self):
|
|
"Test contact_mechanism_get with no type"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
ContactMechanism = pool.get('party.contact_mechanism')
|
|
party, = Party.create([{}])
|
|
ContactMechanism.create([{
|
|
'party': party.id,
|
|
'type': 'email',
|
|
'value': 'test1@example.com',
|
|
}])
|
|
|
|
contact = party.contact_mechanism_get('phone')
|
|
|
|
self.assertEqual(contact, None)
|
|
|
|
@with_transaction()
|
|
def test_contact_mechanism_get_any_type(self):
|
|
"Test contact_mechanism_get with any type"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
ContactMechanism = pool.get('party.contact_mechanism')
|
|
party, = Party.create([{}])
|
|
email1, = ContactMechanism.create([{
|
|
'party': party.id,
|
|
'type': 'email',
|
|
'value': 'test1@example.com',
|
|
}])
|
|
|
|
contact = party.contact_mechanism_get()
|
|
|
|
self.assertEqual(contact, email1)
|
|
|
|
@with_transaction()
|
|
def test_contact_mechanism_get_inactive(self):
|
|
"Test contact_mechanism_get with inactive"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
ContactMechanism = pool.get('party.contact_mechanism')
|
|
party, = Party.create([{}])
|
|
contact1, contact2 = ContactMechanism.create([{
|
|
'party': party.id,
|
|
'sequence': 1,
|
|
'type': 'email',
|
|
'value': 'test1@example.com',
|
|
'active': False,
|
|
}, {
|
|
'party': party.id,
|
|
'sequence': 2,
|
|
'type': 'email',
|
|
'value': 'test2@example.com',
|
|
'active': True,
|
|
}])
|
|
|
|
contact = party.contact_mechanism_get()
|
|
|
|
self.assertEqual(contact, contact2)
|
|
|
|
@with_transaction()
|
|
def test_contact_mechanism_get_usage(self):
|
|
"Test contact_mechanism_get with usage"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
ContactMechanism = pool.get('party.contact_mechanism')
|
|
party, = Party.create([{}])
|
|
contact1, contact2 = ContactMechanism.create([{
|
|
'party': party.id,
|
|
'sequence': 1,
|
|
'type': 'email',
|
|
'value': 'test1@example.com',
|
|
'name': None,
|
|
}, {
|
|
'party': party.id,
|
|
'sequence': 2,
|
|
'type': 'email',
|
|
'value': 'test2@example.com',
|
|
'name': 'email',
|
|
}])
|
|
|
|
contact = party.contact_mechanism_get(usage='name')
|
|
|
|
self.assertEqual(contact, contact2)
|
|
|
|
@with_transaction()
|
|
def test_tax_identifier_types(self):
|
|
"Ensure tax identifier types are in identifier types"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
self.assertFalse(
|
|
set(Party.tax_identifier_types())
|
|
- set(dict(IDENTIFIER_TYPES).keys()))
|
|
|
|
@with_transaction()
|
|
def test_party_distance(self):
|
|
"Test party distance"
|
|
pool = Pool()
|
|
Party = pool.get('party.party')
|
|
|
|
A, B, = Party.create([{
|
|
'name': 'A',
|
|
}, {
|
|
'name': 'B',
|
|
}])
|
|
|
|
parties = Party.search([])
|
|
self.assertEqual([p.distance for p in parties], [None] * 2)
|
|
|
|
with Transaction().set_context(related_party=A.id):
|
|
parties = Party.search([])
|
|
self.assertEqual(
|
|
[(p.name, p.distance) for p in parties],
|
|
[('A', 0), ('B', None)])
|
|
|
|
|
|
def suite():
|
|
suite = trytond.tests.test_tryton.suite()
|
|
suite.addTests(unittest.TestLoader().loadTestsFromTestCase(PartyTestCase))
|
|
suite.addTests(doctest.DocFileSuite(
|
|
'scenario_party_replace.rst',
|
|
tearDown=doctest_teardown, encoding='utf-8',
|
|
checker=doctest_checker,
|
|
optionflags=doctest.REPORT_ONLY_FIRST_FAILURE))
|
|
suite.addTests(doctest.DocFileSuite(
|
|
'scenario_party_erase.rst',
|
|
tearDown=doctest_teardown, encoding='utf-8',
|
|
checker=doctest_checker,
|
|
optionflags=doctest.REPORT_ONLY_FIRST_FAILURE))
|
|
suite.addTests(doctest.DocFileSuite(
|
|
'scenario_party_phone_number.rst',
|
|
tearDown=doctest_teardown, encoding='utf-8',
|
|
checker=doctest_checker,
|
|
optionflags=doctest.REPORT_ONLY_FIRST_FAILURE))
|
|
return suite
|