189 lines
8.4 KiB
Python
189 lines
8.4 KiB
Python
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
||
|
|
|
||
|
|
import pytz
|
||
|
|
|
||
|
|
from datetime import datetime, time, timedelta
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from odoo import fields
|
||
|
|
from odoo.tests import common
|
||
|
|
|
||
|
|
from odoo.addons.lunch.tests.common import TestsCommon
|
||
|
|
|
||
|
|
|
||
|
|
class TestSupplier(TestsCommon):
|
||
|
|
def setUp(self):
|
||
|
|
super(TestSupplier, self).setUp()
|
||
|
|
|
||
|
|
self.monday_1am = datetime(2018, 10, 29, 1, 0, 0)
|
||
|
|
self.monday_10am = datetime(2018, 10, 29, 10, 0, 0)
|
||
|
|
self.monday_1pm = datetime(2018, 10, 29, 13, 0, 0)
|
||
|
|
self.monday_8pm = datetime(2018, 10, 29, 20, 0, 0)
|
||
|
|
|
||
|
|
self.saturday_3am = datetime(2018, 11, 3, 3, 0, 0)
|
||
|
|
self.saturday_10am = datetime(2018, 11, 3, 10, 0, 0)
|
||
|
|
self.saturday_1pm = datetime(2018, 11, 3, 13, 0, 0)
|
||
|
|
self.saturday_8pm = datetime(2018, 11, 3, 20, 0, 0)
|
||
|
|
|
||
|
|
@common.users('cle-lunch-manager')
|
||
|
|
def test_send_email_cron(self):
|
||
|
|
self.supplier_kothai.cron_id.ensure_one()
|
||
|
|
self.assertEqual(self.supplier_kothai.cron_id.nextcall.time(), time(15, 0))
|
||
|
|
self.assertEqual(self.supplier_kothai.cron_id.code, f"""\
|
||
|
|
# This cron is dynamically controlled by Lunch Supplier.
|
||
|
|
# Do NOT modify this cron, modify the related record instead.
|
||
|
|
env['lunch.supplier'].browse([{self.supplier_kothai.id}])._send_auto_email()""")
|
||
|
|
|
||
|
|
cron_id = self.supplier_kothai.cron_id.id
|
||
|
|
self.supplier_kothai.unlink()
|
||
|
|
self.assertFalse(self.env['ir.cron'].sudo().search([('id', '=', cron_id)]))
|
||
|
|
|
||
|
|
@common.users('cle-lunch-manager')
|
||
|
|
def test_compute_available_today(self):
|
||
|
|
tests = [(self.monday_1am, True), (self.monday_10am, True),
|
||
|
|
(self.monday_1pm, True), (self.monday_8pm, True),
|
||
|
|
(self.saturday_3am, False), (self.saturday_10am, False),
|
||
|
|
(self.saturday_1pm, False), (self.saturday_8pm, False)]
|
||
|
|
|
||
|
|
for value, result in tests:
|
||
|
|
with patch.object(fields.Datetime, 'now', return_value=value) as _:
|
||
|
|
assert self.supplier_pizza_inn.available_today == result,\
|
||
|
|
'supplier pizza inn should %s considered available on %s' % ('be' if result else 'not be', value)
|
||
|
|
|
||
|
|
self.supplier_pizza_inn.invalidate_recordset(['available_today'])
|
||
|
|
|
||
|
|
@common.users('cle-lunch-manager')
|
||
|
|
def test_search_available_today(self):
|
||
|
|
'''
|
||
|
|
This test checks that _search_available_today returns a valid domain
|
||
|
|
'''
|
||
|
|
self.env.user.tz = 'Europe/Brussels'
|
||
|
|
Supplier = self.env['lunch.supplier']
|
||
|
|
|
||
|
|
tests = [(self.monday_1am, 1.0, 'mon'), (self.monday_10am, 10.0, 'mon'),
|
||
|
|
(self.monday_1pm, 13.0, 'mon'), (self.monday_8pm, 20.0, 'mon'),
|
||
|
|
(self.saturday_3am, 3.0, 'sat'), (self.saturday_10am, 10.0, 'sat'),
|
||
|
|
(self.saturday_1pm, 13.0, 'sat'), (self.saturday_8pm, 20.0, 'sat')]
|
||
|
|
|
||
|
|
# It should return an empty domain if we compare to values other than datetime
|
||
|
|
assert Supplier._search_available_today('>', 7) == []
|
||
|
|
assert Supplier._search_available_today('>', True) == []
|
||
|
|
|
||
|
|
for value, rvalue, dayname in tests:
|
||
|
|
with patch.object(fields.Datetime, 'now', return_value=value) as _:
|
||
|
|
assert Supplier._search_available_today('=', True) == ['&', '|', ('recurrency_end_date', '=', False),
|
||
|
|
('recurrency_end_date', '>', value.replace(tzinfo=pytz.UTC).astimezone(pytz.timezone(self.env.user.tz))),
|
||
|
|
(dayname, '=', True)],\
|
||
|
|
'Wrong domain generated for values (%s, %s)' % (value, rvalue)
|
||
|
|
|
||
|
|
with patch.object(fields.Datetime, 'now', return_value=self.monday_10am) as _:
|
||
|
|
assert self.supplier_pizza_inn in Supplier.search([('available_today', '=', True)])
|
||
|
|
|
||
|
|
@common.users('cle-lunch-manager')
|
||
|
|
def test_auto_email_send(self):
|
||
|
|
with patch.object(fields.Datetime, 'now', return_value=self.monday_1pm) as _:
|
||
|
|
with patch.object(fields.Date, 'today', return_value=self.monday_1pm.date()) as _:
|
||
|
|
with patch.object(fields.Date, 'context_today', return_value=self.monday_1pm.date()) as _:
|
||
|
|
line = self.env['lunch.order'].create({
|
||
|
|
'product_id': self.product_pizza.id,
|
||
|
|
'date': self.monday_1pm.date(),
|
||
|
|
'supplier_id': self.supplier_pizza_inn.id,
|
||
|
|
})
|
||
|
|
|
||
|
|
line.action_order()
|
||
|
|
assert line.state == 'ordered'
|
||
|
|
|
||
|
|
self.supplier_pizza_inn._send_auto_email()
|
||
|
|
|
||
|
|
assert line.state == 'sent'
|
||
|
|
|
||
|
|
line = self.env['lunch.order'].create({
|
||
|
|
'product_id': self.product_pizza.id,
|
||
|
|
'topping_ids_1': [(6, 0, [self.topping_olives.id])],
|
||
|
|
'date': self.monday_1pm.date(),
|
||
|
|
'supplier_id': self.supplier_pizza_inn.id,
|
||
|
|
})
|
||
|
|
line2 = self.env['lunch.order'].create({
|
||
|
|
'product_id': self.product_sandwich_tuna.id,
|
||
|
|
'date': self.monday_1pm.date(),
|
||
|
|
'supplier_id': self.supplier_coin_gourmand.id,
|
||
|
|
})
|
||
|
|
|
||
|
|
(line | line2).action_order()
|
||
|
|
assert line.state == 'ordered'
|
||
|
|
assert line2.state == 'ordered'
|
||
|
|
|
||
|
|
self.supplier_pizza_inn._send_auto_email()
|
||
|
|
|
||
|
|
assert line.state == 'sent'
|
||
|
|
assert line2.state == 'ordered'
|
||
|
|
|
||
|
|
line_1 = self.env['lunch.order'].create({
|
||
|
|
'product_id': self.product_pizza.id,
|
||
|
|
'quantity': 2,
|
||
|
|
'date': self.monday_1pm.date(),
|
||
|
|
'supplier_id': self.supplier_pizza_inn.id,
|
||
|
|
})
|
||
|
|
|
||
|
|
line_2 = self.env['lunch.order'].create({
|
||
|
|
'product_id': self.product_pizza.id,
|
||
|
|
'topping_ids_1': [(6, 0, [self.topping_olives.id])],
|
||
|
|
'date': self.monday_1pm.date(),
|
||
|
|
'supplier_id': self.supplier_pizza_inn.id,
|
||
|
|
})
|
||
|
|
|
||
|
|
line_3 = self.env['lunch.order'].create({
|
||
|
|
'product_id': self.product_sandwich_tuna.id,
|
||
|
|
'quantity': 2,
|
||
|
|
'date': self.monday_1pm.date(),
|
||
|
|
'supplier_id': self.supplier_coin_gourmand.id,
|
||
|
|
})
|
||
|
|
|
||
|
|
(line_1 | line_2 | line_3).action_order()
|
||
|
|
|
||
|
|
assert all(line.state == 'ordered' for line in [line_1, line_2, line_3])
|
||
|
|
|
||
|
|
self.supplier_pizza_inn._send_auto_email()
|
||
|
|
|
||
|
|
@common.users('cle-lunch-manager')
|
||
|
|
def test_cron_sync_create(self):
|
||
|
|
cron_ny = self.supplier_kothai.cron_id # I am at New-York
|
||
|
|
self.assertTrue(cron_ny.active)
|
||
|
|
self.assertEqual(cron_ny.name, "Lunch: send automatic email to Kothai")
|
||
|
|
self.assertEqual(
|
||
|
|
[line for line in cron_ny.code.splitlines() if not line.lstrip().startswith("#")],
|
||
|
|
["env['lunch.supplier'].browse([%i])._send_auto_email()" % self.supplier_kothai.id])
|
||
|
|
self.assertEqual(cron_ny.nextcall, datetime(2021, 1, 29, 15, 0)) # New-york is UTC-5
|
||
|
|
|
||
|
|
@common.users('cle-lunch-manager')
|
||
|
|
def test_cron_sync_active(self):
|
||
|
|
cron_ny = self.supplier_kothai.cron_id
|
||
|
|
|
||
|
|
self.supplier_kothai.active = False
|
||
|
|
self.assertFalse(cron_ny.active)
|
||
|
|
self.supplier_kothai.active = True
|
||
|
|
self.assertTrue(cron_ny.active)
|
||
|
|
|
||
|
|
self.supplier_kothai.send_by = 'phone'
|
||
|
|
self.assertFalse(cron_ny.active)
|
||
|
|
self.supplier_kothai.send_by = 'mail'
|
||
|
|
self.assertTrue(cron_ny.active)
|
||
|
|
|
||
|
|
@common.users('cle-lunch-manager')
|
||
|
|
def test_cron_sync_nextcall(self):
|
||
|
|
cron_ny = self.supplier_kothai.cron_id
|
||
|
|
old_nextcall = cron_ny.nextcall
|
||
|
|
|
||
|
|
self.supplier_kothai.automatic_email_time -= 5
|
||
|
|
self.assertEqual(cron_ny.nextcall, old_nextcall - timedelta(hours=5) + timedelta(days=1))
|
||
|
|
|
||
|
|
# Simulate cron execution
|
||
|
|
cron_ny.sudo().lastcall = old_nextcall - timedelta(hours=5)
|
||
|
|
cron_ny.sudo().nextcall += timedelta(days=1)
|
||
|
|
|
||
|
|
self.supplier_kothai.automatic_email_time += 7
|
||
|
|
self.assertEqual(cron_ny.nextcall, old_nextcall + timedelta(days=1, hours=2))
|
||
|
|
|
||
|
|
self.supplier_kothai.automatic_email_time -= 1
|
||
|
|
self.assertEqual(cron_ny.nextcall, old_nextcall + timedelta(days=1, hours=1))
|