def test_two_mock_servers(self):
# Add two servers
serv1 = ServerMock(self.hallo)
serv1.name = "server_list_test1"
serv1.auto_connect = True
serv1.nick = "hallo"
serv1.disconnect()
self.hallo.add_server(serv1)
serv2 = ServerMock(self.hallo)
serv2.name = "server_list_test2"
serv2.auto_connect = False
serv2.nick = "yobot"
serv2.start()
self.hallo.add_server(serv2)
# Send command
self.function_dispatcher.dispatch(
EventMessage(self.server, self.test_chan, self.test_user,
"list servers"))
data = self.server.get_send_data(1, self.test_chan, EventMessage)
# Check response
server_list_text = data[0].text.split(": \n")[1]
server_list = server_list_text.split("\n")
assert len(server_list) == 2
if serv1.name in server_list[0]:
server_text1 = server_list[0]
server_text2 = server_list[1]
else:
server_text1 = server_list[1]
server_text2 = server_list[0]
assert serv1.name in server_text1
assert "type=" + serv1.type in server_text1
assert "state=" + serv1.state in server_text1
assert "nick=" + serv1.get_nick() in server_text1
assert "auto_connect=" + str(serv1.auto_connect) in server_text1
assert serv2.name in server_text2
assert "type=" + serv2.type in server_text2
assert "state=" + serv2.state in server_text2
assert "nick=" + serv2.get_nick() in server_text2
assert "auto_connect=" + str(serv2.auto_connect) in server_text2
def test_connect_to_known_server(self):
# Set up an example server
server_name = "known_server_name"
test_server = ServerMock(self.hallo)
test_server.name = server_name
test_server.auto_connect = False
self.hallo.add_server(test_server)
# Call connect function
self.function_dispatcher.dispatch(
EventMessage(self.server, self.test_chan, self.test_user,
"connect " + server_name))
# Ensure response is correct
data = self.server.get_send_data(1, self.test_chan, EventMessage)
assert "error" not in data[0].text.lower(), data[0].text.lower()
assert "connected" in data[0].text.lower(), data[0].text.lower()
assert server_name in data[0].text.lower(), data[0].text.lower()
# Ensure auto connect was set
assert test_server.auto_connect, "Auto connect should have been set to true."
# Ensure server was ran
assert test_server.is_connected(), "Test server was not started."
def test_connect_to_known_server_fail_connected(self):
# Set up example server
server_name = "known_server_name"
test_server = ServerMock(self.hallo)
test_server.name = server_name
test_server.auto_connect = False
test_server.state = Server.STATE_OPEN
self.hallo.add_server(test_server)
# Call connect function
self.function_dispatcher.dispatch(
EventMessage(self.server, self.test_chan, self.test_user,
"connect " + server_name))
# Ensure error response is given
data = self.server.get_send_data(1, self.test_chan, EventMessage)
assert "error" in data[0].text.lower(), data[0].text.lower()
assert "already connected" in data[0].text.lower(), data[0].text.lower(
)
# Ensure auto connect was still set
assert (test_server.auto_connect
), "Auto connect should have still been set to true."
# Ensure server is still running
assert (test_server.state == Server.STATE_OPEN
), "Test server should not have been shut down."