init_test.go 501 B

123456789101112131415161718192021
  1. package controller
  2. import (
  3. "flag"
  4. "go-template/business/util"
  5. "os"
  6. "testing"
  7. "gitea.ckfah.com/cjjy/gocommon/pkg/boot"
  8. "gitea.ckfah.com/cjjy/gocommon/pkg/net/middleware"
  9. )
  10. func TestMain(m *testing.M) {
  11. flag.Parse()
  12. boot.Init(boot.Qconf, boot.DB, boot.Redis, boot.Trace, boot.NacosDiscover, boot.RocketMQProducer)
  13. util.AddMiddleWareHandlerFunc(middleware.AccessLogMiddleware())
  14. util.AddMiddleWareHandlerFunc(middleware.AddTraceId())
  15. boot.Start()
  16. exitCode := m.Run()
  17. os.Exit(exitCode)
  18. }