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

fix: prevent race condition in postgres driver tests

Browse Source 
* add PostgresDriver.waitForPartition
* wait for a partition for 5s in archive driver tests setup
This commit is contained in:
Fabiana Cecin 2025-12-16 15:46:20 -03:00
parent bc5059083e
commit 0305a3c0a0
No known key found for this signature in database
GPG Key ID: BCAB8A55CB51B6C7
3 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/waku_archive/test_driver_postgres.nim
View File

@ -23,6 +23,8 @@ suite "Postgres driver":
driver = PostgresDriver(driverRes.get())
(await driver.waitForPartition()).expect("Test has no DB partition")
asyncTeardown:
let resetRes = await driver.reset()
if resetRes.isErr():

2
tests/waku_archive/test_driver_postgres_query.nim
View File

@ -38,6 +38,8 @@ suite "Postgres driver - queries":
driver = PostgresDriver(driverRes.get())
(await driver.waitForPartition()).expect("Test has no DB partition")
asyncTeardown:
let resetRes = await driver.reset()

16
waku/waku_archive/driver/postgres_driver/postgres_driver.nim
View File

@ -1422,6 +1422,22 @@ proc removeOldestPartition(
proc containsAnyPartition*(self: PostgresDriver): bool =
return not self.partitionMngr.isEmpty()
proc waitForPartition*(
self: PostgresDriver, timeout = chronos.seconds(5)
): Future[ArchiveDriverResult[void]] {.async.} =
let pollInterval = chronos.milliseconds(100)
var elapsed = chronos.milliseconds(0)
while elapsed < timeout:
if self.containsAnyPartition():
return ok()
await sleepAsync(pollInterval)
elapsed += pollInterval
return err("PostgresDriver.waitForPartition() timed out after " & $timeout)
method decreaseDatabaseSize*(
driver: PostgresDriver, targetSizeInBytes: int64, forceRemoval: bool = false
): Future[ArchiveDriverResult[void]] {.async.} =