示例#1
0
 def test_inherit_port(self):
     # Set things up
     test_port = 80
     test_serv_irc = ServerIRC(self.hallo)
     test_serv_irc.name = "test_serv_irc"
     test_serv_irc.server_port = test_port
     test_chan_irc = test_serv_irc.get_channel_by_name("test_chan")
     test_user_irc = test_serv_irc.get_user_by_name("test_user")
     # Run command
     self.function_dispatcher.dispatch("connect irc example.com", test_user_irc, test_chan_irc)
     # Can't check response because I'm using a ServerIRC instead of a ServerMock
     # Find the right server
     assert len(self.hallo.server_list) == 2, "Incorrect number of servers in hallo instance"
     right_server = None  # type: ServerIRC
     for server in self.hallo.server_list:
         if server is not self.server:
             right_server = server
     assert right_server is not None, "New server wasn't found."
     assert right_server.server_port == test_port, "Port incorrect"
示例#2
0
 def test_nickserv_password_inherit(self):
     # Set up
     test_nickserv_pass = "******"
     test_serv_irc = ServerIRC(self.hallo)
     test_serv_irc.name = "test_serv_irc"
     test_serv_irc.nickserv_pass = test_nickserv_pass
     test_chan_irc = test_serv_irc.get_channel_by_name("test_chan")
     test_user_irc = test_serv_irc.get_user_by_name("test_user")
     # Run command
     self.function_dispatcher.dispatch("connect irc example.com:80", test_user_irc, test_chan_irc)
     # Can't check response because I'm using a ServerIRC instead of a ServerMock
     # Find the right server
     assert len(self.hallo.server_list) == 2, "Incorrect number of servers in hallo instance."
     right_server = None  # type: ServerIRC
     for server in self.hallo.server_list:
         if server is not self.server:
             right_server = server
     assert right_server is not None, "New server wasn't found."
     assert right_server.nickserv_pass == test_nickserv_pass, "Nickserv password wasn't inherited"
示例#3
0
 def connect_to_new_server_irc(self, line, user_obj, destination_obj):
     """
     Processes arguments in order to connect to a new IRC server
     :type line: str
     :type user_obj: Destination.User
     :type destination_obj: Destination.Destination
     """
     # Get some handy objects
     current_server = user_obj.get_server()
     hallo_obj = current_server.get_hallo()
     # Set all variables to none as default
     server_address, server_port = None, None
     server_name = None
     # Find the URL, if specified
     url_regex = re.compile("(^|\s)(irc://)?(([a-z.]+\.[a-z]+)(:([0-9]+))?)(\s|$)", re.IGNORECASE)
     url_search = url_regex.search(line)
     if url_search is not None:
         line = line.replace(url_search.group(0), " ")
         server_address = url_search.group(4).lower()
         try:
             server_port = int(url_search.group(6))
         except (ValueError, TypeError):
             server_port = None
     # Find the server_address, if specified with equals notation
     server_address = Commons.find_parameter("server_address", line) or server_address
     # Find the server_port, if specified with equals notation
     server_port_param = Commons.find_parameter("server_port", line)
     if server_port_param is not None:
         try:
             server_port = int(server_port_param)
         except (ValueError, TypeError):
             return "Error, invalid port number."
     # Check server_address and server_port are set
     if server_address is None:
         return "Error, No server address specified."
     if server_port is None and isinstance(current_server, ServerIRC):
         server_port = current_server.get_server_port()
     if server_port is None:
         return "Error, No server port specified."
     # Get server name
     server_name = Commons.find_any_parameter(["server_name", "name"], line) or server_name
     # if server name is null, get it from server_address
     if server_name is None:
         server_name = Commons.get_domain_name(server_address)
     # Get other parameters, if set.
     auto_connect_str = Commons.find_parameter("auto_connect", line)
     auto_connect = True if auto_connect_str is None else Commons.string_to_bool(auto_connect_str)
     server_nick = Commons.find_any_parameter(["server_nick", "nick"], line) or current_server.get_nick()
     server_prefix_arg = Commons.find_any_parameter(["server_prefix", "prefix"], line)
     if not server_prefix_arg:
         server_prefix = current_server.prefix
     else:
         server_prefix = None if Commons.is_string_null(server_prefix_arg) else server_prefix_arg
     full_name = Commons.find_parameter("full_name", line) or current_server.get_full_name()
     nickserv_nick = "nickserv"
     nickserv_identity_command = "status"
     nickserv_identity_resp = "^status [^ ]+ 3$"
     nickserv_password = None
     if isinstance(current_server, ServerIRC):
         nickserv_nick = current_server.get_nickserv_nick()
         nickserv_identity_command = current_server.get_nickserv_ident_command()
         nickserv_identity_resp = current_server.get_nickserv_ident_response()
         nickserv_password = current_server.get_nickserv_pass()
     nickserv_nick = Commons.find_parameter("nickserv_nick", line) or nickserv_nick
     nickserv_identity_command = Commons.find_parameter("nickserv_identity_command",
                                                        line) or nickserv_identity_command
     nickserv_identity_resp = Commons.find_parameter("nickserv_identity_resp", line) or nickserv_identity_resp
     nickserv_password = Commons.find_parameter("nickserv_password", line) or nickserv_password
     # Create this serverIRC object
     new_server_obj = ServerIRC(hallo_obj, server_name, server_address, server_port)
     new_server_obj.set_auto_connect(auto_connect)
     new_server_obj.set_nick(server_nick)
     new_server_obj.set_prefix(server_prefix)
     new_server_obj.set_full_name(full_name)
     new_server_obj.set_nickserv_nick(nickserv_nick)
     new_server_obj.set_nickserv_ident_command(nickserv_identity_command)
     new_server_obj.set_nickserv_ident_response(nickserv_identity_resp)
     new_server_obj.set_nickserv_pass(nickserv_password)
     # Add user with same name on new server to all the same groups as current user
     new_user_nick = Commons.find_any_parameter(["user", "god"], line) or user_obj.get_name()
     new_user = new_server_obj.get_user_by_name(new_user_nick)
     for group in user_obj.user_group_list:
         new_user.add_user_group(group)
     # Add the new object to Hallo's list
     hallo_obj.add_server(new_server_obj)
     # Connect to the new server object.
     Thread(target=new_server_obj.run).start()
     return "Connected to new IRC server: " + new_server_obj.get_name() + "."