from odoo.tests.common import TransactionCase


class TestIntrastatBase(TransactionCase):
    """Tests for this module"""

    def test_10_countries(self):
        # check if only EU countries have the 'intrastat' bit set
        france = self.env.ref("base.fr")
        self.assertTrue(france.intrastat)
        brazil = self.env.ref("base.br")
        self.assertFalse(brazil.intrastat)

    def test_20_company(self):
        # add 'Demo user' to intrastat_remind_user_ids
        demo_user = self.env.ref("base.user_demo")
        demo_company = self.env.ref("base.main_company")
        demo_company.write({"intrastat_remind_user_ids": [(6, False, [demo_user.id])]})
        # then check if intrastat_email_list contains the email of the user
        self.assertEqual(demo_company.intrastat_email_list, demo_user.email)