Go에서 기능 단위 테스트의 미래

package greet

import "testing"

// greet 函数接受一个名字，并返回一个问候信息
func Greet(name string) string {
  return "Hello, " + name + "!"
}

// TestGreet 函数使用 **testing.T** 断言测试 Greet 函数
func TestGreet(t *testing.T) {
  // 断言当传入 "Alice" 时，Greet 函数返回 "Hello, Alice!"
  got := Greet("Alice")
  want := "Hello, Alice!"
  if got != want {
    t.Errorf("Greet(%q) = %q, want %q", "Alice", got, want)
  }

  // 同理，断言其他值
  got = Greet("Bob")
  want = "Hello, Bob!"
  if got != want {
    t.Errorf("Greet(%q) = %q, want %q", "Bob", got, want)
  }
}

package database

import (
  "database/sql"
  "testing"

  // 使用 gomock 创建模拟数据库连接
  "github.com/golang/mock/gomock"
)

// DbClient 类型表示数据库客户端
type DbClient struct {
  conn *sql.DB
}

// Query 函数接受一个查询字符串并返回查询结果
func (c *DbClient) Query(query string) (*sql.Rows, error) {
  return c.conn.Query(query)
}

// TestQuery 函数使用 gomock 创建模拟连接并测试 Query 函数
func TestQuery(t *testing.T) {
  ctrl := gomock.NewController(t)
  defer ctrl.Finish()

  // 创建模拟连接
  mockConn := gomock.NewMockConn(ctrl)

  // 期望模拟连接以特定方式行为
  mockConn.EXPECT().Query("SELECT * FROM users").Return(sql.Rows{}, nil)

  // 创建要测试的 DbClient，并为其指定模拟连接
  client := &DbClient{conn: mockConn}

  // 执行查询并验证结果
  _, err := client.Query("SELECT * FROM users")
  if err != nil {
    t.Errorf("Query() returned error: %v", err)
  }
}