Cleanup

arn/Utils.go

@@ -3,7 +3,6 @@ package arn
 import (
 	"encoding/json"
 	"errors"
-	"flag"
 	"fmt"
 	"os"
 	"path"
@@ -311,11 +310,6 @@ func ListItemStatusName(status string) string {
 	}
 }
 
-// IsTest returns true if the program is currently running in the "go test" tool.
-func IsTest() bool {
-	return flag.Lookup("test.v") != nil
-}
-
 // PanicOnError will panic if the error is not nil.
 func PanicOnError(err error) {
 	if err != nil {

main_test.go

@@ -93,14 +93,14 @@ func TestQuotes(t *testing.T) {
 	}
 }
 
-func TestUsers(t *testing.T) {
+// func TestUsers(t *testing.T) {
-	app := server.New()
+// 	app := server.New()
-	app.BindMiddleware()
+// 	app.BindMiddleware()
 
-	for user := range arn.StreamUsers() {
+// 	for user := range arn.StreamUsers() {
-		fetch(t, app, user.Link())
+// 		fetch(t, app, user.Link())
-	}
+// 	}
-}
+// }
 
 func fetch(t *testing.T, app http.Handler, route string) {
 	request := httptest.NewRequest("GET", strings.ReplaceAll(route, " ", "%20"), nil)

server/New.go

@@ -1,6 +1,7 @@
 package server
 
 import (
+	"flag"
 	"net/http"
 	"strings"
 
@@ -45,14 +46,18 @@ func New() *aero.Application {
 	app.Rewrite(pages.Rewrite)
 
 	// Middleware
-	app.Use(
+	if IsTest() {
-		middleware.Recover,
+		app.Use(middleware.OpenGraph)
-		middleware.HTTPSRedirect,
+	} else {
-		middleware.OpenGraph,
+		app.Use(
-		middleware.Log,
+			middleware.Recover,
-		middleware.Session,
+			middleware.HTTPSRedirect,
-		middleware.UserInfo,
+			middleware.OpenGraph,
-	)
+			middleware.Log,
+			middleware.Session,
+			middleware.UserInfo,
+		)
+	}
 
 	// API
 	arn.API.Install(app)
@@ -86,7 +91,7 @@ func New() *aero.Application {
 	arn.HTMLEmailRenderer = &htmlemail.Renderer{}
 
 	// Check that this is the server
-	if !arn.Node.IsServer() && !arn.IsTest() {
+	if !arn.Node.IsServer() && !IsTest() {
 		panic("Another program is currently running as the database server")
 	}
 
@@ -105,3 +110,8 @@ func New() *aero.Application {
 
 	return app
 }
+
+// IsTest returns true if the program is currently running in the "go test" tool.
+func IsTest() bool {
+	return flag.Lookup("test.v") != nil
+}