43 lines
1015 B
Go
43 lines
1015 B
Go
package common
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
gossipv1 "github.com/certusone/wormhole/node/pkg/proto/gossip/v1"
|
|
"github.com/certusone/wormhole/node/pkg/vaa"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestObsvReqSendLimitEnforced(t *testing.T) {
|
|
obsvReqSendC := make(chan *gossipv1.ObservationRequest, ObsvReqChannelSize)
|
|
|
|
// If the channel overflows, the write hangs, so use a go routine with a timeout.
|
|
done := false
|
|
go func() {
|
|
// Filling the queue up should work.
|
|
for count := 1; count <= ObsvReqChannelSize; count++ {
|
|
req := &gossipv1.ObservationRequest{
|
|
ChainId: uint32(vaa.ChainIDSolana),
|
|
}
|
|
err := PostObservationRequest(obsvReqSendC, req)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
// But one more write should fail.
|
|
req := &gossipv1.ObservationRequest{
|
|
ChainId: uint32(vaa.ChainIDSolana),
|
|
}
|
|
err := PostObservationRequest(obsvReqSendC, req)
|
|
assert.ErrorIs(t, err, ErrChanFull)
|
|
|
|
done = true
|
|
}()
|
|
|
|
time.Sleep(time.Second)
|
|
|
|
// Make sure we didn't hang.
|
|
assert.Equal(t, true, done)
|
|
}
|