74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
import json
|
|
from odoo.tests import HttpCase
|
|
from unittest.mock import patch
|
|
|
|
from odoo.addons.iot.models.iot_channel import IotChannel
|
|
|
|
|
|
class IotCommonTest(HttpCase):
|
|
iot_websocket_messages = []
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().setUpClass()
|
|
cls.shop_iot_box = cls.env['iot.box'].sudo().create({
|
|
'name': 'Shop',
|
|
'identifier': 'test_iot_box',
|
|
'ip': '10.10.10.10',
|
|
'version': '25.07',
|
|
})
|
|
|
|
cls.iot_receipt_printer = cls.env['iot.device'].sudo().create({
|
|
'name': 'Receipt Printer',
|
|
'identifier': 'printer_identifier',
|
|
'iot_id': cls.shop_iot_box.id,
|
|
'type': 'printer',
|
|
'subtype': 'receipt_printer',
|
|
'connection': 'network',
|
|
'connected_status': 'connected',
|
|
})
|
|
|
|
cls.iot_payment_terminal = cls.env['iot.device'].sudo().create({
|
|
'name': 'Payment Terminal',
|
|
'identifier': 'payment_terminal_identifier',
|
|
'iot_id': cls.shop_iot_box.id,
|
|
'type': 'payment',
|
|
'connection': 'network',
|
|
'connected_status': 'connected',
|
|
})
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
original_send_message = IotChannel.send_message
|
|
|
|
def mock_send_message(iot_channel_record, message, message_type='iot_action'):
|
|
self.iot_websocket_messages.append({message_type: message})
|
|
if message_type == 'iot_action':
|
|
# call the websocket response controller to simulate the response from the IoT Box
|
|
return self.url_open(
|
|
'/iot/box/send_websocket',
|
|
headers={'Content-Type': 'application/json'},
|
|
data=json.dumps({
|
|
'params': {
|
|
'session_id': message['session_id'],
|
|
'iot_box_identifier': message['iot_identifiers'][0],
|
|
'device_identifier': message['device_identifiers'][0],
|
|
'status': 'success',
|
|
},
|
|
}),
|
|
)
|
|
return original_send_message(iot_channel_record, message, message_type)
|
|
|
|
def mock_get_iot_channel(_iot_channel_record):
|
|
return "mock_iot_channel"
|
|
|
|
mock_get_iot_channel._api_model = True
|
|
mock_send_message._api_model = True
|
|
|
|
send_message_patcher = patch.object(IotChannel, 'send_message', mock_send_message)
|
|
channel_patcher = patch.object(IotChannel, 'get_iot_channel', mock_get_iot_channel)
|
|
self.addCleanup(send_message_patcher.stop)
|
|
self.addCleanup(channel_patcher.stop)
|
|
send_message_patcher.start()
|
|
channel_patcher.start()
|