def test_6(self, zmq_context, zmq_socket):
'''
There is something wrong with this test as some extra calls to
call().socket().recv().__str__() are added (it seems to be a side
effect of the mock).
'''
thougt_police.main(["list", "robot"])
import socket
address = socket.gethostbyname(socket.getfqdn())
found = list(zmq_socket.mock_calls)
pp = pprint.PrettyPrinter(indent=4)
pp.pprint(found)
assert_equal(
list(zmq_socket.mock_calls)[-14],
mock.call().socket().send("list robot {} 9004".format(address)))
assert_equal(
list(zmq_socket.mock_calls)[-13],
mock.call().socket().recv())
thougt_police.main(["list", "player"])
found = list(zmq_socket.mock_calls)
pp.pprint(found)
assert_equal(
list(zmq_socket.mock_calls)[-15],
mock.call().socket().send("list player {} 9004".format(address)))
assert_equal(
list(zmq_socket.mock_calls)[-14],
mock.call().socket().recv())
def test_1(self, zmq_context, zmq_socket):
# without this empty command the program hangs waiting for input
thougt_police.main([""])
found = self._clean(list(zmq_socket.mock_calls))
assert_equal(found[0], mock.call())
assert_equal(found[1], mock.call().socket(zmq.REQ))
assert_equal(found[2], mock.call().socket().setsockopt(zmq.LINGER, 1))
assert_equal(found[3],
mock.call().socket().connect('tcp://127.0.0.1:9003'))
def test_2_bis(self, zmq_context, zmq_socket):
thougt_police.main(
["-p", "12", "-a", "127.0.0.1", "stop", "application"])
found = self._clean(list(zmq_socket.mock_calls))
# pp = pprint.PrettyPrinter(indent=4)
# pp.pprint(found)
assert_equal(found[0], mock.call())
assert_equal(found[1], mock.call().socket(zmq.REQ))
assert_equal(found[2], mock.call().socket().setsockopt(zmq.LINGER, 1))
assert_equal(mock.call().socket().connect('tcp://127.0.0.1:12'),
found[3])
assert_equal(mock.call().socket().send_string('stop application'),
found[4])
def test_2(self, zmq_context, zmq_socket):
thougt_police.main(["stop", "application"])
found = list(zmq_socket.mock_calls)
pp = pprint.PrettyPrinter(indent=4)
pp.pprint(found)
assert_equal(found[0], mock.call())
# this does not work
#assert_equal(found[1], mock.call().__str__())
assert_equal(found[2], mock.call().socket(1))
assert_equal(found[4], mock.call().socket().setsockopt(17, 1))
assert_equal(found[5],
mock.call().socket().connect('tcp://127.0.0.1:9003'))
assert_equal(found[6], mock.call().socket(7))
assert_equal(found[8], mock.call().socket().setsockopt(17, 1))
assert_equal(found[9], mock.call().socket().bind('tcp://0.0.0.0:9004'))
assert_equal(found[10], mock.call().socket().send('stop application'))
def test_5(self, zmq_context, zmq_socket):
thougt_police.main(["add", "player", "toto"])
assert_equal(
list(zmq_socket.mock_calls)[-1],
mock.call().socket().send("add player toto"))
thougt_police.main(["remove", "player", "toto"])
assert_equal(
list(zmq_socket.mock_calls)[-1],
mock.call().socket().send("remove player toto"))
thougt_police.main(["add", "robot", "Robert"])
assert_equal(
list(zmq_socket.mock_calls)[-1],
mock.call().socket().send("add robot Robert"))
thougt_police.main(["remove", "robot", "Robert"])
assert_equal(
list(zmq_socket.mock_calls)[-1],
mock.call().socket().send("remove robot Robert"))
def test_5(self, zmq_context, zmq_socket):
"""
There is something wrong with this test as some extra calls to
call().socket().recv().__str__() are added (it seems to be a side
effect of the mock).
"""
thougt_police.main(["list", "robot"])
found = self._clean(list(zmq_socket.mock_calls))
# pp = pprint.PrettyPrinter(indent=4)
# pp.pprint(found)
assert_equal(mock.call().socket().send_string("list robot"), found[-2])
assert_equal(mock.call().socket().recv_string(), found[-1])
thougt_police.main(["list", "player"])
found = self._clean(list(zmq_socket.mock_calls))
# pp.pprint(found)
assert_equal(mock.call().socket().send_string("list player"),
found[-2])
assert_equal(mock.call().socket().recv_string(), found[-1])
thougt_police.main(["list", "flag"])
found = self._clean(list(zmq_socket.mock_calls))
# pp.pprint(found)
assert_equal(mock.call().socket().send_string("list flag"), found[-2])
assert_equal(mock.call().socket().recv_string(), found[-1])
def test_4(self, zmq_context, zmq_socket):
thougt_police.main(["stop", "game"])
found = self._clean(list(zmq_socket.mock_calls))
# pp = pprint.PrettyPrinter(indent=4)
# pp.pprint(found)
assert_equal(mock.call().socket().send_string("stop game"), found[-2])
def test_with_spaces(self, zmq_context, zmq_socket):
thougt_police.main(["add", "player", "titi toto"])
assert_equal(
mock.call().socket().send_string('add player "titi toto"'),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["remove", "player", "titi toto"])
assert_equal(
mock.call().socket().send_string('remove player "titi toto"'),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["add", "robot", "Jean Robert"])
assert_equal(
mock.call().socket().send_string('add robot "Jean Robert"'),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["remove", "robot", "Jean Robert"])
assert_equal(
mock.call().socket().send_string('remove robot "Jean Robert"'),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["register", "robot", "Jean Robert"])
assert_equal(
mock.call().socket().send_string('register robot "Jean Robert"'),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["unregister", "robot", "Jean Robert"])
assert_equal(
mock.call().socket().send_string('unregister robot "Jean Robert"'),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(
["set", "robot", "Jean Robert", "video_url", "wrong"])
assert_equal(
mock.call().socket().send_string(
'set robot "Jean Robert" video_url wrong'),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["remove", "robot", "Jean Robert"])
assert_equal(
mock.call().socket().send_string('remove robot "Jean Robert"'),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["add", "team", "A TEAM"])
assert_equal(mock.call().socket().send_string('add team "A TEAM"'),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["remove", "team", "A TEAM"])
assert_equal(mock.call().socket().send_string('remove team "A TEAM"'),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["set", "team", "A TEAM", "score", "2"])
assert_equal(
mock.call().socket().send_string('set team "A TEAM" score 2'),
self._clean(list(zmq_socket.mock_calls))[-2])
def test_6(self, zmq_context, zmq_socket):
thougt_police.main(["add", "player", "toto"])
assert_equal(mock.call().socket().send_string("add player toto"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["remove", "player", "toto"])
assert_equal(mock.call().socket().send_string("remove player toto"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["add", "robot", "Robert"])
assert_equal(mock.call().socket().send_string("add robot Robert"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["remove", "robot", "Robert"])
assert_equal(mock.call().socket().send_string("remove robot Robert"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["register", "robot", "Robert"])
assert_equal(mock.call().socket().send_string("register robot Robert"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["unregister", "robot", "Robert"])
assert_equal(
mock.call().socket().send_string("unregister robot Robert"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["set", "robot", "Robert", "video_url", "wrong"])
assert_equal(
mock.call().socket().send_string(
"set robot Robert video_url wrong"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["remove", "robot", "Robert"])
assert_equal(mock.call().socket().send_string("remove robot Robert"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["set", "game", "duration", "5"])
assert_equal(mock.call().socket().send_string("set game duration 5"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["get", "game", "duration"])
assert_equal(mock.call().socket().send_string("get game duration"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["add", "team", "TEAM"])
assert_equal(mock.call().socket().send_string("add team TEAM"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["remove", "team", "TEAM"])
assert_equal(mock.call().socket().send_string("remove team TEAM"),
self._clean(list(zmq_socket.mock_calls))[-2])
thougt_police.main(["set", "team", "TEAM", "score", "2"])
assert_equal(mock.call().socket().send_string("set team TEAM score 2"),
self._clean(list(zmq_socket.mock_calls))[-2])
def test_4(self, zmq_context, zmq_socket):
thougt_police.main(["stop", "game"])
assert_equal(
list(zmq_socket.mock_calls)[-1],
mock.call().socket().send("stop game"))
def test_1(self):
thougt_police.main()