fixup! fix status

coadler · coadler · commit ec4c78ebf891 · 2023-02-08T23:48:09.000Z
diff --git a/cli/speedtest.go b/cli/speedtest.go
@@ -75,7 +75,7 @@ func speedtest() *cobra.Command {
 					if err != nil {
 						continue
 					}
-					status := conn.Status(true)
+					status := conn.Status()
 					if len(status.Peers()) != 1 {
 						continue
 					}
diff --git a/scaletest/agentconn/run.go b/scaletest/agentconn/run.go
@@ -87,7 +87,7 @@ func (r *Runner) Run(ctx context.Context, _ string, logs io.Writer) error {
 
 	// Ensure DERP for completeness.
 	if r.cfg.ConnectionMode == ConnectionModeDerp {
-		status := conn.Status(true)
+		status := conn.Status()
 		if len(status.Peers()) != 1 {
 			return xerrors.Errorf("check connection mode: expected 1 peer, got %d", len(status.Peers()))
 		}
@@ -172,7 +172,7 @@ func waitForDirectConnection(ctx context.Context, logs io.Writer, conn *codersdk
 
 	for i := 0; i < directConnectionAttempts; i++ {
 		_, _ = fmt.Fprintf(logs, "\tDirect connection check %d/%d...\n", i+1, directConnectionAttempts)
-		status := conn.Status(true)
+		status := conn.Status()
 
 		var err error
 		if len(status.Peers()) != 1 {
diff --git a/tailnet/conn.go b/tailnet/conn.go
@@ -353,7 +353,7 @@ func (c *Conn) RemoveAllPeers() error {
 func (c *Conn) UpdateNodes(nodes []*Node) error {
 	c.mutex.Lock()
 	defer c.mutex.Unlock()
-	status := c.Status(true)
+	status := c.Status()
 	for _, peer := range c.netMap.Peers {
 		peerStatus, ok := status.Peer[peer.Key]
 		if !ok {
@@ -424,8 +424,8 @@ func (c *Conn) UpdateNodes(nodes []*Node) error {
 }
 
 // Status returns the current ipnstate of a connection.
-func (c *Conn) Status(wantPeers bool) *ipnstate.Status {
-	sb := &ipnstate.StatusBuilder{WantPeers: wantPeers}
+func (c *Conn) Status() *ipnstate.Status {
+	sb := &ipnstate.StatusBuilder{WantPeers: true}
 	c.wireguardEngine.UpdateStatus(sb)
 	return sb.Status()
 }

Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,7 @@ func speedtest() *cobra.Command {`
`75`	`75`	`if err != nil {`
`76`	`76`	`continue`
`77`	`77`	`}`
`78`		`- status := conn.Status(true)`
	`78`	`+ status := conn.Status()`
`79`	`79`	`if len(status.Peers()) != 1 {`
`80`	`80`	`continue`
`81`	`81`	`}`