# Copyright 2017 Camptocamp SA
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html)

from odoo.addons.component.core import ComponentRegistry, WorkContext

from .common import SavepointComponentRegistryCase


class TestWorkOn(SavepointComponentRegistryCase):
    """Test on WorkContext

    This model is mostly a container, so we check the access
    to the attributes and properties.

    """

    def test_collection_work_on(self):
        """Create a new instance and test attributes access"""
        collection_record = self.collection.new()
        with collection_record.work_on("res.partner") as work:
            self.assertEqual(collection_record, work.collection)
            self.assertEqual("collection.base", work.collection._name)
            self.assertEqual("res.partner", work.model_name)
            self.assertEqual(self.env["res.partner"], work.model)
            self.assertEqual(self.env, work.env)

    def test_collection_work_on_registry_via_context(self):
        """Test propagation of registry via context"""
        registry = ComponentRegistry()
        collection_record = self.collection.with_context(
            components_registry=registry
        ).new()
        with collection_record.work_on("res.partner") as work:
            self.assertEqual(collection_record, work.collection)
            self.assertEqual("collection.base", work.collection._name)
            self.assertEqual("res.partner", work.model_name)
            self.assertEqual(self.env["res.partner"], work.model)
            self.assertEqual(work.env, collection_record.env)
            self.assertEqual(work.components_registry, registry)

    def test_propagate_work_on(self):
        """Check custom attributes and their propagation"""
        registry = ComponentRegistry()
        work = WorkContext(
            model_name="res.partner",
            collection=self.collection,
            # we can customize the lookup registry, but used mostly for tests
            components_registry=registry,
            # we can pass our own keyword args that will set as attributes
            test_keyword="value",
        )
        self.assertIs(registry, work.components_registry)
        # check that our custom keyword is set as attribute
        self.assertEqual("value", work.test_keyword)

        # when we want to work on another model, work_on() create
        # another instance and propagate the attributes to it
        work2 = work.work_on("res.users")
        self.assertNotEqual(work, work2)
        self.assertEqual(self.env, work2.env)
        self.assertEqual(self.collection, work2.collection)
        self.assertEqual("res.users", work2.model_name)
        self.assertIs(registry, work2.components_registry)
        # test_keyword has been propagated to the new WorkContext instance
        self.assertEqual("value", work2.test_keyword)