mirror of
https://github.com/morpheus65535/bazarr.git
synced 2024-11-10 17:13:35 +08:00
139 lines
3.9 KiB
Python
139 lines
3.9 KiB
Python
import os
|
|
import unittest
|
|
import logging
|
|
import time
|
|
import uuid
|
|
import threading
|
|
|
|
from subprocess import Popen, PIPE
|
|
from signalrcore.hub_connection_builder import HubConnectionBuilder
|
|
from signalrcore.hub.errors import HubConnectionError
|
|
from test.base_test_case import BaseTestCase, Urls
|
|
from signalrcore.transport.websockets.reconnection import RawReconnectionHandler, IntervalReconnectionHandler
|
|
|
|
|
|
class TestReconnectMethods(BaseTestCase):
|
|
|
|
def receive_message(self, args):
|
|
self.assertEqual(args[1], self.message)
|
|
self.received = True
|
|
|
|
def test_reconnect_interval_config(self):
|
|
connection = HubConnectionBuilder()\
|
|
.with_url(self.server_url, options={"verify_ssl": False})\
|
|
.configure_logging(logging.ERROR)\
|
|
.with_automatic_reconnect({
|
|
"type": "interval",
|
|
"intervals": [1, 2, 4, 45, 6, 7, 8, 9, 10]
|
|
})\
|
|
.build()
|
|
_lock = threading.Lock()
|
|
connection.on_open(lambda: _lock.release())
|
|
connection.on_close(lambda: _lock.release())
|
|
|
|
self.assertTrue(_lock.acquire(timeout=10))
|
|
|
|
connection.start()
|
|
|
|
self.assertTrue(_lock.acquire(timeout=10))
|
|
|
|
connection.stop()
|
|
|
|
self.assertTrue(_lock.acquire(timeout=10))
|
|
|
|
del _lock
|
|
|
|
def test_reconnect_interval(self):
|
|
connection = HubConnectionBuilder()\
|
|
.with_url(self.server_url, options={"verify_ssl": False})\
|
|
.configure_logging(logging.ERROR)\
|
|
.with_automatic_reconnect({
|
|
"type": "interval",
|
|
"intervals": [1, 2, 4, 45, 6, 7, 8, 9, 10],
|
|
"keep_alive_interval": 3
|
|
})\
|
|
.build()
|
|
self.reconnect_test(connection)
|
|
|
|
def test_no_reconnect(self):
|
|
connection = HubConnectionBuilder()\
|
|
.with_url(self.server_url, options={"verify_ssl": False})\
|
|
.configure_logging(logging.ERROR)\
|
|
.build()
|
|
|
|
_lock = threading.Lock()
|
|
|
|
_lock.acquire(timeout=10)
|
|
|
|
connection.on_open(lambda: _lock.release())
|
|
|
|
connection.on("ReceiveMessage", lambda _: _lock.release())
|
|
|
|
connection.start()
|
|
|
|
self.assertTrue(_lock.acquire(timeout=10)) # Released on ReOpen
|
|
|
|
connection.send("DisconnectMe", [])
|
|
|
|
self.assertTrue(_lock.acquire(timeout=10))
|
|
|
|
time.sleep(10)
|
|
|
|
self.assertRaises(
|
|
HubConnectionError,
|
|
lambda: connection.send("DisconnectMe", []))
|
|
|
|
connection.stop()
|
|
del _lock
|
|
|
|
def reconnect_test(self, connection):
|
|
_lock = threading.RLock()
|
|
|
|
connection.on_open(lambda: _lock.release())
|
|
|
|
connection.start()
|
|
|
|
self.assertTrue(_lock.acquire(timeout=10)) # Release on Open
|
|
|
|
connection.send("DisconnectMe", [])
|
|
|
|
self.assertTrue(_lock.acquire(timeout=10)) # released on open
|
|
|
|
connection.stop()
|
|
del _lock
|
|
|
|
def test_raw_reconnection(self):
|
|
connection = HubConnectionBuilder()\
|
|
.with_url(self.server_url, options={"verify_ssl": False})\
|
|
.configure_logging(logging.ERROR)\
|
|
.with_automatic_reconnect({
|
|
"type": "raw",
|
|
"keep_alive_interval": 10,
|
|
"max_attempts": 4
|
|
})\
|
|
.build()
|
|
|
|
self.reconnect_test(connection)
|
|
|
|
def test_raw_handler(self):
|
|
handler = RawReconnectionHandler(5, 10)
|
|
attemp = 0
|
|
|
|
while attemp <= 10:
|
|
self.assertEqual(handler.next(), 5)
|
|
attemp = attemp + 1
|
|
|
|
self.assertRaises(ValueError, handler.next)
|
|
|
|
def test_interval_handler(self):
|
|
intervals = [1, 2, 4, 5, 6]
|
|
handler = IntervalReconnectionHandler(intervals)
|
|
for interval in intervals:
|
|
self.assertEqual(handler.next(), interval)
|
|
self.assertRaises(ValueError, handler.next)
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def setUp(self):
|
|
pass
|