mirror of https://github.com/waku-org/nwaku.git
228 lines
8.5 KiB
Nim
228 lines
8.5 KiB
Nim
{.used.}
|
|
|
|
import std/options, testutils/unittests, chronos, chronicles, libp2p/crypto/crypto
|
|
|
|
import
|
|
../../../waku/
|
|
[node/peer_manager, waku_core, waku_store, waku_store/client, common/paging],
|
|
../testlib/[common, wakucore, testasync, futures],
|
|
./store_utils
|
|
|
|
suite "Store Client":
|
|
var message1 {.threadvar.}: WakuMessage
|
|
var message2 {.threadvar.}: WakuMessage
|
|
var message3 {.threadvar.}: WakuMessage
|
|
var hash1 {.threadvar.}: WakuMessageHash
|
|
var hash2 {.threadvar.}: WakuMessageHash
|
|
var hash3 {.threadvar.}: WakuMessageHash
|
|
var messageSeq {.threadvar.}: seq[WakuMessageKeyValue]
|
|
var handlerFuture {.threadvar.}: Future[StoreQueryRequest]
|
|
var handler {.threadvar.}: StoreQueryRequestHandler
|
|
var storeQuery {.threadvar.}: StoreQueryRequest
|
|
|
|
var serverSwitch {.threadvar.}: Switch
|
|
var clientSwitch {.threadvar.}: Switch
|
|
|
|
var server {.threadvar.}: WakuStore
|
|
var client {.threadvar.}: WakuStoreClient
|
|
|
|
var serverPeerInfo {.threadvar.}: RemotePeerInfo
|
|
var clientPeerInfo {.threadvar.}: RemotePeerInfo
|
|
|
|
asyncSetup:
|
|
message1 = fakeWakuMessage(contentTopic = DefaultContentTopic)
|
|
message2 = fakeWakuMessage(contentTopic = DefaultContentTopic)
|
|
message3 = fakeWakuMessage(contentTopic = DefaultContentTopic)
|
|
hash1 = computeMessageHash(DefaultPubsubTopic, message1)
|
|
hash2 = computeMessageHash(DefaultPubsubTopic, message2)
|
|
hash3 = computeMessageHash(DefaultPubsubTopic, message3)
|
|
messageSeq =
|
|
@[
|
|
WakuMessageKeyValue(
|
|
messageHash: hash1,
|
|
message: some(message1),
|
|
pubsubTopic: some(DefaultPubsubTopic),
|
|
),
|
|
WakuMessageKeyValue(
|
|
messageHash: hash2,
|
|
message: some(message2),
|
|
pubsubTopic: some(DefaultPubsubTopic),
|
|
),
|
|
WakuMessageKeyValue(
|
|
messageHash: hash3,
|
|
message: some(message3),
|
|
pubsubTopic: some(DefaultPubsubTopic),
|
|
),
|
|
]
|
|
handlerFuture = newHistoryFuture()
|
|
handler = proc(req: StoreQueryRequest): Future[StoreQueryResult] {.async, gcsafe.} =
|
|
var request = req
|
|
request.requestId = ""
|
|
handlerFuture.complete(request)
|
|
return ok(StoreQueryResponse(messages: messageSeq))
|
|
storeQuery = StoreQueryRequest(
|
|
pubsubTopic: some(DefaultPubsubTopic),
|
|
contentTopics: @[DefaultContentTopic],
|
|
paginationForward: PagingDirection.FORWARD,
|
|
)
|
|
|
|
serverSwitch = newTestSwitch()
|
|
clientSwitch = newTestSwitch()
|
|
|
|
server = await newTestWakuStore(serverSwitch, handler = handler)
|
|
client = newTestWakuStoreClient(clientSwitch)
|
|
|
|
await allFutures(serverSwitch.start(), clientSwitch.start())
|
|
|
|
## The following sleep is aimed to prevent macos failures in CI
|
|
#[
|
|
2024-05-16T13:24:45.5106200Z INF 2024-05-16 13:24:45.509+00:00 Stopping AutonatService topics="libp2p autonatservice" tid=53712 file=service.nim:203
|
|
2024-05-16T13:24:45.5107960Z WRN 2024-05-16 13:24:45.509+00:00 service is already stopped topics="libp2p switch" tid=53712 file=switch.nim:86
|
|
2024-05-16T13:24:45.5109010Z . (1.68s)
|
|
2024-05-16T13:24:45.5109320Z Store Client (0.00s)
|
|
2024-05-16T13:24:45.5109870Z SIGSEGV: Illegal storage access. (Attempt to read from nil?)
|
|
2024-05-16T13:24:45.5111470Z stack trace: (most recent call last)
|
|
]#
|
|
await sleepAsync(500.millis)
|
|
|
|
serverPeerInfo = serverSwitch.peerInfo.toRemotePeerInfo()
|
|
clientPeerInfo = clientSwitch.peerInfo.toRemotePeerInfo()
|
|
|
|
asyncTeardown:
|
|
await allFutures(serverSwitch.stop(), clientSwitch.stop())
|
|
|
|
suite "StoreQueryRequest Creation and Execution":
|
|
asyncTest "Valid Queries":
|
|
# When a valid query is sent to the server
|
|
let queryResponse = await client.query(storeQuery, peer = serverPeerInfo)
|
|
|
|
# Then the query is processed successfully
|
|
assert await handlerFuture.withTimeout(FUTURE_TIMEOUT)
|
|
check:
|
|
handlerFuture.read() == storeQuery
|
|
queryResponse.get().messages == messageSeq
|
|
|
|
asyncTest "Invalid Queries":
|
|
# TODO: IMPROVE: We can't test "actual" invalid queries because
|
|
# it directly depends on the handler implementation, to achieve
|
|
# proper coverage we'd need an example implementation.
|
|
|
|
# Given some invalid queries
|
|
let
|
|
invalidQuery1 = StoreQueryRequest(
|
|
pubsubTopic: some(DefaultPubsubTopic),
|
|
contentTopics: @[],
|
|
paginationForward: PagingDirection.FORWARD,
|
|
)
|
|
invalidQuery2 = StoreQueryRequest(
|
|
pubsubTopic: PubsubTopic.none(),
|
|
contentTopics: @[DefaultContentTopic],
|
|
paginationForward: PagingDirection.FORWARD,
|
|
)
|
|
invalidQuery3 = StoreQueryRequest(
|
|
pubsubTopic: some(DefaultPubsubTopic),
|
|
contentTopics: @[DefaultContentTopic],
|
|
paginationLimit: some(uint64(0)),
|
|
)
|
|
invalidQuery4 = StoreQueryRequest(
|
|
pubsubTopic: some(DefaultPubsubTopic),
|
|
contentTopics: @[DefaultContentTopic],
|
|
paginationLimit: some(uint64(0)),
|
|
)
|
|
invalidQuery5 = StoreQueryRequest(
|
|
pubsubTopic: some(DefaultPubsubTopic),
|
|
contentTopics: @[DefaultContentTopic],
|
|
startTime: some(0.Timestamp),
|
|
endTime: some(0.Timestamp),
|
|
)
|
|
invalidQuery6 = StoreQueryRequest(
|
|
pubsubTopic: some(DefaultPubsubTopic),
|
|
contentTopics: @[DefaultContentTopic],
|
|
startTime: some(0.Timestamp),
|
|
endTime: some(-1.Timestamp),
|
|
)
|
|
|
|
# When the query is sent to the server
|
|
let queryResponse1 = await client.query(invalidQuery1, peer = serverPeerInfo)
|
|
|
|
# Then the query is not processed
|
|
assert await handlerFuture.withTimeout(FUTURE_TIMEOUT)
|
|
check:
|
|
handlerFuture.read() == invalidQuery1
|
|
queryResponse1.get().messages == messageSeq
|
|
|
|
# When the query is sent to the server
|
|
handlerFuture = newHistoryFuture()
|
|
let queryResponse2 = await client.query(invalidQuery2, peer = serverPeerInfo)
|
|
|
|
# Then the query is not processed
|
|
assert await handlerFuture.withTimeout(FUTURE_TIMEOUT)
|
|
check:
|
|
handlerFuture.read() == invalidQuery2
|
|
queryResponse2.get().messages == messageSeq
|
|
|
|
# When the query is sent to the server
|
|
handlerFuture = newHistoryFuture()
|
|
let queryResponse3 = await client.query(invalidQuery3, peer = serverPeerInfo)
|
|
|
|
# Then the query is not processed
|
|
assert await handlerFuture.withTimeout(FUTURE_TIMEOUT)
|
|
check:
|
|
handlerFuture.read() == invalidQuery3
|
|
queryResponse3.get().messages == messageSeq
|
|
|
|
# When the query is sent to the server
|
|
handlerFuture = newHistoryFuture()
|
|
let queryResponse4 = await client.query(invalidQuery4, peer = serverPeerInfo)
|
|
|
|
# Then the query is not processed
|
|
assert await handlerFuture.withTimeout(FUTURE_TIMEOUT)
|
|
check:
|
|
handlerFuture.read() == invalidQuery4
|
|
queryResponse4.get().messages == messageSeq
|
|
|
|
# When the query is sent to the server
|
|
handlerFuture = newHistoryFuture()
|
|
let queryResponse5 = await client.query(invalidQuery5, peer = serverPeerInfo)
|
|
|
|
# Then the query is not processed
|
|
assert await handlerFuture.withTimeout(FUTURE_TIMEOUT)
|
|
check:
|
|
handlerFuture.read() == invalidQuery5
|
|
queryResponse5.get().messages == messageSeq
|
|
|
|
# When the query is sent to the server
|
|
handlerFuture = newHistoryFuture()
|
|
let queryResponse6 = await client.query(invalidQuery6, peer = serverPeerInfo)
|
|
|
|
# Then the query is not processed
|
|
assert await handlerFuture.withTimeout(FUTURE_TIMEOUT)
|
|
check:
|
|
handlerFuture.read() == invalidQuery6
|
|
queryResponse6.get().messages == messageSeq
|
|
|
|
suite "Verification of StoreQueryResponse Payload":
|
|
asyncTest "Positive Responses":
|
|
# When a valid query is sent to the server
|
|
let queryResponse = await client.query(storeQuery, peer = serverPeerInfo)
|
|
|
|
# Then the query is processed successfully, and is of the expected type
|
|
check:
|
|
await handlerFuture.withTimeout(FUTURE_TIMEOUT)
|
|
type(queryResponse.get()) is StoreQueryResponse
|
|
|
|
asyncTest "Negative Responses - PeerDialFailure":
|
|
# Given a stopped peer
|
|
let
|
|
otherServerSwitch = newTestSwitch()
|
|
otherServerPeerInfo = otherServerSwitch.peerInfo.toRemotePeerInfo()
|
|
|
|
# When a query is sent to the stopped peer
|
|
let queryResponse = await client.query(storeQuery, peer = otherServerPeerInfo)
|
|
|
|
# Then the query is not processed
|
|
check:
|
|
not await handlerFuture.withTimeout(FUTURE_TIMEOUT)
|
|
queryResponse.isErr()
|
|
queryResponse.error.kind == ErrorCode.PEER_DIAL_FAILURE
|