Test messages

Author: Pithikos

tests/_bootstrap_.py

@@ -1,4 +1,5 @@
 #Bootstrap
 import sys, os
-if 'python-websockets' in  os.getcwd():
+if 'python-websockets-server' in  os.getcwd():
 	sys.path.insert(0, '..')
+	sys.path.insert(0, 'testsuit')

tests/message_lengths.py

@@ -0,0 +1,57 @@
+import _bootstrap_
+from websocket import WebSocketsServer
+from time import sleep
+from messages import *
+
+'''
+This creates just a server that will send a different message to every new connection: 
+    
+    1. A message of length less than 126
+    2. A message of length 126
+    3. A message of length 127
+    4. A message of length bigger than 127
+    5. A message above 1024
+    6. A message above 65K
+    7. An enormous message (well beyond 65K)
+    
+    
+Reconnect to get the next message
+'''
+
+
+counter = 0
+
+# Called for every client connecting (after handshake)
+def new_client(client, server):
+	print("New client connected and was given id %d" % client['id'])
+	global counter
+	if counter == 0:
+		print("Sending message 1 of length %d" % len(msg_125b))
+		server.send_message(client, msg_125b)
+	elif counter == 1:
+		print("Sending message 2 of length %d" % len(msg_126b))
+		server.send_message(client, msg_126b)
+	elif counter == 2:
+		print("Sending message 3 of length %d" % len(msg_127b))
+		server.send_message(client, msg_127b)
+	elif counter == 3:
+		print("Sending message 4 of length %d" % len(msg_208b))
+		server.send_message(client, msg_208b)
+	elif counter == 4:
+		print("Sending message 5 of length %d" % len(msg_1251b))
+		server.send_message(client, msg_1251b)
+	elif counter == 5:
+		print("Sending message 6 of length %d" % len(msg_68Kb))
+		server.send_message(client, msg_68Kb)
+	elif counter == 6:
+		print("Sending message 7 of length %d" % len(msg_1500Kb))
+		server.send_message(client, msg_1500Kb)
+	else:
+		print("No errors")
+	counter += 1
+
+
+PORT=13254
+server = WebSocketsServer(PORT)
+server.set_fn_new_client(new_client)
+server.run_forever()

tests/messages.py

@@ -0,0 +1,21 @@
+#
+# Fixed messages by length
+# Every message ends with its length..
+#
+
+msg_125b   = 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
+             'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
+             'abcdefghijklmnopqr125'                                  # 125
+msg_126b   = 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
+             'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
+             'abcdefghijklmnopqrs126'                                 # 126
+msg_127b   = 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
+             'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
+             'abcdefghijklmnopqrst127'                                # 127
+msg_208b   = 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
+             'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
+             'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
+             'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvw208'   # 208
+msg_1251b  = (msg_125b*10)+'1'       # 1251
+msg_68Kb   = ('a'*67995)+'68000' # 68000
+msg_1500Kb = ('a'*1500000)+'1500000' # 1.5Mb