46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
||
|
from odoo.tests import common, new_test_user
|
||
|
from odoo import fields
|
||
|
|
||
|
|
||
|
class TestFleet(common.TransactionCase):
|
||
|
|
||
|
def test_search_renewal(self):
|
||
|
"""
|
||
|
Should find the car with overdue contract or renewal due soon
|
||
|
"""
|
||
|
user = new_test_user(self.env, "test base user", groups="base.group_user")
|
||
|
brand = self.env["fleet.vehicle.model.brand"].create({
|
||
|
"name": "Audi",
|
||
|
})
|
||
|
model = self.env["fleet.vehicle.model"].create({
|
||
|
"brand_id": brand.id,
|
||
|
"name": "A3",
|
||
|
})
|
||
|
car_1 = self.env["fleet.vehicle"].create({
|
||
|
"model_id": model.id,
|
||
|
"driver_id": user.partner_id.id,
|
||
|
"plan_to_change_car": False
|
||
|
})
|
||
|
|
||
|
car_2 = self.env["fleet.vehicle"].create({
|
||
|
"model_id": model.id,
|
||
|
"driver_id": user.partner_id.id,
|
||
|
"plan_to_change_car": False
|
||
|
})
|
||
|
Log = self.env['fleet.vehicle.log.contract']
|
||
|
Log.create({
|
||
|
'vehicle_id': car_2.id,
|
||
|
'expiration_date': fields.Date.add(fields.Date.today(), days=10)
|
||
|
})
|
||
|
res = self.env["fleet.vehicle"].search([('contract_renewal_due_soon', '=', True), ('id', '=', car_2.id)])
|
||
|
self.assertEqual(res, car_2)
|
||
|
|
||
|
Log.create({
|
||
|
'vehicle_id': car_1.id,
|
||
|
'expiration_date': fields.Date.add(fields.Date.today(), days=-10)
|
||
|
})
|
||
|
res = self.env["fleet.vehicle"].search([('contract_renewal_overdue', '=', True), ('id', '=', car_1.id)])
|
||
|
self.assertEqual(res, car_1)
|