mirror of https://github.com/waku-org/nwaku.git
60 lines
1.4 KiB
Nim
60 lines
1.4 KiB
Nim
{.used.}
|
|
|
|
import std/sequtils, testutils/unittests, chronos
|
|
import
|
|
../../../waku/common/databases/db_sqlite,
|
|
../../../waku/waku_archive,
|
|
../../../waku/waku_archive/driver/sqlite_driver,
|
|
../../../waku/waku_core,
|
|
../waku_archive/archive_utils,
|
|
../testlib/common,
|
|
../testlib/wakucore
|
|
|
|
suite "SQLite driver":
|
|
test "init driver and database":
|
|
## Given
|
|
let database = newSqliteDatabase()
|
|
|
|
## When
|
|
let driverRes = SqliteDriver.new(database)
|
|
|
|
## Then
|
|
check:
|
|
driverRes.isOk()
|
|
|
|
let driver: ArchiveDriver = driverRes.tryGet()
|
|
check:
|
|
not driver.isNil()
|
|
|
|
## Cleanup
|
|
(waitFor driver.close()).expect("driver to close")
|
|
|
|
test "insert a message":
|
|
## Given
|
|
const contentTopic = "test-content-topic"
|
|
|
|
let driver = newSqliteArchiveDriver()
|
|
|
|
let msg = fakeWakuMessage(contentTopic = contentTopic)
|
|
let msgHash = computeMessageHash(DefaultPubsubTopic, msg)
|
|
|
|
## When
|
|
let putRes = waitFor driver.put(
|
|
DefaultPubsubTopic, msg, computeDigest(msg), msgHash, msg.timestamp
|
|
)
|
|
|
|
## Then
|
|
check:
|
|
putRes.isOk()
|
|
|
|
let storedMsg = (waitFor driver.getAllMessages()).tryGet()
|
|
check:
|
|
storedMsg.len == 1
|
|
storedMsg.all do(item: auto) -> bool:
|
|
let (pubsubTopic, msg, _, _, hash) = item
|
|
msg.contentTopic == contentTopic and pubsubTopic == DefaultPubsubTopic and
|
|
hash == msgHash
|
|
|
|
## Cleanup
|
|
(waitFor driver.close()).expect("driver to close")
|