feat: implement collection library filter with WebSocket improvements and test coverage
This commit adds comprehensive functionality for filtering collections by library, improves WebSocket real-time updates with user activity detection, and adds extensive test coverage. ## Core Features ### Collection Library Filter - Added library_id parameter to media-items search API - Collections can now be filtered by specific library - Toggle UI component for enabling/disabling library filter - Default state is "checked" when library_id is present - Consistent behavior across partial and fuzzy search modes ### WebSocket Auto-Reload Mitigation - Added user activity detection to prevent disruptive page reloads - Checks if user is actively typing in INPUT/TEXTAREA/SELECT elements - Skips auto-reload when user is interacting with form elements - Toast notifications still show for awareness - Prevents data loss during editing operations ## Implementation Changes ### Backend - internal/database/queries.sql.go: Added library filter support to search queries - internal/handlers/media.go: Enhanced search with library_id parameter validation - internal/handlers/collections.go: Updated collection handlers with library filtering - internal/sync/websocket.go: Improved broadcast mechanism with user-scoped updates - internal/router/frontend.go: Pass libraryID to collection templates ### Frontend - templates/collections.templ: Added library filter toggle UI component - web/src/collections.ts: TypeScript implementation with WebSocket integration - templates/collections_templ.go: Generated template code ### Testing - cmd/server/tests/search_test.go: Added TestCollectionSearchLibraryFilter - cmd/server/tests/websocket_test.go: Added TestWebSocketUserScopedBroadcast - New helper functions for creating libraries and media items via API - Comprehensive test coverage for library filtering and user-scoped broadcasts ## API Documentation Updates ### Bruno Tests (Comprehensive Documentation) - bruno/collections/*: Added detailed API documentation for all collection endpoints - bruno/devices/*: Added device management and sync API documentation - bruno/devices/kobo/api.yml: Kobo-specific sync protocol docs - bruno/devices/koreader/api.yml: KOReader-specific sync protocol docs - bruno/opds/*: Added OPDS feed and download endpoint documentation - bruno/library/browse-folders.yml: Library folder browsing API docs ### New Bruno Tests - bruno/media-items/Search All Libraries.yml: Test search without library filter - bruno/media-items/Search Specific Library.yml: Test search with library filter - bruno/media-items/Search Invalid Library ID.yml: Test error handling ## Documentation - docs/developer/api/media-items/search_media_items.md: Updated with library_id parameter - IMPLEMENTATION_COLLECTION_FIX.md: Comprehensive implementation guide with test scenarios ## Testing ### Integration Tests - Library filter tests verify correct filtering across multiple libraries - Invalid library_id tests ensure proper error handling - WebSocket tests verify user-scoped broadcast behavior - User A no longer receives User B's collection updates ### Manual Testing Scenarios - Open collection in multiple tabs - updates propagate correctly - Type in search box while another tab adds books - no disruptive reload - Add/remove books from collection - toast notifications appear - Toggle library filter - results update dynamically ## Technical Details - WebSocket broadcasts are now user-scoped for privacy - Active element detection uses tagName and contenteditable attributes - Library ID validation uses UUID format checking - Progressive enhancement maintained - page works without JavaScript - All changes follow PROJECT_GUIDELINES.md conventions - TypeScript only for frontend logic - TailwindCSS only for styling - Procedural programming style throughout ## Breaking Changes None - all changes are additive and backward compatible.
This commit is contained in:
@@ -2,10 +2,13 @@ package main
|
||||
|
||||
import (
|
||||
"bookhoard/internal/database"
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/gorilla/websocket"
|
||||
@@ -238,3 +241,102 @@ func createTestMediaItem(t *testing.T, db *database.Queries, userID uuid.UUID) s
|
||||
|
||||
return uuid.UUID(mediaID.ID.Bytes).String()
|
||||
}
|
||||
|
||||
// TestWebSocketUserScopedBroadcast tests that broadcasts only go to the user who made changes
|
||||
func TestWebSocketUserScopedBroadcast(t *testing.T) {
|
||||
setup := setupTestServer(t)
|
||||
client := &http.Client{}
|
||||
|
||||
// Use pre-created admin user (setup.Token) and regular user (setup.RegularToken)
|
||||
// Both users already created by setupTestServer()
|
||||
|
||||
// Create a collection for admin user
|
||||
collectionReq := map[string]interface{}{
|
||||
"name": "Admin Collection",
|
||||
"description": "Test collection for WebSocket test",
|
||||
}
|
||||
collectionBody, _ := json.Marshal(collectionReq)
|
||||
|
||||
collectionHTTP, _ := http.NewRequest("POST", setup.Server.URL+"/api/collections", bytes.NewBuffer(collectionBody))
|
||||
collectionHTTP.Header.Set("Content-Type", "application/json")
|
||||
collectionHTTP.Header.Set("Authorization", "Bearer "+setup.Token)
|
||||
|
||||
collectionResp, err := client.Do(collectionHTTP)
|
||||
require.NoError(t, err)
|
||||
defer collectionResp.Body.Close()
|
||||
require.Equal(t, http.StatusCreated, collectionResp.StatusCode)
|
||||
|
||||
var collectionResult map[string]interface{}
|
||||
json.NewDecoder(collectionResp.Body).Decode(&collectionResult)
|
||||
collectionID := collectionResult["id"].(string)
|
||||
|
||||
// Create a test book via API
|
||||
bookID := createTestMediaItemID(t, setup)
|
||||
|
||||
// Connect admin user via WebSocket
|
||||
wsAdmin := connectWebSocketToServer(t, setup.Server.URL, setup.Token)
|
||||
defer wsAdmin.Close()
|
||||
|
||||
// Connect regular user via WebSocket
|
||||
wsRegular := connectWebSocketToServer(t, setup.Server.URL, setup.RegularToken)
|
||||
defer wsRegular.Close()
|
||||
|
||||
// Admin adds book to collection
|
||||
addReq := map[string]interface{}{
|
||||
"book_ids": []string{bookID},
|
||||
}
|
||||
addBody, _ := json.Marshal(addReq)
|
||||
|
||||
addHTTP, _ := http.NewRequest("POST", setup.Server.URL+"/api/collections/"+collectionID+"/books", bytes.NewBuffer(addBody))
|
||||
addHTTP.Header.Set("Content-Type", "application/json")
|
||||
addHTTP.Header.Set("Authorization", "Bearer "+setup.Token)
|
||||
|
||||
addResp, err := client.Do(addHTTP)
|
||||
require.NoError(t, err)
|
||||
defer addResp.Body.Close()
|
||||
require.Equal(t, http.StatusNoContent, addResp.StatusCode)
|
||||
|
||||
// Admin should receive collection_updated message
|
||||
msgAdmin := readWebSocketMessage(t, wsAdmin, 2*time.Second)
|
||||
if msgAdmin["type"] != "collection_updated" {
|
||||
t.Errorf("Admin should receive collection_updated, got %s", msgAdmin["type"])
|
||||
}
|
||||
|
||||
// Regular user should NOT receive collection_updated message
|
||||
wsRegular.SetReadDeadline(time.Now().Add(500 * time.Millisecond))
|
||||
_, _, err = wsRegular.ReadMessage()
|
||||
if err == nil {
|
||||
t.Errorf("Regular user should not receive collection_updated message")
|
||||
}
|
||||
}
|
||||
|
||||
// Helper: connectWebSocketToServer establishes WebSocket connection with auth token
|
||||
func connectWebSocketToServer(t *testing.T, serverURL string, token string) *websocket.Conn {
|
||||
wsURL := "ws" + strings.TrimPrefix(serverURL, "http") + "/ws/sync?token=" + token
|
||||
|
||||
ws, resp, err := websocket.DefaultDialer.Dial(wsURL, nil)
|
||||
require.NoError(t, err, "WebSocket connection should succeed")
|
||||
if resp != nil {
|
||||
resp.Body.Close()
|
||||
}
|
||||
require.NotNil(t, ws, "WebSocket connection should be established")
|
||||
|
||||
// Wait for connection to be ready
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
|
||||
return ws
|
||||
}
|
||||
|
||||
// Helper: readWebSocketMessage reads a message from WebSocket with timeout
|
||||
func readWebSocketMessage(t *testing.T, ws *websocket.Conn, timeout time.Duration) map[string]interface{} {
|
||||
ws.SetReadDeadline(time.Now().Add(timeout))
|
||||
|
||||
_, message, err := ws.ReadMessage()
|
||||
require.NoError(t, err, "Should receive WebSocket message")
|
||||
|
||||
var msg map[string]interface{}
|
||||
err = json.Unmarshal(message, &msg)
|
||||
require.NoError(t, err, "Message should be valid JSON")
|
||||
|
||||
return msg
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user