wip

cmd/tendermint/commands/rollback_test.go

@@ -21,20 +21,18 @@ func TestRollbackIntegration(t *testing.T) {
 	cfg, err := rpctest.CreateConfig(t.Name())
 	require.NoError(t, err)
 	cfg.BaseConfig.DBBackend = "goleveldb"
-	app, err := e2e.NewApplication(e2e.DefaultConfig(dir))
+	e2eCfg := e2e.DefaultConfig(dir)
+	app, err := e2e.NewApplication(e2eCfg)
 
-	t.Run("First run", func(t *testing.T) {
-		ctx, cancel := context.WithCancel(ctx)
-		defer cancel()
-		require.NoError(t, err)
-		node, _, err := rpctest.StartTendermint(ctx, cfg, app, rpctest.SuppressStdout)
-		require.NoError(t, err)
+	cancelCtx, cancel := context.WithCancel(ctx)
+	require.NoError(t, err)
+	node, _, err := rpctest.StartTendermint(cancelCtx, cfg, app, rpctest.SuppressStdout)
+	require.NoError(t, err)
+	// we run a height here.
 
-		time.Sleep(3 * time.Second)
-		cancel()
-		node.Wait()
-		require.False(t, node.IsRunning())
-	})
+	cancel()
+	node.Wait()
+	require.False(t, node.IsRunning())
 
 	t.Run("Rollback", func(t *testing.T) {
 		require.NoError(t, app.Rollback())