feat: seperate close function into tiny helper function

2026-07-12 01:20:22 +07:00
parent efbff0f7f7
commit 0a5f673c68
5 changed files with 67 additions and 35 deletions
@@ -86,7 +86,7 @@ func (ml *mockLifecycle) PortRegistry() port.Port {
 	return args.Get(0).(port.Port)
 }

-func (ml *mockLifecycle) SetChannel(channel ssh.Channel)       { ml.Called(channel) }
+func (ml *mockLifecycle) SetChannel(channel ssh.Channel) error { return ml.Called(channel).Error(0) }
 func (ml *mockLifecycle) SetStatus(status types.SessionStatus) { ml.Called(status) }
 func (ml *mockLifecycle) IsActive() bool                       { return ml.Called().Bool(0) }
 func (ml *mockLifecycle) StartedAt() time.Time                 { return ml.Called().Get(0).(time.Time) }