diff options
Diffstat (limited to 'testrig/db.go')
| -rw-r--r-- | testrig/db.go | 12 |
1 files changed, 3 insertions, 9 deletions
diff --git a/testrig/db.go b/testrig/db.go index 4304050cf..65ec391a7 100644 --- a/testrig/db.go +++ b/testrig/db.go @@ -73,15 +73,11 @@ var testModels = []interface{}{ // value as the port instead. func NewTestDB() db.DB { if alternateAddress := os.Getenv("GTS_DB_ADDRESS"); alternateAddress != "" { - config.Config(func(cfg *config.Configuration) { - cfg.DbAddress = alternateAddress - }) + config.SetDbAddress(alternateAddress) } if alternateDBType := os.Getenv("GTS_DB_TYPE"); alternateDBType != "" { - config.Config(func(cfg *config.Configuration) { - cfg.DbType = alternateDBType - }) + config.SetDbType(alternateDBType) } if alternateDBPort := os.Getenv("GTS_DB_PORT"); alternateDBPort != "" { @@ -89,9 +85,7 @@ func NewTestDB() db.DB { if err != nil { panic(err) } - config.Config(func(cfg *config.Configuration) { - cfg.DbPort = int(port) - }) + config.SetDbPort(int(port)) } var state state.State |