Revert dropped test context

gnolang · Apr 25, 2024 · eb5a3a2 · eb5a3a2
1 parent b898d26
commit eb5a3a2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/gno.land/cmd/gnoland/start_test.go b/gno.land/cmd/gnoland/start_test.go
@@ -5,6 +5,7 @@ import (
 	"context"
 	"path/filepath"
 	"testing"
+	"time"
 
 	"github.com/gnolang/gno/tm2/pkg/commands"
 	"github.com/stretchr/testify/assert"
@@ -42,8 +43,12 @@ func TestStartInitialize(t *testing.T) {
 	io.SetOut(commands.WriteNopCloser(mockOut))
 	io.SetErr(commands.WriteNopCloser(mockErr))
 
+	// Create and run the command
+	ctx, cancelFn := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancelFn()
+
 	cmd := newRootCmd(io)
-	require.NoError(t, cmd.ParseAndRun(context.Background(), args))
+	require.NoError(t, cmd.ParseAndRun(ctx, args))
 
 	// Make sure the directory is created
 	assert.DirExists(t, nodeDir)