mirror of
https://github.com/ZoiteChat/zoitechat.git
synced 2026-10-17 18:37:14 +00:00
86 lines
3.7 KiB
Python
86 lines
3.7 KiB
Python
|
"""Check that visual-test evidence requires a real IRC handshake and reply."""
|
||
|
|
import importlib.util
|
||
|
|
from pathlib import Path
|
||
|
|
import socket
|
||
|
|
import threading
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
spec = importlib.util.spec_from_file_location(
|
||
|
|
"visual_smoke", Path(__file__).resolve().parents[1] / "visual-smoke.py")
|
||
|
|
visual = importlib.util.module_from_spec(spec)
|
||
|
|
spec.loader.exec_module(visual)
|
||
|
|
|
||
|
|
|
||
|
|
class VisualSmokeTests(unittest.TestCase):
|
||
|
|
def test_tls_clienthello_is_rejected_without_waiting_for_a_newline(self):
|
||
|
|
with visual.IRCServer() as server:
|
||
|
|
thread = threading.Thread(target=server.serve_forever, daemon=True)
|
||
|
|
thread.start()
|
||
|
|
try:
|
||
|
|
with socket.create_connection(server.server_address, timeout=3) as client:
|
||
|
|
# A TLS handshake record, with no IRC line terminator.
|
||
|
|
client.sendall(b"\x16\x03\x03\x00\x00")
|
||
|
|
self.assertEqual(client.recv(1), b"")
|
||
|
|
self.assertIn("Received TLS", server.error)
|
||
|
|
self.assertFalse(server.joined.is_set())
|
||
|
|
self.assertFalse(server.pong.is_set())
|
||
|
|
finally:
|
||
|
|
server.shutdown()
|
||
|
|
thread.join(timeout=3)
|
||
|
|
|
||
|
|
def test_registration_join_and_challenge_reply(self):
|
||
|
|
with visual.IRCServer() as server:
|
||
|
|
thread = threading.Thread(target=server.serve_forever, daemon=True)
|
||
|
|
thread.start()
|
||
|
|
try:
|
||
|
|
with socket.create_connection(server.server_address, timeout=3) as client:
|
||
|
|
stream = client.makefile("rb")
|
||
|
|
|
||
|
|
def until(marker):
|
||
|
|
lines = []
|
||
|
|
for _ in range(30):
|
||
|
|
line = stream.readline().decode("utf-8").rstrip()
|
||
|
|
self.assertTrue(line, "server closed before expected response")
|
||
|
|
lines.append(line)
|
||
|
|
if marker in line:
|
||
|
|
return lines
|
||
|
|
self.fail("expected response missing: " + marker)
|
||
|
|
|
||
|
|
client.sendall(b"CAP LS 302\r\nNICK MacTester\r\nUSER ci 0 * :CI\r\n")
|
||
|
|
until(" CAP * LS :")
|
||
|
|
self.assertFalse(server.joined.is_set())
|
||
|
|
client.sendall(b"CAP END\r\n")
|
||
|
|
until(" 376 MacTester ")
|
||
|
|
client.sendall(b"JOIN #ci-smoke\r\n")
|
||
|
|
messages = until("PING :" + server.token)
|
||
|
|
self.assertTrue(any("PRIVMSG #ci-smoke" in line for line in messages))
|
||
|
|
self.assertTrue(server.joined.wait(timeout=1))
|
||
|
|
self.assertFalse(server.pong.is_set())
|
||
|
|
client.sendall(b"PONG :wrong-token\r\nPING :barrier\r\n")
|
||
|
|
until("PONG :barrier")
|
||
|
|
self.assertFalse(server.pong.is_set())
|
||
|
|
client.sendall(("PONG :" + server.token + "\r\n").encode())
|
||
|
|
self.assertTrue(server.pong.wait(timeout=1))
|
||
|
|
client.sendall(b"QUIT\r\n")
|
||
|
|
stream.close()
|
||
|
|
finally:
|
||
|
|
server.shutdown()
|
||
|
|
thread.join(timeout=3)
|
||
|
|
|
||
|
|
def test_early_disconnect_is_not_success(self):
|
||
|
|
with visual.IRCServer() as server:
|
||
|
|
thread = threading.Thread(target=server.serve_forever, daemon=True)
|
||
|
|
thread.start()
|
||
|
|
try:
|
||
|
|
with socket.create_connection(server.server_address, timeout=3) as client:
|
||
|
|
client.sendall(b"QUIT\r\n")
|
||
|
|
self.assertFalse(server.joined.is_set())
|
||
|
|
self.assertFalse(server.pong.is_set())
|
||
|
|
finally:
|
||
|
|
server.shutdown()
|
||
|
|
thread.join(timeout=3)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|