# -*-coding:utf-8-*-
from odoo.tests.common import TransactionCase


class TestEquipmentMaintenanceStandards(TransactionCase):
    def setUp(self, *args, **kwargs):
        result = super().setUp(*args, **kwargs)

        # Define global variables to be used in the test cases
        self.user = self.env.ref('base.user_admin')
        self.equipment_category = self.env['maintenance.equipment.category'].create({
            'name': 'Test Equipment Category'
        })
        self.equipment = self.env['maintenance.equipment'].create({
            'name': 'Test Equipment',
            'category_id': self.equipment_category.id,
        })

        return result

    def test_create_equipment_maintenance_standard(self):
        """Test creating a new equipment maintenance standard"""
        maintenance_standard = self.env['equipment.maintenance.standards'].create({
            'name': 'Test Maintenance Standard',
            'maintenance_type': '保养',
            'maintenance_equipment_category_id': self.equipment_category.id,
            'maintenance_equipment_ids': [(6, 0, [self.equipment.id])],
            'maintenance_projects': 'Test Maintenance Project',
            'maintenance_standards': 'Test Maintenance Standard',
        })
        self.assertTrue(maintenance_standard)

    def test_maintenance_standard_code(self):
        """Test that the maintenance standard code is generated correctly"""
        maintenance_standard_1 = self.env['equipment.maintenance.standards'].create({
            'name': 'Test Maintenance Standard 1',
            'maintenance_type': '保养',
            'maintenance_equipment_category_id': self.equipment_category.id,
            'maintenance_equipment_ids': [(6, 0, [self.equipment.id])],
            'maintenance_projects': 'Test Maintenance Project',
            'maintenance_standards': 'Test Maintenance Standard',
        })
        maintenance_standard_2 = self.env['equipment.maintenance.standards'].create({
            'name': 'Test Maintenance Standard 2',
            'maintenance_type': '保养',
            'maintenance_equipment_category_id': self.equipment_category.id,
            'maintenance_equipment_ids': [(6, 0, [self.equipment.id])],
            'maintenance_projects': 'Test Maintenance Project',
            'maintenance_standards': 'Test Maintenance Standard',
        })
        self.assertEqual(maintenance_standard_1.code, '0001')
        self.assertEqual(maintenance_standard_2.code, '0002')

    def test_maintenance_standard_fields(self):
        """Test that the maintenance standard fields are correctly set"""
        maintenance_standard = self.env['equipment.maintenance.standards'].create({
            'name': 'Test Maintenance Standard',
            'maintenance_type': '保养',
            'maintenance_equipment_category_id': self.equipment_category.id,
            'maintenance_equipment_ids': [(6, 0, [self.equipment.id])],
            'maintenance_projects': 'Test Maintenance Project',
            'maintenance_standards': 'Test Maintenance Standard',
        })
        self.assertEqual(maintenance_standard.name, 'Test Maintenance Standard')
        self.assertEqual(maintenance_standard.maintenance_type, '保养')
        self.assertEqual(maintenance_standard.maintenance_equipment_category_id.id, self.equipment_category.id)
        self.assertIn(self.equipment, maintenance_standard.maintenance_equipment_ids)
        self.assertEqual(maintenance_standard.maintenance_projects, 'Test Maintenance Project')
        self.assertEqual(maintenance_standard.maintenance_standards, 'Test Maintenance Standard')