coder
diff --git a/‎agent/agent.go
Lines changed: 0 additions & 1 deletion b/‎agent/agent.go
Lines changed: 0 additions & 1 deletion
diff --git a/‎agent/agenttest/client.go
Lines changed: 11 additions & 1 deletion b/‎agent/agenttest/client.go
Lines changed: 11 additions & 1 deletion
@@ -372,7 +372,6 @@ func (a *agent) collectMetadata(ctx context.Context, md codersdk.WorkspaceAgentM
 	// Important: if the command times out, we may see a misleading error like
 	// "exit status 1", so it's important to include the context error.
 	err = errors.Join(err, ctx.Err())
-
 	if err != nil {
 		result.Error = fmt.Sprintf("run cmd: %+v", err)
 	}
 
@@ -15,6 +15,7 @@ import (
 	"golang.org/x/exp/slices"
 	"golang.org/x/xerrors"
 	"google.golang.org/protobuf/types/known/durationpb"
+	"google.golang.org/protobuf/types/known/emptypb"
 	"storj.io/drpc/drpcmux"
 	"storj.io/drpc/drpcserver"
 	"tailscale.com/tailcfg"
@@ -170,6 +171,7 @@ type FakeAgentAPI struct {
 	lifecycleStates []codersdk.WorkspaceAgentLifecycle
 	metadata        map[string]agentsdk.Metadata
 	timings         []*agentproto.Timing
+	connections     []*agentproto.Connection
 
 	getAnnouncementBannersFunc              func() ([]codersdk.BannerConfig, error)
 	getResourcesMonitoringConfigurationFunc func() (*agentproto.GetResourcesMonitoringConfigurationResponse, error)
@@ -338,12 +340,20 @@ func (f *FakeAgentAPI) BatchCreateLogs(ctx context.Context, req *agentproto.Batc
 
 func (f *FakeAgentAPI) ScriptCompleted(_ context.Context, req *agentproto.WorkspaceAgentScriptCompletedRequest) (*agentproto.WorkspaceAgentScriptCompletedResponse, error) {
 	f.Lock()
-	f.timings = append(f.timings, req.Timing)
+	f.timings = append(f.timings, req.GetTiming())
 	f.Unlock()
 
 	return &agentproto.WorkspaceAgentScriptCompletedResponse{}, nil
 }
 
+func (f *FakeAgentAPI) ReportConnection(_ context.Context, req *agentproto.ReportConnectionRequest) (*emptypb.Empty, error) {
+	f.Lock()
+	f.connections = append(f.connections, req.GetConnection())
+	f.Unlock()
+
+	return &emptypb.Empty{}, nil
+}
+
 func NewFakeAgentAPI(t testing.TB, logger slog.Logger, manifest *agentproto.Manifest, statsCh chan *agentproto.Stats) *FakeAgentAPI {
 	return &FakeAgentAPI{
 		t:           t,
Original file line number	Diff line number	Diff line change
`@@ -372,7 +372,6 @@ func (a *agent) collectMetadata(ctx context.Context, md codersdk.WorkspaceAgentM`
`372`	`372`	`// Important: if the command times out, we may see a misleading error like`
`373`	`373`	`// "exit status 1", so it's important to include the context error.`
`374`	`374`	`err = errors.Join(err, ctx.Err())`
`375`		`-`
`376`	`375`	`if err != nil {`
`377`	`376`	`result.Error = fmt.Sprintf("run cmd: %+v", err)`
`378`	`377`	`}`