initial commit

1 files changed, 89 insertions, 0 deletions

diff --git a/api/api_test.go b/api/api_test.go
new file mode 100644
index 0000000..f737413
--- /dev/null
+++ b/api/api_test.go
@@ -0,0 +1,89 @@
+package api_test
+
+import (
+	"database/sql"
+	"io"
+	"net/http/httptest"
+	"os"
+	"strings"
+	"testing"
+
+	"git.simponic.xyz/simponic/phoneof/api"
+	"git.simponic.xyz/simponic/phoneof/args"
+	"git.simponic.xyz/simponic/phoneof/database"
+	"git.simponic.xyz/simponic/phoneof/utils"
+)
+
+func setup(t *testing.T) (*sql.DB, *httptest.Server) {
+	randomDb := utils.RandomId()
+
+	testDb := database.MakeConn(&randomDb)
+	database.Migrate(testDb)
+
+	arguments := &args.Arguments{
+		TemplatePath: "../templates",
+		StaticPath:   "../static",
+	}
+
+	mux := api.MakeMux(arguments, testDb)
+	testServer := httptest.NewServer(mux)
+
+	t.Cleanup(func() {
+		testServer.Close()
+		testDb.Close()
+		os.Remove(randomDb)
+	})
+	return testDb, testServer
+}
+
+func assertResponseCode(t *testing.T, resp *httptest.ResponseRecorder, statusCode int) {
+	if resp.Code != statusCode {
+		t.Errorf("code is unexpected: %d, expected %d", resp.Code, statusCode)
+	}
+}
+
+func assertResponseBody(t *testing.T, resp *httptest.ResponseRecorder, body string) {
+	buf := new(strings.Builder)
+	_, err := io.Copy(buf, resp.Body)
+	if err != nil {
+		panic("could not read response body")
+	}
+	bodyStr := buf.String()
+	if bodyStr != body {
+		t.Errorf("body is unexpected: %s, expected %s", bodyStr, body)
+	}
+}
+
+func TestHealthcheck(t *testing.T) {
+	_, testServer := setup(t)
+
+	req := httptest.NewRequest("GET", "/health", nil)
+	resp := httptest.NewRecorder()
+	testServer.Config.Handler.ServeHTTP(resp, req)
+
+	assertResponseCode(t, resp, 200)
+	assertResponseBody(t, resp, "healthy")
+}
+
+func TestHello(t *testing.T) {
+	_, testServer := setup(t)
+
+	req := httptest.NewRequest("GET", "/", nil)
+	resp := httptest.NewRecorder()
+	testServer.Config.Handler.ServeHTTP(resp, req)
+
+	assertResponseCode(t, resp, 200)
+}
+
+func TestCachingStaticFiles(t *testing.T) {
+	_, testServer := setup(t)
+
+	req := httptest.NewRequest("GET", "/static/css/styles.css", nil)
+	resp := httptest.NewRecorder()
+	testServer.Config.Handler.ServeHTTP(resp, req)
+
+	assertResponseCode(t, resp, 200)
+	if resp.Header().Get("Cache-Control") != "public, max-age=3600" {
+		t.Errorf("client cache will live indefinitely for static files, which is probably not great! %s", resp.Header().Get("Cache-Control"))
+	}
+}