测试Go中的嵌套输入方法

本篇文章给大家分享《测试Go中的嵌套输入方法》，覆盖了Golang的常见基础知识，其实一个语言的全部知识点一篇文章是不可能说完的，但希望通过这些问题，让读者对自己的掌握程度有一定的认识(B 数)，从而弥补自己的不足，更好的掌握它。

我有使用 os.stdin 获取用户输入的功能

func (i input) getinput(stdin io.reader) (string, error) {
    reader := bufio.newreader(stdin)

    data, err := reader.readstring('\n')
    if err != nil {
        return "", fmt.errorf("get input error: %w", err)
    }

    return strings.replaceall(data, "\n", ""), nil
}

在我的程序中，我需要有 2 个输入：

• 首先用于获取基本信息，例如用户名
• 第二个用于获取取决于第一个输入的附加信息

name,err := getinput(os.stdin)
if err != nil {
 // error handling.....
}

switch name {
  case "test":
    //do something...
    age, err := getinput(os.stdin)
    if err != nil {
      // error handling.....
    }
    fmt.println(age)

  case "another":
    // here another input
}

是否可以为这种情况编写单元测试？ 为了测试一个用户输入，我使用了这个代码片段，它有效：

    var stdin bytes.Buffer
    stdin.Write([]byte(fmt.Sprintf("%s\n", tt.input)))
    GetInput(stdin)

但它不适用于 2 个嵌套输入

正确答案

也许可以考虑使用一个返回特定类型结果的函数并将其放入单独的包中。

由于我看到提到了 name 和 age ，也许我们可以假设一个具体类型，例如 person 来进行说明。

需要注意的是，我们希望将实际的阅读器作为参数包含在内，而不是对 os.stdin 进行硬编码引用。这使得嵌套输入的模拟首先成为可能。

这样，方法的签名可能如下所示：

func nestedinput(input io.reader) (*person, error)

对应的类型可以是：

type person struct {
    name string
    age  int
}

如果现在将您的代码片段合并到一个完整的 go 文件，并在一个单独的目录中使用名称 input.go，它可能看起来像这样：

package input

import (
    "bufio"
    "fmt"
    "io"
    "strconv"
    "strings"
)

func getinput(reader *bufio.reader) (string, error) {
    data, err := reader.readstring('\n')
    if err != nil {
        return "", fmt.errorf("get input error: %w", err)
    }

    return strings.replaceall(data, "\n", ""), nil
}

type person struct {
    name string
    age  int
}

func nestedinput(input io.reader) (*person, error) {
    reader := bufio.newreader(input)
    name, err := getinput(reader)
    if err != nil {
        return nil, err
    }

    switch name {
    case "q":
        return nil, nil
    default:
        agestr, err := getinput(reader)
        if err != nil {
            return nil, err
        }
        age, err := strconv.atoi(agestr)
        if err != nil {
            return nil, err
        }
        return &person{name: name, age: age}, nil
    }
}

q 的输入返回 nil、nil 并可用于终止输入，例如如果查询是在循环中进行的。

单元测试

单元测试

func test_nestedinput(t *testing.t)

名为 input_test.go 的文件现在应该提供输入数据。

由于 nestedinput 函数现在需要 io.reader 作为参数，因此我们可以简单地生成所需的输入，例如，

input := strings.newreader("george\n26\n")

所以测试可能看起来像这样：

package input

import (
    "strings"
    "testing"
)

func Test_nestedInput(t *testing.T) {
    input := strings.NewReader("George\n26\n")
    person, err := NestedInput(input)
    if err != nil {
        t.Error("nested input failed")
    }
    if person == nil {
        t.Errorf("expected person, but got nil")
        return
    }
    if person.Name != "George" {
        t.Errorf("wrong name %s, expected 'George'", person.Name)
    }
    if person.Age != 26 {
        t.Errorf("wrong age %d, expected 26", person.Age)
    }
}

当然，可以通过更多细节来扩展测试。但是，正如您所看到的，这模拟了嵌套输入。

今天关于《测试Go中的嵌套输入方法》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！