81 lines
2.6 KiB
Go
81 lines
2.6 KiB
Go
package whisper
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/syndtr/goleveldb/leveldb/errors"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/p2p/enode"
|
|
)
|
|
|
|
func checkValidErrorPayload(t *testing.T, id []byte, errorMsg string) {
|
|
requestID := common.BytesToHash(id)
|
|
errPayload := CreateMailServerRequestFailedPayload(requestID, errors.New(errorMsg))
|
|
nid := enode.ID{1}
|
|
event, err := CreateMailServerEvent(nid, errPayload)
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, event)
|
|
require.Equal(t, nid, event.Peer)
|
|
require.Equal(t, requestID, event.Hash)
|
|
|
|
eventData, ok := event.Data.(*MailServerResponse)
|
|
if !ok {
|
|
require.FailNow(t, "Unexpected data in event: %v, expected a MailServerResponse", event.Data)
|
|
}
|
|
require.EqualError(t, eventData.Error, errorMsg)
|
|
}
|
|
|
|
func checkValidSuccessPayload(t *testing.T, id []byte, lastHash []byte, timestamp uint32, envHash []byte) {
|
|
requestID := common.BytesToHash(id)
|
|
lastEnvelopeHash := common.BytesToHash(lastHash)
|
|
timestampBytes := make([]byte, 4)
|
|
binary.LittleEndian.PutUint32(timestampBytes, timestamp)
|
|
envelopeHash := common.BytesToHash(envHash)
|
|
cursor := append(timestampBytes, envelopeHash[:]...)
|
|
successPayload := CreateMailServerRequestCompletedPayload(common.BytesToHash(id), lastEnvelopeHash, cursor)
|
|
nid := enode.ID{1}
|
|
event, err := CreateMailServerEvent(nid, successPayload)
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, event)
|
|
require.Equal(t, nid, event.Peer)
|
|
require.Equal(t, requestID, event.Hash)
|
|
|
|
eventData, ok := event.Data.(*MailServerResponse)
|
|
if !ok {
|
|
require.FailNow(t, "Unexpected data in event: %v, expected a MailServerResponse", event.Data)
|
|
}
|
|
require.Equal(t, lastEnvelopeHash, eventData.LastEnvelopeHash)
|
|
require.Equal(t, cursor, eventData.Cursor)
|
|
require.NoError(t, eventData.Error)
|
|
}
|
|
|
|
func TestCreateMailServerEvent(t *testing.T) {
|
|
// valid cases
|
|
longErrorMessage := "longMessage|"
|
|
for i := 0; i < 5; i++ {
|
|
longErrorMessage = longErrorMessage + longErrorMessage
|
|
}
|
|
checkValidErrorPayload(t, []byte{0x01}, "test error 1")
|
|
checkValidErrorPayload(t, []byte{0x02}, "test error 2")
|
|
checkValidErrorPayload(t, []byte{0x02}, "")
|
|
checkValidErrorPayload(t, []byte{0x00}, "test error 3")
|
|
checkValidErrorPayload(t, []byte{}, "test error 4")
|
|
|
|
checkValidSuccessPayload(t, []byte{0x01}, []byte{0x02}, 123, []byte{0x03})
|
|
// invalid payloads
|
|
|
|
// too small
|
|
_, err := CreateMailServerEvent(enode.ID{}, []byte{0x00})
|
|
require.Error(t, err)
|
|
|
|
// too big and not error payload
|
|
payloadTooBig := make([]byte, common.HashLength*2+cursorSize+100)
|
|
_, err = CreateMailServerEvent(enode.ID{}, payloadTooBig)
|
|
require.Error(t, err)
|
|
}
|