Added shutdown method to containers.

config/config_test.go

@@ -3,10 +3,18 @@ package config
 import (
 	"testing"
 
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
 
 func TestGetConfig(t *testing.T) {
 	_, err := GetConfig()
 	require.NoError(t, err)
+
+	var env Environment
+	env = "abc"
+	SwitchEnvironment(env)
+	cfg, err := GetConfig()
+	require.NoError(t, err)
+	assert.Equal(t, env, cfg.App.Environment)
 }

controller/controller_test.go

@@ -32,12 +32,14 @@ func TestMain(m *testing.M) {
 
 	// Create a new container
 	c = services.NewContainer()
+	defer func() {
+		if err := c.Shutdown(); err != nil {
+			c.Web.Logger.Fatal(err)
+		}
+	}()
 
 	// Run tests
 	exitVal := m.Run()
-	if err := c.Shutdown(); err != nil {
-		panic(err)
-	}
 	os.Exit(exitVal)
 }
 

main.go

@@ -13,7 +13,13 @@ import (
 )
 
 func main() {
+	// Start a new container
 	c := services.NewContainer()
+	defer func() {
+		if err := c.Shutdown(); err != nil {
+			c.Web.Logger.Fatal(err)
+		}
+	}()
 
 	// Build the router
 	routes.BuildRouter(c)

middleware/cache.go

@@ -54,7 +54,7 @@ func ServeCachedPage(ch *cache.Cache) echo.MiddlewareFunc {
 					c.Response().Header().Set(k, v)
 				}
 			}
-			c.Logger().Infof("serving cached page")
+			c.Logger().Info("serving cached page")
 
 			return c.HTMLBlob(page.StatusCode, page.HTML)
 		}

routes/routes_test.go

@@ -25,17 +25,21 @@ func TestMain(m *testing.M) {
 	// Set the environment to test
 	config.SwitchEnvironment(config.EnvTest)
 
-	// Start a test HTTP server
+	// Start a new container
 	c = services.NewContainer()
+	defer func() {
+		if err := c.Shutdown(); err != nil {
+			c.Web.Logger.Fatal(err)
+		}
+	}()
+
+	// Start a test HTTP server
 	BuildRouter(c)
 	srv = httptest.NewServer(c.Web)
 
 	// Run tests
 	exitVal := m.Run()
 	srv.Close()
-	if err := c.Shutdown(); err != nil {
-		panic(err)
-	}
 	os.Exit(exitVal)
 }