115 lines
3.4 KiB
Python
115 lines
3.4 KiB
Python
#!/usr/bin/env 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.
|
|
from __future__ import with_statement
|
|
import sys, os
|
|
DIR = os.path.abspath(os.path.normpath(os.path.join(__file__,
|
|
'..', '..', '..', '..', '..', 'trytond')))
|
|
if os.path.isdir(DIR):
|
|
sys.path.insert(0, os.path.dirname(DIR))
|
|
|
|
import unittest
|
|
import trytond.tests.test_tryton
|
|
from trytond.tests.test_tryton import POOL, DB_NAME, USER, CONTEXT, test_view
|
|
from trytond.transaction import Transaction
|
|
|
|
|
|
class PartyTestCase(unittest.TestCase):
|
|
'''
|
|
Test Party module.
|
|
'''
|
|
|
|
def setUp(self):
|
|
trytond.tests.test_tryton.install_module('party')
|
|
self.category = POOL.get('party.category')
|
|
self.party = POOL.get('party.party')
|
|
self.address = POOL.get('party.address')
|
|
|
|
def test0005views(self):
|
|
'''
|
|
Test views.
|
|
'''
|
|
test_view('party')
|
|
|
|
def test0010category(self):
|
|
'''
|
|
Create category.
|
|
'''
|
|
with Transaction().start(DB_NAME, USER, CONTEXT) as transaction:
|
|
category1_id = self.category.create({
|
|
'name': 'Category 1',
|
|
})
|
|
self.assert_(category1_id)
|
|
transaction.cursor.commit()
|
|
|
|
def test0020category_recursion(self):
|
|
'''
|
|
Test category recursion.
|
|
'''
|
|
with Transaction().start(DB_NAME, USER, CONTEXT) as transaction:
|
|
category1_id = self.category.search([
|
|
('name', '=', 'Category 1'),
|
|
], limit=1)[0]
|
|
|
|
category2_id = self.category.create({
|
|
'name': 'Category 2',
|
|
'parent': category1_id,
|
|
})
|
|
self.assert_(category2_id)
|
|
|
|
self.failUnlessRaises(Exception, self.category.write,
|
|
category1_id, {
|
|
'parent': category2_id,
|
|
})
|
|
|
|
def test0030party(self):
|
|
'''
|
|
Create party.
|
|
'''
|
|
with Transaction().start(DB_NAME, USER, CONTEXT) as transaction:
|
|
party1_id = self.party.create({
|
|
'name': 'Party 1',
|
|
})
|
|
self.assert_(party1_id)
|
|
transaction.cursor.commit()
|
|
|
|
def test0040party_code(self):
|
|
'''
|
|
Test party code constraint.
|
|
'''
|
|
with Transaction().start(DB_NAME, USER, CONTEXT) as transaction:
|
|
party1_id = self.party.search([], limit=1)[0]
|
|
|
|
code = self.party.read(party1_id, ['code'])['code']
|
|
|
|
party2_id = self.party.create({
|
|
'name': 'Party 2',
|
|
})
|
|
|
|
self.failUnlessRaises(Exception, self.party.write,
|
|
party2_id, {
|
|
'code': code,
|
|
})
|
|
|
|
def test0050address(self):
|
|
'''
|
|
Create address.
|
|
'''
|
|
with Transaction().start(DB_NAME, USER, CONTEXT) as transaction:
|
|
party1_id = self.party.search([], limit=1)[0]
|
|
|
|
address1_id = self.address.create({
|
|
'party': party1_id,
|
|
'street': 'St sample, 15',
|
|
'city': 'City',
|
|
})
|
|
self.assert_(party1_id)
|
|
|
|
def suite():
|
|
suite = trytond.tests.test_tryton.suite()
|
|
suite.addTests(unittest.TestLoader().loadTestsFromTestCase(PartyTestCase))
|
|
return suite
|
|
|
|
if __name__ == '__main__':
|
|
unittest.TextTestRunner(verbosity=2).run(suite())
|