Greg Steiert / pegasus_dev

Development mbed library for MAX32630FTHR

Dependents:   blinky_max32630fthr

tools/host_tests/tcpecho_server_auto.py@0:5c4d7b2438d3, 2016-11-11 (annotated)

Committer:
switches
Date:
Fri Nov 11 20:59:50 2016 +0000
Revision:
0:5c4d7b2438d3
Initial commit

Who changed what in which revision?

UserRevisionLine numberNew contents of line
switches 0:5c4d7b2438d3 1 """
switches 0:5c4d7b2438d3 2 mbed SDK
switches 0:5c4d7b2438d3 3 Copyright (c) 2011-2013 ARM Limited
switches 0:5c4d7b2438d3 4
switches 0:5c4d7b2438d3 5 Licensed under the Apache License, Version 2.0 (the "License");
switches 0:5c4d7b2438d3 6 you may not use this file except in compliance with the License.
switches 0:5c4d7b2438d3 7 You may obtain a copy of the License at
switches 0:5c4d7b2438d3 8
switches 0:5c4d7b2438d3 9 http://www.apache.org/licenses/LICENSE-2.0
switches 0:5c4d7b2438d3 10
switches 0:5c4d7b2438d3 11 Unless required by applicable law or agreed to in writing, software
switches 0:5c4d7b2438d3 12 distributed under the License is distributed on an "AS IS" BASIS,
switches 0:5c4d7b2438d3 13 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
switches 0:5c4d7b2438d3 14 See the License for the specific language governing permissions and
switches 0:5c4d7b2438d3 15 limitations under the License.
switches 0:5c4d7b2438d3 16 """
switches 0:5c4d7b2438d3 17
switches 0:5c4d7b2438d3 18 import re
switches 0:5c4d7b2438d3 19 import sys
switches 0:5c4d7b2438d3 20 import uuid
switches 0:5c4d7b2438d3 21 import socket
switches 0:5c4d7b2438d3 22 from sys import stdout
switches 0:5c4d7b2438d3 23
switches 0:5c4d7b2438d3 24 class TCPEchoServerTest():
switches 0:5c4d7b2438d3 25 ECHO_SERVER_ADDRESS = ""
switches 0:5c4d7b2438d3 26 ECHO_PORT = 0
switches 0:5c4d7b2438d3 27 ECHO_LOOPs = 100
switches 0:5c4d7b2438d3 28 s = None # Socket
switches 0:5c4d7b2438d3 29
switches 0:5c4d7b2438d3 30 PATTERN_SERVER_IP = "Server IP Address is (\d+).(\d+).(\d+).(\d+):(\d+)"
switches 0:5c4d7b2438d3 31 re_detect_server_ip = re.compile(PATTERN_SERVER_IP)
switches 0:5c4d7b2438d3 32
switches 0:5c4d7b2438d3 33 def test(self, selftest):
switches 0:5c4d7b2438d3 34 result = False
switches 0:5c4d7b2438d3 35 c = selftest.mbed.serial_readline()
switches 0:5c4d7b2438d3 36 if c is None:
switches 0:5c4d7b2438d3 37 return selftest.RESULT_IO_SERIAL
switches 0:5c4d7b2438d3 38 selftest.notify(c)
switches 0:5c4d7b2438d3 39
switches 0:5c4d7b2438d3 40 m = self.re_detect_server_ip.search(c)
switches 0:5c4d7b2438d3 41 if m and len(m.groups()):
switches 0:5c4d7b2438d3 42 self.ECHO_SERVER_ADDRESS = ".".join(m.groups()[:4])
switches 0:5c4d7b2438d3 43 self.ECHO_PORT = int(m.groups()[4]) # must be integer for socket.connect method
switches 0:5c4d7b2438d3 44 selftest.notify("HOST: TCP Server found at: " + self.ECHO_SERVER_ADDRESS + ":" + str(self.ECHO_PORT))
switches 0:5c4d7b2438d3 45
switches 0:5c4d7b2438d3 46 # We assume this test fails so can't send 'error' message to server
switches 0:5c4d7b2438d3 47 try:
switches 0:5c4d7b2438d3 48 self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
switches 0:5c4d7b2438d3 49 self.s.connect((self.ECHO_SERVER_ADDRESS, self.ECHO_PORT))
switches 0:5c4d7b2438d3 50 except Exception, e:
switches 0:5c4d7b2438d3 51 self.s = None
switches 0:5c4d7b2438d3 52 selftest.notify("HOST: Socket error: %s"% e)
switches 0:5c4d7b2438d3 53 return selftest.RESULT_ERROR
switches 0:5c4d7b2438d3 54
switches 0:5c4d7b2438d3 55 print 'HOST: Sending %d echo strings...'% self.ECHO_LOOPs,
switches 0:5c4d7b2438d3 56 for i in range(0, self.ECHO_LOOPs):
switches 0:5c4d7b2438d3 57 TEST_STRING = str(uuid.uuid4())
switches 0:5c4d7b2438d3 58 try:
switches 0:5c4d7b2438d3 59 self.s.sendall(TEST_STRING)
switches 0:5c4d7b2438d3 60 data = self.s.recv(128)
switches 0:5c4d7b2438d3 61 except Exception, e:
switches 0:5c4d7b2438d3 62 self.s = None
switches 0:5c4d7b2438d3 63 selftest.notify("HOST: Socket error: %s"% e)
switches 0:5c4d7b2438d3 64 return selftest.RESULT_ERROR
switches 0:5c4d7b2438d3 65
switches 0:5c4d7b2438d3 66 received_str = repr(data)[1:-1]
switches 0:5c4d7b2438d3 67 if TEST_STRING == received_str: # We need to cut not needed single quotes from the string
switches 0:5c4d7b2438d3 68 sys.stdout.write('.')
switches 0:5c4d7b2438d3 69 stdout.flush()
switches 0:5c4d7b2438d3 70 result = True
switches 0:5c4d7b2438d3 71 else:
switches 0:5c4d7b2438d3 72 print "Expected: "
switches 0:5c4d7b2438d3 73 print "'%s'"% TEST_STRING
switches 0:5c4d7b2438d3 74 print "received: "
switches 0:5c4d7b2438d3 75 print "'%s'"% received_str
switches 0:5c4d7b2438d3 76 result = False
switches 0:5c4d7b2438d3 77 break
switches 0:5c4d7b2438d3 78
switches 0:5c4d7b2438d3 79 if self.s is not None:
switches 0:5c4d7b2438d3 80 self.s.close()
switches 0:5c4d7b2438d3 81 else:
switches 0:5c4d7b2438d3 82 selftest.notify("HOST: TCP Server not found")
switches 0:5c4d7b2438d3 83 result = False
switches 0:5c4d7b2438d3 84 return selftest.RESULT_SUCCESS if result else selftest.RESULT_FAILURE