# Copyright 2020 VentorTech OU
# License LGPL-3.0 or later (https://www.gnu.org/licenses/lgpl-3.0).

from odoo.tests.common import TransactionCase
from datetime import datetime


class TestMerpOutgoingRouting(TransactionCase):

    def setUp(self):
        super(TestMerpOutgoingRouting, self).setUp()

        self.location_1 = self.env['stock.location'].create({
            'name': 'test_location_1',
            'removal_prio': 2
        })
        self.location_2 = self.env['stock.location'].create({
            'name': 'test_location_2',
            'removal_prio': 3
        })
        self.location_3 = self.env['stock.location'].create({
            'name': 'test_location_3',
            'removal_prio': 1
        })
        self.location_4 = self.env['stock.location'].create({
            'name': 'test_location_4',
            'removal_prio': 4
        })

        company = self.env.user.company_id
        picking_type = self.env['stock.picking.type'].search([], limit=1)
        self.stock_picking = self.env['stock.picking'].create({
            'name': 'test_stock_picking',
            'location_id': self.location_1.id,
            'location_dest_id': self.location_2.id,
            'move_type': 'one',
            'company_id': company.id,
            'picking_type_id': picking_type.id
        })

        products = self.env['product.product']
        for suf in range(1, 5):
            products += self.create_product(suf)

        self.move_line_1 = self.env['stock.move.line'].create({
            'picking_id': self.stock_picking.id,
            'qty_done': 1.0,
            'location_id': self.location_1.id,
            'date': datetime.now(),
            'location_dest_id': self.location_2.id,
            'product_uom_qty': 20.0,
            'product_uom_id': products[1].uom_id.id,
            'product_id': products[1].id
        })
        self.move_line_2 = self.env['stock.move.line'].create({
            'picking_id': self.stock_picking.id,
            'qty_done': 2.0,
            'location_id': self.location_2.id,
            'date': datetime.now(),
            'location_dest_id': self.location_3.id,
            'product_uom_qty': 25.0,
            'product_uom_id': products[0].uom_id.id,
            'product_id': products[0].id
        })
        self.move_line_3 = self.env['stock.move.line'].create({
            'picking_id': self.stock_picking.id,
            'qty_done': 3.0,
            'location_id': self.location_3.id,
            'date': datetime.now(),
            'location_dest_id': self.location_2.id,
            'product_uom_qty': 15.0,
            'product_uom_id': products[2].uom_id.id,
            'product_id': products[2].id
        })
        self.move_line_4 = self.env['stock.move.line'].create({
            'picking_id': self.stock_picking.id,
            'qty_done': 10.0,
            'location_id': self.location_4.id,
            'date': datetime.now(),
            'location_dest_id': self.location_2.id,
            'product_uom_qty': 10.0,
            'product_uom_id': products[3].uom_id.id,
            'product_id': products[3].id
        })

    def create_product(self, suf):
        name = 'test_product_{}'.format(suf)
        product_data = {
            'name': name,
        }
        return self.env['product.product'].create(product_data)

    def test_sort_alphabet_a_z(self):
        self.set_way_outgoing_routing('location_id.name', '0')

        locations_name = self.stock_picking.mapped('operations_to_pick.location_id.name')
        self.assertEqual(locations_name[0], 'test_location_1')
        self.assertEqual(locations_name[1], 'test_location_2')
        self.assertEqual(locations_name[2], 'test_location_3')

    def test_sort_alphabet_z_a(self):
        self.set_way_outgoing_routing('location_id.name', '1')

        locations_name = self.stock_picking.mapped('operations_to_pick.location_id.name')
        self.assertEqual(locations_name[0], 'test_location_3')
        self.assertEqual(locations_name[1], 'test_location_2')
        self.assertEqual(locations_name[2], 'test_location_1')

    def test_sort_removal_priority_a_z(self):
        self.set_way_outgoing_routing('location_id.removal_prio', '0')

        locations_removal_prio = self.stock_picking\
            .mapped('operations_to_pick.location_id.removal_prio')
        self.assertEqual(locations_removal_prio[0], 1)
        self.assertEqual(locations_removal_prio[1], 2)
        self.assertEqual(locations_removal_prio[2], 3)

    def test_sort_removal_priority_z_a(self):
        self.set_way_outgoing_routing('location_id.removal_prio', '1')

        locations_removal_prio = self.stock_picking\
            .mapped('operations_to_pick.location_id.removal_prio')
        self.assertEqual(locations_removal_prio[0], 3)
        self.assertEqual(locations_removal_prio[1], 2)
        self.assertEqual(locations_removal_prio[2], 1)

    def test_sort_by_products_name_a_z(self):
        self.set_way_outgoing_routing('product_id.name', '0')

        products_name = self.stock_picking \
            .mapped('operations_to_pick.product_id.name')
        self.assertEqual(products_name[0], 'test_product_1')
        self.assertEqual(products_name[1], 'test_product_2')
        self.assertEqual(products_name[2], 'test_product_3')

    def test_sort_by_products_name_z_a(self):
        self.set_way_outgoing_routing('product_id.name', '1')

        products_name = self.stock_picking \
            .mapped('operations_to_pick.product_id.name')
        self.assertEqual(products_name[0], 'test_product_3')
        self.assertEqual(products_name[1], 'test_product_2')
        self.assertEqual(products_name[2], 'test_product_1')

    def set_way_outgoing_routing(self, outgoing_routing_strategy, outgoing_routing_order):
        config = self.env['res.config.settings'].create({
            'outgoing_routing_strategy': outgoing_routing_strategy,
            'outgoing_routing_order': outgoing_routing_order
        })
        config.execute()
