Go: mock return value of a function in unit test

huangapple
go评论148阅读模式
英文:

Go: mock return value of a function in unit test

问题

我已经在golang中创建了一个小脚本（我的第一个golang项目）。

示例：

  1. package main
  3. import (
  4. 	"fmt"
  5. 	"math/rand"
  6. )
  8. func main() {
  9. 	i := rand.Intn(10)
  10. 	foo := foo(i)
  12. 	if foo {
  13. 		fmt.Printf("%d是偶数！", i)
  14. 		// 更多代码...
  15. 	} else {
  16. 		fmt.Printf("%d是奇数！", i)
  17. 		// 更多代码...
  18. 	}
  19. }
  21. func foo(i int) bool {
  22. 	if i%2 == 0 {
  23. 		return true
  24. 	} else {
  25. 		return false
  26. 	}
  27. }

我想为每个函数创建一个小的单元测试。
对于"main()"函数，我想模拟函数"foo()"的返回值，因为我不想测试"foo()"，而是测试main()函数的其余代码。

我正在寻找一种简单的方法来存根/模拟返回值。
我只找到了使用结构体或接口等进行模拟。但是我在代码中没有使用这些元素（这是一个简单的项目）。

英文:

I have created a small script in golang (my first golang project).

Example:

  1. package main
  3. import (
  4. 	"fmt"
  5. 	"math/rand"
  6. )
  8. func main() {
  9. 	i := rand.Intn(10)
  10. 	foo := foo(i)
  12. 	if foo {
  13. 		fmt.Printf("%d is even!", i)
  14. 		// more code ... 
  15. 	} else {
  16. 		fmt.Printf("%d is odd!", i)
  17. 		// more code ... 
  18. 	}
  19. }
  21. func foo(i int) bool {
  22. 	if i%2 == 0 {
  23. 		return true
  24. 	} else {
  25. 		return false
  26. 	}
  27. }

I would like to create a small unit test for each function.
For "main()", I want to mock the return value of function "foo()", because I won't test "foo()", but the rest of main()-code.

I'm looking for an easy way to stub/mock the return value.
I just found mocks with struct or interaces etc. But I didn't use these elements in code (it's a simple project).

答案1

得分: 3

使用一个真实、简洁、可重现的示例：如何创建一个简洁、可重现的示例。

例如，在Go语言中：

  1. package main
  3. import (
  4. 	"fmt"
  5. 	"math/rand"
  6. )
  8. func isEven(i int) bool {
  9. 	return i%2 == 0
  10. }
  12. func side(n int) string {
  13. 	if isEven(n) {
  14. 		return "right"
  15. 	} else {
  16. 		return "left"
  17. 	}
  18. }
  20. func main() {
  21. 	n := 1 + rand.Intn(10)
  22. 	hand := side(n)
  23. 	fmt.Printf("Number %d is on the %s-hand side of the street.\n", n, hand)
  24. }

https://go.dev/play/p/a0SudgaRCnu

输出结果为：

  1. Number 9 is on the left-hand side of the street.

使用Go的testing包对side函数进行单元测试。你也可以直接对isEven函数进行单元测试。main函数不应包含任何你想要进行单元测试的代码。

  1. package main
  3. import (
  4. 	"testing"
  5. )
  7. func TestSide(t *testing.T) {
  8. 	n := 7
  9. 	got := side(n)
  10. 	want := "left"
  11. 	if got != want {
  12. 		t.Errorf("side(%d) = %s; want %s", n, got, want)
  13. 	}
  14. }
英文:

Use a realistic, minimal, reproducible example: How to create a Minimal, Reproducible Example.

For example, in Go,

  1. package main
  3. import (
  4. 	"fmt"
  5. 	"math/rand"
  6. )
  8. func isEven(i int) bool {
  9. 	return i%2 == 0
  10. }
  12. func side(n int) string {
  13. 	if isEven(n) {
  14. 		return "right"
  15. 	} else {
  16. 		return "left"
  17. 	}
  18. }
  20. func main() {
  21. 	n := 1 + rand.Intn(10)
  22. 	hand := side(n)
  23. 	fmt.Printf("Number %d is on the %s-hand side of the street.\n", n, hand)
  24. }

https://go.dev/play/p/a0SudgaRCnu

  1. Number 9 is on the left-hand side of the street.

Use the Go testing package to unit test the side function. You might also unit test the isEven function directly. The main function should not contain any code that you want to unit test.

  1. package main
  3. import (
  4. 	"testing"
  5. )
  7. func TestSide(t *testing.T) {
  8. 	n := 7
  9. 	got := side(n)
  10. 	want := "left"
  11. 	if got != want {
  12. 		t.Errorf("side(%d) = %s; want %s", n, got, want)
  13. 	}
  14. }

huangapple
  • 本文由 发表于 2023年4月28日 04:11:01
  • 转载请务必保留本文链接：https://go.coder-hub.com/76124276.html
  • go
  • mocking
  • stub
  • unit-testing
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定