You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

122 lines
3.0 KiB

package mbt
import (
"io/ioutil"
"path/filepath"
"testing"
"time"
"github.com/stretchr/testify/require"
tmjson "github.com/tendermint/tendermint/libs/json"
"github.com/tendermint/tendermint/light"
"github.com/tendermint/tendermint/types"
)
const jsonDir = "./json"
func TestVerify(t *testing.T) {
filenames := jsonFilenames(t)
for _, filename := range filenames {
filename := filename
t.Run(filename, func(t *testing.T) {
jsonBlob, err := ioutil.ReadFile(filename)
if err != nil {
t.Fatal(err)
}
var tc testCase
err = tmjson.Unmarshal(jsonBlob, &tc)
if err != nil {
t.Fatal(err)
}
t.Log(tc.Description)
var (
trustedSignedHeader = tc.Initial.SignedHeader
trustedNextVals = tc.Initial.NextValidatorSet
trustingPeriod = time.Duration(tc.Initial.TrustingPeriod) * time.Nanosecond
)
for _, input := range tc.Input {
var (
newSignedHeader = input.LightBlock.SignedHeader
newVals = input.LightBlock.ValidatorSet
)
err = light.Verify(
&trustedSignedHeader,
&trustedNextVals,
newSignedHeader,
newVals,
trustingPeriod,
input.Now,
1*time.Second,
light.DefaultTrustLevel,
)
t.Logf("%d -> %d", trustedSignedHeader.Height, newSignedHeader.Height)
switch input.Verdict {
case "SUCCESS":
require.NoError(t, err)
case "NOT_ENOUGH_TRUST":
require.IsType(t, light.ErrNewValSetCantBeTrusted{}, err)
case "INVALID":
switch err.(type) {
case light.ErrOldHeaderExpired:
case light.ErrInvalidHeader:
default:
t.Fatalf("expected either ErrInvalidHeader or ErrOldHeaderExpired, but got %v", err)
}
default:
t.Fatalf("unexpected verdict: %q", input.Verdict)
}
if err == nil { // advance
trustedSignedHeader = *newSignedHeader
trustedNextVals = *input.LightBlock.NextValidatorSet
}
}
})
}
}
// jsonFilenames returns a list of files in jsonDir directory
func jsonFilenames(t *testing.T) []string {
matches, err := filepath.Glob(filepath.Join(jsonDir, "*.json"))
if err != nil {
t.Fatal(err)
}
return matches
}
type testCase struct {
Description string `json:"description"`
Initial initialData `json:"initial"`
Input []inputData `json:"input"`
}
type initialData struct {
SignedHeader types.SignedHeader `json:"signed_header"`
NextValidatorSet types.ValidatorSet `json:"next_validator_set"`
TrustingPeriod uint64 `json:"trusting_period"`
Now time.Time `json:"now"`
}
type inputData struct {
LightBlock lightBlockWithNextValidatorSet `json:"block"`
Now time.Time `json:"now"`
Verdict string `json:"verdict"`
}
// In tendermint-rs, NextValidatorSet is used to verify new blocks (opposite to
// Go tendermint).
type lightBlockWithNextValidatorSet struct {
*types.SignedHeader `json:"signed_header"`
ValidatorSet *types.ValidatorSet `json:"validator_set"`
NextValidatorSet *types.ValidatorSet `json:"next_validator_set"`
}