Noise: complete handshake process

TODO
- Figure out why `state.rs` is erased at some moment(even handshake
is not done).
- Refactor
- Add tests
This commit is contained in:
mhchia
2020-02-16 00:42:49 +08:00
parent 8a4ebd4cbb
commit d0290d2b5a
6 changed files with 176 additions and 24 deletions

View File

@ -1,6 +1,7 @@
import pytest
from libp2p.tools.factories import noise_conn_factory
from libp2p.security.noise.messages import NoiseHandshakePayload
from libp2p.tools.factories import noise_conn_factory, noise_handshake_payload_factory
DATA = b"testing_123"
@ -18,3 +19,10 @@ async def test_noise_connection(nursery):
await local_conn.write(DATA)
read_data = await remote_conn.read(len(DATA))
assert read_data == DATA
def test_noise_handshake_payload():
payload = noise_handshake_payload_factory()
payload_serialized = payload.serialize()
payload_deserialized = NoiseHandshakePayload.deserialize(payload_serialized)
assert payload == payload_deserialized