1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
#! /usr/bin/env python
"""
This script tests some of the base functionalities of MORSE.
"""
import sys
from morse.testing.testing import MorseTestCase
# Include this import to be able to use your test file as a regular
# builder script, ie, usable with: 'morse [run|exec] base_testing.py
try:
from morse.builder import *
except ImportError:
pass
class BaseTest(MorseTestCase):
def setUpEnv(self):
""" Defines the test scenario, using the Builder API.
"""
# Adding 4 robots
# no name provided, use the name of the associated variable
jido = Jido()
# use explicitly name provided
robot2 = ATRV('mana')
# setup the name using explicitly robot3.name
robot3 = ATRV()
robot3.name = 'dala'
# no name provided, use variable name, old school style
atrv = ATRV()
env = Environment('empty', fastmode = True)
def test_list_robots(self):
""" Tests the simulator can return the list of robots
This test is guaranteed to be started only when the simulator
is ready.
"""
# Initialize a socket connection to the simulator
import socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(("localhost", 4000))
sockf = s.makefile()
# Queries for the list of robots
s.send(b"id1 simulation list_robots\n")
result = sockf.readline()
id, success, robots = result.strip().split(' ', 2)
self.assertEquals(success, "SUCCESS")
import ast
robotsset = set(ast.literal_eval(robots))
self.assertEquals(robotsset, {'jido', 'mana', 'dala', 'atrv'})
sockf.close()
s.close()
def test_socket_request_parser_resilience(self):
""" Tests that the socket request parser is resilient to
useless whitespaces.
"""
# Initialize a socket connection to the simulator
import socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(("localhost", 4000))
sockf = s.makefile()
queries = [b"id1 simulation list_robots\n",
b"id1 simulation list_robots\n",
b"id1\tsimulation\tlist_robots\n",
b"id1 \t simulation \t list_robots\n",
b" id1 simulation list_robots \n"]
for q in queries:
s.send(q)
result = sockf.readline()
id, success, robots = result.strip().split(' ', 2)
self.assertEquals(success, "SUCCESS")
sockf.close()
s.close()
########################## Run these tests ##########################
if __name__ == "__main__":
from morse.testing.testing import main
main(BaseTest)
|
