vectors_test.go

package ics23

import (
	"bytes"
	"fmt"
	"testing"
)

func TestVectors(t *testing.T) {
	cases := VectorsTestData()
	for _, tc := range cases {
		tc := tc
		name := fmt.Sprintf("%s/%s", tc.Dir, tc.Filename)
		t.Run(name, func(t *testing.T) {
			proof, ref := LoadFile(t, tc.Dir, tc.Filename)

			// Test Calculate method
			calculatedRoot, err := proof.Calculate()
			if err != nil {
				t.Fatal("proof.Calculate() returned error")
			}
			if !bytes.Equal(ref.RootHash, calculatedRoot) {
				t.Fatalf("calculated root: %X did not match expected root: %X", calculatedRoot, ref.RootHash)
			}
			// Test Verify method
			if ref.Value == nil {
				// non-existence
				valid := VerifyNonMembership(tc.Spec, ref.RootHash, proof, ref.Key)
				if !valid {
					t.Fatalf("Invalid proof: %v", err)
				}
			} else {
				valid := VerifyMembership(tc.Spec, ref.RootHash, proof, ref.Key, ref.Value)
				if !valid {
					t.Fatalf("Invalid proof: %v", err)
				}
			}
		})
	}
}