@@ -139,20 +139,20 @@ func TestCLICoverageGapBranches(t *testing.T) {
139139 }
140140
141141 cmd := newDiagAutoCommand ()
142- cmd .SetArgs ([]string {"on" , "--socket " , "/tmp/diag.sock " })
142+ cmd .SetArgs ([]string {"on" , "--session " , "shell-session-9 " })
143143 if err := cmd .ExecuteContext (context .Background ()); err != nil {
144144 t .Fatalf ("ExecuteContext() error = %v" , err )
145145 }
146- if captured .SocketPath != "/tmp/diag.sock " || ! captured .Enabled {
147- t .Fatalf ("captured = %+v, want socket=/tmp/diag.sock enabled=true" , captured )
146+ if captured .SessionID != "shell-session-9 " || ! captured .Enabled {
147+ t .Fatalf ("captured = %+v, want session=shell-session-9 enabled=true" , captured )
148148 }
149149 })
150150
151- t .Run ("diag diagnose command socket flag registration " , func (t * testing.T ) {
151+ t .Run ("diag diagnose command socket flag removed " , func (t * testing.T ) {
152152 cmd := newDiagDiagnoseCommand ()
153153 flag := cmd .Flags ().Lookup ("socket" )
154- if flag = = nil {
155- t .Fatal ("expected --socket flag on diagnose subcommand" )
154+ if flag ! = nil {
155+ t .Fatal ("expected --socket flag to be removed from diagnose subcommand" )
156156 }
157157 })
158158
@@ -164,8 +164,8 @@ func TestCLICoverageGapBranches(t *testing.T) {
164164 return errors .New ("send auto failed" )
165165 }
166166 err := defaultDiagAutoCommandRunner (context .Background (), diagAutoCommandOptions {
167- SocketPath : "/tmp/diag.sock " ,
168- Enabled : true ,
167+ SessionID : "shell-session-10 " ,
168+ Enabled : true ,
169169 }, & bytes.Buffer {})
170170 if err == nil || ! strings .Contains (err .Error (), "send auto failed" ) {
171171 t .Fatalf ("err = %v, want contains send auto failed" , err )
0 commit comments