1

这是一个小golang程序，它通过连接到 localhostssh并执行一些sftp类似的操作。它工作正常，但我想让该doTheWork()功能更可测试。

我已经阅读了使用gomock为接口创建模拟的内容。所以我已经创建了两个接口，Sftper和Walker，我正在使用它们而不是sftp直接调用 's 方法。

我想做的是从stfp里面模拟出所有的东西doTheWork()。Close()很简单，因为它只返回一个error（或nil）。但Create()返回*sftp.File，这是一个结构，而不是一个接口。与Walk()which 返回另一个结构相同。我必须在我的测试代码中创建这些结构之一，如果不实际调用真正的sftp库并建立远程连接并创建远程文件，这实际上是不可能的，而这正是我想要模拟的。

我错过了什么？

package main

import (
    "log"
    "os"

    "github.com/kr/fs"
    "github.com/pkg/sftp"
    "golang.org/x/crypto/ssh"
)

// Sftper helps make things testable
type Sftper interface {
    Close() error
    Create(path string) (*sftp.File, error)
    Lstat(p string) (os.FileInfo, error)
    Walk(root string) *fs.Walker
}

// Walker helps make things testable
type Walker interface {
    Step() bool
    Err() error
    Path() string
}

var sftpclient Sftper
var w Walker

func doTheWork(sftpclient Sftper) {
    defer sftpclient.Close()

    // walk a directory
    w = sftpclient.Walk("/tmp/")
    for w.Step() {
        if w.Err() != nil {
            continue
        }
        log.Println(w.Path())
    }

    // leave your mark
    f, err := sftpclient.Create("hello.txt")
    if err != nil {
        log.Fatal(err)
    }
    if _, err1 := f.Write([]byte("Hello world!")); err1 != nil {
        log.Fatal(err1)
    }

    // check it's there
    fi, err := sftpclient.Lstat("hello.txt")
    if err != nil {
        log.Fatal(err)
    }
    log.Println(fi)

}

func main() {
    config := &ssh.ClientConfig{
        User: "myusername",
        Auth: []ssh.AuthMethod{
            ssh.Password("mysupersecretpassword"),
        },
        HostKeyCallback: ssh.InsecureIgnoreHostKey(),
    }
    conn, err0 := ssh.Dial("tcp", "localhost:22", config)
    if err0 != nil {
        log.Fatal("Failed to dial: ", err0)
    }

    // open an SFTP session over an existing ssh connection.
    var err error
    sftpclient, err = sftp.NewClient(conn)
    if err != nil {
        log.Fatal(err)
    }
    doTheWork(sftpclient)
}
4

0 回答 0