mirror of https://github.com/vacp2p/nim-libp2p.git
51 lines
1.0 KiB
Nim
51 lines
1.0 KiB
Nim
|
import unittest
|
||
|
|
import chronos, nimcrypto/utils
|
||
|
|
import ../libp2p/[connection,
|
||
|
|
stream/lpstream,
|
||
|
|
stream/bufferstream]
|
||
|
|
|
||
|
|
suite "Connection":
|
||
|
|
test "close":
|
||
|
|
proc test(): Future[bool] {.async.} =
|
||
|
|
var conn = newConnection(newBufferStream())
|
||
|
|
await conn.close()
|
||
|
|
check:
|
||
|
|
conn.closed == true
|
||
|
|
|
||
|
|
result = true
|
||
|
|
|
||
|
|
check:
|
||
|
|
waitFor(test()) == true
|
||
|
|
|
||
|
|
test "parent close":
|
||
|
|
proc test(): Future[bool] {.async.} =
|
||
|
|
var buf = newBufferStream()
|
||
|
|
var conn = newConnection(buf)
|
||
|
|
|
||
|
|
await conn.close()
|
||
|
|
check:
|
||
|
|
conn.closed == true
|
||
|
|
buf.closed == true
|
||
|
|
|
||
|
|
await sleepAsync(1.seconds)
|
||
|
|
result = true
|
||
|
|
|
||
|
|
check:
|
||
|
|
waitFor(test()) == true
|
||
|
|
|
||
|
|
test "child close":
|
||
|
|
proc test(): Future[bool] {.async.} =
|
||
|
|
var buf = newBufferStream()
|
||
|
|
var conn = newConnection(buf)
|
||
|
|
|
||
|
|
await buf.close()
|
||
|
|
check:
|
||
|
|
conn.closed == true
|
||
|
|
buf.closed == true
|
||
|
|
|
||
|
|
await sleepAsync(1.seconds)
|
||
|
|
result = true
|
||
|
|
|
||
|
|
check:
|
||
|
|
waitFor(test()) == true
|