mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-06-06 21:24:10 +00:00
09f84782b0
- Add tests for config, database, model, util/common, util/crypto, util/random, web/middleware, web/service, and xray packages - Fix redirect middleware using slice instead of map to guarantee deterministic longest-prefix-first matching order
100 lines
2.4 KiB
Go
100 lines
2.4 KiB
Go
package service
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestIsValidGeofileName_Valid(t *testing.T) {
|
|
svc := &ServerService{}
|
|
valid := []string{
|
|
"geoip.dat",
|
|
"geosite.dat",
|
|
"geoip_IR.dat",
|
|
"geoip_RU.dat",
|
|
"geosite_IR.dat",
|
|
"custom-file_v2.dat",
|
|
}
|
|
for _, name := range valid {
|
|
if !svc.IsValidGeofileName(name) {
|
|
t.Errorf("IsValidGeofileName(%q) should return true", name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsValidGeofileName_PathTraversal(t *testing.T) {
|
|
svc := &ServerService{}
|
|
invalid := []string{
|
|
"../geoip.dat",
|
|
"../../etc/passwd",
|
|
"subdir/geoip.dat",
|
|
"geoip.dat/../../../etc",
|
|
"..\\geoip.dat",
|
|
}
|
|
for _, name := range invalid {
|
|
if svc.IsValidGeofileName(name) {
|
|
t.Errorf("IsValidGeofileName(%q) should return false (path traversal)", name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsValidGeofileName_Empty(t *testing.T) {
|
|
svc := &ServerService{}
|
|
if svc.IsValidGeofileName("") {
|
|
t.Error("IsValidGeofileName(\"\") should return false")
|
|
}
|
|
}
|
|
|
|
func TestIsValidGeofileName_NoDatExtension(t *testing.T) {
|
|
svc := &ServerService{}
|
|
invalid := []string{
|
|
"geoip.txt",
|
|
"geosite",
|
|
"file.exe",
|
|
"script.sh",
|
|
}
|
|
for _, name := range invalid {
|
|
if svc.IsValidGeofileName(name) {
|
|
t.Errorf("IsValidGeofileName(%q) should return false (no .dat extension)", name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsValidGeofileName_SpecialChars(t *testing.T) {
|
|
svc := &ServerService{}
|
|
invalid := []string{
|
|
"geoip$.dat",
|
|
"geoip!.dat",
|
|
"geoip;.dat",
|
|
"geoip .dat",
|
|
"geoip@attack.dat",
|
|
}
|
|
for _, name := range invalid {
|
|
if svc.IsValidGeofileName(name) {
|
|
t.Errorf("IsValidGeofileName(%q) should return false (special chars)", name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestLogEntryContains(t *testing.T) {
|
|
tests := []struct {
|
|
line string
|
|
suffixes []string
|
|
want bool
|
|
}{
|
|
// The implementation checks strings.Contains(line, sfx+"]")
|
|
{"line with freedom]", []string{"freedom"}, true},
|
|
{"line with blackhole]", []string{"blackhole"}, true},
|
|
{"freedom outbound", []string{"freedom"}, false},
|
|
{"blackhole outbound", []string{"blackhole"}, false},
|
|
{"freedom outbound", []string{"blackhole"}, false},
|
|
{"some log line", []string{}, false},
|
|
{"line with freedom] and blackhole]", []string{"freedom", "blackhole"}, true},
|
|
{"line with freedom] and blackhole]", []string{"other"}, false},
|
|
}
|
|
for _, tt := range tests {
|
|
got := logEntryContains(tt.line, tt.suffixes)
|
|
if got != tt.want {
|
|
t.Errorf("logEntryContains(%q, %v) = %v, want %v", tt.line, tt.suffixes, got, tt.want)
|
|
}
|
|
}
|
|
}
|