# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.

from datetime import timedelta

from odoo import fields
from odoo.addons.sale_timesheet.tests.common import TestCommonSaleTimesheet
from odoo.tests import tagged


@tagged('-at_install', 'post_install')
class TestInvoiceTimesheet(TestCommonSaleTimesheet):

    def test_timesheet_transfer_sol(self):
        """ Test transfer of timesheet between sale order line + Test cancel invoice
        """
        self.env['ir.config_parameter'].sudo().set_param('sale.invoiced_timesheet', 'approved')

        # create SO
        sale_order = self.env['sale.order'].create({
            'partner_id': self.partner_a.id,
            'partner_invoice_id': self.partner_a.id,
            'partner_shipping_id': self.partner_a.id,
            'pricelist_id': self.company_data['default_pricelist'].id,
        })
        so_line_product_1 = self.env['sale.order.line'].create({
            'product_id': self.product_delivery_timesheet3.id,
            'order_id': sale_order.id,
        })

        # confirm SO
        sale_order.action_confirm()

        task_id = self.env['project.task'].search([('sale_line_id', '=', so_line_product_1.id)])
        project_id = self.env['project.project'].search([('sale_line_id', '=', so_line_product_1.id)])
        self.assertEqual(len(task_id), 1, "On SO confirmation, a task should have been created")
        self.assertEqual(len(project_id), 1, "On SO confirmation, a project should have been created")

        so_line_product_2 = self.env['sale.order.line'].create({
            'product_id': self.product_delivery_timesheet1.id,
            'order_id': sale_order.id,
        })

        so_line_product_3 = self.env['sale.order.line'].create({
            'product_id': self.product_delivery_timesheet2.id,
            'order_id': sale_order.id,
        })

        # let's log some timesheets
        timesheet_1 = self.env['account.analytic.line'].create({
            'date': fields.Date.today() - timedelta(days=1),
            'name': 'Line 1',
            'project_id': project_id.id,
            'task_id': task_id.id,
            'unit_amount': 2,
            'employee_id': self.employee_manager.id,
        })
        timesheet_2 = self.env['account.analytic.line'].create({
            'date': fields.Date.today() - timedelta(days=1),
            'name': 'Line 2',
            'project_id': project_id.id,
            'task_id': task_id.id,
            'unit_amount': 3,
            'employee_id': self.employee_user.id,
        })

        # Validate first timesheet
        timesheet_1.action_validate_timesheet()
        self.assertEqual(so_line_product_1.qty_delivered, 2, "Timesheet 1 is validated, so 2 hours must be delivered")

        # Create an invoice, cancel it
        invoice_1 = sale_order._create_invoices()
        invoice_1.button_cancel()

        self.assertEqual(so_line_product_1.qty_invoiced, 0, "No hours must be invoiced")
        self.assertEqual(timesheet_1.timesheet_invoice_id, invoice_1, "Timesheet 1 is linked to invoice 1")
        self.assertFalse(timesheet_2.timesheet_invoice_id, "Timesheet 2 is not linked to invoice")

        # Create a second invoice for the first timesheet
        invoice_2 = sale_order._create_invoices()

        self.assertEqual(timesheet_1.timesheet_invoice_id, invoice_2, "Timesheet 1 is linked to invoice 2")
        self.assertFalse(timesheet_2.timesheet_invoice_id, "Timesheet 2 is not linked to invoice")
        self.assertEqual(so_line_product_1.qty_invoiced, 2, "2 hours must be invoiced")
        self.assertEqual(so_line_product_1.qty_delivered, 2, "Timesheet 1 is validated, so 2 hours must be delivered")

        timesheet_2.action_validate_timesheet()
        self.assertEqual(so_line_product_1.qty_invoiced, 2, "2 hours must be invoiced (only timsheet 1)")
        self.assertEqual(so_line_product_1.qty_delivered, 5, "Timesheet 1 and 2 are validated, so 5 hours must be delivered")
        self.assertFalse(so_line_product_2.qty_invoiced, "No hours invoiced (as no timsheet linked to this so_line)")
        self.assertFalse(so_line_product_2.qty_delivered, "No hours delivered (as no timsheet linked to this so_line)")

        # Change the SO line on the task
        task_id.sale_line_id = so_line_product_3
        self.assertEqual(timesheet_2.so_line, so_line_product_1, "SO line must remain same for validated Timesheet")

        # Make validated Timesheet to Draft to change SO line
        timesheet_2.action_invalidate_timesheet()

        # Change the SO line on the task
        task_id.sale_line_id = so_line_product_2
        self.assertEqual(timesheet_2.so_line, so_line_product_2, "SO line must change for Timesheet which are Draft")

        timesheet_2.action_validate_timesheet()
        self.assertEqual(so_line_product_1.qty_invoiced, 2, "2 hours must be invoiced (only timsheet 1)")
        self.assertEqual(so_line_product_1.qty_delivered, 2, "Timesheet 1 stay on this so_line (as already invoiced), so 2 hours must be delivered")
        self.assertFalse(so_line_product_2.qty_invoiced, "No hours yet invoiced")
        self.assertEqual(so_line_product_2.qty_delivered, 3, "Timesheet 2 is linked to this new so_line (3 hours delivered)")

        # Create a third invoice for the remaining
        invoice_3 = sale_order._create_invoices()
        self.assertEqual(so_line_product_1.qty_invoiced, 2, "2 hours must be invoiced (only timsheet 1)")
        self.assertEqual(so_line_product_1.qty_delivered, 2, "Timesheet 1 stay on this so_line (as already invoiced), so 2 hours must be delivered")
        self.assertEqual(so_line_product_2.qty_invoiced, 3, "Timesheet 2 is linked to this so_line (3 hours invoiced)")
        self.assertEqual(so_line_product_2.qty_delivered, 3, "Timesheet 2 is linked to this so_line (3 hours delivered)")

        self.assertEqual(timesheet_1.timesheet_invoice_id, invoice_2, "Timesheet 1 is always linked to invoice 2")
        self.assertEqual(timesheet_2.timesheet_invoice_id, invoice_3, "Timesheet 2 is linked to invoice 3")