import odoo

from ..export import export
from .util.odoo_tests import TestBase


@odoo.tests.common.at_install(False)
@odoo.tests.common.post_install(True)
class Test(TestBase):
    def test_export_char_field(self):
        record = self.env["res.partner"].create({"name": "my name"})
        self.assertEqual(
            {"name": "my name", "company": None},
            export(
                record, {"name": {}, "company": {"source": "company_name"}}
            ),
        )

    def test_export_integer_field(self):
        record = self.env["res.partner"].create({"name": "my name"})
        self.assertEqual({"color": 0}, export(record, {"color": {}}))
        record.color = 10
        self.assertEqual({"color": 10}, export(record, {"color": {}}))