Skip to content

Commit 189e165

Browse files
authored
Merge pull request #1816 from lightninglabs/wip/fix-flake-unit-test-TestQueryMessages
tapdb: fix flaky `TestQueryMessages` due to close timestamps
2 parents db328e5 + 92ac1c2 commit 189e165

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

tapdb/authmailbox_test.go

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -69,25 +69,29 @@ func TestQueryMessages(t *testing.T) {
6969
mailboxStore, _ := newMailboxStore(t)
7070
ctx := context.Background()
7171

72+
// Use a fixed base timestamp to avoid flaky tests.
73+
baseTime := time.Date(2024, 1, 1, 12, 0, 0, 0, time.UTC)
74+
7275
const numMessages = 5
7376
for i := 0; i < numMessages; i++ {
7477
txProof := proof.MockTxProof(t)
7578
msg := &authmailbox.Message{
7679
ReceiverKey: *receiverKey,
7780
EncryptedPayload: []byte("payload"),
78-
ArrivalTimestamp: time.Now().Add(
79-
time.Duration(i) * time.Second,
81+
ArrivalTimestamp: baseTime.Add(
82+
time.Duration(i) * time.Hour,
8083
),
8184
}
8285

8386
_, err := mailboxStore.StoreMessage(ctx, *txProof, msg)
8487
require.NoError(t, err)
8588
}
8689

87-
// Query messages created after the second message.
90+
// Query messages created after the second message (after 1 hour from
91+
// base time).
8892
filter := authmailbox.MessageFilter{
8993
ReceiverKey: *receiverKey,
90-
After: time.Now().Add(time.Second),
94+
After: baseTime.Add(time.Hour),
9195
}
9296
messages, err := mailboxStore.QueryMessages(ctx, filter)
9397
require.NoError(t, err)

0 commit comments

Comments
 (0)