from .test_bemade_fsm_common import FSMManagerUserTransactionCase
from odoo.tests.common import HttpCase, tagged
from odoo.tools import mute_logger
from odoo.exceptions import MissingError
from odoo import Command
from psycopg2.errors import ForeignKeyViolation


class TestTaskTemplateCommon(FSMManagerUserTransactionCase):

    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        hours_uom = cls.env['uom.uom'].search([('name', '=', 'Hour')]) or False
        # Test product to use with the various tests
        cls.task1 = cls.env['project.task.template'].create({
            'name': 'Template 1',
        })

        cls.project = cls.env['project.project'].create({
            'name': 'Test Project',
        })
        cls.product_task_global_project = cls.env['product.product'].create({
            'name': 'Test Product 1',
            'type': 'service',
            'service_tracking': 'task_global_project',
            'project_id': cls.project.id,
            'task_template_id': cls.task1.id,
            'uom_id': hours_uom.id,
            'uom_po_id': hours_uom.id,
        })
        cls.project_template = cls.env['project.project'].create({
            'name': 'Test Project Template',
        })
        cls.product_task_in_project = cls.env['product.product'].create({
            'name': 'Test Product 2',
            'type': 'service',
            'service_tracking': 'task_in_project',
            'task_template_id': cls.task1.id,
            'project_template_id': cls.project_template.id,
            'uom_po_id': hours_uom.id,
            'uom_id': hours_uom.id,
        })

        # Set up a task template tree with 2 children and 1 grandchild
        cls.parent_task = cls.env['project.task.template'].create({
            'name': 'Parent Template',
        })
        cls.child_task_1 = cls.env['project.task.template'].create({
            'name': 'Child Template 1',
            'parent': cls.parent_task.id,
        })
        cls.child_task_2 = cls.env['project.task.template'].create({
            'name': 'Child Template 2',
            'parent': cls.parent_task.id,
        })
        cls.parent_task.write({'subtasks': [Command.set([cls.child_task_1.id, cls.child_task_2.id])]})
        cls.grandchild_task = cls.env['project.task.template'].create({
            'name': 'Grandchild Template',
            'parent': cls.child_task_2.id
        })
        cls.child_task_2.write({'subtasks': [Command.set([cls.grandchild_task.id])]})

        # Create products using the task tree we just created
        cls.product_task_tree_global_project = cls.env['product.product'].create({
            'name': 'Test Product 3',
            'type': 'service',
            'service_tracking': 'task_global_project',
            'project_id': cls.project.id,
            'task_template_id': cls.parent_task.id,
            'uom_id': hours_uom.id,
            'uom_po_id': hours_uom.id,
        })
        cls.product_task_tree_in_project = cls.env['product.product'].create({
            'name': 'Test Product 2',
            'type': 'service',
            'service_tracking': 'task_in_project',
            'task_template_id': cls.parent_task.id,
            'project_template_id': cls.project_template.id,
            'uom_po_id': hours_uom.id,
            'uom_id': hours_uom.id,
        })


@tagged('-at_install', 'post_install')
class TestTaskTemplate(TestTaskTemplateCommon):

    def test_delete_task_template(self):
        """User should never be able to delete a task template used on a product"""
        with self.assertRaises(ForeignKeyViolation):
            self.task1.unlink()

    def test_delete_subtask_template(self):
        """ Deletion of a child task should be OK even if the parent is on a product. Children of the deleted
        subtask should be deleted."""
        self.child_task_2.unlink()
        # Reading deleted child's name field should be impossible
        with self.assertRaises(MissingError):
            test = self.grandchild_task.name


@tagged('-at_install', 'post_install')
class TestTaskTemplateTour(HttpCase, TestTaskTemplateCommon):

    def test_task_template_tour(self):
        self.start_tour('/web', 'task_template_tour',
                        login='misterpm', )