@@ -5,11 +5,13 @@ import (
55"testing"
66
77"github.com/stretchr/testify/assert"
8+ "github.com/stretchr/testify/require"
89
910"cdr.dev/slog/sloggers/slogtest"
1011"github.com/coder/coder/agent"
1112"github.com/coder/coder/cli/clitest"
1213"github.com/coder/coder/coderd/coderdtest"
14+ "github.com/coder/coder/codersdk"
1315"github.com/coder/coder/codersdk/agentsdk"
1416"github.com/coder/coder/pty/ptytest"
1517"github.com/coder/coder/testutil"
@@ -30,13 +32,27 @@ func TestSpeedtest(t *testing.T) {
3032defer agentCloser .Close ()
3133coderdtest .AwaitWorkspaceAgents (t ,client ,workspace .ID )
3234
35+ ctx ,cancel := context .WithTimeout (context .Background (),testutil .WaitLong )
36+ defer cancel ()
37+
38+ require .Eventually (t ,func ()bool {
39+ ws ,err := client .Workspace (ctx ,workspace .ID )
40+ if ! assert .NoError (t ,err ) {
41+ return false
42+ }
43+ a := ws .LatestBuild .Resources [0 ].Agents [0 ]
44+ return a .Status == codersdk .WorkspaceAgentConnected &&
45+ a .LifecycleState == codersdk .WorkspaceAgentLifecycleReady
46+ },testutil .WaitLong ,testutil .IntervalFast ,"agent is not ready" )
47+
3348cmd ,root := clitest .New (t ,"speedtest" ,workspace .Name )
3449clitest .SetupConfig (t ,client ,root )
3550pty := ptytest .New (t )
3651cmd .SetOut (pty .Output ())
3752
38- ctx ,cancel : =context .WithTimeout (context .Background (),testutil .WaitLong )
53+ ctx ,cancel = context .WithTimeout (context .Background (),testutil .WaitLong )
3954defer cancel ()
55+
4056cmdDone := tGo (t ,func () {
4157err := cmd .ExecuteContext (ctx )
4258assert .NoError (t ,err )