mirror of
https://git.vectorsigma.ru/public/photoview.git
synced 2026-08-03 18:19:11 +00:00
2b1240b88b
* Refactored code; temp disable workflows on push * remove printf leftovers * A better sanitizing code * extend auth routes tests * more tests, fix 2 bugs in token processing * more refactoring and fixes * optimize video mediaURLs DB query and make testCachePath in utils thread-safe * fix always passing test * and more refactoring; 1 more test, but cleanup still has to be fixed * more debug code in the test * fix the test * Final commit * Adding a defer function just in case of panic * Address some of review comments * Protect the shared var by mutex; implement correct context-aware scanning * Better error; better initial func wrap/capture; better log messages in `photos.go`; * Addressing a few more review comments * Order the query results; a better error type catch; setting the MP4 content type header explicitly * try to fix the cancelation detection condition * Extract a function and call it by name --------- Co-authored-by: Konstantin Koval
124 lines
3.6 KiB
Go
124 lines
3.6 KiB
Go
package routes
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/photoview/photoview/api/graphql/auth"
|
|
"github.com/photoview/photoview/api/graphql/models"
|
|
"github.com/photoview/photoview/api/scanner"
|
|
"github.com/photoview/photoview/api/test_utils"
|
|
"github.com/photoview/photoview/api/utils"
|
|
"github.com/stretchr/testify/assert"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func TestPhotoRoutes(t *testing.T) {
|
|
db := test_utils.DatabaseTest(t)
|
|
|
|
user, err := models.RegisterUser(db, "testuser", nil, false)
|
|
assert.NoError(t, err)
|
|
album := models.Album{Title: "test_album", Path: "/photos"}
|
|
assert.NoError(t, db.Model(&user).Association("Albums").Append(&album))
|
|
|
|
media := models.Media{
|
|
Title: "test_media",
|
|
Path: "/photos/test_image.jpg",
|
|
AlbumID: album.ID,
|
|
DateShot: time.Now(),
|
|
Type: models.MediaTypePhoto,
|
|
}
|
|
assert.NoError(t, db.Save(&media).Error)
|
|
|
|
mediaURL := models.MediaURL{
|
|
MediaID: media.ID,
|
|
Media: &media,
|
|
MediaName: "test_image.jpg",
|
|
Width: 1000,
|
|
Height: 800,
|
|
Purpose: models.PhotoThumbnail,
|
|
ContentType: "image/jpeg",
|
|
FileSize: 1024,
|
|
}
|
|
assert.NoError(t, db.Save(&mediaURL).Error)
|
|
|
|
tempDir := t.TempDir()
|
|
orig := utils.MediaCachePath()
|
|
utils.ConfigureTestCache(tempDir)
|
|
defer utils.ConfigureTestCache(orig)
|
|
|
|
router := mux.NewRouter()
|
|
RegisterPhotoRoutes(db, router)
|
|
|
|
// -- Test cases --
|
|
|
|
// Non-existent media_name => 404 (no auth required)
|
|
t.Run("media not found", func(t *testing.T) {
|
|
req := httptest.NewRequest("GET", "/does_not_exist.jpg", nil)
|
|
rec := httptest.NewRecorder()
|
|
router.ServeHTTP(rec, req)
|
|
assert.Equal(t, http.StatusNotFound, rec.Code)
|
|
assert.Equal(t, "404 - Media not found", rec.Body.String())
|
|
})
|
|
|
|
// Missing auth => 403
|
|
t.Run("auth failure", func(t *testing.T) {
|
|
req := httptest.NewRequest("GET", "/test_image.jpg", nil)
|
|
rec := httptest.NewRecorder()
|
|
router.ServeHTTP(rec, req)
|
|
assert.Equal(t, http.StatusForbidden, rec.Code)
|
|
assert.Contains(t, rec.Body.String(), "unauthorized")
|
|
})
|
|
|
|
// Cache miss + scan error => 500
|
|
t.Run("scan failure yields 500", func(t *testing.T) {
|
|
req := httptest.NewRequest("GET", "/test_image.jpg", nil)
|
|
ctx := auth.AddUserToContext(req.Context(), user)
|
|
req = req.WithContext(ctx)
|
|
|
|
// ensure no cached file
|
|
cachedPath, err := mediaURL.CachedPath()
|
|
assert.NoError(t, err)
|
|
os.Remove(cachedPath)
|
|
|
|
// mock scan to fail
|
|
origScan := scanner.ProcessSingleMediaFunc
|
|
scanner.ProcessSingleMediaFunc = func(ctx context.Context, db *gorm.DB, m *models.Media) error {
|
|
return fmt.Errorf("scan error")
|
|
}
|
|
defer func() { scanner.ProcessSingleMediaFunc = origScan }()
|
|
|
|
rec := httptest.NewRecorder()
|
|
router.ServeHTTP(rec, req)
|
|
assert.Equal(t, http.StatusInternalServerError, rec.Code)
|
|
assert.Contains(t, rec.Body.String(), "server error")
|
|
})
|
|
|
|
// Cache hit => 200 with correct body and headers
|
|
t.Run("cache hit serves file", func(t *testing.T) {
|
|
req := httptest.NewRequest("GET", "/test_image.jpg", nil)
|
|
ctx := auth.AddUserToContext(req.Context(), user)
|
|
req = req.WithContext(ctx)
|
|
|
|
// pre-create cached file
|
|
cachedPath, err := mediaURL.CachedPath()
|
|
assert.NoError(t, err)
|
|
assert.NoError(t, os.MkdirAll(path.Dir(cachedPath), 0755))
|
|
content := []byte("cached-binary")
|
|
assert.NoError(t, os.WriteFile(cachedPath, content, 0644))
|
|
|
|
rec := httptest.NewRecorder()
|
|
router.ServeHTTP(rec, req)
|
|
assert.Equal(t, http.StatusOK, rec.Code)
|
|
assert.Equal(t, "cached-binary", rec.Body.String())
|
|
assert.Equal(t, "private, max-age=86400, immutable", rec.Header().Get("Cache-Control"))
|
|
})
|
|
}
|