mirror of
https://github.com/logos-messaging/logos-messaging-go-bindings.git
synced 2026-04-30 15:03:12 +00:00
Add page store test
This commit is contained in:
aya
parent
26b6b34aa4
commit
56d90642a5
@ -247,3 +247,215 @@ func TestStoreQueryWith5Pagination(t *testing.T) {
|
||||
|
||||
Debug("Test successfully verified store query with pagination limit")
|
||||
}
|
||||
|
||||
func TestStoreQueryWithPaginationMultiplePages(t *testing.T) {
|
||||
Debug("Starting test to verify store query with pagination across multiple pages")
|
||||
|
||||
node1Config := DefaultWakuConfig
|
||||
node1Config.Relay = true
|
||||
|
||||
Debug("Creating Node1 (Relay enabled)")
|
||||
node1, err := StartWakuNode("Node1", &node1Config)
|
||||
require.NoError(t, err, "Failed to start Node1")
|
||||
|
||||
node2Config := DefaultWakuConfig
|
||||
node2Config.Relay = true
|
||||
node2Config.Store = true
|
||||
|
||||
Debug("Creating Node2 (Relay & Store enabled)")
|
||||
node2, err := StartWakuNode("Node2", &node2Config)
|
||||
require.NoError(t, err, "Failed to start Node2")
|
||||
|
||||
node3Config := DefaultWakuConfig
|
||||
node3Config.Relay = false
|
||||
|
||||
Debug("Creating Node3 (Peer connected to Node2)")
|
||||
node3, err := StartWakuNode("Node3", &node3Config)
|
||||
require.NoError(t, err, "Failed to start Node3")
|
||||
|
||||
defer func() {
|
||||
Debug("Stopping and destroying all Waku nodes")
|
||||
node1.StopAndDestroy()
|
||||
node2.StopAndDestroy()
|
||||
node3.StopAndDestroy()
|
||||
}()
|
||||
|
||||
Debug("Connecting Node1 to Node2")
|
||||
err = node1.ConnectPeer(node2)
|
||||
require.NoError(t, err, "Failed to connect Node1 to Node2")
|
||||
|
||||
Debug("Connecting Node3 to Node2")
|
||||
err = node3.ConnectPeer(node2)
|
||||
require.NoError(t, err, "Failed to connect Node3 to Node2")
|
||||
|
||||
Debug("Waiting for peer connections to stabilize")
|
||||
time.Sleep(2 * time.Second)
|
||||
|
||||
numMessages := 8
|
||||
var sentHashes []common.MessageHash
|
||||
defaultPubsubTopic := DefaultPubsubTopic
|
||||
|
||||
Debug("Publishing %d messages from Node1 using RelayPublish", numMessages)
|
||||
for i := 0; i < numMessages; i++ {
|
||||
message := node1.CreateMessage(&pb.WakuMessage{
|
||||
Payload: []byte(fmt.Sprintf("message-%d", i)),
|
||||
ContentTopic: "test-content-topic",
|
||||
Timestamp: proto.Int64(time.Now().UnixNano()),
|
||||
})
|
||||
|
||||
msgHash, err := node1.RelayPublishNoCTX(defaultPubsubTopic, message)
|
||||
require.NoError(t, err, "Failed to publish message from Node1")
|
||||
|
||||
sentHashes = append(sentHashes, msgHash)
|
||||
}
|
||||
|
||||
Debug("Waiting for message delivery to Node2")
|
||||
time.Sleep(5 * time.Second)
|
||||
|
||||
Debug("Node3 querying first page of stored messages from Node2")
|
||||
storeRequest1 := common.StoreQueryRequest{
|
||||
IncludeData: true,
|
||||
ContentTopics: &[]string{"test-content-topic"},
|
||||
PaginationLimit: proto.Uint64(5),
|
||||
PaginationForward: true,
|
||||
}
|
||||
|
||||
res1, err := node3.GetStoredMessages(node2, &storeRequest1)
|
||||
require.NoError(t, err, "Failed to retrieve first page of stored messages from Node2")
|
||||
require.NotNil(t, res1.Messages, "Expected stored messages but received nil")
|
||||
|
||||
storedMessages1 := *res1.Messages
|
||||
require.Len(t, storedMessages1, 5, "Expected to retrieve exactly 5 messages from first query")
|
||||
|
||||
for i := 0; i < 5; i++ {
|
||||
require.Equal(t, sentHashes[i], storedMessages1[i].MessageHash, "Message order mismatch in first query")
|
||||
}
|
||||
|
||||
Debug("Node3 querying second page of stored messages from Node2")
|
||||
storeRequest2 := common.StoreQueryRequest{
|
||||
IncludeData: true,
|
||||
ContentTopics: &[]string{"test-content-topic"},
|
||||
PaginationLimit: proto.Uint64(5),
|
||||
PaginationForward: true,
|
||||
PaginationCursor: &res1.PaginationCursor,
|
||||
}
|
||||
|
||||
res2, err := node3.GetStoredMessages(node2, &storeRequest2)
|
||||
require.NoError(t, err, "Failed to retrieve second page of stored messages from Node2")
|
||||
require.NotNil(t, res2.Messages, "Expected stored messages but received nil")
|
||||
|
||||
storedMessages2 := *res2.Messages
|
||||
require.Len(t, storedMessages2, 3, "Expected to retrieve exactly 3 messages from second query")
|
||||
|
||||
for i := 0; i < 3; i++ {
|
||||
require.Equal(t, sentHashes[i+5], storedMessages2[i].MessageHash, "Message order mismatch in second query")
|
||||
}
|
||||
|
||||
Debug("Test successfully verified store query pagination across multiple pages")
|
||||
}
|
||||
|
||||
func TestStoreQueryWithPaginationReverseOrder(t *testing.T) {
|
||||
Debug("Starting test to verify store query with pagination in reverse order")
|
||||
|
||||
node1Config := DefaultWakuConfig
|
||||
node1Config.Relay = true
|
||||
|
||||
Debug("Creating Node1 (Relay enabled)")
|
||||
node1, err := StartWakuNode("Node1", &node1Config)
|
||||
require.NoError(t, err, "Failed to start Node1")
|
||||
|
||||
node2Config := DefaultWakuConfig
|
||||
node2Config.Relay = true
|
||||
node2Config.Store = true
|
||||
|
||||
Debug("Creating Node2 (Relay & Store enabled)")
|
||||
node2, err := StartWakuNode("Node2", &node2Config)
|
||||
require.NoError(t, err, "Failed to start Node2")
|
||||
|
||||
node3Config := DefaultWakuConfig
|
||||
node3Config.Relay = false
|
||||
|
||||
Debug("Creating Node3 (Peer connected to Node2)")
|
||||
node3, err := StartWakuNode("Node3", &node3Config)
|
||||
require.NoError(t, err, "Failed to start Node3")
|
||||
|
||||
defer func() {
|
||||
Debug("Stopping and destroying all Waku nodes")
|
||||
node1.StopAndDestroy()
|
||||
node2.StopAndDestroy()
|
||||
node3.StopAndDestroy()
|
||||
}()
|
||||
|
||||
Debug("Connecting Node1 to Node2")
|
||||
err = node1.ConnectPeer(node2)
|
||||
require.NoError(t, err, "Failed to connect Node1 to Node2")
|
||||
|
||||
Debug("Connecting Node3 to Node2")
|
||||
err = node3.ConnectPeer(node2)
|
||||
require.NoError(t, err, "Failed to connect Node3 to Node2")
|
||||
|
||||
Debug("Waiting for peer connections to stabilize")
|
||||
time.Sleep(2 * time.Second)
|
||||
|
||||
numMessages := 8
|
||||
var sentHashes []common.MessageHash
|
||||
defaultPubsubTopic := DefaultPubsubTopic
|
||||
|
||||
Debug("Publishing %d messages from Node1 using RelayPublish", numMessages)
|
||||
for i := 0; i < numMessages; i++ {
|
||||
message := node1.CreateMessage(&pb.WakuMessage{
|
||||
Payload: []byte(fmt.Sprintf("message-%d", i)),
|
||||
ContentTopic: "test-content-topic",
|
||||
Timestamp: proto.Int64(time.Now().UnixNano()),
|
||||
})
|
||||
|
||||
msgHash, err := node1.RelayPublishNoCTX(defaultPubsubTopic, message)
|
||||
require.NoError(t, err, "Failed to publish message from Node1")
|
||||
|
||||
sentHashes = append(sentHashes, msgHash)
|
||||
}
|
||||
|
||||
Debug("Waiting for message delivery to Node2")
|
||||
time.Sleep(5 * time.Second)
|
||||
|
||||
Debug("Node3 querying first page of stored messages from Node2 (Newest first)")
|
||||
storeRequest1 := common.StoreQueryRequest{
|
||||
IncludeData: true,
|
||||
ContentTopics: &[]string{"test-content-topic"},
|
||||
PaginationLimit: proto.Uint64(5),
|
||||
PaginationForward: false,
|
||||
}
|
||||
|
||||
res1, err := node3.GetStoredMessages(node2, &storeRequest1)
|
||||
require.NoError(t, err, "Failed to retrieve first page of stored messages from Node2")
|
||||
require.NotNil(t, res1.Messages, "Expected stored messages but received nil")
|
||||
|
||||
storedMessages1 := *res1.Messages
|
||||
require.Len(t, storedMessages1, 5, "Expected to retrieve exactly 5 messages from first query")
|
||||
|
||||
for i := 0; i < 5; i++ {
|
||||
require.Equal(t, sentHashes[numMessages-1-i], storedMessages1[i].MessageHash, "Message order mismatch in first query")
|
||||
}
|
||||
|
||||
Debug("Node3 querying second page of stored messages from Node2")
|
||||
storeRequest2 := common.StoreQueryRequest{
|
||||
IncludeData: true,
|
||||
ContentTopics: &[]string{"test-content-topic"},
|
||||
PaginationLimit: proto.Uint64(5),
|
||||
PaginationForward: false,
|
||||
PaginationCursor: &res1.PaginationCursor,
|
||||
}
|
||||
|
||||
res2, err := node3.GetStoredMessages(node2, &storeRequest2)
|
||||
require.NoError(t, err, "Failed to retrieve second page of stored messages from Node2")
|
||||
require.NotNil(t, res2.Messages, "Expected stored messages but received nil")
|
||||
|
||||
storedMessages2 := *res2.Messages
|
||||
require.Len(t, storedMessages2, 3, "Expected to retrieve exactly 3 messages from second query")
|
||||
|
||||
for i := 0; i < 3; i++ {
|
||||
require.Equal(t, sentHashes[numMessages-6-i], storedMessages2[i].MessageHash, "Message order mismatch in second query")
|
||||
}
|
||||
|
||||
Debug("Test successfully verified store query pagination in reverse order")
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user