@@ -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