package hashes

import (
	"fmt"
	"math/rand"
	"testing"
)

func TestNewBlockHash(t *testing.T) {
	datas := [][]byte{
		{},
		{1},
		{5, 199, 126, 44, 71, 32, 82, 139, 122, 217, 43, 48, 52, 112, 40, 209, 180, 83, 139, 231, 72, 48, 136, 48, 168, 226, 133, 7, 60, 4, 160, 205},
		{42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42},
		{0, 0, 0, 0, 0, 0, 0, 0},
	}
	tests := []struct {
		hasher   HashWriter
		expected []string
	}{
		{NewTransactionHashWriter(), []string{
			"50272a9e37c728026f93d0eda6ab4467f627338b879076483c88d291193cb3bf",
			"f9bf7e04c712621a0f4bb75d763f9ef5f73af6c438fd15b80744393bc96398ad",
			"8e791f3edcc92b71b8de2778efbc4666ee5bd146acbe8723a55bca26b022b0e0",
			"a6dab1a3088548c62d13a082fa28e870fdbbe51adcd8c364e2ea37e473c04d81",
			"3b79b78b967233843ad30f707b165eb3d6a91af8338076be8755c46a963c3d1d",
		}},
		{NewTransactionIDWriter(), []string{
			"e5f65efda0894d2b0590c2e9e46e9acc03032f505a1522f5e8c78c5ec70b1d9c",
			"aea52cf5e5a13da13a52dd69abd636eb1b0f86e58bc1dda6b17886b94593415a",
			"a50a2f87bdce075740189e9e23907ae22b5addbd875ccb70c116811b1fa5fb18",
			"0db7a485f7013a346a8f7f5caf73d52ca3c3b5ee101ad8753adedd4235b7236b",
			"2afc9c855854b0a6e94a722c3451d0cdfc8c11748b78ef65b9786f87b48d0d07",
		}},
		{NewTransactionSigningHashWriter(), []string{
			"34c75037ad62740d4b3228f88f844f7901c07bfacd55a045be518eabc15e52ce",
			"8523b0471bcbea04575ccaa635eef9f9114f2890bda54367e5ff8caa3878bf82",
			"a51c49d9eb3d13f9de16e1aa8d1ff17668d55633ce00f36a643ac714b0fb137f",
			"487f199ef74c3e893e85bd37770e6334575a2d4d113b2e10474593c49807de93",
			"6392adc33a8e24e9a0a0c4c5f07f9c1cc958ad40c16d7a9a276e374cebb4e32b",
		}},
		{NewTransactionSigningHashECDSAWriter(), []string{
			"b31ad1fbbe41b0e2a90e07c84708b38ba581f0c0e9185416913a04fb6d342027",
			"c43e1f75ea9df6379b56a95074c2b6289ed8c5a01fff2d49d9d44ad5575c164b",
			"49085f99fa0084b5436663f757a5916b1e4290c3321707fb76921ed4e47844ec",
			"3f887e866428de813c1d0463b14eef3ca1363c8187e917dda1eee0ec5996490b",
			"56de89a8c75f0fee2de61b11ab05d0d42e29ed50879467cf128dd80800a52ada",
		}},
		{NewBlockHashWriter(), []string{
			"a80b6aa20f20b15ebabe2b1949527f78a257594a732e774de637d85e6973a768",
			"5643023add641f9421187b8c9aa3c6c73227d5ec34131c61a08d35b43e7e4b65",
			"4dc3bf72045431e46f8839a7d390898f27c887fddd8637149bfb70f732f04334",
			"15d7648e69023dca65c949a61ea166192049f449c604523494813873b19918a7",
			"3ac41af8385ea5d902ce6d47f509b7accc9c631f1d57a719d777874467f6d877",
		}},
		{NewMerkleBranchHashWriter(), []string{
			"4de3617db456d01248173f17ec58196e92fbd994b636476db4b875ed2ec84054",
			"5737cd8b6fca5a30c19a491323a14e6b7021641cb3f8875f10c7a2eafd3cf43f",
			"a49eeda61cc75e0a8e5915829752fe0ad97620d6d32de7c9883595b0810ca33e",
			"28f33681dcff1313674e07dacc2d74c3089f6d8cea7a4f8792a71fd870988ee5",
			"2d53a43a42020a5091c125230bcd8a4cf0eeb188333e68325d4bce58a1c75ca3",
		}},
	}

	for _, testVector := range tests {
		hasher := testVector.hasher
		for i, data := range datas {
			hasher.InfallibleWrite(data)
			res := hasher.Finalize().String()
			if res != testVector.expected[i] {
				panic(fmt.Sprintf("expected: %s, got: %s", testVector.expected[i], res))
			}
		}
	}

}

func BenchmarkNewBlockHashWriterSmall(b *testing.B) {
	r := rand.New(rand.NewSource(0))
	var someBytes [32]byte
	r.Read(someBytes[:])
	for i := 0; i < b.N; i++ {
		hasher := NewBlockHashWriter()
		hasher.InfallibleWrite(someBytes[:])
		hasher.Finalize()
	}
}

func BenchmarkNewBlockHashWriterBig(b *testing.B) {
	r := rand.New(rand.NewSource(0))
	var someBytes [1024]byte
	r.Read(someBytes[:])
	for i := 0; i < b.N; i++ {
		hasher := NewBlockHashWriter()
		hasher.InfallibleWrite(someBytes[:])
		hasher.Finalize()
	}
}

func BenchmarkNewHeavyHashWriterSmall(b *testing.B) {
	r := rand.New(rand.NewSource(0))
	var someBytes [32]byte
	r.Read(someBytes[:])
	for i := 0; i < b.N; i++ {
		hasher := NewHeavyHashWriter()
		hasher.InfallibleWrite(someBytes[:])
		hasher.Finalize()
	}
}

func BenchmarkNewHeavyHashWriterBig(b *testing.B) {
	r := rand.New(rand.NewSource(0))
	var someBytes [1024]byte
	r.Read(someBytes[:])
	for i := 0; i < b.N; i++ {
		hasher := NewHeavyHashWriter()
		hasher.InfallibleWrite(someBytes[:])
		hasher.Finalize()
	}
}