mirror of https://github.com/waku-org/nwaku.git
41 lines
1.1 KiB
Nim
41 lines
1.1 KiB
Nim
{.used.}
|
|
|
|
import
|
|
std/[unittest, options, tables, sets],
|
|
chronos, chronicles,
|
|
../../waku/v2/node/storage/message/waku_message_store,
|
|
../../waku/v2/protocol/waku_store/waku_store,
|
|
./utils
|
|
|
|
suite "Message Store":
|
|
test "set and get works":
|
|
let
|
|
database = SqliteDatabase.init("", inMemory = true)[]
|
|
store = WakuMessageStore.init(database)[]
|
|
topic = ContentTopic("/waku/2/default-content/proto")
|
|
pubsubTopic = "/waku/2/default-waku/proto"
|
|
|
|
var msgs = @[
|
|
WakuMessage(payload: @[byte 1, 2, 3], contentTopic: topic),
|
|
WakuMessage(payload: @[byte 1, 2, 3, 4], contentTopic: topic),
|
|
WakuMessage(payload: @[byte 1, 2, 3, 4, 5], contentTopic: topic),
|
|
]
|
|
|
|
defer: store.close()
|
|
|
|
for msg in msgs:
|
|
let output = store.put(computeIndex(msg), msg, pubsubTopic)
|
|
check output.isOk
|
|
|
|
var responseCount = 0
|
|
proc data(timestamp: uint64, msg: WakuMessage, psTopic: string) =
|
|
responseCount += 1
|
|
check msg in msgs
|
|
check psTopic == pubsubTopic
|
|
|
|
let res = store.getAll(data)
|
|
|
|
check:
|
|
res.isErr == false
|
|
responseCount == 3
|