def __init__(self): self.logger = logging.getLogger() self.logger.debug("init") self.status = None self.address = None self.password = None self.pam_username = None self.pam_password = None self.port = None self.dirname = None self.socket = None self.auth_credentials_valid = False self.max_reconnects = 0 self.tls_psk_enable = True self.tls_psk_require = False try: self.tls_version = ssl.PROTOCOL_TLS except AttributeError: self.tls_version = ssl.PROTOCOL_SSLv23 self.connection_type = None self.requested_protocol_version = None self.protocol_messages = ProtocolMessages() # identity_prefix have to be set in each class self.identity_prefix = u"R_NONE" self.receive_buffer = b""
def test_protocol_message(self): logger = logging.getLogger() name = u"Testname" expected_regex_str = r"Hello {} calling version (.*)".format(name) expected_regex = bytes(bytearray(expected_regex_str, "utf-8")) hello_message = ProtocolMessages().hello(name) logger.debug(hello_message) self.assertRegexpMatches(hello_message, expected_regex) version = re.search(expected_regex, hello_message).group(1).decode("utf-8") logger.debug(u"version: {} ({})".format(version, type(version))) self.assertGreaterEqual(self.versiontuple(version), self.versiontuple(u"18.2.5"))