dao_test.go 642 B

123456789101112131415161718192021222324252627282930313233
  1. package dao
  2. import (
  3. "flag"
  4. "go-common/app/job/main/account-summary/conf"
  5. "os"
  6. "testing"
  7. )
  8. var (
  9. d *Dao
  10. )
  11. func TestMain(m *testing.M) {
  12. if os.Getenv("DEPLOY_ENV") != "" {
  13. flag.Set("app_id", "main.account.account-summary")
  14. flag.Set("conf_token", "5761c1b876ef2941fe4b63f621b3d74a")
  15. flag.Set("tree_id", "51697")
  16. flag.Set("conf_version", "docker-1")
  17. flag.Set("deploy_env", "uat")
  18. flag.Set("conf_host", "config.bilibili.co")
  19. flag.Set("conf_path", "/tmp")
  20. flag.Set("region", "sh")
  21. flag.Set("zone", "sh001")
  22. }
  23. flag.Parse()
  24. if err := conf.Init(); err != nil {
  25. panic(err)
  26. }
  27. d = New(conf.Conf)
  28. m.Run()
  29. os.Exit(0)
  30. }