Disroot
/
gpg-lacre
9
15
Fork 4
Code Issues 29 Pull Requests 1 Releases 2 Wiki Activity

Make test relay slightly cleaner

This commit is contained in:
Piotr F. Mieszkowski 2023-03-16 22:30:59 +01:00
parent b2bd6a9926
commit 12b7c3394d
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test/utils/relay.py
View File

@ -41,6 +41,9 @@ def receive_and_confirm(session):
session.recv(BUFFER_SIZE)
session.sendall(ok())
def receive_and_ignore(session):
session.recv(BUFFER_SIZE)
def localhost_at(port):
return ('127.0.0.1', port)
@ -65,8 +68,7 @@ def serve(port) -> bytes:
receive_and_confirm(conn) # Ignore sender address
receive_and_confirm(conn) # Ignore recipient address
b = conn.recv(BUFFER_SIZE)
logging.debug('Got data: %s', repr(b))
receive_and_ignore(conn)
conn.sendall(provide_message())
# Consume until we get <CR><LF>.<CR><LF>, the end-of-message marker.