os.stdin
获取用户输入的功能,在我的程序中,我需要有 2 个输入:,首先用于获取基本信息,例如用户名第二个用于获取取决于第一个输入的附加信息,是否可以为这种情况编写单元测试?
为了测试一个用户输入,我使用了这个代码片段,它有效:,但它不适用于 2 个嵌套输入,也许可以考虑使用一个返回特定类型结果的函数并将其放入单独的包中。,由于我看到提到了
name
和 age
,也许我们可以假设一个具体类型,例如 person
来进行说明。,需要注意的是,我们希望将实际的阅读器作为参数包含在内,而不是对 os.stdin
进行硬编码引用。这使得嵌套输入的模拟首先成为可能。,这样,方法的签名可能如下所示:,对应的类型可以是:,如果现在将您的代码片段合并到一个完整的 go 文件,并在一个单独的目录中使用名称 input.go
,它可能看起来像这样:,q
的输入返回 nil、nil
并可用于终止输入,例如如果查询是在循环中进行的。,单元测试,单元测试,名为 input_test.go
的文件现在应该提供输入数据。,由于 nestedinput
函数现在需要 io.reader
作为参数,因此我们可以简单地生成所需的输入,例如,,所以测试可能看起来像这样:,当然,可以通过更多细节来扩展测试。但是,正如您所看到的,这模拟了嵌套输入。,