Go: 基于 Ginkgo 框架进行单元测试实践

// TruncateString truncate string to specific length
func TruncateString(s string, n int) string {
	if n > len(s) {
		return s
	}
	return s[:n]
}

func TestTruncateString(t *testing.T) {
	type args struct {
		s string
		n int
	}
	tests := []struct {
		name string
		args args
		want string
	}{
		// TODO: Add test cases.
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := util.TruncateString(tt.args.s, tt.args.n); got != tt.want {
				t.Errorf("TruncateString() = %v, want %v", got, tt.want)
			}
		})
	}
}

package util_test

import (
	"fmt"
	"strconv"
	"testing"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/ginkgo/extensions/table"
	"github.com/stretchr/testify/assert"

	"example/pkg/util"
)

var _ = Describe("String", func() {

	Describe("TruncateBytes", func() {
		var s = []byte("helloworld")

		DescribeTable("TruncateBytes cases", func(expected []byte, truncatedSize int) {
			assert.Equal(GinkgoT(), expected, util.TruncateBytes(s, truncatedSize))
		},
			Entry("truncated size less than real size", []byte("he"), 2),
			Entry("truncated size equals to real size", s, 10),
			Entry("truncated size greater than real size", s, 20),
		)
	})
})

		Describe("WithoutResourceType", func() {
			var a types.Action
			BeforeEach(func() {
				a = types.NewAction()
			})

			It("true", func() {
				assert.True(GinkgoT(), a.WithoutResourceType())
			})

			It("true, empty ResourceType", func() {
				a.Attribute.SetResourceTypes([]types.ActionResourceType{})
				assert.True(GinkgoT(), a.WithoutResourceType())
			})

			It("false", func() {
				a.Attribute.SetResourceTypes([]types.ActionResourceType{
					{
						System: "test",
						Type:   "test",
					},
				})
				assert.False(GinkgoT(), a.WithoutResourceType())
			})

		}

	Describe("GetPolicesAttrKeys", func() {
		var resource *types.Resource
		var policies []types.AuthPolicy
		BeforeEach(func() {
			resource = &types.Resource{
				System:    "test",
				Type:      "host",
				ID:        "1",
				Attribute: nil,
			}
			policies = []types.AuthPolicy{}
		})

		It("fail", func() {
			errExpr := `[{"system": "test", "type": "host", "expression":
{"OR": {"content": [{"NotExists": {"id": []}}]}}}]`
			policies = []types.AuthPolicy{
				{
					Expression: errExpr,
				},
			}
			_, err := GetPolicesAttrKeys(resource, policies)
			assert.Error(GinkgoT(), err)
		})
})

	Describe("FillSubjectDetail", func() {
		var r *request.Request
		var ctl *gomock.Controller
		var patches *gomonkey.Patches
		BeforeEach(func() {
			ctl = gomock.NewController(GinkgoT())
			r = request.NewRequest()
		})
		AfterEach(func() {
			ctl.Finish()
			if patches != nil {
				patches.Reset()
			}
		})

		It("pip.GetSubjectPK fail", func() {
			patches = gomonkey.ApplyFunc(pip.GetSubjectPK, func(_type, id string) (pk int64, err error) {
				return -1, errors.New("get subject_pk fail")
			})
			err := r.FillSubjectDetail()
			assert.Error(GinkgoT(), err)
			assert.Contains(GinkgoT(), err.Error(), "get subject_pk fail")

		})
})

# 进入模块
$ cd pkg/util

# 生成一个模块的 {module_name}_suite_test.go
$ ginkgo bootstrap

# 生成模块下某个文件 {file_name}_test.go
$ ginkgo generate set

# 运行当前目录下的
$ ginkgo

# 带tags和gcflags (如果做了mock)
$ ginkgo -tags=jsoniter -gcflags=all=-l

# 递归执行当前目录及所有子目录下的
$ ginkgo -r -v -cover

# 只执行某几个目录
$ ginkgo  -r pkg/util pkg/cache

# in Makefile
test:
	go test -mod=vendor -gcflags=all=-l $(shell go list ./... | grep -v mock | grep -v docs) -covermode=count -coverprofile .coverage.cov

cov:
	go tool cover -html=.coverage.cov