nim-libp2p/tests/testswitch.nim

79 lines
2.5 KiB
Nim
Raw Normal View History

import unittest, tables, options
2019-09-09 17:33:32 +00:00
import chronos, chronicles
2019-09-06 07:13:47 +00:00
import ../libp2p/switch,
../libp2p/multistream,
../libp2p/protocols/identify,
../libp2p/connection,
../libp2p/transports/[transport, tcptransport],
../libp2p/multiaddress,
../libp2p/peerinfo,
../libp2p/crypto/crypto,
../libp2p/peer,
../libp2p/protocols/protocol,
../libp2p/muxers/muxer,
2019-09-04 22:00:39 +00:00
../libp2p/muxers/mplex/mplex,
2019-09-13 20:04:46 +00:00
../libp2p/muxers/mplex/types,
../libp2p/protocols/secure/secio,
../libp2p/protocols/secure/secure
2019-08-31 17:58:49 +00:00
const TestCodec = "/test/proto/1.0.0"
type
TestProto = ref object of LPProtocol
method init(p: TestProto) {.gcsafe.} =
proc handle(conn: Connection, proto: string) {.async, gcsafe.} =
let msg = cast[string](await conn.readLp())
check "Hello!" == msg
await conn.writeLp("Hello!")
2019-09-08 07:43:33 +00:00
await conn.close()
2019-08-31 17:58:49 +00:00
2019-08-31 18:52:56 +00:00
p.codec = TestCodec
2019-08-31 17:58:49 +00:00
p.handler = handle
suite "Switch":
test "e2e use switch":
2019-09-04 22:00:39 +00:00
proc createSwitch(ma: MultiAddress): (Switch, PeerInfo) =
let seckey = PrivateKey.random(RSA)
var peerInfo: PeerInfo
peerInfo.peerId = some(PeerID.init(seckey))
2019-09-04 22:00:39 +00:00
peerInfo.addrs.add(ma)
let identify = newIdentify(peerInfo)
proc createMplex(conn: Connection): Muxer =
result = newMplex(conn)
let mplexProvider = newMuxerProvider(createMplex, MplexCodec)
let transports = @[Transport(newTransport(TcpTransport))]
let muxers = [(MplexCodec, mplexProvider)].toTable()
2019-09-13 20:04:46 +00:00
# let secureManagers = [(SecioCodec, Secure(newSecio(seckey)))].toTable()
2019-09-04 22:00:39 +00:00
let switch = newSwitch(peerInfo, transports, identify, muxers)
result = (switch, peerInfo)
2019-08-31 17:58:49 +00:00
proc testSwitch(): Future[bool] {.async, gcsafe.} =
let ma1: MultiAddress = Multiaddress.init("/ip4/127.0.0.1/tcp/53370")
2019-09-06 07:13:47 +00:00
let ma2: MultiAddress = Multiaddress.init("/ip4/127.0.0.1/tcp/53371")
2019-08-31 17:58:49 +00:00
var peerInfo1, peerInfo2: PeerInfo
var switch1, switch2: Switch
(switch1, peerInfo1) = createSwitch(ma1)
2019-09-05 15:19:39 +00:00
2019-09-04 22:00:39 +00:00
let testProto = new TestProto
2019-09-05 15:19:39 +00:00
testProto.init()
2019-09-04 22:00:39 +00:00
testProto.codec = TestCodec
2019-08-31 17:58:49 +00:00
switch1.mount(testProto)
2019-09-13 00:57:11 +00:00
var switch1Fut = await switch1.start()
2019-08-31 17:58:49 +00:00
(switch2, peerInfo2) = createSwitch(ma2)
2019-09-13 00:57:11 +00:00
var switch2Fut = await switch2.start()
2019-08-31 17:58:49 +00:00
let conn = await switch2.dial(peerInfo1, TestCodec)
await conn.writeLp("Hello!")
let msg = cast[string](await conn.readLp())
check "Hello!" == msg
2019-09-13 00:57:11 +00:00
discard allFutures(switch1.stop(), switch2.stop())
2019-08-31 17:58:49 +00:00
result = true
check:
waitFor(testSwitch()) == true