mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
63a24da9ee
Result: {"status":"keep","total_issues":8,"golint_canonicalheader":0,"golint_errname":0,"golint_errorlint":1,"golint_forcetypeassert":0,"golint_gosec":0,"golint_intrange":0,"golint_modernize":3,"golint_nilnil":3,"golint_perfsprint":0,"golint_prealloc":0,"golint_recvcheck":1,"golint_usestdlibvars":0,"golint_wastedassign":0,"golint_total":8,"golint_test_testifylint":0,"golint_test_thelper":0,"golint_test_usetesting":0,"golint_test_total":0,"eslint_problems":0,"eslint_errors":0,"eslint_warnings":0,"tsc_errors":0,"measure_s":39}
120 lines
2.6 KiB
Go
120 lines
2.6 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package logs
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
|
|
db "github.com/Rain-kl/Wavelet/internal/infra/persistence"
|
|
"github.com/Rain-kl/Wavelet/internal/model"
|
|
)
|
|
|
|
func setupTestDB(t *testing.T) *gorm.DB {
|
|
t.Helper()
|
|
dbConn, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatalf("failed to open sqlite in memory: %v", err)
|
|
}
|
|
err = dbConn.AutoMigrate(&model.SystemConfig{})
|
|
if err != nil {
|
|
t.Fatalf("failed to migrate schema: %v", err)
|
|
}
|
|
db.SetDB(dbConn)
|
|
return dbConn
|
|
}
|
|
|
|
func TestWebSocketCheckOrigin(t *testing.T) {
|
|
dbConn := setupTestDB(t)
|
|
|
|
// Clean up global DB after test
|
|
defer db.SetDB(nil)
|
|
|
|
// Seed ConfigKeyServerAddress with allowed frontend origin
|
|
allowedOrigin := "http://localhost:3000"
|
|
if err := dbConn.Create(&model.SystemConfig{
|
|
Key: model.ConfigKeyServerAddress,
|
|
Value: allowedOrigin,
|
|
}).Error; err != nil {
|
|
t.Fatalf("failed to seed server address config: %v", err)
|
|
}
|
|
|
|
upgrader := getUpgrader()
|
|
if upgrader.CheckOrigin == nil {
|
|
t.Fatal("expected CheckOrigin to be defined")
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
origin string
|
|
host string
|
|
wantOK bool
|
|
}{
|
|
{
|
|
name: "empty origin (non-browser clients)",
|
|
origin: "",
|
|
host: "localhost:8000",
|
|
wantOK: true,
|
|
},
|
|
{
|
|
name: "same-origin request",
|
|
origin: "http://localhost:8000",
|
|
host: "localhost:8000",
|
|
wantOK: true,
|
|
},
|
|
{
|
|
name: "same-origin request case insensitive",
|
|
origin: "HTTP://LOCALHOST:8000",
|
|
host: "localhost:8000",
|
|
wantOK: true,
|
|
},
|
|
{
|
|
name: "configured allowed origin request",
|
|
origin: "http://localhost:3000",
|
|
host: "localhost:8000",
|
|
wantOK: true,
|
|
},
|
|
{
|
|
name: "configured allowed origin request with trailing slash",
|
|
origin: "http://localhost:3000/",
|
|
host: "localhost:8000",
|
|
wantOK: true,
|
|
},
|
|
{
|
|
name: "unauthorized third-party origin",
|
|
origin: "http://evil.com",
|
|
host: "localhost:8000",
|
|
wantOK: false,
|
|
},
|
|
{
|
|
name: "invalid origin format",
|
|
origin: "::not-a-valid-url",
|
|
host: "localhost:8000",
|
|
wantOK: false,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
req, err := http.NewRequestWithContext(context.Background(), "GET", "/api/v1/admin/logs/ws", nil)
|
|
if err != nil {
|
|
t.Fatalf("failed to create request: %v", err)
|
|
}
|
|
req.Host = tt.host
|
|
if tt.origin != "" {
|
|
req.Header.Set("Origin", tt.origin)
|
|
}
|
|
|
|
got := upgrader.CheckOrigin(req)
|
|
if got != tt.wantOK {
|
|
t.Errorf("CheckOrigin() = %v, want %v", got, tt.wantOK)
|
|
}
|
|
})
|
|
}
|
|
}
|