1package mimc7
 2
 3import (
 4	"encoding/hex"
 5	"math/big"
 6	"testing"
 7
 8	"github.com/ethereum/go-ethereum/crypto"
 9	"github.com/iden3/go-iden3-crypto/field"
10	"github.com/stretchr/testify/assert"
11)
12
13func TestKeccak256(t *testing.T) {
14	res := crypto.Keccak256([]byte(SEED))
15	assert.Equal(t, "b6e489e6b37224a50bebfddbe7d89fa8fdcaa84304a70bd13f79b5d9f7951e9e", hex.EncodeToString(res))
16	c := new(big.Int).SetBytes(crypto.Keccak256([]byte(SEED)))
17	assert.Equal(t, "82724731331859054037315113496710413141112897654334566532528783843265082629790", c.String())
18}
19
20func TestMIMC7Generic(t *testing.T) {
21	b1 := big.NewInt(int64(1))
22	b2 := big.NewInt(int64(2))
23	b3 := big.NewInt(int64(3))
24
25	r, ok := new(big.Int).SetString("21888242871839275222246405745257275088548364400416034343698204186575808495617", 10)
26	assert.True(t, ok)
27	fqR := field.NewFq(r)
28
29	bigArray := []*big.Int{b1, b2, b3}
30
31	// Generic Hash
32	mhg := MIMC7HashGeneric(fqR, b1, b2, 91)
33	assert.Equal(t, "10594780656576967754230020536574539122676596303354946869887184401991294982664", mhg.String())
34	hg, err := HashGeneric(fqR.Zero(), bigArray, fqR, 91)
35	assert.Nil(t, err)
36	assert.Equal(t, "6464402164086696096195815557694604139393321133243036833927490113253119343397", (*big.Int)(hg).String())
37}
38
39func TestMIMC7(t *testing.T) {
40	b12 := big.NewInt(int64(12))
41	b45 := big.NewInt(int64(45))
42	b78 := big.NewInt(int64(78))
43	b41 := big.NewInt(int64(41))
44
45	// h1, hash of 1 elements
46	bigArray1 := []*big.Int{b12}
47
48	h1, err := Hash(bigArray1, nil)
49	assert.Nil(t, err)
50	// same hash value than the iden3js and circomlib tests:
51	assert.Equal(t, "0x"+hex.EncodeToString((*big.Int)(h1).Bytes()), "0x237c92644dbddb86d8a259e0e923aaab65a93f1ec5758b8799988894ac0958fd")
52
53	// h2a, hash of 2 elements
54	bigArray2a := []*big.Int{b78, b41}
55
56	h2a, err := Hash(bigArray2a, nil)
57	assert.Nil(t, err)
58	// same hash value than the iden3js and circomlib tests:
59	assert.Equal(t, "0x"+hex.EncodeToString((*big.Int)(h2a).Bytes()), "0x067f3202335ea256ae6e6aadcd2d5f7f4b06a00b2d1e0de903980d5ab552dc70")
60
61	// h2b, hash of 2 elements
62	bigArray2b := []*big.Int{b12, b45}
63
64	mh2b := MIMC7Hash(b12, b45)
65	assert.Nil(t, err)
66	assert.Equal(t, "0x"+hex.EncodeToString((*big.Int)(mh2b).Bytes()), "0x2ba7ebad3c6b6f5a20bdecba2333c63173ca1a5f2f49d958081d9fa7179c44e4")
67
68	h2b, err := Hash(bigArray2b, nil)
69	assert.Nil(t, err)
70	// same hash value than the iden3js and circomlib tests:
71	assert.Equal(t, "0x"+hex.EncodeToString((*big.Int)(h2b).Bytes()), "0x15ff7fe9793346a17c3150804bcb36d161c8662b110c50f55ccb7113948d8879")
72
73	// h4, hash of 4 elements
74	bigArray4 := []*big.Int{b12, b45, b78, b41}
75
76	h4, err := Hash(bigArray4, nil)
77	assert.Nil(t, err)
78	// same hash value than the iden3js and circomlib tests:
79	assert.Equal(t, "0x"+hex.EncodeToString((*big.Int)(h4).Bytes()), "0x284bc1f34f335933a23a433b6ff3ee179d682cd5e5e2fcdd2d964afa85104beb")
80}
81
82func BenchmarkMIMC7(b *testing.B) {
83	b12 := big.NewInt(int64(12))
84	b45 := big.NewInt(int64(45))
85	b78 := big.NewInt(int64(78))
86	b41 := big.NewInt(int64(41))
87	bigArray4 := []*big.Int{b12, b45, b78, b41}
88
89	var h4 *big.Int
90	for i := 0; i < b.N; i++ {
91		h4, _ = Hash(bigArray4, nil)
92	}
93	println(h4)
94}