Add a test for issue #45; NFC

Fixes #45

router.go

@@ -79,23 +79,8 @@ func (w *statusResponseWriter) WriteHeader(statusCode int) {
 	w.Inner.WriteHeader(statusCode)
 }
 
-// StartMainServer starts the main http server listening on addr.
-func StartMainServer(addr string, db *db.Database, fs *db.FileStore, rawRootURL string) {
-	var rootURL *url.URL
-	if rawRootURL != "" {
-		var err error
-		rootURL, err = url.Parse(rawRootURL)
-		if err != nil {
-			log.Fatalln(errors.Wrap(err, "could not parse rootURL flag"))
-		}
-	}
-	rl := rushlink{
-		db:      db,
-		fs:      fs,
-		rootURL: rootURL,
-	}
-
-	// Initialize Gorilla router
+// CreateMainRouter creates the main Gorilla router for the application.
+func CreateMainRouter(rl *rushlink) *mux.Router {
 	router := mux.NewRouter()
 	router.Use(rl.recoveryMiddleware)
 	router.Use(rl.metricsMiddleware)
@@ -114,9 +99,27 @@ func StartMainServer(addr string, db *db.Database, fs *db.FileStore, rawRootURL
 	router.HandleFunc("/"+urlKeyWithExtExpr, rl.deletePasteHandler).Methods("DELETE")
 	router.HandleFunc("/"+urlKeyExpr+"/delete", rl.deletePasteHandler).Methods("POST")
 	router.HandleFunc("/"+urlKeyWithExtExpr+"/delete", rl.deletePasteHandler).Methods("POST")
+	return router
+}
+
+// StartMainServer starts the main http server listening on addr.
+func StartMainServer(addr string, db *db.Database, fs *db.FileStore, rawRootURL string) {
+	var rootURL *url.URL
+	if rawRootURL != "" {
+		var err error
+		rootURL, err = url.Parse(rawRootURL)
+		if err != nil {
+			log.Fatalln(errors.Wrap(err, "could not parse rootURL flag"))
+		}
+	}
+	rl := rushlink{
+		db:      db,
+		fs:      fs,
+		rootURL: rootURL,
+	}
 
 	srv := &http.Server{
-		Handler:      router,
+		Handler:      CreateMainRouter(&rl),
 		Addr:         addr,
 		WriteTimeout: 15 * time.Second,
 		ReadTimeout:  15 * time.Second,