mirror of https://github.com/status-im/go-waku.git
37 lines
1.2 KiB
Go
37 lines
1.2 KiB
Go
|
package store
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/status-im/go-waku/tests"
|
||
|
|
"github.com/status-im/go-waku/waku/v2/protocol/pb"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMessageQueue(t *testing.T) {
|
||
|
|
msg1 := tests.CreateWakuMessage("1", 1)
|
||
|
|
msg2 := tests.CreateWakuMessage("2", 2)
|
||
|
|
msg3 := tests.CreateWakuMessage("3", 3)
|
||
|
|
msg4 := tests.CreateWakuMessage("3", 3)
|
||
|
|
msg5 := tests.CreateWakuMessage("3", 3)
|
||
|
|
|
||
|
|
msgQ := NewMessageQueue(3)
|
||
|
|
msgQ.Push(IndexedWakuMessage{msg: msg1, index: &pb.Index{}, pubsubTopic: "test"})
|
||
|
|
msgQ.Push(IndexedWakuMessage{msg: msg2, index: &pb.Index{}, pubsubTopic: "test"})
|
||
|
|
msgQ.Push(IndexedWakuMessage{msg: msg3, index: &pb.Index{}, pubsubTopic: "test"})
|
||
|
|
|
||
|
|
require.Len(t, msgQ.messages, 3)
|
||
|
|
|
||
|
|
msgQ.Push(IndexedWakuMessage{msg: msg4, index: &pb.Index{}, pubsubTopic: "test"})
|
||
|
|
|
||
|
|
require.Len(t, msgQ.messages, 3)
|
||
|
|
require.Equal(t, msg2.Payload, msgQ.messages[0].msg.Payload)
|
||
|
|
require.Equal(t, msg4.Payload, msgQ.messages[2].msg.Payload)
|
||
|
|
|
||
|
|
msgQ.Push(IndexedWakuMessage{msg: msg5, index: &pb.Index{}, pubsubTopic: "test"})
|
||
|
|
|
||
|
|
require.Len(t, msgQ.messages, 3)
|
||
|
|
require.Equal(t, msg3.Payload, msgQ.messages[0].msg.Payload)
|
||
|
|
require.Equal(t, msg5.Payload, msgQ.messages[2].msg.Payload)
|
||
|
|
}
|