It builds!

ammario · ammario · commit 510524d3cb07 · 2023-03-24T19:25:40.000Z
diff --git a/coderd/database/dbauthz/querier.go b/coderd/database/dbauthz/querier.go
@@ -1500,6 +1500,9 @@ func (q *querier) InsertWorkspaceAgentStat(ctx context.Context, arg database.Ins
 
 func (q *querier) InsertWorkspaceAgentMetadata(ctx context.Context, arg database.InsertWorkspaceAgentMetadataParams) error {
 	workspace, err := q.db.GetWorkspaceByAgentID(ctx, arg.WorkspaceAgentID)
+	if err != nil {
+		return err
+	}
 
 	err = q.authorizeContext(ctx, rbac.ActionUpdate, workspace)
 	if err != nil {
@@ -1511,6 +1514,9 @@ func (q *querier) InsertWorkspaceAgentMetadata(ctx context.Context, arg database
 
 func (q *querier) UpdateWorkspaceAgentMetadata(ctx context.Context, arg database.UpdateWorkspaceAgentMetadataParams) error {
 	workspace, err := q.db.GetWorkspaceByAgentID(ctx, arg.WorkspaceAgentID)
+	if err != nil {
+		return err
+	}
 
 	err = q.authorizeContext(ctx, rbac.ActionUpdate, workspace)
 	if err != nil {
diff --git a/coderd/workspaceagents_test.go b/coderd/workspaceagents_test.go
@@ -1316,7 +1316,7 @@ func TestWorkspaceAgent_Metadata(t *testing.T) {
 	agentClient := agentsdk.New(client.URL)
 	agentClient.SetSessionToken(authToken)
 
-	ctx, _ := testutil.Context(t)
+	ctx := testutil.Context(t, testutil.WaitMedium)
 
 	manifest, err := agentClient.Manifest(ctx)
 	require.NoError(t, err)
@@ -1328,9 +1328,6 @@ func TestWorkspaceAgent_Metadata(t *testing.T) {
 	require.EqualValues(t, 10, manifest.Metadata[0].Interval)
 	require.EqualValues(t, 3, manifest.Metadata[0].Timeout)
 
-	ctx, cancel := testutil.Context(t)
-	defer cancel()
-
 	post := func(key string, mr codersdk.WorkspaceAgentMetadataResult) {
 		err := agentClient.PostMetadata(ctx, key, mr)
 		require.NoError(t, err, "post metadata", t)
