Golang单元测试实战

vscode 工具生成测试文件、测试代码

package main

import (
	"os"
	"strings"
)

func ExtractDir(isDIr bool, p string) string {
	if isDIr {
		return p
	}
	ps := "/"
	l := strings.Split(p, ps)
	n := len(l) - 1
	p = strings.Join(l[0:n], ps)
	return p
}

func CreateDirIfIsNotExist(isDIr bool, p string, perm os.FileMode) error {
	// isDIr 由调用者声明是从文件或者目录，来创建目录（由具体的业务场景判断，因为无法从字符串判断）。
	d := ExtractDir(isDIr, p)
	_, err := os.Open(p)
	if os.IsNotExist(err) {
		// 注意路径穿越检查
		// secguide/Go安全指南.md at main · Tencent/secguide · GitHub https://github.com/Tencent/secguide/blob/main/Go%E5%AE%89%E5%85%A8%E6%8C%87%E5%8D%97.md#121%E5%BF%85%E9%A1%BB-%E8%B7%AF%E5%BE%84%E7%A9%BF%E8%B6%8A%E6%A3%80%E6%9F%A5
		err := os.MkdirAll(d, os.ModeDir)
		return err
	}
	return nil
}

func NewFile(name string, data []byte, perm os.FileMode) error {
	err := os.WriteFile(name, data, perm)
	return err
}

　

package main

import (
	"os"
	"testing"
)

func TestCreateDirIfIsNotExist(t *testing.T) {
	type args struct {
		isDIr bool
		p     string
		perm  os.FileMode
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{name: "", args: args{isDIr: true, p: "test2/atxt", perm: 0777}},
		{name: "", args: args{isDIr: true, p: "test3/t1/atxt", perm: 0777}},
		{name: "", args: args{isDIr: true, p: "../../testa/t1/atxt", perm: 0777}},
		{name: "", args: args{isDIr: true, p: "../../../testb/t1/atxt", perm: 0777}},
		{name: "", args: args{isDIr: false, p: "atest2/a/b.txt", perm: 0777}},
		{name: "", args: args{isDIr: false, p: "atest3/t1/a/b.txt", perm: 0777}},
		{name: "", args: args{isDIr: false, p: "../../atest/t1/a/b.txt", perm: 0777}},
		{name: "", args: args{isDIr: false, p: "../../../atest/t1/a/b.txt", perm: 0777}},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := CreateDirIfIsNotExist(tt.args.isDIr, tt.args.p, tt.args.perm); (err != nil) != tt.wantErr {
				t.Errorf("CreateDirIfIsNotExist() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}

func TestNewFile(t *testing.T) {
	type args struct {
		name string
		data []byte
		perm os.FileMode
	}
	tests := []struct {
		name    string
		args    args
		wantErr bool
	}{
		{name: "", args: args{name: "atest2/a/b.txt", data: []byte("1"), perm: 0777}},
		{name: "", args: args{name: "atest3/t1/a/b.txt", data: []byte("1"), perm: 0777}},
		{name: "", args: args{name: "../../atest/t1/a/b.txt", data: []byte("1"), perm: 0777}},
		{name: "", args: args{name: "../../../atest/t1/a/b.txt", data: []byte("1"), perm: 0777}},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if err := NewFile(tt.args.name, tt.args.data, tt.args.perm); (err != nil) != tt.wantErr {
				t.Errorf("NewFile() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}

　　

　

一文说尽Golang单元测试实战的那些事儿

原创 张力 云加社区 2021-07-27

一文说尽Golang单元测试实战的那些事儿 https://mp.weixin.qq.com/s/AuUd-S0g4QX_68CDO6jJWg