62 lines
1.8 KiB
Go
62 lines
1.8 KiB
Go
package proof
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestProofFromBytesSlices(t *testing.T) {
|
|
tests := []struct {
|
|
keys []string
|
|
values []string
|
|
want string
|
|
}{
|
|
{[]string{}, []string{}, "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855"},
|
|
{[]string{"key1"}, []string{"value1"}, "a44d3cc7daba1a4600b00a2434b30f8b970652169810d6dfa9fb1793a2189324"},
|
|
{[]string{"key1"}, []string{"value2"}, "0638e99b3445caec9d95c05e1a3fc1487b4ddec6a952ff337080360b0dcc078c"},
|
|
// swap order with 2 keys
|
|
{
|
|
[]string{"key1", "key2"},
|
|
[]string{"value1", "value2"},
|
|
"8fd19b19e7bb3f2b3ee0574027d8a5a4cec370464ea2db2fbfa5c7d35bb0cff3",
|
|
},
|
|
{
|
|
[]string{"key2", "key1"},
|
|
[]string{"value2", "value1"},
|
|
"55d4bce1c53b7d394bd41bbfc2b239cc2e1c7e36423612a97181c47e79bb713c",
|
|
},
|
|
// swap order with 3 keys
|
|
{
|
|
[]string{"key1", "key2", "key3"},
|
|
[]string{"value1", "value2", "value3"},
|
|
"1dd674ec6782a0d586a903c9c63326a41cbe56b3bba33ed6ff5b527af6efb3dc",
|
|
},
|
|
{
|
|
[]string{"key1", "key3", "key2"},
|
|
[]string{"value1", "value3", "value2"},
|
|
"443382fbb629e0d50e86d6ea49e22aa4e27ba50262730b0122cec36860c903a2",
|
|
},
|
|
}
|
|
for i, tc := range tests {
|
|
var err error
|
|
leaves := make([][]byte, len(tc.keys))
|
|
for j, key := range tc.keys {
|
|
leaves[j], err = LeafHash([]byte(key), []byte(tc.values[j]))
|
|
require.NoError(t, err)
|
|
}
|
|
for j := range leaves {
|
|
buf := make([][]byte, len(leaves))
|
|
copy(buf, leaves)
|
|
rootHash, inners := ProofFromByteSlices(buf, j)
|
|
require.Equal(t, tc.want, fmt.Sprintf("%x", rootHash), "test case %d", i)
|
|
commitmentOp := ConvertCommitmentOp(inners, []byte(tc.keys[j]), []byte(tc.values[j]))
|
|
expRoots, err := commitmentOp.Run([][]byte{[]byte(tc.values[j])})
|
|
require.NoError(t, err)
|
|
require.Equal(t, tc.want, fmt.Sprintf("%x", expRoots[0]), "test case %d", i)
|
|
}
|
|
|
|
}
|
|
}
|