Question

这是一个通过golang连接到localhost的小ssh程序，并执行一些sftp类操作。它工作正常，但我想使doTheWork()函数更易于测试。

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

我想做的是嘲笑stfp内doTheWork()内的所有内容。 Close()很简单，因为它只返回error（或nil）。但Create()返回*sftp.File，这是一个结构，而不是一个接口。与返回另一个结构的Walk()相同。我必须在我的测试代码中创建这些结构中的一个，如果没有真正调用真正的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)
}

如何模拟返回结构的方法？

0 个答案: