github.com/easysoft/zendata@v0.0.0-20240513203326-705bd5a7fd67/cmd/test/helper/comm.go (about)

     1  package testHelper
     2  
     3  import (
     4  	"log"
     5  	"os"
     6  
     7  	"github.com/easysoft/zendata/cmd/test/consts"
     8  	commandConfig "github.com/easysoft/zendata/internal/command/config"
     9  	configUtils "github.com/easysoft/zendata/internal/pkg/config"
    10  	"github.com/easysoft/zendata/pkg/utils/vari"
    11  	"github.com/fatih/color"
    12  )
    13  
    14  func BeforeAll() {
    15  	configUtils.InitConfig("")
    16  	vari.DB, _ = commandConfig.NewGormDB()
    17  }
    18  
    19  func PreCase() {
    20  	log.SetOutput(&consts.Buf)
    21  	color.Output = &consts.Buf
    22  
    23  	vari.GlobalVars.Total = 10
    24  	vari.GlobalVars.ExportFields = []string{""}
    25  	vari.GlobalVars.Output = ""
    26  	vari.GlobalVars.Trim = false
    27  	vari.GlobalVars.Human = false
    28  	vari.GlobalVars.Recursive = false
    29  
    30  	vari.GlobalVars.DBType = ""
    31  	vari.GlobalVars.Table = ""
    32  	vari.ProtoCls = ""
    33  }
    34  
    35  func PostCase() {
    36  	consts.Buf.Reset()
    37  	log.SetOutput(os.Stdout)
    38  }