# Part of Odoo. See LICENSE file for full copyright and licensing details.

from odoo.fields import Command
from odoo.tests import HttpCase, tagged

from odoo.addons.website_sale_subscription.tests.common import WebsiteSaleSubscriptionCommon


@tagged('post_install', '-at_install')
class TestWebsiteSaleSubscriptionConfigurators(HttpCase, WebsiteSaleSubscriptionCommon):

    def test_website_sale_subscription_product_configurator(self):
        optional_product = self._create_product(
            name="Optional product",
            subscription_rule_ids=[
                Command.create({'plan_id': self.plan_week.id, 'fixed_price': 6}),
                Command.create({'plan_id': self.plan_month.id, 'fixed_price': 16}),
            ],
        )
        self._create_product(
            name="Main product",
            optional_product_ids=[Command.set(optional_product.product_tmpl_id.ids)],
            subscription_rule_ids=[
                Command.create({'plan_id': self.plan_week.id, 'fixed_price': 5}),
                Command.create({'plan_id': self.plan_month.id, 'fixed_price': 15}),
            ],
        )
        self.start_tour('/', 'website_sale_subscription_product_configurator')

    def test_website_sale_subscription_combo_configurator(self):
        combo = self.env['product.combo'].create({
            'name': "Test combo",
            'combo_item_ids': [
                Command.create({'product_id': self._create_product().id}),
                Command.create({'product_id': self._create_product().id}),
            ],
        })
        self._create_product(
            name="Combo product",
            type='combo',
            combo_ids=[Command.link(combo.id)],
            subscription_rule_ids=[
                Command.create({'plan_id': self.plan_week.id, 'fixed_price': 5}),
                Command.create({'plan_id': self.plan_month.id, 'fixed_price': 15}),
            ],
        )
        self.start_tour('/', 'website_sale_subscription_combo_configurator')

    @classmethod
    def _create_product(cls, **kwargs):
        if 'recurring_invoice' not in kwargs:
            kwargs['recurring_invoice'] = True
        return super()._create_product(**kwargs)