package light_test import ( "context" "os" "testing" "time" "github.com/stretchr/testify/require" dbm "github.com/tendermint/tm-db" "github.com/tendermint/tendermint/abci/example/kvstore" "github.com/tendermint/tendermint/libs/log" "github.com/tendermint/tendermint/light" "github.com/tendermint/tendermint/light/provider" httpp "github.com/tendermint/tendermint/light/provider/http" dbs "github.com/tendermint/tendermint/light/store/db" rpctest "github.com/tendermint/tendermint/rpc/test" "github.com/tendermint/tendermint/types" ) // NOTE: these are ports of the tests from example_test.go but // rewritten as more conventional tests. // Automatically getting new headers and verifying them. func TestClientIntegration_Update(t *testing.T) { t.Parallel() ctx, cancel := context.WithCancel(context.Background()) defer cancel() conf, err := rpctest.CreateConfig(t.Name()) require.NoError(t, err) // Start a test application app := kvstore.NewApplication() _, closer, err := rpctest.StartTendermint(ctx, conf, app, rpctest.SuppressStdout) require.NoError(t, err) defer func() { require.NoError(t, closer(ctx)) }() // give Tendermint time to generate some blocks time.Sleep(5 * time.Second) dbDir, err := os.MkdirTemp("", "light-client-test-update-example") require.NoError(t, err) defer os.RemoveAll(dbDir) chainID := conf.ChainID() primary, err := httpp.New(chainID, conf.RPC.ListenAddress) require.NoError(t, err) // give Tendermint time to generate some blocks block, err := waitForBlock(ctx, primary, 2) require.NoError(t, err) db, err := dbm.NewGoLevelDB("light-client-db", dbDir) require.NoError(t, err) c, err := light.NewClient( ctx, chainID, light.TrustOptions{ Period: 504 * time.Hour, // 21 days Height: 2, Hash: block.Hash(), }, primary, []provider.Provider{primary}, // NOTE: primary should not be used here dbs.New(db), light.Logger(log.TestingLogger()), ) require.NoError(t, err) defer func() { require.NoError(t, c.Cleanup()) }() // ensure Tendermint is at height 3 or higher _, err = waitForBlock(ctx, primary, 3) require.NoError(t, err) h, err := c.Update(ctx, time.Now()) require.NoError(t, err) require.NotNil(t, h) require.True(t, h.Height > 2) } // Manually getting light blocks and verifying them. func TestClientIntegration_VerifyLightBlockAtHeight(t *testing.T) { t.Parallel() ctx, cancel := context.WithCancel(context.Background()) defer cancel() conf, err := rpctest.CreateConfig(t.Name()) require.NoError(t, err) // Start a test application app := kvstore.NewApplication() _, closer, err := rpctest.StartTendermint(ctx, conf, app, rpctest.SuppressStdout) require.NoError(t, err) defer func() { require.NoError(t, closer(ctx)) }() dbDir, err := os.MkdirTemp("", "light-client-test-verify-example") require.NoError(t, err) defer os.RemoveAll(dbDir) chainID := conf.ChainID() primary, err := httpp.New(chainID, conf.RPC.ListenAddress) require.NoError(t, err) // give Tendermint time to generate some blocks block, err := waitForBlock(ctx, primary, 2) require.NoError(t, err) db, err := dbm.NewGoLevelDB("light-client-db", dbDir) require.NoError(t, err) c, err := light.NewClient(ctx, chainID, light.TrustOptions{ Period: 504 * time.Hour, // 21 days Height: 2, Hash: block.Hash(), }, primary, []provider.Provider{primary}, // NOTE: primary should not be used here dbs.New(db), light.Logger(log.TestingLogger()), ) require.NoError(t, err) defer func() { require.NoError(t, c.Cleanup()) }() // ensure Tendermint is at height 3 or higher _, err = waitForBlock(ctx, primary, 3) require.NoError(t, err) _, err = c.VerifyLightBlockAtHeight(ctx, 3, time.Now()) require.NoError(t, err) h, err := c.TrustedLightBlock(3) require.NoError(t, err) require.EqualValues(t, 3, h.Height) } func waitForBlock(ctx context.Context, p provider.Provider, height int64) (*types.LightBlock, error) { for { block, err := p.LightBlock(ctx, height) switch err { case nil: return block, nil // node isn't running yet, wait 1 second and repeat case provider.ErrNoResponse, provider.ErrHeightTooHigh: timer := time.NewTimer(1 * time.Second) select { case <-ctx.Done(): return nil, ctx.Err() case <-timer.C: } default: return nil, err } } }