mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package upload
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"Wavelet/core"
|
|
"Wavelet/core/contracts"
|
|
)
|
|
|
|
type stubDBService struct{ contracts.DBService }
|
|
|
|
type stubStorageService struct{ contracts.StorageService }
|
|
|
|
type stubAuthService struct{ contracts.AuthService }
|
|
|
|
func (stubAuthService) RequireAuthMiddleware() any {
|
|
return gin.HandlerFunc(func(c *gin.Context) { c.Next() })
|
|
}
|
|
|
|
func (stubAuthService) RequireAdminMiddleware() any {
|
|
return gin.HandlerFunc(func(c *gin.Context) { c.Next() })
|
|
}
|
|
|
|
func TestUserUploadRoutes(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
ctx := core.NewContext(context.Background())
|
|
core.Provide[contracts.DBService](ctx, stubDBService{})
|
|
core.Provide[contracts.StorageService](ctx, stubStorageService{})
|
|
core.Provide[contracts.AuthService](ctx, stubAuthService{})
|
|
if err := New().Apply(ctx); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
want := []string{
|
|
"POST /api/v1/upload",
|
|
"DELETE /api/v1/upload/:id",
|
|
"GET /api/v1/upload/my",
|
|
"PUT /api/v1/upload/:id",
|
|
"GET /api/v1/upload/download/:id",
|
|
"POST /api/v1/upload/download/batch",
|
|
"GET /api/v1/admin/uploads",
|
|
"GET /api/v1/admin/uploads/stats",
|
|
"DELETE /api/v1/admin/uploads/:id",
|
|
"GET /api/v1/admin/uploads/download/:id",
|
|
"POST /api/v1/admin/uploads/download/batch",
|
|
"GET /api/v1/admin/uploads/types",
|
|
}
|
|
found := make(map[string]bool, len(want))
|
|
for _, rd := range ctx.Router().Routes() {
|
|
found[rd.Method+" "+rd.Path] = true
|
|
}
|
|
for _, key := range want {
|
|
if !found[key] {
|
|
t.Errorf("missing route %s", key)
|
|
}
|
|
}
|
|
}
|