nim-libp2p/tests/testdaemon.nim

148 lines
4.5 KiB
Nim
Raw Normal View History

2018-11-19 04:34:05 +00:00
import unittest
import asyncdispatch2
import ../libp2p/daemon/daemonapi, ../libp2p/multiaddress, ../libp2p/multicodec
2018-11-19 04:34:05 +00:00
proc identitySpawnTest(): Future[bool] {.async.} =
2018-12-10 20:55:06 +00:00
var api = await newDaemonApi()
2018-11-19 04:34:05 +00:00
var data = await api.identity()
await api.close()
result = true
proc connectStreamTest(): Future[bool] {.async.} =
2018-12-10 20:55:06 +00:00
var api1 = await newDaemonApi()
var api2 = await newDaemonApi()
var id1 = await api1.identity()
var id2 = await api2.identity()
var protos = @["/test-stream"]
var test = "TEST STRING"
var testFuture = newFuture[string]("test.future")
proc streamHandler(api: DaemonAPI, stream: P2PStream) {.async.} =
var line = await stream.transp.readLine()
testFuture.complete(line)
await api2.addHandler(protos, streamHandler)
await api1.connect(id2.peer, id2.addresses)
# echo await api1.listPeers()
var stream = await api1.openStream(id2.peer, protos)
let sent = await stream.transp.write(test & "\r\n")
doAssert(sent == len(test) + 2)
var check = await wait(testFuture, 10000)
doAssert(check == test)
2018-12-10 10:38:12 +00:00
await stream.close()
2018-11-19 20:53:20 +00:00
await api1.close()
await api2.close()
result = true
proc provideBadCidTest(): Future[bool] {.async.} =
var cid = newSeq[byte](10)
var api = await newDaemonApi({DHTFull})
try:
await api.dhtProvide(cid)
result = false
except DaemonRemoteError:
result = true
finally:
await api.close()
proc getOnlyOneIPv4Address(addresses: seq[MultiAddress]): seq[MultiAddress] =
## We doing this becuase of bug in `go-pubsub`
## https://github.com/libp2p/go-libp2p-pubsub/issues/130
if len(addresses) > 0:
result = newSeqOfCap[MultiAddress](len(addresses))
let ip4 = multiCodec("ip4")
for item in addresses:
if item.protoCode() == ip4:
result.add(item)
break
proc pubsubTest(f: set[P2PDaemonFlags]): Future[bool] {.async.} =
var pubsubData = "TEST MESSAGE"
var msgData = cast[seq[byte]](pubsubData)
2018-12-10 20:55:06 +00:00
var api1, api2: DaemonAPI
2018-12-12 14:20:26 +00:00
api1 = await newDaemonApi(f + {Verbose, Logging})
api2 = await newDaemonApi(f + {Verbose, Logging})
var id1 = await api1.identity()
var id2 = await api2.identity()
var resultsCount = 0
2018-12-10 20:55:06 +00:00
var handlerFuture1 = newFuture[void]()
var handlerFuture2 = newFuture[void]()
proc pubsubHandler1(api: DaemonAPI,
ticket: PubsubTicket,
message: PubSubMessage): Future[bool] {.async.} =
let smsg = cast[string](message.data)
if smsg == pubsubData:
inc(resultsCount)
2018-12-10 20:55:06 +00:00
handlerFuture1.complete()
# Callback must return `false` to close subscription channel.
2018-12-10 20:55:06 +00:00
result = false
proc pubsubHandler2(api: DaemonAPI,
ticket: PubsubTicket,
message: PubSubMessage): Future[bool] {.async.} =
let smsg = cast[string](message.data)
if smsg == pubsubData:
inc(resultsCount)
2018-12-10 20:55:06 +00:00
handlerFuture2.complete()
# Callback must return `false` to close subscription channel.
result = false
# Not subscribed to any topics everything must be 0.
# We are making only one connection, because of bug
# https://github.com/libp2p/go-libp2p-pubsub/issues/130
await api1.connect(id2.peer, getOnlyOneIPv4Address(id2.addresses))
var ticket1 = await api1.pubsubSubscribe("test-topic", pubsubHandler1)
var ticket2 = await api2.pubsubSubscribe("test-topic", pubsubHandler2)
2018-12-10 20:55:06 +00:00
await sleepAsync(2000)
var topics1 = await api1.pubsubGetTopics()
var topics2 = await api2.pubsubGetTopics()
2018-12-10 20:55:06 +00:00
if len(topics1) == 1 and len(topics2) == 1:
var peers1 = await api1.pubsubListPeers("test-topic")
var peers2 = await api2.pubsubListPeers("test-topic")
if len(peers1) == 1 and len(peers2) == 1:
# Publish test data via api1.
await sleepAsync(500)
await api1.pubsubPublish("test-topic", msgData)
var andfut = handlerFuture1 and handlerFuture2
await andfut or sleepAsync(10000)
await api1.close()
await api2.close()
if resultsCount == 2:
result = true
2018-12-12 14:20:26 +00:00
else:
echo " -- CLIENT1 -- "
echo api1.log
echo " -- CLIENT2 -- "
echo api2.log
2018-11-19 04:34:05 +00:00
when isMainModule:
suite "libp2p-daemon test suite":
test "Simple spawn and get identity test":
2018-11-19 04:42:50 +00:00
check:
waitFor(identitySpawnTest()) == true
test "Connect/Accept peer/stream test":
check:
2018-11-19 20:53:20 +00:00
waitFor(connectStreamTest()) == true
test "DHT provide bad CID test":
check:
2018-11-23 10:20:53 +00:00
waitFor(provideBadCidTest()) == true
test "GossipSub test":
check:
waitFor(pubsubTest({PSGossipSub})) == true
test "FloodSub test":
check:
waitFor(pubsubTest({PSFloodSub})) == true