Add pagination for booksearch

internal/apitest/search_book_test.go

@@ -6,6 +6,7 @@ import (
 
 	"net/http"
 	"net/http/httptest"
+	"net/url"
 	"testing"
 
 	"git.artlef.fr/PersonalLibraryManager/internal/testutils"
@@ -19,33 +20,96 @@ type bookSearchGet struct {
 }
 
 func TestSearchBook_MultipleBooks(t *testing.T) {
-	books := testSearchBook(t, "san")
+	books := testSearchBook(t, "san", "", "")
 
 	assert.Equal(t, 2, len(books))
 }
 
 func TestSearchBook_AllFields(t *testing.T) {
-	books := testSearchBook(t, "de san")
+	books := testSearchBook(t, "de san", "", "")
 	assert.Equal(t, 1, len(books))
 	assert.Equal(t,
 		[]bookSearchGet{{Title: "De sang-froid", Author: "Truman Capote", Id: 18}},
 		books)
 }
 
-func testSearchBook(t *testing.T, searchterm string) []bookSearchGet {
+func TestSearchBook_Limit(t *testing.T) {
+	books := testSearchBook(t, "a", "10", "")
+	assert.Equal(t, 10, len(books))
+}
+
+func TestSearchBook_Offset(t *testing.T) {
+	books := testSearchBook(t, "sa", "", "2")
+	assert.Equal(t, 3, len(books))
+}
+
+func testSearchBook(t *testing.T, searchterm string, limit string, offset string) []bookSearchGet {
 	router := testutils.TestSetup()
 
+	u, err := url.Parse("/search/" + searchterm)
+	if err != nil {
+		t.Error(err)
+	}
+	if limit != "" {
+		q := u.Query()
+		q.Set("limit", limit)
+		u.RawQuery = q.Encode()
+	}
+	if offset != "" {
+		q := u.Query()
+		q.Set("offset", offset)
+		u.RawQuery = q.Encode()
+	}
+
 	token := testutils.ConnectDemoUser(router)
-	req, _ := http.NewRequest("GET", "/search/"+searchterm, nil)
+	req, _ := http.NewRequest("GET", u.String(), nil)
 	req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", token))
 	w := httptest.NewRecorder()
 	router.ServeHTTP(w, req)
 
 	var books []bookSearchGet
-	err := json.Unmarshal(w.Body.Bytes(), &books)
+	err = json.Unmarshal(w.Body.Bytes(), &books)
 	if err != nil {
 		t.Error(err)
 	}
 	assert.Equal(t, 200, w.Code)
 	return books
 }
+
+func TestSearchBookCount_OK(t *testing.T) {
+	router := testutils.TestSetup()
+
+	token := testutils.ConnectDemoUser(router)
+	req, _ := http.NewRequest("GET", "/search/sa/count", nil)
+	req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", token))
+	w := httptest.NewRecorder()
+	router.ServeHTTP(w, req)
+
+	var count countResponse
+
+	assert.Equal(t, 200, w.Code)
+	err := json.Unmarshal(w.Body.Bytes(), &count)
+	if err != nil {
+		t.Error(err)
+	}
+	assert.Equal(t, 5, count.Count)
+}
+
+func TestSearchBookCount_Zero(t *testing.T) {
+	router := testutils.TestSetup()
+
+	token := testutils.ConnectDemoUser(router)
+	req, _ := http.NewRequest("GET", "/search/dsfsfdsdfsfd/count", nil)
+	req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", token))
+	w := httptest.NewRecorder()
+	router.ServeHTTP(w, req)
+
+	var count countResponse
+
+	assert.Equal(t, 200, w.Code)
+	err := json.Unmarshal(w.Body.Bytes(), &count)
+	if err != nil {
+		t.Error(err)
+	}
+	assert.Equal(t, 0, count.Count)
+}