34 lines
777 B
Python
34 lines
777 B
Python
|
"""
|
||
|
|
This is just a script for testing that the dumb TCP server actually works
|
||
|
|
correctly, for verifying that problems with tests are in Vim. Pass the
|
||
|
|
same port number given to the test server to check that it's working.
|
||
|
|
"""
|
||
|
|
import socket
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
|
|
result = sock.connect_ex(('127.0.0.1', int(sys.argv[1])))
|
||
|
|
|
||
|
|
if result:
|
||
|
|
sock.close()
|
||
|
|
sys.exit("Couldn't connect to the socket!")
|
||
|
|
|
||
|
|
data_sent = 'x' * 1024
|
||
|
|
|
||
|
|
sock.send(data_sent)
|
||
|
|
data_received = sock.recv(1024)
|
||
|
|
|
||
|
|
if data_sent != data_received:
|
||
|
|
sock.close()
|
||
|
|
sys.exit("Data sent didn't match data received.")
|
||
|
|
|
||
|
|
sock.close()
|
||
|
|
|
||
|
|
print("Everything was just fine.")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|