1package poseidon
 2
 3import (
 4	"encoding/hex"
 5	"math/big"
 6	"testing"
 7
 8	"github.com/iden3/go-iden3-crypto/utils"
 9	"github.com/stretchr/testify/assert"
10	"golang.org/x/crypto/blake2b"
11)
12
13func TestBlake2bVersion(t *testing.T) {
14	h := blake2b.Sum256([]byte("poseidon_constants"))
15	assert.Equal(t, "e57ba154fb2c47811dc1a2369b27e25a44915b4e4ece4eb8ec74850cb78e01b1", hex.EncodeToString(h[:]))
16}
17
18func TestPoseidonHash(t *testing.T) {
19	b0 := big.NewInt(0)
20	b1 := big.NewInt(1)
21	b2 := big.NewInt(2)
22
23	h, err := Hash([]*big.Int{b1})
24	assert.Nil(t, err)
25	assert.Equal(t, "11043376183861534927536506085090418075369306574649619885724436265926427398571", h.String())
26
27	h, err = Hash([]*big.Int{b1, b2})
28	assert.Nil(t, err)
29	assert.Equal(t, "17117985411748610629288516079940078114952304104811071254131751175361957805920", h.String())
30
31	h, err = Hash([]*big.Int{b1, b2, b0, b0, b0})
32	assert.Nil(t, err)
33	assert.Equal(t, "3975478831357328722254985704342968745327876719981393787143845259590563829094", h.String())
34	h, err = Hash([]*big.Int{b1, b2, b0, b0, b0, b0})
35	assert.Nil(t, err)
36	assert.Equal(t, "19772360636270345724087386688434825760738403416279047262510528378903625000110", h.String())
37
38	b3 := big.NewInt(3)
39	b4 := big.NewInt(4)
40	h, err = Hash([]*big.Int{b3, b4, b0, b0, b0})
41	assert.Nil(t, err)
42	assert.Equal(t, "3181200837746671699652342497997860344148947482942465819251904554707352676086", h.String())
43	h, err = Hash([]*big.Int{b3, b4, b0, b0, b0, b0})
44	assert.Nil(t, err)
45	assert.Equal(t, "8386348873272147968934270337233829407378789978142456170950021426339096575008", h.String())
46
47	b5 := big.NewInt(5)
48	b6 := big.NewInt(6)
49	h, err = Hash([]*big.Int{b1, b2, b3, b4, b5, b6})
50	assert.Nil(t, err)
51	assert.Equal(t, "5202465217520500374834597824465244016759843635092906214933648999760272616044", h.String())
52}
53
54func TestErrorInputs(t *testing.T) {
55	b0 := big.NewInt(0)
56	b1 := big.NewInt(1)
57	b2 := big.NewInt(2)
58
59	_, err := Hash([]*big.Int{b1, b2, b0, b0, b0, b0})
60	assert.Nil(t, err)
61
62	_, err = Hash([]*big.Int{b1, b2, b0, b0, b0, b0, b0})
63	assert.NotNil(t, err)
64	assert.Equal(t, "invalid inputs length 7, max 7", err.Error())
65
66	_, err = Hash([]*big.Int{b1, b2, b0, b0, b0, b0, b0, b0})
67	assert.NotNil(t, err)
68	assert.Equal(t, "invalid inputs length 8, max 7", err.Error())
69}
70
71func BenchmarkPoseidonHash(b *testing.B) {
72	b0 := big.NewInt(0)
73	b1 := utils.NewIntFromString("12242166908188651009877250812424843524687801523336557272219921456462821518061")
74	b2 := utils.NewIntFromString("12242166908188651009877250812424843524687801523336557272219921456462821518061")
75
76	bigArray4 := []*big.Int{b1, b2, b0, b0, b0, b0}
77
78	for i := 0; i < b.N; i++ {
79		Hash(bigArray4) //nolint:errcheck
80	}
81}