Add MustTempDir (#1069)

tools/goctl/rpc/generator/prototmpl_test.go

@@ -5,16 +5,18 @@ import (
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+
+	"github.com/tal-tech/go-zero/tools/goctl/util"
 )
 
 func TestProtoTmpl(t *testing.T) {
 	_ = Clean()
 	// exists dir
-	err := ProtoTmpl(t.TempDir())
+	err := ProtoTmpl(util.MustTempDir())
 	assert.Nil(t, err)
 
 	// not exist dir
-	dir := filepath.Join(t.TempDir(), "test")
+	dir := filepath.Join(util.MustTempDir(), "test")
 	err = ProtoTmpl(dir)
 	assert.Nil(t, err)
 }