mirror of
https://github.com/OCA/contract.git
synced 2025-02-13 17:57:24 +02:00
2cddefbdc6
* Use SavepointCase for making the setup only once for all tests * Make them inheritable, creating a base class with only the setup, so that it can be inherited without the need of executing all tests contained here each time you inherit it, and adding other class in the same module that inherits from the base class that actually performs the tests. * Removed duplicated test method
235 lines
9.6 KiB
Python
235 lines
9.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2016 Tecnativa - Carlos Dauden
|
|
# Copyright 2017 Tecnativa - Pedro M. Baeza
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
|
|
from odoo.exceptions import ValidationError
|
|
from odoo.tests import common
|
|
|
|
|
|
class TestContractBase(common.SavepointCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(TestContractBase, cls).setUpClass()
|
|
cls.partner = cls.env.ref('base.res_partner_2')
|
|
cls.product = cls.env.ref('product.product_product_2')
|
|
cls.product.taxes_id += cls.env['account.tax'].search(
|
|
[('type_tax_use', '=', 'sale')], limit=1)
|
|
cls.product.description_sale = 'Test description sale'
|
|
cls.template_vals = {
|
|
'recurring_rule_type': 'yearly',
|
|
'recurring_interval': 12345,
|
|
'name': 'Test Contract Template',
|
|
}
|
|
cls.template = cls.env['account.analytic.contract'].create(
|
|
cls.template_vals,
|
|
)
|
|
cls.contract = cls.env['account.analytic.account'].create({
|
|
'name': 'Test Contract',
|
|
'partner_id': cls.partner.id,
|
|
'pricelist_id': cls.partner.property_product_pricelist.id,
|
|
'recurring_invoices': True,
|
|
'date_start': '2016-02-15',
|
|
'recurring_next_date': '2016-02-29',
|
|
})
|
|
cls.line_vals = {
|
|
'analytic_account_id': cls.contract.id,
|
|
'product_id': cls.product.id,
|
|
'name': 'Services from #START# to #END#',
|
|
'quantity': 1,
|
|
'uom_id': cls.product.uom_id.id,
|
|
'price_unit': 100,
|
|
'discount': 50,
|
|
}
|
|
cls.acct_line = cls.env['account.analytic.invoice.line'].create(
|
|
cls.line_vals,
|
|
)
|
|
|
|
|
|
class TestContract(TestContractBase):
|
|
def _add_template_line(self, overrides=None):
|
|
if overrides is None:
|
|
overrides = {}
|
|
vals = self.line_vals.copy()
|
|
vals['analytic_account_id'] = self.template.id
|
|
vals.update(overrides)
|
|
return self.env['account.analytic.contract.line'].create(vals)
|
|
|
|
def test_check_discount(self):
|
|
with self.assertRaises(ValidationError):
|
|
self.acct_line.write({'discount': 120})
|
|
|
|
def test_contract(self):
|
|
self.assertAlmostEqual(self.acct_line.price_subtotal, 50.0)
|
|
res = self.acct_line._onchange_product_id()
|
|
self.assertIn('uom_id', res['domain'])
|
|
self.acct_line.price_unit = 100.0
|
|
|
|
self.contract.partner_id = False
|
|
with self.assertRaises(ValidationError):
|
|
self.contract.recurring_create_invoice()
|
|
self.contract.partner_id = self.partner.id
|
|
|
|
self.contract.recurring_create_invoice()
|
|
self.invoice_monthly = self.env['account.invoice'].search(
|
|
[('contract_id', '=', self.contract.id)])
|
|
self.assertTrue(self.invoice_monthly)
|
|
self.assertEqual(self.contract.recurring_next_date, '2016-03-29')
|
|
|
|
self.inv_line = self.invoice_monthly.invoice_line_ids[0]
|
|
self.assertTrue(self.inv_line.invoice_line_tax_ids)
|
|
self.assertAlmostEqual(self.inv_line.price_subtotal, 50.0)
|
|
self.assertEqual(self.contract.partner_id.user_id,
|
|
self.invoice_monthly.user_id)
|
|
|
|
def test_contract_daily(self):
|
|
self.contract.recurring_next_date = '2016-02-29'
|
|
self.contract.recurring_rule_type = 'daily'
|
|
self.contract.pricelist_id = False
|
|
self.contract.cron_recurring_create_invoice()
|
|
invoice_daily = self.env['account.invoice'].search(
|
|
[('contract_id', '=', self.contract.id)])
|
|
self.assertTrue(invoice_daily)
|
|
self.assertEqual(self.contract.recurring_next_date, '2016-03-01')
|
|
|
|
def test_contract_weekly(self):
|
|
self.contract.recurring_next_date = '2016-02-29'
|
|
self.contract.recurring_rule_type = 'weekly'
|
|
self.contract.recurring_invoicing_type = 'post-paid'
|
|
self.contract.recurring_create_invoice()
|
|
invoices_weekly = self.env['account.invoice'].search(
|
|
[('contract_id', '=', self.contract.id)])
|
|
self.assertTrue(invoices_weekly)
|
|
self.assertEqual(
|
|
self.contract.recurring_next_date, '2016-03-07')
|
|
|
|
def test_contract_yearly(self):
|
|
self.contract.recurring_next_date = '2016-02-29'
|
|
self.contract.recurring_rule_type = 'yearly'
|
|
self.contract.recurring_create_invoice()
|
|
invoices_weekly = self.env['account.invoice'].search(
|
|
[('contract_id', '=', self.contract.id)])
|
|
self.assertTrue(invoices_weekly)
|
|
self.assertEqual(
|
|
self.contract.recurring_next_date, '2017-02-28')
|
|
|
|
def test_contract_monthly_lastday(self):
|
|
self.contract.recurring_next_date = '2016-02-29'
|
|
self.contract.recurring_invoicing_type = 'post-paid'
|
|
self.contract.recurring_rule_type = 'monthlylastday'
|
|
self.contract.recurring_create_invoice()
|
|
invoices_monthly_lastday = self.env['account.invoice'].search(
|
|
[('contract_id', '=', self.contract.id)])
|
|
self.assertTrue(invoices_monthly_lastday)
|
|
self.assertEqual(self.contract.recurring_next_date, '2016-03-31')
|
|
|
|
def test_onchange_partner_id(self):
|
|
self.contract._onchange_partner_id()
|
|
self.assertEqual(self.contract.pricelist_id,
|
|
self.contract.partner_id.property_product_pricelist)
|
|
|
|
def test_onchange_recurring_invoices(self):
|
|
self.contract.recurring_next_date = False
|
|
self.contract._onchange_recurring_invoices()
|
|
self.assertEqual(self.contract.recurring_next_date,
|
|
self.contract.date_start)
|
|
|
|
def test_uom(self):
|
|
uom_litre = self.env.ref('product.product_uom_litre')
|
|
self.acct_line.uom_id = uom_litre.id
|
|
self.acct_line._onchange_product_id()
|
|
self.assertEqual(self.acct_line.uom_id,
|
|
self.acct_line.product_id.uom_id)
|
|
|
|
def test_onchange_product_id(self):
|
|
line = self.env['account.analytic.invoice.line'].new()
|
|
res = line._onchange_product_id()
|
|
self.assertFalse(res['domain']['uom_id'])
|
|
|
|
def test_no_pricelist(self):
|
|
self.contract.pricelist_id = False
|
|
self.acct_line.quantity = 2
|
|
self.assertAlmostEqual(self.acct_line.price_subtotal, 100.0)
|
|
|
|
def test_check_journal(self):
|
|
contract_no_journal = self.contract.copy()
|
|
contract_no_journal.journal_id = False
|
|
journal = self.env['account.journal'].search([('type', '=', 'sale')])
|
|
journal.write({'type': 'general'})
|
|
with self.assertRaises(ValidationError):
|
|
contract_no_journal.recurring_create_invoice()
|
|
|
|
def test_onchange_contract_template_id(self):
|
|
"""It should change the contract values to match the template."""
|
|
self.contract.contract_template_id = self.template
|
|
self.contract._onchange_contract_template_id()
|
|
res = {
|
|
'recurring_rule_type': self.contract.recurring_rule_type,
|
|
'recurring_interval': self.contract.recurring_interval,
|
|
}
|
|
del self.template_vals['name']
|
|
self.assertDictEqual(res, self.template_vals)
|
|
|
|
def test_onchange_contract_template_id_lines(self):
|
|
"""It should create invoice lines for the contract lines."""
|
|
|
|
self.acct_line.unlink()
|
|
self.line_vals['analytic_account_id'] = self.template.id
|
|
self.env['account.analytic.contract.line'].create(self.line_vals)
|
|
self.contract.contract_template_id = self.template
|
|
|
|
self.assertFalse(self.contract.recurring_invoice_line_ids,
|
|
'Recurring lines were not removed.')
|
|
|
|
self.contract._onchange_contract_template_id()
|
|
del self.line_vals['analytic_account_id']
|
|
|
|
self.assertEqual(len(self.contract.recurring_invoice_line_ids), 1)
|
|
|
|
for key, value in self.line_vals.items():
|
|
test_value = self.contract.recurring_invoice_line_ids[0][key]
|
|
try:
|
|
test_value = test_value.id
|
|
except AttributeError:
|
|
pass
|
|
self.assertEqual(test_value, value)
|
|
|
|
def test_send_mail_contract(self):
|
|
result = self.contract.action_contract_send()
|
|
self.assertEqual(result['res_model'], 'mail.compose.message')
|
|
|
|
def test_contract_onchange_product_id_domain_blank(self):
|
|
"""It should return a blank UoM domain when no product."""
|
|
line = self.env['account.analytic.contract.line'].new()
|
|
res = line._onchange_product_id()
|
|
self.assertFalse(res['domain']['uom_id'])
|
|
|
|
def test_contract_onchange_product_id_domain(self):
|
|
"""It should return UoM category domain."""
|
|
line = self._add_template_line()
|
|
res = line._onchange_product_id()
|
|
self.assertEqual(
|
|
res['domain']['uom_id'][0],
|
|
('category_id', '=', self.product.uom_id.category_id.id),
|
|
)
|
|
|
|
def test_contract_onchange_product_id_uom(self):
|
|
"""It should update the UoM for the line."""
|
|
line = self._add_template_line(
|
|
{'uom_id': self.env.ref('product.product_uom_litre').id}
|
|
)
|
|
line.product_id.uom_id = self.env.ref('product.product_uom_day').id
|
|
line._onchange_product_id()
|
|
self.assertEqual(line.uom_id,
|
|
line.product_id.uom_id)
|
|
|
|
def test_contract_onchange_product_id_name(self):
|
|
"""It should update the name for the line."""
|
|
line = self._add_template_line()
|
|
line.product_id.description_sale = 'Test'
|
|
line._onchange_product_id()
|
|
self.assertEqual(line.name,
|
|
'\n'.join([line.product_id.name,
|
|
line.product_id.description_sale,
|
|
]))
|