mail-addons/mail_multi_website/tests/test_mail_model.py

# 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 models, fields


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')))