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,
16		"e57ba154fb2c47811dc1a2369b27e25a44915b4e4ece4eb8ec74850cb78e01b1",
17		hex.EncodeToString(h[:]))
18}
19
20func TestPoseidonHash(t *testing.T) {
21	b0 := big.NewInt(0)
22	b1 := big.NewInt(1)
23	b2 := big.NewInt(2)
24
25	h, err := Hash([]*big.Int{b1})
26	assert.Nil(t, err)
27	assert.Equal(t,
28		"18586133768512220936620570745912940619677854269274689475585506675881198879027",
29		h.String())
30
31	h, err = Hash([]*big.Int{b1, b2})
32	assert.Nil(t, err)
33	assert.Equal(t,
34		"7853200120776062878684798364095072458815029376092732009249414926327459813530",
35		h.String())
36
37	h, err = Hash([]*big.Int{b1, b2, b0, b0, b0})
38	assert.Nil(t, err)
39	assert.Equal(t,
40		"1018317224307729531995786483840663576608797660851238720571059489595066344487",
41		h.String())
42	h, err = Hash([]*big.Int{b1, b2, b0, b0, b0, b0})
43	assert.Nil(t, err)
44	assert.Equal(t,
45		"15336558801450556532856248569924170992202208561737609669134139141992924267169",
46		h.String())
47
48	b3 := big.NewInt(3)
49	b4 := big.NewInt(4)
50	h, err = Hash([]*big.Int{b3, b4, b0, b0, b0})
51	assert.Nil(t, err)
52	assert.Equal(t,
53		"5811595552068139067952687508729883632420015185677766880877743348592482390548",
54		h.String())
55	h, err = Hash([]*big.Int{b3, b4, b0, b0, b0, b0})
56	assert.Nil(t, err)
57	assert.Equal(t,
58		"12263118664590987767234828103155242843640892839966517009184493198782366909018",
59		h.String())
60
61	b5 := big.NewInt(5)
62	b6 := big.NewInt(6)
63	h, err = Hash([]*big.Int{b1, b2, b3, b4, b5, b6})
64	assert.Nil(t, err)
65	assert.Equal(t,
66		"20400040500897583745843009878988256314335038853985262692600694741116813247201",
67		h.String())
68}
69
70func TestErrorInputs(t *testing.T) {
71	b0 := big.NewInt(0)
72	b1 := big.NewInt(1)
73	b2 := big.NewInt(2)
74
75	_, err := Hash([]*big.Int{b1, b2, b0, b0, b0, b0})
76	assert.Nil(t, err)
77
78	_, err = Hash([]*big.Int{b1, b2, b0, b0, b0, b0, b0})
79	assert.NotNil(t, err)
80	assert.Equal(t, "invalid inputs length 7, max 7", err.Error())
81
82	_, err = Hash([]*big.Int{b1, b2, b0, b0, b0, b0, b0, b0})
83	assert.NotNil(t, err)
84	assert.Equal(t, "invalid inputs length 8, max 7", err.Error())
85}
86
87func BenchmarkPoseidonHash(b *testing.B) {
88	b0 := big.NewInt(0)
89	b1 := utils.NewIntFromString("12242166908188651009877250812424843524687801523336557272219921456462821518061") //nolint:lll
90	b2 := utils.NewIntFromString("12242166908188651009877250812424843524687801523336557272219921456462821518061") //nolint:lll
91
92	bigArray4 := []*big.Int{b1, b2, b0, b0, b0, b0}
93
94	for i := 0; i < b.N; i++ {
95		Hash(bigArray4) //nolint:errcheck,gosec
96	}
97}