mirror of https://github.com/vacp2p/nim-libp2p.git
55 lines
1.7 KiB
Nim
55 lines
1.7 KiB
Nim
|
import unittest
|
||
|
|
import chronos
|
||
|
|
import ../libp2p/switch, ../libp2p/multistream,
|
||
|
|
../libp2p/identify, ../libp2p/connection,
|
||
|
|
../libp2p/transport, ../libp2p/tcptransport,
|
||
|
|
../libp2p/multiaddress, ../libp2p/peerinfo,
|
||
|
|
../libp2p/crypto/crypto, ../libp2p/peer,
|
||
|
|
../libp2p/protocol
|
||
|
|
|
||
|
|
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!")
|
||
|
|
|
||
|
|
p.handler = handle
|
||
|
|
|
||
|
|
suite "Switch":
|
||
|
|
test "e2e use switch":
|
||
|
|
proc testSwitch(): Future[bool] {.async, gcsafe.} =
|
||
|
|
let ma1: MultiAddress = Multiaddress.init("/ip4/127.0.0.1/tcp/53370")
|
||
|
|
let ma2: MultiAddress = Multiaddress.init("/ip4/127.0.0.1/tcp/53371")
|
||
|
|
|
||
|
|
var peerInfo1, peerInfo2: PeerInfo
|
||
|
|
var switch1, switch2: Switch
|
||
|
|
proc createSwitch(ma: MultiAddress): (Switch, PeerInfo) =
|
||
|
|
let seckey = PrivateKey.random(RSA)
|
||
|
|
var peerInfo: PeerInfo
|
||
|
|
peerInfo.peerId = PeerID.init(seckey)
|
||
|
|
peerInfo.addrs.add(ma)
|
||
|
|
let switch = newSwitch(peerInfo, @[Transport(newTransport(TcpTransport))])
|
||
|
|
result = (switch, peerInfo)
|
||
|
|
|
||
|
|
(switch1, peerInfo1) = createSwitch(ma1)
|
||
|
|
let testProto = newProtocol(TestProto, peerInfo1)
|
||
|
|
switch1.mount(testProto)
|
||
|
|
await switch1.start()
|
||
|
|
|
||
|
|
(switch2, peerInfo2) = createSwitch(ma2)
|
||
|
|
await switch2.start()
|
||
|
|
let conn = await switch2.dial(peerInfo1, TestCodec)
|
||
|
|
await conn.writeLp("Hello!")
|
||
|
|
let msg = cast[string](await conn.readLp())
|
||
|
|
check "Hello!" == msg
|
||
|
|
|
||
|
|
result = true
|
||
|
|
|
||
|
|
check:
|
||
|
|
waitFor(testSwitch()) == true
|