@@ -28,6 +28,7 @@ import (
2828"github.com/coder/coder/v2/agent/agenttest"
2929"github.com/coder/coder/v2/coderd/coderdtest"
3030"github.com/coder/coder/v2/coderd/database"
31+ "github.com/coder/coder/v2/coderd/database/dbauthz"
3132"github.com/coder/coder/v2/coderd/database/dbfake"
3233"github.com/coder/coder/v2/coderd/database/dbgen"
3334"github.com/coder/coder/v2/coderd/httpapi"
@@ -106,8 +107,9 @@ func TestTools(t *testing.T) {
106107})
107108
108109t .Run ("ReportTask" ,func (t * testing.T ) {
110+ ctx := testutil .Context (t ,testutil .WaitShort )
109111tb ,err := toolsdk .NewDeps (memberClient ,toolsdk .WithTaskReporter (func (args toolsdk.ReportTaskArgs )error {
110- return agentClient .PatchAppStatus (setupCtx , agentsdk.PatchAppStatus {
112+ return agentClient .PatchAppStatus (ctx , agentsdk.PatchAppStatus {
111113AppSlug :"some-agent-app" ,
112114Message :args .Summary ,
113115URI :args .Link ,
@@ -1393,7 +1395,17 @@ func TestTools(t *testing.T) {
13931395task := ws .Task
13941396
13951397_ = agenttest .New (t ,client .URL ,ws .AgentToken )
1396- coderdtest .NewWorkspaceAgentWaiter (t ,client ,ws .Workspace .ID ).Wait ()
1398+ coderdtest .NewWorkspaceAgentWaiter (t ,client ,ws .Workspace .ID ).
1399+ WaitFor (coderdtest .AgentsReady )
1400+
1401+ ctx := testutil .Context (t ,testutil .WaitShort )
1402+
1403+ // Ensure the app is healthy (required to send task input).
1404+ err = store .UpdateWorkspaceAppHealthByID (dbauthz .AsSystemRestricted (ctx ), database.UpdateWorkspaceAppHealthByIDParams {
1405+ ID :task .WorkspaceAppID .UUID ,
1406+ Health :database .WorkspaceAppHealthHealthy ,
1407+ })
1408+ require .NoError (t ,err )
13971409
13981410tests := []struct {
13991411name string
@@ -1454,8 +1466,6 @@ func TestTools(t *testing.T) {
14541466
14551467for _ ,tt := range tests {
14561468t .Run (tt .name ,func (t * testing.T ) {
1457- t .Parallel ()
1458-
14591469tb ,err := toolsdk .NewDeps (memberClient )
14601470require .NoError (t ,err )
14611471
@@ -1525,7 +1535,17 @@ func TestTools(t *testing.T) {
15251535task := ws .Task
15261536
15271537_ = agenttest .New (t ,client .URL ,ws .AgentToken )
1528- coderdtest .NewWorkspaceAgentWaiter (t ,client ,ws .Workspace .ID ).Wait ()
1538+ coderdtest .NewWorkspaceAgentWaiter (t ,client ,ws .Workspace .ID ).
1539+ WaitFor (coderdtest .AgentsReady )
1540+
1541+ ctx := testutil .Context (t ,testutil .WaitShort )
1542+
1543+ // Ensure the app is healthy (required to read task logs).
1544+ err = store .UpdateWorkspaceAppHealthByID (dbauthz .AsSystemRestricted (ctx ), database.UpdateWorkspaceAppHealthByIDParams {
1545+ ID :task .WorkspaceAppID .UUID ,
1546+ Health :database .WorkspaceAppHealthHealthy ,
1547+ })
1548+ require .NoError (t ,err )
15291549
15301550tests := []struct {
15311551name string
@@ -1577,8 +1597,6 @@ func TestTools(t *testing.T) {
15771597
15781598for _ ,tt := range tests {
15791599t .Run (tt .name ,func (t * testing.T ) {
1580- t .Parallel ()
1581-
15821600tb ,err := toolsdk .NewDeps (memberClient )
15831601require .NoError (t ,err )
15841602