# -*- coding: utf-8 -*- # Part of Odoo. See LICENSE file for full copyright and licensing details. import time from odoo.tests import Form from odoo.tests.common import tagged, TransactionCase from odoo import fields from odoo.exceptions import ValidationError class TestEquipmentCommon(TransactionCase): def setUp(self): super().setUp() self.equipment = self.env['maintenance.equipment'] self.maintenance_request = self.env['maintenance.request'] self.res_users = self.env['res.users'] self.maintenance_team = self.env['maintenance.team'] self.main_company = self.env.ref('base.main_company') res_user = self.env.ref('base.group_user') res_manager = self.env.ref('maintenance.group_equipment_manager') self.user = self.res_users.create(dict( name="Normal User/Employee", company_id=self.main_company.id, login="emp", email="empuser@yourcompany.example.com", group_ids=[(6, 0, [res_user.id])] )) self.manager = self.res_users.create(dict( name="Equipment Manager", company_id=self.main_company.id, login="hm", email="eqmanager@yourcompany.example.com", group_ids=[(6, 0, [res_manager.id])] )) self.equipment_monitor = self.env['maintenance.equipment.category'].create({ 'name': 'Monitors - Test', }) @tagged('at_install', '-post_install') # LEGACY at_install class TestEquipment(TestEquipmentCommon): def test_10_equipment_request_category(self): # Create a new equipment equipment_01 = self.equipment.with_user(self.manager).create({ 'name': 'Samsung Monitor "15', 'category_id': self.equipment_monitor.id, 'technician_user_id': self.ref('base.user_root'), 'owner_user_id': self.user.id, 'assign_date': time.strftime('%Y-%m-%d'), 'serial_no': 'MT/127/18291015', 'model': 'NP355E5X', 'color': 3, }) # Check that equipment is created or not assert equipment_01, "Equipment not created" # Create new maintenance request maintenance_request_01 = self.maintenance_request.with_user(self.user).create({ 'name': 'Resolution is bad', 'user_id': self.user.id, 'owner_user_id': self.user.id, 'equipment_id': equipment_01.id, 'color': 7, 'stage_id': self.ref('maintenance.stage_0'), 'maintenance_team_id': self.ref('maintenance.equipment_team_maintenance') }) # I check that maintenance_request is created or not assert maintenance_request_01, "Maintenance Request not created" # I check that Initially maintenance request is in the "New Request" stage self.assertEqual(maintenance_request_01.stage_id.id, self.ref('maintenance.stage_0')) # I check that change the maintenance_request stage on click statusbar maintenance_request_01.with_user(self.user).write({'stage_id': self.ref('maintenance.stage_1')}) # I check that maintenance request is in the "In Progress" stage self.assertEqual(maintenance_request_01.stage_id.id, self.ref('maintenance.stage_1')) def test_forever_maintenance_repeat_type(self): """ Test that a maintenance request with repeat_type = forever will be duplicated when it is marked as 'done', and the state of the new request will be set to 'normal'. """ maintenance_request = self.env['maintenance.request'].create({ 'name': 'Test forever maintenance', 'repeat_type': 'forever', 'maintenance_type': 'preventive', 'recurring_maintenance': True, }) maintenance_request.state = 'done' new_maintenance_count = self.env['maintenance.request'].search_count([('name', '=', 'Test forever maintenance'), ('state', '=', 'normal')]) self.assertGreater(new_maintenance_count, 0) @tagged("post_install", "-at_install") class TestEquipmentPostInstall(TestEquipmentCommon): def test_basic_access_and_new_equipment(self): """ Ensure that - a maintenance manager can create an equipment and assign it to a specific user - the user can open it """ equipment_name = "Super Equipment" with self.with_user('hm'): form = Form(self.env['maintenance.equipment']) form.name = equipment_name equipment = form.save() self.assertTrue(equipment) equipment.owner_user_id = self.user with self.with_user('emp'): # Using browse to avoid the env of record `equipment` form = Form(self.env['maintenance.equipment'].browse(equipment.id)) self.assertEqual(form.name, equipment_name) def test_done_maintenance_no_close_or_schedule_date(self): """ Ensure equipment with done maintenance requests that have `close_date` or `schedule_date` set to False can still be opened. In theory this should never happen, but we should fail gracefully in case these dates are forced set to False. """ form = Form(self.env['maintenance.equipment'].with_user(self.manager)) form.name = "brain" equipment = form.save() form = Form(self.env['maintenance.request'].with_user(self.manager)) form.name = "improve efficiency" form.equipment_id = equipment form.maintenance_type = 'corrective' maintenance = form.save() self.assertFalse(maintenance.schedule_date) self.assertFalse(maintenance.close_date) maintenance.state = 'done' self.assertFalse(maintenance.schedule_date) self.assertTrue(maintenance.close_date) form = Form(equipment) # this shouldn't happen unless it's forced maintenance.close_date = False form = Form(equipment) maintenance.close_date = fields.Date.today() form = Form(equipment) maintenance.close_date = False form = Form(equipment) def test_schedule_date_and_end_must_be_set_together(self): request = self.maintenance_request.create({ 'name': 'Valid maintenance request', 'user_id': self.user.id, 'owner_user_id': self.user.id, 'maintenance_team_id': self.ref('maintenance.equipment_team_maintenance'), 'equipment_id': self.env['maintenance.equipment'].create({ 'name': 'Samsung Monitor "17', 'category_id': self.equipment_monitor.id, }).id, 'schedule_date': fields.Datetime.now(), }) with self.assertRaises(ValidationError): self.maintenance_request.create({ 'name': 'Invalid maintenance request', 'user_id': self.user.id, 'owner_user_id': self.user.id, 'maintenance_team_id': self.ref('maintenance.equipment_team_maintenance'), 'equipment_id': request.equipment_id.id, 'schedule_end': fields.Datetime.now(), }) with self.assertRaises(ValidationError): request.write({ 'schedule_date': False, 'schedule_end': fields.Datetime.now(), })