User book API: use a single query with result and count
This commit is contained in:
@@ -13,7 +13,7 @@ import (
|
||||
)
|
||||
|
||||
type bookAuthorGetResult struct {
|
||||
Count int64 `json:"count"`
|
||||
Count int64 `json:"count"`
|
||||
Books []bookAuthorGet `json:"books"`
|
||||
}
|
||||
|
||||
|
||||
@@ -1,73 +0,0 @@
|
||||
package apitest
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"git.artlef.fr/PersonalLibraryManager/internal/testutils"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
type countResponse struct {
|
||||
Count int
|
||||
}
|
||||
|
||||
func TestGetReadBooksCountHandler_Demo(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemoUser(router)
|
||||
|
||||
req, _ := http.NewRequest("GET", "/mybooks/read/count", nil)
|
||||
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", token))
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
var c countResponse
|
||||
err := json.Unmarshal(w.Body.Bytes(), &c)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
assert.Equal(t, 22, c.Count)
|
||||
}
|
||||
|
||||
func TestGetWantReadBooksCountHandler_Demo(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemoUser(router)
|
||||
|
||||
req, _ := http.NewRequest("GET", "/mybooks/wantread/count", nil)
|
||||
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", token))
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
var c countResponse
|
||||
err := json.Unmarshal(w.Body.Bytes(), &c)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
assert.Equal(t, 2, c.Count)
|
||||
}
|
||||
|
||||
func TestGetReadingBooksCountHandler_Demo(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemoUser(router)
|
||||
|
||||
req, _ := http.NewRequest("GET", "/mybooks/reading/count", nil)
|
||||
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", token))
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
var c countResponse
|
||||
err := json.Unmarshal(w.Body.Bytes(), &c)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
assert.Equal(t, 2, c.Count)
|
||||
}
|
||||
@@ -13,55 +13,60 @@ func TestGetReadBooksHandler_Demo(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemoUser(router)
|
||||
books := testGetReadBooksHandler(t, router, token, 200, "", "")
|
||||
assert.Equal(t, 22, len(books))
|
||||
result := testGetReadBooksHandler(t, router, token, 200, "", "")
|
||||
assert.Equal(t, 22, len(result.Books))
|
||||
assert.Equal(t, int64(22), result.Count)
|
||||
}
|
||||
|
||||
func TestGetReadBooksHandler_DemoNoLimit(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemoUser(router)
|
||||
books := testGetReadBooksHandler(t, router, token, 200, "100", "")
|
||||
assert.Equal(t, 22, len(books))
|
||||
result := testGetReadBooksHandler(t, router, token, 200, "100", "")
|
||||
assert.Equal(t, 22, len(result.Books))
|
||||
assert.Equal(t, int64(22), result.Count)
|
||||
}
|
||||
|
||||
func TestGetReadBooksHandler_DemoLowerLimit(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemoUser(router)
|
||||
books := testGetReadBooksHandler(t, router, token, 200, "5", "")
|
||||
assert.Equal(t, 5, len(books))
|
||||
result := testGetReadBooksHandler(t, router, token, 200, "5", "")
|
||||
assert.Equal(t, 5, len(result.Books))
|
||||
assert.Equal(t, int64(22), result.Count)
|
||||
}
|
||||
|
||||
func TestGetReadBooksHandler_DemoOffset(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemoUser(router)
|
||||
books := testGetReadBooksHandler(t, router, token, 200, "20", "10")
|
||||
assert.Equal(t, 12, len(books))
|
||||
result := testGetReadBooksHandler(t, router, token, 200, "20", "10")
|
||||
assert.Equal(t, 12, len(result.Books))
|
||||
assert.Equal(t, int64(22), result.Count)
|
||||
}
|
||||
|
||||
func TestGetReadBooksHandler_Demo2(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemo2User(router)
|
||||
books := testGetReadBooksHandler(t, router, token, 200, "", "")
|
||||
assert.Equal(t, 3, len(books))
|
||||
result := testGetReadBooksHandler(t, router, token, 200, "", "")
|
||||
assert.Equal(t, 3, len(result.Books))
|
||||
assert.Equal(t, int64(3), result.Count)
|
||||
}
|
||||
|
||||
func TestGetReadBooksHandler_CheckOneBook(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemo2User(router)
|
||||
books := testGetReadBooksHandler(t, router, token, 200, "100", "")
|
||||
var book bookUserGet
|
||||
for _, b := range books {
|
||||
result := testGetReadBooksHandler(t, router, token, 200, "100", "")
|
||||
var book bookUserGetBook
|
||||
for _, b := range result.Books {
|
||||
if b.Title == "De sang-froid" {
|
||||
book = b
|
||||
}
|
||||
}
|
||||
assert.Equal(t,
|
||||
bookUserGet{
|
||||
bookUserGetBook{
|
||||
BookId: 18,
|
||||
Title: "De sang-froid",
|
||||
Author: "Truman Capote",
|
||||
@@ -70,7 +75,7 @@ func TestGetReadBooksHandler_CheckOneBook(t *testing.T) {
|
||||
}, book)
|
||||
}
|
||||
|
||||
func testGetReadBooksHandler(t *testing.T, router *gin.Engine, userToken string, expectedCode int, limit string, offset string) []bookUserGet {
|
||||
func testGetReadBooksHandler(t *testing.T, router *gin.Engine, userToken string, expectedCode int, limit string, offset string) bookUserGet {
|
||||
u, err := url.Parse("/mybooks/read")
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
|
||||
@@ -13,19 +13,21 @@ func TestGetReadingBooksHandler_Demo(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemoUser(router)
|
||||
books := testGetReadingBooksHandler(t, router, token, 200, "", "")
|
||||
assert.Equal(t, 2, len(books))
|
||||
result := testGetReadingBooksHandler(t, router, token, 200, "", "")
|
||||
assert.Equal(t, 2, len(result.Books))
|
||||
assert.Equal(t, int64(2), result.Count)
|
||||
}
|
||||
|
||||
func TestGetReadingBooksHandler_Demo2(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemo2User(router)
|
||||
books := testGetReadingBooksHandler(t, router, token, 200, "", "")
|
||||
assert.Equal(t, 0, len(books))
|
||||
result := testGetReadingBooksHandler(t, router, token, 200, "", "")
|
||||
assert.Equal(t, 0, len(result.Books))
|
||||
assert.Equal(t, int64(0), result.Count)
|
||||
}
|
||||
|
||||
func testGetReadingBooksHandler(t *testing.T, router *gin.Engine, userToken string, expectedCode int, limit string, offset string) []bookUserGet {
|
||||
func testGetReadingBooksHandler(t *testing.T, router *gin.Engine, userToken string, expectedCode int, limit string, offset string) bookUserGet {
|
||||
u, err := url.Parse("/mybooks/reading")
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
|
||||
@@ -13,6 +13,11 @@ import (
|
||||
)
|
||||
|
||||
type bookUserGet struct {
|
||||
Count int64 `json:"count"`
|
||||
Books []bookUserGetBook `json:"books"`
|
||||
}
|
||||
|
||||
type bookUserGetBook struct {
|
||||
BookId uint `json:"id"`
|
||||
Title string `json:"title" binding:"required,max=300"`
|
||||
Author string `json:"author" binding:"max=100"`
|
||||
@@ -21,14 +26,14 @@ type bookUserGet struct {
|
||||
WantRead bool `json:"wantread"`
|
||||
}
|
||||
|
||||
func testGetbooksHandler(t *testing.T, router *gin.Engine, userToken string, expectedCode int, url string) []bookUserGet {
|
||||
func testGetbooksHandler(t *testing.T, router *gin.Engine, userToken string, expectedCode int, url string) bookUserGet {
|
||||
req, _ := http.NewRequest("GET", url, nil)
|
||||
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", userToken))
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
router.ServeHTTP(w, req)
|
||||
|
||||
var parsedResponse []bookUserGet
|
||||
var parsedResponse bookUserGet
|
||||
err := json.Unmarshal(w.Body.Bytes(), &parsedResponse)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
|
||||
@@ -12,18 +12,20 @@ func TestGetWantReadBooksHandler_Demo(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemoUser(router)
|
||||
books := testGetWantReadBooksHandler(t, router, token, 200)
|
||||
assert.Equal(t, 2, len(books))
|
||||
result := testGetWantReadBooksHandler(t, router, token, 200)
|
||||
assert.Equal(t, 2, len(result.Books))
|
||||
assert.Equal(t, int64(2), result.Count)
|
||||
}
|
||||
|
||||
func TestGetWantReadBooksHandler_Demo2(t *testing.T) {
|
||||
router := testutils.TestSetup()
|
||||
|
||||
token := testutils.ConnectDemo2User(router)
|
||||
books := testGetWantReadBooksHandler(t, router, token, 200)
|
||||
assert.Equal(t, 0, len(books))
|
||||
result := testGetWantReadBooksHandler(t, router, token, 200)
|
||||
assert.Equal(t, 0, len(result.Books))
|
||||
assert.Equal(t, int64(0), result.Count)
|
||||
}
|
||||
|
||||
func testGetWantReadBooksHandler(t *testing.T, router *gin.Engine, userToken string, expectedCode int) []bookUserGet {
|
||||
func testGetWantReadBooksHandler(t *testing.T, router *gin.Engine, userToken string, expectedCode int) bookUserGet {
|
||||
return testGetbooksHandler(t, router, userToken, expectedCode, "/mybooks/wantread")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user