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)
}