You are an expert in writing code in many different languages. Your goal is to perform code completion for the following code, keeping in mind the rest of the code and the file meta data.
    Metadata details:
    filename: main_test.go
    path of file: ./repos/dtm/test
    
    the code of the file until where you have to start completion: /*
 * Copyright (c) 2021 yedf. All rights reserved.
 * Use of this source code is governed by a BSD-style
 * license that can be found in the LICENSE file.
 */

package test

import (
	"os"
	"testing"
	"time"

	"github.com/dtm-labs/dtm/client/dtmcli"
	"github.com/dtm-labs/dtm/client/dtmcli/dtmimp"
	"github.com/dtm-labs/dtm/client/workflow"
	"github.com/dtm-labs/dtm/dtmsvr"
	"github.com/dtm-labs/dtm/dtmsvr/config"
	"github.com/dtm-labs/dtm/dtmsvr/storage/registry"
	"github.com/dtm-labs/dtm/dtmutil"
	"github.com/dtm-labs/dtm/test/busi"
	"github.com/dtm-labs/dtmdriver"
	"github.com/dtm-labs/logger"
)

func TestMain(m *testing.M) {
	config.MustLoadConfig("")
	logger.InitLog("debug")
	dtmsvr.TransProcessedTestChan = make(chan string, 1)
	dtmsvr.NowForwardDuration = 0 * time.Second
	dtmsvr.CronForwardDuration = 180 * time.Second
	conf.UpdateBranchSync = 1
	conf.ConfigUpdateInterval = 1
	conf.AlertWebHook = busi.Busi + "/AlertWebHook"

	dtmdriver.Middlewares.HTTP = append(dtmdriver.Middlewares.HTTP, busi.SetHTTPHeaderForHeadersYes)
	dtmdriver.Middlewares.Grpc = append(dtmdriver.Middlewares.Grpc, busi.SetGrpcHeaderForHeadersYes)

	tenv := dtmimp.OrString(os.Getenv("TEST_STORE"), config.Redis)
	conf.ConfigUpdateInterval = 1
	conf.Store.Host = "localhost"
	conf.Store.Driver = tenv
	if tenv == "boltdb" {
	} else if tenv == config.Mysql {
		conf.Store.Port = 3306
		conf.Store.User = "root"
		conf.Store.Password = ""
	} else if tenv == config.Postgres {
		conf.Store.Port = 5432
		conf.Store.User = "postgres"
		conf.Store.Password = "mysecretpassword"
	} else if tenv == config.Redis {
		conf.Store.User = ""
		conf.Store.Password = ""
		conf.Store.Port = 6379
	} else if tenv == config.SQLServer {
		conf.Store.User = "sa"
		conf.Store.Password = "p@ssw0rd"
		conf.Store.Port = 1433
	}
	conf.Store.Db = ""
	registry.WaitStoreUp()

	dtmsvr.PopulateDB(false)
	conf.Store.Db = "dtm" // after populat
    
func TestMain(m *testing.M) {
	config.MustLoadConfig("")
	logger.InitLog("debug")
	dtmsvr.TransProcessedTestChan = make(chan string, 1)
	dtmsvr.NowForwardDuration = 0 * time.Second
	dtmsvr.CronForwardDuration = 180 * time.Second
	conf.UpdateBranchSync = 1
	conf.ConfigUpdateInterval = 1
	conf.AlertWebHook = busi.Busi + "/AlertWebHook"

	dtmdriver.Middlewares.HTTP = append(dtmdriver.Middlewares.HTTP, busi.SetHTTPHeaderForHeadersYes)
	dtmdriver.Middlewares.Grpc = append(dtmdriver.Middlewares.Grpc, busi.SetGrpcHeaderForHeadersYes)

	tenv := dtmimp.OrString(os.Getenv("TEST_STORE"), config.Redis)
	conf.ConfigUpdateInterval = 1
	conf.Store.Host = "localhost"
	conf.Store.Driver = tenv
	if tenv == "boltdb" {
	} else if tenv == config.Mysql {
		conf.Store.Port = 3306
		conf.Store.User = "root"
		conf.Store.Password = ""
	} else if tenv == config.Postgres {
		conf.Store.Port = 5432
		conf.Store.User = "postgres"
		conf.Store.Password = "mysecretpassword"
	} else if tenv == config.Redis {
		conf.Store.User = ""
		conf.Store.Password = ""
		conf.Store.Port = 6379
	} else if tenv == config.SQLServer {
		conf.Store.User = "sa"
		conf.Store.Password = "p@ssw0rd"
		conf.Store.Port = 1433
	}
	conf.Store.Db = ""
	registry.WaitStoreUp()

	dtmsvr.PopulateDB(false)
	conf.Store.Db = "dtm" // after populateDB, set current db to dtm
	if tenv == "postgres" {
		busi.BusiConf = conf.Store.GetDBConf()
		dtmcli.SetCurrentDBType(tenv)
	}
	go dtmsvr.StartSvr()

	busi.PopulateDB(false)
	hsvr, gsvr := busi.Startup()
	// WorkflowStarup 1
	workflow.InitHTTP(dtmutil.DefaultHTTPServer, Busi+"/workflow/resume")
	workflow.InitGrpc(dtmutil.DefaultGrpcServer, busi.BusiGrpc, gsvr)
	go busi.RunGrpc(gsvr)
	go busi.RunHTTP(hsvr)

	subscribeTopic()
	subscribeGrpcTopic()
	dtmsvr.CronUpdateTopicsMapOnce()
	logger.Debugf("unit main test inited")
	r := m.Run()
	if r != 0 {
		os.Exit(r)
	}
	close(dtmsvr.TransProcessedTestChan)
	gid, more := <-dtmsvr.TransProcessedTestChan
	logger.FatalfIf(more, "extra gid: %s in test chan", gid)
	os.Exit(0)
}