105 lines
3.8 KiB
Go
105 lines
3.8 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"yingce/backend/internal/app"
|
|
"yingce/backend/internal/database"
|
|
"yingce/backend/internal/model"
|
|
"yingce/backend/internal/repository"
|
|
"yingce/backend/internal/service"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func TestAppearanceRoutesAreRegistered(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
router := gin.New()
|
|
RegisterAppearanceRoutes(router.Group("/api"), &service.Service{})
|
|
wanted := map[string]bool{
|
|
"GET /api/public/appearance/updates/:id": false,
|
|
"GET /api/admin/settings/appearance/updates/:id": false,
|
|
"POST /api/admin/settings/appearance/live2d": false,
|
|
"GET /api/admin/settings/appearance/live2d/:id/*file": false,
|
|
"GET /api/public/appearance/live2d/:id/*file": false,
|
|
"GET /api/public/appearance": false,
|
|
"GET /api/public/appearance/assets/:slot": false,
|
|
"GET /api/admin/settings/appearance": false,
|
|
"PATCH /api/admin/settings/appearance": false,
|
|
"DELETE /api/admin/settings/appearance": false,
|
|
"POST /api/admin/settings/appearance/assets/:slot": false,
|
|
}
|
|
for _, route := range router.Routes() {
|
|
key := route.Method + " " + route.Path
|
|
if _, exists := wanted[key]; exists {
|
|
wanted[key] = true
|
|
}
|
|
}
|
|
for route, found := range wanted {
|
|
if !found {
|
|
t.Errorf("route %s is not registered", route)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAppearanceUpdateAnnouncementMediaRangeAndDisabledAccess(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
db, err := gorm.Open(sqlite.Open("file:appearance-handler-range?mode=memory&cache=shared"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := database.MigrateSchema(db); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dir := t.TempDir()
|
|
if err := os.MkdirAll(filepath.Join(dir, "resources"), 0750); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(filepath.Join(dir, "resources", "fixture.mp4"), []byte("video-data"), 0640); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
resource := model.Resource{ID: "fixture-video", UserID: "admin", Kind: "video", Status: model.ResourceStatusReady, Provider: "local", ObjectKey: "fixture.mp4", MimeType: "video/mp4", Size: 10}
|
|
if err := db.Create(&resource).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
svc := service.New(repository.New(db), dir)
|
|
admin := &model.User{ID: "admin", Role: model.UserRoleAdmin, Status: model.UserStatusActive}
|
|
setting, err := svc.AdminAppearance(admin)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
setting.Updates = app.UpdateAnnouncement{Enabled: true, CurrentVersion: "v2", Releases: []app.UpdateRelease{{ID: "r1", Version: "v2", Blocks: []app.UpdateBlock{{ID: "b1", Type: "video", ResourceID: resource.ID, Layout: "full"}}}}}
|
|
if _, err := svc.UpdateAppearance(admin, setting.AppearanceSetting); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
router := gin.New()
|
|
RegisterAppearanceRoutes(router.Group("/api"), svc)
|
|
req := httptest.NewRequest(http.MethodGet, "/api/public/appearance/updates/fixture-video", nil)
|
|
req.Header.Set("Range", "bytes=0-3")
|
|
rec := httptest.NewRecorder()
|
|
router.ServeHTTP(rec, req)
|
|
if rec.Code != 206 || rec.Body.String() != "vide" || rec.Header().Get("Content-Range") != "bytes 0-3/10" {
|
|
t.Fatalf("Range HTTP response: %d %q %v", rec.Code, rec.Body.String(), rec.Header())
|
|
}
|
|
preview := httptest.NewRecorder()
|
|
router.ServeHTTP(preview, httptest.NewRequest(http.MethodGet, "/api/admin/settings/appearance/updates/fixture-video", nil))
|
|
if preview.Code != 401 {
|
|
t.Fatalf("anonymous admin preview: %d", preview.Code)
|
|
}
|
|
setting.Updates.Enabled = false
|
|
if _, err := svc.UpdateAppearance(admin, setting.AppearanceSetting); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
disabled := httptest.NewRecorder()
|
|
router.ServeHTTP(disabled, req)
|
|
if disabled.Code != 404 {
|
|
t.Fatalf("disabled public media: %d", disabled.Code)
|
|
}
|
|
}
|