logos-messaging/logos-messaging-nim
1
0
Fork 0
mirror of https://github.com/logos-messaging/logos-messaging-nim.git synced 2026-01-07 16:33:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_driver_postgres: enhance test coverage, multiple and single topic (#2301)

Browse Source 
Co-authored-by: Abhimanyu <ABresting@users.noreply.github.com>
This commit is contained in:
Ivan FB 2023-12-19 10:41:50 +01:00 committed by GitHub
parent 51199214f9
commit 88f2a9f89b
1 changed files with 23 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tests/waku_archive/test_driver_postgres_query.nim
View File

@ -201,18 +201,36 @@ suite "Postgres driver - query by content topic":
require (await driver.put(DefaultPubsubTopic, msg, computeDigest(msg), computeMessageHash(DefaultPubsubTopic, msg), msg.timestamp)).isOk() require (await driver.put(DefaultPubsubTopic, msg, computeDigest(msg), computeMessageHash(DefaultPubsubTopic, msg), msg.timestamp)).isOk()
## When ## When
let res = await driver.getMessages( var res = await driver.getMessages(
contentTopic= @[contentTopic1, contentTopic2], contentTopic= @[contentTopic1, contentTopic2],
pubsubTopic=some(DefaultPubsubTopic),
maxPageSize=2, maxPageSize=2,
ascendingOrder=true ascendingOrder=true,
startTime=some(ts(00)),
endTime=some(ts(40))
) )
## Then ## Then
assert res.isOk(), res.error assert res.isOk(), res.error
var filteredMessages = res.tryGet().mapIt(it[1])
check filteredMessages == expected[2..3]
let filteredMessages = res.tryGet().mapIt(it[1]) ## When
check: ## This is very similar to the previous one but we enforce to use the prepared
filteredMessages == expected[2..3] ## statement by querying one single content topic
res = await driver.getMessages(
contentTopic= @[contentTopic1],
pubsubTopic=some(DefaultPubsubTopic),
maxPageSize=2,
ascendingOrder=true,
startTime=some(ts(00)),
endTime=some(ts(40))
)
## Then
assert res.isOk(), res.error
filteredMessages = res.tryGet().mapIt(it[1])
check filteredMessages == @[expected[2]]
## Cleanup ## Cleanup
(await driver.close()).expect("driver to close") (await driver.close()).expect("driver to close")