summaryrefslogtreecommitdiff
path: root/test_client.py
diff options
context:
space:
mode:
Diffstat (limited to 'test_client.py')
-rw-r--r--test_client.py34
1 files changed, 34 insertions, 0 deletions
diff --git a/test_client.py b/test_client.py
new file mode 100644
index 0000000..7b29101
--- /dev/null
+++ b/test_client.py
@@ -0,0 +1,34 @@
+import enet
+import os
+
+SHUTDOWN_MSG = b"SHUTDOWN"
+MSG_NUMBER = 10
+
+host = enet.Host(None, 1, 0, 0, 0)
+peer = host.connect(enet.Address(b"localhost", 54301), 1)
+
+counter = 0
+run = True
+while run:
+ event = host.service(1000)
+ if event.type == enet.EVENT_TYPE_CONNECT:
+ print("%s: CONNECT" % event.peer.address)
+ elif event.type == enet.EVENT_TYPE_DISCONNECT:
+ print("%s: DISCONNECT" % event.peer.address)
+ run = False
+ continue
+ elif event.type == enet.EVENT_TYPE_RECEIVE:
+ print("%s: IN: %r" % (event.peer.address, event.packet.data))
+ continue
+ msg = os.urandom(40)
+ packet = enet.Packet(msg)
+ peer.send(0, packet)
+
+ counter += 1
+ if counter >= MSG_NUMBER:
+ msg = SHUTDOWN_MSG
+ peer.send(0, enet.Packet(msg))
+ host.service(0)
+ peer.disconnect()
+
+ print("%s: OUT: %r" % (peer.address, msg))