refactor(tests): clean up websocket test helper and fix broadcast test

- Remove createTestMediaItem helper function and replace with createTestMediaItemID
- Update TestWebSocketProgressBroadcast to use simplified helper
- Add read deadline and initial message read in TestWebSocketUserScopedBroadcast to properly consume initial connection messages
- This reduces code duplication and improves test reliability by properly handling WebSocket connection setup

cmd/server/tests/websocket_test.go

@@ -104,8 +104,7 @@ func TestWebSocketProgressBroadcast(t *testing.T) {
 	token := setup.Token
 
 	// Create a test media item
-	userID := getTestUserID(t, setup.DB)
-	mediaID := createTestMediaItem(t, setup.DB, userID)
+	mediaID := createTestMediaItemID(t, setup)
 
 	// Connect WebSocket client
 	wsURL := strings.Replace(setup.Server.URL, "http", "ws", 1) + "/ws/sync?token=" + token
@@ -233,30 +232,6 @@ func TestWebSocketInvalidToken(t *testing.T) {
 	assert.Error(t, err)
 }
 
-// Helper function to create a test media item
-func createTestMediaItem(t *testing.T, db *database.Queries, userID uuid.UUID) string {
-	// First create a test library
-	libID, err := db.CreateLibrary(context.Background(), database.CreateLibraryParams{
-		Name:             "Test Library",
-		LibraryTypeID:    pgtype.UUID{Bytes: [16]byte(uuid.UUID{}), Valid: true},
-		CreatedByAdminID: pgtype.UUID{Bytes: [16]byte(userID), Valid: true},
-	})
-	require.NoError(t, err)
-
-	// Create a test media item
-	mediaID, err := db.CreateMediaItem(context.Background(), database.CreateMediaItemParams{
-		LibraryID:      libID.ID,
-		Title:          "Test Book",
-		FilePath:       "/tmp/test.epub",
-		FileSize:       pgtype.Int8{Int64: 1024, Valid: true},
-		MimeType:       pgtype.Text{String: "application/epub+zip", Valid: true},
-		AddedByAdminID: pgtype.UUID{Bytes: [16]byte(userID), Valid: true},
-	})
-	require.NoError(t, err)
-
-	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)
@@ -292,10 +267,16 @@ func TestWebSocketUserScopedBroadcast(t *testing.T) {
 	wsAdmin := connectWebSocketToServer(t, setup.Server.URL, setup.Token)
 	defer wsAdmin.Close()
 
+	wsAdmin.SetReadDeadline(time.Now().Add(2 * time.Second))
+	_, _, _ = wsAdmin.ReadMessage()
+
 	// Connect regular user via WebSocket
 	wsRegular := connectWebSocketToServer(t, setup.Server.URL, setup.RegularToken)
 	defer wsRegular.Close()
 
+	wsRegular.SetReadDeadline(time.Now().Add(2 * time.Second))
+	_, _, _ = wsRegular.ReadMessage()
+
 	// Admin adds book to collection
 	addReq := map[string]interface{}{
 		"book_ids": []string{bookID},