Kodlo-chan
119 lines
4.0 KiB
Python
119 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
|
"""Тест задачи 07: поднимает сервер, проверяет эхо на нескольких соединениях и SIGTERM."""
|
|
import os
|
|
import signal
|
|
import socket
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
|
|
HERE = os.path.dirname(os.path.abspath(__file__))
|
|
BIN = os.path.join(HERE, "solution")
|
|
failures = 0
|
|
|
|
|
|
def check(cond, name):
|
|
global failures
|
|
if cond:
|
|
print(f"ok {name}")
|
|
else:
|
|
print(f"FAIL {name}")
|
|
failures += 1
|
|
|
|
|
|
def free_port():
|
|
s = socket.socket()
|
|
s.bind(("127.0.0.1", 0))
|
|
p = s.getsockname()[1]
|
|
s.close()
|
|
return p
|
|
|
|
|
|
def main():
|
|
port = free_port()
|
|
proc = subprocess.Popen([BIN, str(port)], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
up = False
|
|
for _ in range(100): # до 5 секунд на старт
|
|
if proc.poll() is not None:
|
|
break
|
|
try:
|
|
s = socket.create_connection(("127.0.0.1", port), timeout=0.3)
|
|
s.close()
|
|
up = True
|
|
break
|
|
except OSError:
|
|
time.sleep(0.05)
|
|
check(up, "сервер принял соединение и слушает порт")
|
|
|
|
if up:
|
|
# три одновременных соединения, большие и разорванные сообщения
|
|
socks = []
|
|
for i in range(3):
|
|
s = socket.create_connection(("127.0.0.1", port), timeout=2)
|
|
s.settimeout(5)
|
|
socks.append(s)
|
|
payloads = [b"hello\n", b"x" * 70000, b"chunk-" + bytes(range(32, 127))]
|
|
for s, p in zip(socks, payloads):
|
|
s.sendall(p)
|
|
ok = True
|
|
for i, (s, p) in enumerate(zip(socks, payloads)):
|
|
got = b""
|
|
while len(got) < len(p):
|
|
chunk = s.recv(65536)
|
|
if not chunk:
|
|
break
|
|
got += chunk
|
|
if got != p:
|
|
ok = False
|
|
print(f" соединение {i}: получено {len(got)} из {len(p)} байт")
|
|
check(ok, "эхо на трёх одновременных соединениях (включая 70 КБ)")
|
|
|
|
# эхо в двух шагах по одному соединению
|
|
s = socks[0]
|
|
s.sendall(b"part1")
|
|
first = s.recv(5)
|
|
s.sendall(b"part2")
|
|
second = b""
|
|
while len(second) < 5:
|
|
chunk = s.recv(5 - len(second))
|
|
if not chunk:
|
|
break
|
|
second += chunk
|
|
check(first == b"part1" and second == b"part2", "эхо по частям (несколько циклов read/write)")
|
|
|
|
# клиент закрывается — сервер не падает
|
|
for s in socks:
|
|
s.close()
|
|
time.sleep(0.3)
|
|
check(proc.poll() is None, "сервер жив после закрытия всех клиентов")
|
|
|
|
# новое соединение после закрытия старых
|
|
try:
|
|
s = socket.create_connection(("127.0.0.1", port), timeout=2)
|
|
s.settimeout(3)
|
|
s.sendall(b"again")
|
|
check(s.recv(5) == b"again", "новое соединение после отключения старых")
|
|
s.close()
|
|
except OSError as e:
|
|
check(False, f"новое соединение после отключения старых ({e})")
|
|
|
|
proc.send_signal(signal.SIGTERM)
|
|
try:
|
|
rc = proc.wait(timeout=5)
|
|
check(rc == 0, f"аккуратное завершение по SIGTERM (exit={rc})")
|
|
except subprocess.TimeoutExpired:
|
|
proc.kill()
|
|
check(False, "аккуратное завершение по SIGTERM (не завершился за 5 с)")
|
|
|
|
if failures:
|
|
err = proc.stderr.read().decode(errors="replace")[:2000]
|
|
if err:
|
|
print("--- stderr сервера ---")
|
|
print(err)
|
|
print(f"\n{'FAILURES: %d' % failures if failures else 'ALL PASS'}")
|
|
sys.exit(1 if failures else 0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|