Go语言教程：为函数编写测试代码

1
package main
2

3
import (
4
  "errors"
5
  "flag"
6
  "fmt"
7
)
8

9
var name string
10

11
func init() {
12
  flag.StringVar(&name, "name", "everyone", "The greeting object.")
13
}
14

15
func main() {
16
  flag.Parse()
17
  greeting, err := hello(name)
18
  if err != nil {
19
    fmt.Printf("error: %s\\n", err)
20
    return
21
  }
22
  fmt.Println(greeting, introduce())
23
}
24

25
// hello 用于生成问候内容。
26
func hello(name string) (string, error) {
27
  if name == "" {
28
    return "", errors.New("empty name")
29
  }
30
  return fmt.Sprintf("Hello, %s!", name), nil
31
}
32

33
// introduce 用于生成介绍内容。
34
func introduce() string {
35
  return "Welcome to my Golang column."
36
}

1
package main
2

3
import (
4
  "fmt"
5
  "testing"
6
)
7

8
func TestHello(t *testing.T) {
9
  var name string
10
  greeting, err := hello(name)
11
  if err == nil {
12
    t.Errorf("The error is nil, but it should not be. (name=%q)", name)
13
  }
14
  if greeting != "" {
15
    t.Errorf("Nonempty greeting, but it should not be. (name=%q)", name)
16
  }
17
  name = "Robert"
18
  greeting, err = hello(name)
19
  if err != nil {
20
    t.Errorf("The error is not nil, but it should be. (name=%q)", name)
21
  }
22
  if greeting == "" {
23
    t.Errorf("Empty greeting, but it should not be. (name=%q)", name)
24
  }
25
  expected := fmt.Sprintf("Hello, %s!", name)
26
  if greeting != expected {
27
    t.Errorf("The actual greeting %q is not the expected. (name=%q)", greeting, name)
28
  }
29
  t.Logf("The expected greeting is %q.\\n", expected)
30
}
31

32
func testIntroduce(t *testing.T) { // 请注意这个测试函数的名称。
33
  intro := introduce()
34
  expected := "Welcome to my Golang column."
35
  if intro != expected {
36
    t.Errorf("The actual introduce %q is not the expected.", intro)
37
  }
38
  t.Logf("The expected introduce is %q.\\n", expected)
39
}