quartzgun/quartzgun_test.go

package main

import (
	"context"
	"fmt"
	"hacklab.nilfm.cc/quartzgun/indentalUserDB"
	"hacklab.nilfm.cc/quartzgun/middleware"
	"hacklab.nilfm.cc/quartzgun/rateLimiter"
	"hacklab.nilfm.cc/quartzgun/renderer"
	"hacklab.nilfm.cc/quartzgun/router"
	"html/template"
	"net/http"
	"testing"
)

func AddContent(next http.Handler) http.Handler {
	handlerFunc := func(w http.ResponseWriter, req *http.Request) {
		if !req.Form.Has("Content") {
			req.Form.Add("Content", "Yesssssss")
		}
		next.ServeHTTP(w, req)
	}
	return http.HandlerFunc(handlerFunc)
}

func ApiSomething(next http.Handler) http.Handler {
	handlerFunc := func(w http.ResponseWriter, req *http.Request) {
		*req = *req.WithContext(context.WithValue(req.Context(), "apiData", "something"))
		next.ServeHTTP(w, req)
	}

	return http.HandlerFunc(handlerFunc)
}

func TestMain(m *testing.M) {
	udb := indentalUserDB.CreateIndentalUserDB("testData/userDB.ndtl")
	udb.AddUser("nilix", "questing")
	sesh, _ := udb.InitiateSession("nilix", "questing", 60)

	bouncer := rateLimiter.IpRateLimiter{
		map[string]*rateLimiter.RateLimitData{},
		5,
		2,
	}

	fmt.Printf("%s // %s\n", sesh, sesh)
	rtr := &router.Router{
		StaticPaths: map[string]string{
			"/static": "testData/static",
		},
		Fallback: *template.Must(template.ParseFiles("testData/templates/error.html", "testData/templates/footer.html")),
	}

	rtr.Get("/login", renderer.Template(
		"testData/templates/login.html"))

	rtr.Post("/login", middleware.Authorize("/", udb, "/login?tryagain=1", 120))

	rtr.Post("/provision", middleware.Provision(udb, 60))
	rtr.Get("/protected", middleware.Validate(renderer.Template("testData/templates/test.html"), udb, map[string]string{}))

	rtr.Get("/", middleware.Protected(
		renderer.Template(
			"testData/templates/test.html"), http.MethodGet, udb, "/login"))

	rtr.Get("/json", middleware.Throttle(ApiSomething(renderer.JSON("apiData")), bouncer.RateLimit))

	rtr.Get(`/thing/(?P<Thing>\w+)`, renderer.Template("testData/templates/paramTest.html"))

	http.ListenAndServe(":8080", rtr)
}