31 lines
712 B
Go
31 lines
712 B
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
|
|
"github.com/cli/cli/v2/pkg/liveshare"
|
|
"golang.org/x/crypto/ssh"
|
|
)
|
|
|
|
type Session struct {
|
|
channel ssh.Channel
|
|
}
|
|
|
|
func (s *Session) KeepAlive(reason string) {
|
|
}
|
|
|
|
func (s *Session) StartSharing(ctx context.Context, sessionName string, port int) (liveshare.ChannelID, error) {
|
|
conn, err := net.Dial("tcp", fmt.Sprintf("127.0.0.1:%d", ServerPort))
|
|
if err != nil {
|
|
return liveshare.ChannelID{}, err
|
|
}
|
|
s.channel = &Channel{conn}
|
|
return liveshare.ChannelID{}, nil
|
|
}
|
|
|
|
// Creates mock SSH channel connected to the mock gRPC server
|
|
func (s *Session) OpenStreamingChannel(ctx context.Context, id liveshare.ChannelID) (ssh.Channel, error) {
|
|
return s.channel, nil
|
|
}
|