67 lines
2.7 KiB
Python
67 lines
2.7 KiB
Python
# Part of Talisman . See LICENSE file for full copyright and licensing details.
|
|
|
|
import json
|
|
import time
|
|
|
|
try:
|
|
from websocket._exceptions import WebSocketProtocolException
|
|
except ImportError:
|
|
pass
|
|
|
|
from odoo.tests import common
|
|
from .common import WebsocketCase
|
|
from ..websocket import CloseCode, Websocket
|
|
|
|
@common.tagged('post_install', '-at_install')
|
|
class TestWebsocketRateLimiting(WebsocketCase):
|
|
def test_rate_limiting_base_ok(self):
|
|
ws = self.websocket_connect()
|
|
|
|
for _ in range(Websocket.RL_BURST + 1):
|
|
ws.send(json.dumps({'event_name': 'test_rate_limiting'}))
|
|
time.sleep(Websocket.RL_DELAY)
|
|
|
|
def test_rate_limiting_base_ko(self):
|
|
ws = self.websocket_connect()
|
|
|
|
# Websocket client's close codes are not up to date. Indeed, the
|
|
# 1013 close code results in a protocol exception while it is a
|
|
# valid, registered close code ("TRY LATER") :
|
|
# https://www.iana.org/assignments/websocket/websocket.xhtml
|
|
with self.assertRaises(WebSocketProtocolException) as cm:
|
|
for _ in range(Websocket.RL_BURST + 1):
|
|
ws.send(json.dumps({'event_name': 'test_rate_limiting'}))
|
|
self.assert_close_with_code(ws, CloseCode.TRY_LATER)
|
|
self.assertEqual(str(cm.exception), 'Invalid close opcode.')
|
|
|
|
def test_rate_limiting_opening_burst(self):
|
|
ws = self.websocket_connect()
|
|
|
|
# first RL_BURST requests are accepted.
|
|
for _ in range(Websocket.RL_BURST):
|
|
ws.send(json.dumps({'event_name': 'test_rate_limiting'}))
|
|
|
|
# sending at a correct rate after burst should be accepted.
|
|
for _ in range(2):
|
|
time.sleep(Websocket.RL_DELAY)
|
|
ws.send(json.dumps({'event_name': 'test_rate_limiting'}))
|
|
|
|
def test_rate_limiting_start_ok_end_ko(self):
|
|
ws = self.websocket_connect()
|
|
|
|
# first requests are legit and should be accepted
|
|
for _ in range(Websocket.RL_BURST + 1):
|
|
ws.send(json.dumps({'event_name': 'test_rate_limiting'}))
|
|
time.sleep(Websocket.RL_DELAY)
|
|
|
|
# Websocket client's close codes are not up to date. Indeed, the
|
|
# 1013 close code results in a protocol exception while it is a
|
|
# valid, registered close code ("TRY LATER") :
|
|
# https://www.iana.org/assignments/websocket/websocket.xhtml
|
|
with self.assertRaises(WebSocketProtocolException) as cm:
|
|
# those requests are illicit and should not be accepted.
|
|
for _ in range(Websocket.RL_BURST * 2):
|
|
ws.send(json.dumps({'event_name': 'test_rate_limiting'}))
|
|
self.assert_close_with_code(ws, CloseCode.TRY_LATER)
|
|
self.assertEqual(str(cm.exception), 'Invalid close opcode.')
|