Files
tendermint/light/mbt/driver_test.go
Anton Kaliaev 8e6194626e light: model-based tests (#5461)
This is the first iteration of model-based testing in Go Tendermint. The test runner is using the static JSON fixtures located under the ./json directory. In the future, the Rust tensgen binary will be used to generate those (given the static intermediate scenarios and the test seed, which will be published along with each testgen release).

Closes: #5322
2020-11-02 12:07:18 +04:00

123 lines
3.0 KiB
Go

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"`
}