mirror of https://github.com/waku-org/nwaku.git
69 lines
1.5 KiB
Nim
69 lines
1.5 KiB
Nim
|
{.used.}
|
||
|
|
|
||
|
|
import
|
||
|
|
std/sequtils,
|
||
|
|
testutils/unittests,
|
||
|
|
chronos
|
||
|
|
import
|
||
|
|
../../../waku/common/sqlite,
|
||
|
|
../../../waku/v2/protocol/waku_archive,
|
||
|
|
../../../waku/v2/protocol/waku_archive/driver/sqlite_driver,
|
||
|
|
../../../waku/v2/protocol/waku_message,
|
||
|
|
../utils,
|
||
|
|
../testlib/common
|
||
|
|
|
||
|
|
|
||
|
|
proc newTestDatabase(): SqliteDatabase =
|
||
|
|
SqliteDatabase.new(":memory:").tryGet()
|
||
|
|
|
||
|
|
proc newTestSqliteDriver(): ArchiveDriver =
|
||
|
|
let db = newTestDatabase()
|
||
|
|
SqliteDriver.new(db).tryGet()
|
||
|
|
|
||
|
|
|
||
|
|
suite "SQLite driver":
|
||
|
|
|
||
|
|
test "init driver and database":
|
||
|
|
## Given
|
||
|
|
let database = newTestDatabase()
|
||
|
|
|
||
|
|
## When
|
||
|
|
let driverRes = SqliteDriver.new(database)
|
||
|
|
|
||
|
|
## Then
|
||
|
|
check:
|
||
|
|
driverRes.isOk()
|
||
|
|
|
||
|
|
let driver: ArchiveDriver = driverRes.tryGet()
|
||
|
|
check:
|
||
|
|
not driver.isNil()
|
||
|
|
|
||
|
|
## Cleanup
|
||
|
|
driver.close().expect("driver to close")
|
||
|
|
|
||
|
|
test "insert a message":
|
||
|
|
## Given
|
||
|
|
const contentTopic = "test-content-topic"
|
||
|
|
|
||
|
|
let driver = newTestSqliteDriver()
|
||
|
|
|
||
|
|
let msg = fakeWakuMessage(contentTopic=contentTopic)
|
||
|
|
|
||
|
|
## When
|
||
|
|
let putRes = driver.put(DefaultPubsubTopic, msg, computeDigest(msg), msg.timestamp)
|
||
|
|
|
||
|
|
## Then
|
||
|
|
check:
|
||
|
|
putRes.isOk()
|
||
|
|
|
||
|
|
let storedMsg = driver.getAllMessages().tryGet()
|
||
|
|
check:
|
||
|
|
storedMsg.len == 1
|
||
|
|
storedMsg.all do (item: auto) -> bool:
|
||
|
|
let (pubsubTopic, msg, digest, storeTimestamp) = item
|
||
|
|
msg.contentTopic == contentTopic and
|
||
|
|
pubsubTopic == DefaultPubsubTopic
|
||
|
|
|
||
|
|
## Cleanup
|
||
|
|
driver.close().expect("driver to close")
|