Skip to content

Commit 4bf65df

Browse files
committed
Fix test TestStoreQueryWithPaginationReverseOrder
1 parent e39e2e2 commit 4bf65df

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

waku/store_test.go

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -404,6 +404,7 @@ func TestStoreQueryWithPaginationReverseOrder(t *testing.T) {
404404
var sentHashes []common.MessageHash
405405
defaultPubsubTopic := DefaultPubsubTopic
406406

407+
queryTimestamp := proto.Int64(time.Now().UnixNano())
407408
Debug("Publishing %d messages from Node1 using RelayPublish", numMessages)
408409
for i := 0; i < numMessages; i++ {
409410
message := node1.CreateMessage(&pb.WakuMessage{
@@ -428,6 +429,7 @@ func TestStoreQueryWithPaginationReverseOrder(t *testing.T) {
428429
ContentTopics: &[]string{"test-content-topic"},
429430
PaginationLimit: proto.Uint64(5),
430431
PaginationForward: false,
432+
TimeStart: queryTimestamp,
431433
}
432434

433435
res1, err := node3.GetStoredMessages(node2, &storeRequest1)
@@ -436,6 +438,9 @@ func TestStoreQueryWithPaginationReverseOrder(t *testing.T) {
436438

437439
storedMessages1 := *res1.Messages
438440
require.Len(t, storedMessages1, 5, "Expected to retrieve exactly 5 messages from first query")
441+
for i := 0; i < 5; i++ {
442+
Debug("stored hashes round 2 iteration %i is %s", i, storedMessages1[i].MessageHash)
443+
}
439444

440445
for i := 0; i < 5; i++ {
441446
require.Equal(t, sentHashes[i+3], storedMessages1[i].MessageHash, "Message order mismatch in first query")
@@ -445,9 +450,10 @@ func TestStoreQueryWithPaginationReverseOrder(t *testing.T) {
445450
storeRequest2 := common.StoreQueryRequest{
446451
IncludeData: true,
447452
ContentTopics: &[]string{"test-content-topic"},
448-
PaginationLimit: proto.Uint64(5),
453+
PaginationLimit: proto.Uint64(3),
449454
PaginationForward: false,
450455
PaginationCursor: &res1.PaginationCursor,
456+
TimeStart: queryTimestamp,
451457
}
452458

453459
res2, err := node3.GetStoredMessages(node2, &storeRequest2)
@@ -457,8 +463,8 @@ func TestStoreQueryWithPaginationReverseOrder(t *testing.T) {
457463
storedMessages2 := *res2.Messages
458464
require.Len(t, storedMessages2, 3, "Expected to retrieve exactly 3 messages from second query")
459465

460-
for i := 2; i < 5; i++ {
461-
require.Equal(t, sentHashes[i-2], storedMessages2[i].MessageHash, "Message order mismatch in second query")
466+
for i := 0; i < 3; i++ {
467+
require.Equal(t, sentHashes[i], storedMessages2[i].MessageHash, "Message order mismatch in second query")
462468

463469
}
464470

0 commit comments

Comments
 (0)