fixing linting errors with protocol_muxer and adding to Travis

This commit is contained in:
Robert Zajac 2019-01-09 20:24:41 -05:00
parent ab1aa0e7a8
commit 1dbe490623
4 changed files with 27 additions and 25 deletions

View File

@ -9,7 +9,7 @@ install:
script: script:
- pytest --cov=./ -v - pytest --cov=./ -v
- pylint --rcfile=.pylintrc encryption host libp2p network peer stream_muxer transport tests - pylint --rcfile=.pylintrc encryption host libp2p network peer protocol_muxer stream_muxer transport tests
after_success: after_success:
- codecov - codecov

View File

@ -72,21 +72,22 @@ class Multiselect(IMultiselectMuxer):
handshake_contents = await communicator.read_stream_until_eof() handshake_contents = await communicator.read_stream_until_eof()
# Confirm that the protocols are the same # Confirm that the protocols are the same
if not self.validate_handshake(handshake_contents): if not validate_handshake(handshake_contents):
raise MultiselectError("multiselect protocol ID mismatch") raise MultiselectError("multiselect protocol ID mismatch")
# Handshake succeeded if this point is reached # Handshake succeeded if this point is reached
def validate_handshake(self, handshake_contents):
"""
Determine if handshake is valid and should be confirmed
:param handshake_contents: contents of handshake message
:return: true if handshake is complete, false otherwise
"""
# TODO: Modify this when format used by go repo for messages def validate_handshake(handshake_contents):
# is added """
return handshake_contents == MULTISELECT_PROTOCOL_ID Determine if handshake is valid and should be confirmed
:param handshake_contents: contents of handshake message
:return: true if handshake is complete, false otherwise
"""
# TODO: Modify this when format used by go repo for messages
# is added
return handshake_contents == MULTISELECT_PROTOCOL_ID
class MultiselectError(ValueError): class MultiselectError(ValueError):
"""Raised when an error occurs in multiselect process""" """Raised when an error occurs in multiselect process"""

View File

@ -9,7 +9,7 @@ class MultiselectClient(IMultiselectClient):
Client for communicating with receiver's multiselect Client for communicating with receiver's multiselect
module in order to select a protocol id to communicate over module in order to select a protocol id to communicate over
""" """
def __init__(self): def __init__(self):
pass pass
@ -30,22 +30,11 @@ class MultiselectClient(IMultiselectClient):
handshake_contents = await communicator.read_stream_until_eof() handshake_contents = await communicator.read_stream_until_eof()
# Confirm that the protocols are the same # Confirm that the protocols are the same
if not self.validate_handshake(handshake_contents): if not validate_handshake(handshake_contents):
raise MultiselectClientError("multiselect protocol ID mismatch") raise MultiselectClientError("multiselect protocol ID mismatch")
# Handshake succeeded if this point is reached # Handshake succeeded if this point is reached
def validate_handshake(self, handshake_contents):
"""
Determine if handshake is valid and should be confirmed
:param handshake_contents: contents of handshake message
:return: true if handshake is complete, false otherwise
"""
# TODO: Modify this when format used by go repo for messages
# is added
return handshake_contents == MULTISELECT_PROTOCOL_ID
async def select_protocol_or_fail(self, protocol, stream): async def select_protocol_or_fail(self, protocol, stream):
""" """
Send message to multiselect selecting protocol Send message to multiselect selecting protocol
@ -117,5 +106,17 @@ class MultiselectClient(IMultiselectClient):
else: else:
raise MultiselectClientError("unrecognized response: " + response) raise MultiselectClientError("unrecognized response: " + response)
def validate_handshake(handshake_contents):
"""
Determine if handshake is valid and should be confirmed
:param handshake_contents: contents of handshake message
:return: true if handshake is complete, false otherwise
"""
# TODO: Modify this when format used by go repo for messages
# is added
return handshake_contents == MULTISELECT_PROTOCOL_ID
class MultiselectClientError(ValueError): class MultiselectClientError(ValueError):
"""Raised when an error occurs in protocol selection process""" """Raised when an error occurs in protocol selection process"""

View File

@ -6,7 +6,7 @@ class MultiselectCommunicator(IMultiselectCommunicator):
and multistream module will follow the same multistream protocol, and multistream module will follow the same multistream protocol,
which is necessary for them to work which is necessary for them to work
""" """
def __init__(self, stream): def __init__(self, stream):
self.stream = stream self.stream = stream