blob: d37c37d491ba58e9beb7bbecbec2a1a4df0bfc83 [file] [log] [blame]
// Copyright 2021 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
package dutssh
import (
// This file only exists because go cannot mock structs and the ssh client
// library does not provide interfaces for testing.
type ClientInterface interface {
Close() error
NewSession() (SessionInterface, error)
Wait() error
type SSHClient struct {
func (c *SSHClient) Close() error {
return c.Client.Close()
func (c *SSHClient) NewSession() (SessionInterface, error) {
session, err := c.Client.NewSession()
if err != nil {
return nil, err
return &SSHSession{Session: session}, nil
func (c *SSHClient) Wait() error {
return c.Client.Wait()
type SessionInterface interface {
Close() error
SetStdout(writer io.Writer)
SetStderr(writer io.Writer)
SetStdin(reader io.Reader)
Run(cmd string) error
Start(cmd string) error
Output(cmd string) ([]byte, error)
StdoutPipe() (io.Reader, error)
StderrPipe() (io.Reader, error)
type SSHSession struct {
func (s *SSHSession) Close() error {
return s.Session.Close()
func (s *SSHSession) SetStdout(writer io.Writer) {
s.Session.Stdout = writer
func (s *SSHSession) SetStderr(writer io.Writer) {
s.Session.Stderr = writer
func (s *SSHSession) SetStdin(reader io.Reader) {
s.Session.Stdin = reader
func (s *SSHSession) Run(cmd string) error {
return s.Session.Run(cmd)
func (s *SSHSession) Start(cmd string) error {
return s.Session.Start(cmd)
func (s *SSHSession) Output(cmd string) ([]byte, error) {
return s.Session.Output(cmd)
func (s *SSHSession) StdoutPipe() (io.Reader, error) {
return s.Session.StdoutPipe()
func (s *SSHSession) StderrPipe() (io.Reader, error) {
return s.Session.StderrPipe()