Files
intrastat-extrastat/intrastat_base/tests/test_all.py

80 lines
2.7 KiB
Python

from odoo.exceptions import UserError, ValidationError
from odoo.tests.common import SavepointCase
from .common import IntrastatCommon
class TestIntrastatBase(IntrastatCommon):
"""Tests for this module"""
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.declaration_test_obj = cls.env["intrastat.declaration.test"]
cls._load_xml("intrastat_base", "tests/data/mail_template.xml")
cls.mail_template_id = (
"intrastat_base.base_intrastat_product_reminder_email_template"
)
def test_company(self):
# add 'Demo user' to intrastat_remind_user_ids
self.demo_company.write(
{"intrastat_remind_user_ids": [(6, False, [self.demo_user.id])]}
)
# then check if intrastat_email_list contains the email of the user
self.assertEqual(self.demo_company.intrastat_email_list, self.demo_user.email)
def test_no_email(self):
self.demo_user.email = False
with self.assertRaises(ValidationError):
self.demo_company.write(
{"intrastat_remind_user_ids": [(6, False, [self.demo_user.id])]}
)
def test_accessory(self):
with self.assertRaises(ValidationError):
self.shipping_cost.type = "consu"
def test_declaration_no_country(self):
self.demo_company.country_id = False
with self.assertRaises(ValidationError):
self._create_declaration()
self.declaration.flush()
def test_declaration_no_vat(self):
self.demo_company.partner_id.vat = False
with self.assertRaises(UserError):
self._create_declaration()
self.declaration.flush()
self.declaration._check_generate_xml()
def test_declaration_send_mail(self):
self._create_declaration()
mail_before = self.mail_obj.search([])
self.declaration.send_reminder_email(self.mail_template_id)
mail_after = self.mail_obj.search([]) - mail_before
self.assertEqual(0, len(mail_after))
self.demo_company.write(
{"intrastat_remind_user_ids": [(6, False, [self.demo_user.id])]}
)
self.declaration.send_reminder_email(self.mail_template_id)
mail_after = self.mail_obj.search([]) - mail_before
self.assertEqual(1, len(mail_after))
self.assertIn(
mail_after.email_to,
self.demo_user.email,
)
def test_declaration_state(self):
self._create_declaration()
self.declaration.unlink()
self._create_declaration()
self.declaration.state = "done"
with self.assertRaises(UserError):
self.declaration.unlink()
class TestIntrastat(TestIntrastatBase, SavepointCase):
""" Test Intrastat """