mirror of
https://github.com/go-shiori/shiori.git
synced 2025-09-11 23:45:51 +08:00
d41981861c
* feat: add tag search api * feat: add apis to add/remove bookmark tags * chore: removed debug logger * docs: updated swagger * test: added tests * test: invalid ids * feat: webapp v2 * chore: updated swagger * fix: route params missing * feat: added cors middleware * feat: built api implementation * feat: implemented login in webapp * feat; dark mode in web app * feat: labels + i18n * refactor: remove custom message output in json apis * docs: updated swagger * chore: make lint * chore: make styles * fix: include new webapp dist files (for now)
97 lines
3 KiB
Go
97 lines
3 KiB
Go
package testutil
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/go-shiori/shiori/internal/http/response"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type testResponse struct {
|
|
Response response.Response
|
|
}
|
|
|
|
func (r *testResponse) AssertMessageIsEmptyList(t *testing.T) {
|
|
var jsonData []any
|
|
err := json.Unmarshal(r.Response.GetData().([]byte), &jsonData)
|
|
require.NoError(t, err)
|
|
require.Equal(t, []any{}, jsonData)
|
|
}
|
|
|
|
func (r *testResponse) AssertMessageIsNotEmptyList(t *testing.T) {
|
|
var jsonData []any
|
|
err := json.Unmarshal(r.Response.GetData().([]byte), &jsonData)
|
|
require.NoError(t, err)
|
|
require.Greater(t, len(jsonData), 0)
|
|
}
|
|
|
|
func (r *testResponse) AssertMessageIsListLength(t *testing.T, length int) {
|
|
var jsonData []any
|
|
err := json.Unmarshal(r.Response.GetData().([]byte), &jsonData)
|
|
require.NoError(t, err)
|
|
require.Len(t, jsonData, length)
|
|
}
|
|
|
|
// ForEach iterates over the items in the response and calls the provided function
|
|
// with each item.
|
|
func (r *testResponse) ForEach(t *testing.T, fn func(item map[string]any)) {
|
|
var jsonData []any
|
|
err := json.Unmarshal(r.Response.GetData().([]byte), &jsonData)
|
|
require.NoError(t, err)
|
|
for _, item := range jsonData {
|
|
fn(item.(map[string]any))
|
|
}
|
|
}
|
|
|
|
func (r *testResponse) AssertNilMessage(t *testing.T) {
|
|
require.Equal(t, nil, r.Response.GetData())
|
|
}
|
|
|
|
func (r testResponse) AssertMessageEquals(t *testing.T, expected any) {
|
|
require.Equal(t, expected, r.Response.GetData())
|
|
}
|
|
|
|
func (r testResponse) AssertMessageJSONContains(t *testing.T, expected string) {
|
|
require.JSONEq(t, expected, string(r.Response.GetData().([]byte)))
|
|
}
|
|
|
|
// AssertMessageJSONContainsKey asserts that the response message contains a key
|
|
// and returns the value of the key to be used in other comparisons depending on the
|
|
// value type.
|
|
func (r testResponse) AssertMessageJSONContainsKey(t *testing.T, key string) any {
|
|
var jsonData map[string]any
|
|
err := json.Unmarshal(r.Response.GetData().([]byte), &jsonData)
|
|
require.NoError(t, err)
|
|
require.Contains(t, jsonData, key)
|
|
return jsonData[key]
|
|
}
|
|
|
|
// AssertMessageJSONKeyValue asserts that the response message contains a key
|
|
// and calls the provided function with the value of the key to be used in other
|
|
// comparisons depending on the value type.
|
|
func (r *testResponse) AssertMessageJSONKeyValue(t *testing.T, key string, valueAssertFunc func(t *testing.T, value any)) {
|
|
value := r.AssertMessageJSONContainsKey(t, key)
|
|
valueAssertFunc(t, value)
|
|
}
|
|
|
|
func (r *testResponse) AssertMessageContains(t *testing.T, expected string) {
|
|
require.Contains(t, r.Response.GetData(), expected)
|
|
}
|
|
|
|
func (r *testResponse) AssertMessageIsBytes(t *testing.T, expected []byte) {
|
|
require.Equal(t, expected, r.Response.GetData().([]byte))
|
|
}
|
|
|
|
func (r *testResponse) AssertOk(t *testing.T) {
|
|
require.False(t, r.Response.IsError())
|
|
}
|
|
|
|
func (r *testResponse) AssertNotOk(t *testing.T) {
|
|
require.True(t, r.Response.IsError())
|
|
}
|
|
|
|
func NewTestResponseFromRecorder(w *httptest.ResponseRecorder) *testResponse {
|
|
return &testResponse{Response: *response.NewResponse(w.Body.Bytes(), w.Code)}
|
|
}
|