Added helper to switch environments.
This commit is contained in:
parent
6b537c5626
commit
0a8d08d8bd
@ -1,6 +1,7 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/joeshaw/envdecode"
|
||||
@ -85,3 +86,9 @@ func GetConfig() (Config, error) {
|
||||
err := envdecode.StrictDecode(&cfg)
|
||||
return cfg, err
|
||||
}
|
||||
|
||||
func SwitchEnv(env Env) {
|
||||
if err := os.Setenv("APP_ENV", string(env)); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
|
@ -23,15 +23,14 @@ var (
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
// Set the environment to test
|
||||
if err := os.Setenv("APP_ENV", string(config.EnvTest)); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
config.SwitchEnv(config.EnvTest)
|
||||
|
||||
// Start a test HTTP server
|
||||
c = container.NewContainer()
|
||||
BuildRouter(c)
|
||||
srv = httptest.NewServer(c.Web)
|
||||
|
||||
// Run tests
|
||||
exitVal := m.Run()
|
||||
srv.Close()
|
||||
os.Exit(exitVal)
|
Loading…
Reference in New Issue
Block a user