fix tests

book_test.go

@@ -1,6 +1,9 @@
 package main
 
 import (
+	"encoding/json"
+	"fmt"
+	"log"
 	"net/http"
 	"net/http/httptest"
 	"strings"
@@ -17,10 +20,36 @@ func testSetup() *gin.Engine {
 	return setup(&c)
 }
 
+type loginResponse struct {
+	Message string `json:"message"`
+	Token   string `json:"token"`
+}
+
+func connectDemoUser(router *gin.Engine) string {
+	loginJson :=
+		`{
+		"username": "demo",
+		"password":"demopw"
+	}`
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/auth/login", strings.NewReader(loginJson))
+	router.ServeHTTP(w, req)
+	var parsedResponse loginResponse
+	body := w.Body.String()
+	err := json.Unmarshal([]byte(body), &parsedResponse)
+	if err != nil {
+		log.Fatal(err)
+	}
+	return parsedResponse.Token
+}
+
 func TestGetBooksHandler(t *testing.T) {
 	router := testSetup()
 	w := httptest.NewRecorder()
+
+	token := connectDemoUser(router)
 	req, _ := http.NewRequest("GET", "/books", nil)
+	req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", token))
 	router.ServeHTTP(w, req)
 
 	assert.Equal(t, 200, w.Code)
@@ -87,8 +116,10 @@ func testPostBookHandler(t *testing.T, bookJson string, expectedCode int) {
 	router := testSetup()
 	w := httptest.NewRecorder()
 
+	token := connectDemoUser(router)
 	req, _ := http.NewRequest("POST", "/book",
 		strings.NewReader(string(bookJson)))
+	req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", token))
 	router.ServeHTTP(w, req)
 
 	assert.Equal(t, expectedCode, w.Code)