package checksum

import (
	"crypto/rand"
	"crypto/sha256"
	"testing"

	"github.com/nspcc-dev/neofs-api-go/v2/refs"
	"github.com/nspcc-dev/tzhash/tz"
	"github.com/stretchr/testify/require"
)

func TestChecksum(t *testing.T) {
	var c Checksum

	cSHA256 := [sha256.Size]byte{}
	_, _ = rand.Read(cSHA256[:])

	c.SetSHA256(cSHA256)

	require.Equal(t, SHA256, c.Type())
	require.Equal(t, cSHA256[:], c.Value())

	var cV2 refs.Checksum
	c.WriteToV2(&cV2)

	require.Equal(t, refs.SHA256, cV2.GetType())
	require.Equal(t, cSHA256[:], cV2.GetSum())

	cTZ := [tz.Size]byte{}
	_, _ = rand.Read(cSHA256[:])

	c.SetTillichZemor(cTZ)

	require.Equal(t, TZ, c.Type())
	require.Equal(t, cTZ[:], c.Value())

	c.WriteToV2(&cV2)

	require.Equal(t, refs.TillichZemor, cV2.GetType())
	require.Equal(t, cTZ[:], cV2.GetSum())
}

func TestNewChecksum(t *testing.T) {
	t.Run("default values", func(t *testing.T) {
		var chs Checksum

		// check initial values
		require.Equal(t, Unknown, chs.Type())
		require.Nil(t, chs.Value())

		// convert to v2 message
		var chsV2 refs.Checksum
		chs.WriteToV2(&chsV2)

		require.Equal(t, refs.UnknownChecksum, chsV2.GetType())
		require.Nil(t, chsV2.GetSum())
	})
}

func TestCalculation(t *testing.T) {
	var c Checksum
	payload := []byte{0, 1, 2, 3, 4, 5}

	t.Run("SHA256", func(t *testing.T) {
		orig := sha256.Sum256(payload)

		Calculate(&c, SHA256, payload)

		require.Equal(t, orig[:], c.Value())
	})

	t.Run("TZ", func(t *testing.T) {
		orig := tz.Sum(payload)

		Calculate(&c, TZ, payload)

		require.Equal(t, orig[:], c.Value())
	})
}