add `ls` test
This commit is contained in:
parent
6e5641dc48
commit
4dcb542ae8
|
@ -1,9 +1,9 @@
|
||||||
import unittest, strutils, sequtils
|
import unittest, strutils, sequtils, sugar
|
||||||
import chronos
|
import chronos
|
||||||
import ../libp2p/connection, ../libp2p/multistreamselect,
|
import ../libp2p/connection, ../libp2p/multistreamselect,
|
||||||
../libp2p/readerwriter, ../libp2p/connection
|
../libp2p/readerwriter, ../libp2p/connection
|
||||||
|
|
||||||
## Stream for select test
|
## Mock stream for select test
|
||||||
type
|
type
|
||||||
TestSelectStream = ref object of ReadWrite
|
TestSelectStream = ref object of ReadWrite
|
||||||
step*: int
|
step*: int
|
||||||
|
@ -34,7 +34,7 @@ proc newTestSelectStream(): TestSelectStream =
|
||||||
new result
|
new result
|
||||||
result.step = 1
|
result.step = 1
|
||||||
|
|
||||||
## Stream for handles test
|
## Mock stream for handles test
|
||||||
type
|
type
|
||||||
TestHandlesStream = ref object of ReadWrite
|
TestHandlesStream = ref object of ReadWrite
|
||||||
step*: int
|
step*: int
|
||||||
|
@ -65,15 +65,54 @@ proc newTestHandlesStream(): TestHandlesStream =
|
||||||
new result
|
new result
|
||||||
result.step = 1
|
result.step = 1
|
||||||
|
|
||||||
suite "Multistream select":
|
## Mock stream for handles test
|
||||||
# test "test select custom proto":
|
type
|
||||||
# proc testSelect(): Future[bool] {.async.} =
|
LsHandler = proc(procs: seq[byte]): Future[void]
|
||||||
# let ms = newMultistream()
|
|
||||||
# let conn = newConnection(newTestSelectStream())
|
|
||||||
# result = await ms.select(conn, "/test/proto/1.0.0")
|
|
||||||
|
|
||||||
# check:
|
TestLsStream = ref object of ReadWrite
|
||||||
# waitFor(testSelect()) == true
|
step*: int
|
||||||
|
ls*: proc(procs: seq[byte]): Future[void]
|
||||||
|
|
||||||
|
method readExactly*(s: TestLsStream, pbytes: pointer, nbytes: int): Future[void] {.async.} =
|
||||||
|
case s.step:
|
||||||
|
of 1:
|
||||||
|
var buf = newSeq[byte](1)
|
||||||
|
buf[0] = 19
|
||||||
|
copyMem(cast[pointer](cast[uint](pbytes)), addr buf[0], buf.len())
|
||||||
|
s.step = 2
|
||||||
|
of 2:
|
||||||
|
var buf = "/multistream/1.0.0\n"
|
||||||
|
copyMem(cast[pointer](cast[uint](pbytes)), addr buf[0], buf.len())
|
||||||
|
s.step = 3
|
||||||
|
of 3:
|
||||||
|
var buf = newSeq[byte](1)
|
||||||
|
buf[0] = 3
|
||||||
|
copyMem(cast[pointer](cast[uint](pbytes)), addr buf[0], buf.len())
|
||||||
|
s.step = 4
|
||||||
|
of 4:
|
||||||
|
var buf = "ls\n"
|
||||||
|
copyMem(cast[pointer](cast[uint](pbytes)), addr buf[0], buf.len())
|
||||||
|
else:
|
||||||
|
copyMem(cast[pointer](cast[uint](pbytes)), cstring("\0x3na\n"), "\0x3na\n".len())
|
||||||
|
|
||||||
|
method write*(s: TestLsStream, msg: seq[byte], msglen = -1) {.async.} =
|
||||||
|
if s.step == 4:
|
||||||
|
await s.ls(msg)
|
||||||
|
|
||||||
|
proc newTestLsStream(ls: LsHandler): TestLsStream =
|
||||||
|
new result
|
||||||
|
result.ls = ls
|
||||||
|
result.step = 1
|
||||||
|
|
||||||
|
suite "Multistream select":
|
||||||
|
test "test select custom proto":
|
||||||
|
proc testSelect(): Future[bool] {.async.} =
|
||||||
|
let ms = newMultistream()
|
||||||
|
let conn = newConnection(newTestSelectStream())
|
||||||
|
result = await ms.select(conn, "/test/proto/1.0.0")
|
||||||
|
|
||||||
|
check:
|
||||||
|
waitFor(testSelect()) == true
|
||||||
|
|
||||||
test "test handle custom proto":
|
test "test handle custom proto":
|
||||||
proc testHandle(): Future[bool] {.async.} =
|
proc testHandle(): Future[bool] {.async.} =
|
||||||
|
@ -88,4 +127,26 @@ suite "Multistream select":
|
||||||
result = true
|
result = true
|
||||||
|
|
||||||
check:
|
check:
|
||||||
waitFor(testHandle()) == true
|
waitFor(testHandle()) == true
|
||||||
|
|
||||||
|
test "test handle `ls`":
|
||||||
|
proc testLs(): Future[bool] {.async.} =
|
||||||
|
let ms = newMultistream()
|
||||||
|
|
||||||
|
proc testLs(proto: seq[byte]): Future[void] {.async.}
|
||||||
|
let conn = newConnection(newTestLsStream(testLs))
|
||||||
|
|
||||||
|
let protos: seq[string] = @["\x13/test/proto1/1.0.0\n", "\x13/test/proto2/1.0.0\n"]
|
||||||
|
proc testLs(proto: seq[byte]): Future[void] {.async.} =
|
||||||
|
var strProto: string = cast[string](proto)
|
||||||
|
check strProto in protos
|
||||||
|
await conn.close()
|
||||||
|
|
||||||
|
proc testHandler(conn: Connection, proto: string): Future[void] {.async.} = discard
|
||||||
|
ms.addHandler("/test/proto1/1.0.0", testHandler)
|
||||||
|
ms.addHandler("/test/proto2/1.0.0", testHandler)
|
||||||
|
await ms.handle(conn)
|
||||||
|
result = true
|
||||||
|
|
||||||
|
check:
|
||||||
|
waitFor(testLs()) == true
|
||||||
|
|
Loading…
Reference in New Issue