# Copyright 2018 Ivan Yelizariev <https://it-projects.info/team/yelizariev>
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html).
from odoo import fields, models


class MailTest(models.Model):
    _inherit = "mail.test"

    company_id = fields.Many2one(
        "res.company",
        default=lambda self: self.env["res.company"]._company_default_get(),
    )
    website_id = fields.Many2one(
        "website",
        default=lambda self: self.env["website"].browse(
            self.env.context.get("website_id")
        ),
    )