package hashtools

import (
	"encoding/hex"
	"testing"
)

func TestAll(t *testing.T) {
	t.Parallel()

	testData := []byte("The quick brown fox jumps over the lazy dog.")

	all := AsList()
	for _, hashTool := range all {
		// Test hash usage.
		hash, err := New(hashTool.Name)
		if err != nil {
			t.Fatalf("failed to get HashTool %s", hashTool.Name)
		}

		if hash.BlockSize() != hashTool.BlockSize {
			t.Errorf("hashTool %s is broken or reports invalid block size. Expected %d, got %d.", hashTool.Name, hashTool.BlockSize, hash.BlockSize())
		}

		_, err = hash.Write(testData)
		if err != nil {
			t.Errorf("hashTool %s failed to write: %s", hashTool.Name, err)
		}

		sum := hash.Sum(nil)
		if len(sum) != hashTool.DigestSize {
			t.Errorf("hashTool %s is broken or reports invalid digest size. Expected %d, got %d.", hashTool.Name, hashTool.DigestSize, len(sum))
		}

		// Check hash outputs.
		expectedOutputs, ok := testOutputs[hashTool.Name]
		if !ok {
			t.Errorf("no test outputs available for %s", hashTool.Name)
			continue
		}

		// Test empty string.
		hash.Reset()
		_, _ = hash.Write(testInputEmpty)
		hexSum := hex.EncodeToString(hash.Sum(nil))
		if hexSum != expectedOutputs[0] {
			t.Errorf("hash tool %s: test empty: digest mismatch, expected %+v, got %+v",
				hashTool.Name, expectedOutputs[0], hexSum)
		}

		// Test fox string.
		hash.Reset()
		_, _ = hash.Write(testInputFox)
		hexSum = hex.EncodeToString(hash.Sum(nil))
		if hexSum != expectedOutputs[1] {
			t.Errorf("hash tool %s: test empty: digest mismatch, expected %+v, got %+v",
				hashTool.Name, expectedOutputs[1], hexSum)
		}
	}
}

var (
	testInputEmpty = []byte("")
	testInputFox   = []byte("The quick brown fox jumps over the lazy dog.")
)

var testOutputs = map[string][2]string{
	"SHA2-224": {
		"d14a028c2a3a2bc9476102bb288234c415a2b01f828ea62ac5b3e42f",
		"619cba8e8e05826e9b8c519c0a5c68f4fb653e8a3d8aa04bb2c8cd4c",
	},
	"SHA2-256": {
		"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
		"ef537f25c895bfa782526529a9b63d97aa631564d5d789c2b765448c8635fb6c",
	},
	"SHA2-384": {
		"38b060a751ac96384cd9327eb1b1e36a21fdb71114be07434c0cc7bf63f6e1da274edebfe76f65fbd51ad2f14898b95b",
		"ed892481d8272ca6df370bf706e4d7bc1b5739fa2177aae6c50e946678718fc67a7af2819a021c2fc34e91bdb63409d7",
	},
	"SHA2-512": {
		"cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e",
		"91ea1245f20d46ae9a037a989f54f1f790f0a47607eeb8a14d12890cea77a1bbc6c7ed9cf205e67b7f2b8fd4c7dfd3a7a8617e45f3c463d481c7e586c39ac1ed",
	},
	"SHA2-512-224": {
		"6ed0dd02806fa89e25de060c19d3ac86cabb87d6a0ddd05c333b84f4",
		"6d6a9279495ec4061769752e7ff9c68b6b0b3c5a281b7917ce0572de",
	},
	"SHA2-512-256": {
		"c672b8d1ef56ed28ab87c3622c5114069bdd3ad7b8f9737498d0c01ecef0967a",
		"1546741840f8a492b959d9b8b2344b9b0eb51b004bba35c0aebaac86d45264c3",
	},
	"SHA3-224": {
		"6b4e03423667dbb73b6e15454f0eb1abd4597f9a1b078e3f5b5a6bc7",
		"2d0708903833afabdd232a20201176e8b58c5be8a6fe74265ac54db0",
	},
	"SHA3-256": {
		"a7ffc6f8bf1ed76651c14756a061d662f580ff4de43b49fa82d80a4b80f8434a",
		"a80f839cd4f83f6c3dafc87feae470045e4eb0d366397d5c6ce34ba1739f734d",
	},
	"SHA3-384": {
		"0c63a75b845e4f7d01107d852e4c2485c51a50aaaa94fc61995e71bbee983a2ac3713831264adb47fb6bd1e058d5f004",
		"1a34d81695b622df178bc74df7124fe12fac0f64ba5250b78b99c1273d4b080168e10652894ecad5f1f4d5b965437fb9",
	},
	"SHA3-512": {
		"a69f73cca23a9ac5c8b567dc185a756e97c982164fe25859e0d1dcc1475c80a615b2123af1f5f94c11e3e9402c3ac558f500199d95b6d3e301758586281dcd26",
		"18f4f4bd419603f95538837003d9d254c26c23765565162247483f65c50303597bc9ce4d289f21d1c2f1f458828e33dc442100331b35e7eb031b5d38ba6460f8",
	},
	"BLAKE2s-256": {
		"69217a3079908094e11121d042354a7c1f55b6482ca1a51e1b250dfd1ed0eef9",
		"95bca6e1b761dca1323505cc629949a0e03edf11633cc7935bd8b56f393afcf2",
	},
	"BLAKE2b-256": {
		"0e5751c026e543b2e8ab2eb06099daa1d1e5df47778f7787faab45cdf12fe3a8",
		"69d7d3b0afba81826d27024c17f7f183659ed0812cf27b382eaef9fdc29b5712",
	},
	"BLAKE2b-384": {
		"b32811423377f52d7862286ee1a72ee540524380fda1724a6f25d7978c6fd3244a6caf0498812673c5e05ef583825100",
		"16d65de1a3caf1c26247234c39af636284c7e19ca448c0de788272081410778852c94d9cef6b939968d4f872c7f78337",
	},
	"BLAKE2b-512": {
		"786a02f742015903c6c6fd852552d272912f4740e15847618a86e217f71f5419d25e1031afee585313896444934eb04b903a685b1448b755d56f701afe9be2ce",
		"87af9dc4afe5651b7aa89124b905fd214bf17c79af58610db86a0fb1e0194622a4e9d8e395b352223a8183b0d421c0994b98286cbf8c68a495902e0fe6e2bda2",
	},
	"BLAKE3": {
		"af1349b9f5f9a1a6a0404dea36dcc9499bcb25c9adc112b7cc9a93cae41f3262",
		"4c9bd68d7f0baa2e167cef98295eb1ec99a3ec8f0656b33dbae943b387f31d5d",
	},
}