"""
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()