# Copyright (C) 2011 Julius Network Solutions SARL <contact@julius.fr>
# Copyright 2018 Camptocamp SA
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl)

from odoo.tests import common


class TestsCommon(common.SavepointCase):

    @classmethod
    def setUpClass(cls):
        super(TestsCommon, cls).setUpClass()
        cls.env = cls.env(context=dict(cls.env.context, tracking_disable=True))
        cls.location_obj = cls.env["stock.location"]
        product_obj = cls.env["product.product"]
        cls.wizard_obj = cls.env["wiz.stock.move.location"]
        cls.quant_obj = cls.env["stock.quant"]

        cls.internal_loc_1 = cls.location_obj.create({
            "name": "INT_1",
            "usage": "internal",
            "active": True,
        })
        cls.internal_loc_2 = cls.location_obj.create({
            "name": "INT_2",
            "usage": "internal",
            "active": True,
        })
        cls.uom_unit = cls.env.ref('product.product_uom_unit')
        cls.product_no_lots = product_obj.create({
            "name": "Pineapple",
            "type": "product",
            "tracking": "none",
            'categ_id': cls.env.ref('product.product_category_all').id,
        })
        cls.product_lots = product_obj.create({
            "name": "Pineapple",
            "type": "product",
            "tracking": "lot",
            'categ_id': cls.env.ref('product.product_category_all').id,
        })
        cls.lot1 = cls.env['stock.production.lot'].create({
            'product_id': cls.product_lots.id,
        })
        cls.lot2 = cls.env['stock.production.lot'].create({
            'product_id': cls.product_lots.id,
        })
        cls.lot3 = cls.env['stock.production.lot'].create({
            'product_id': cls.product_lots.id,
        })

    def set_product_amount(self, product, location, amount, lot_id=None):
        self.env['stock.quant']._update_available_quantity(
            product,
            location,
            amount,
            lot_id=lot_id,
        )

    def check_product_amount(self, product, location, amount, lot_id=None):
        self.assertEqual(
            self.env['stock.quant']._get_available_quantity(
                product,
                location,
                lot_id=lot_id,
            ),
            amount,
        )