47 lines
1.4 KiB
Go
47 lines
1.4 KiB
Go
package media_shard
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
|
|
"git.ilapage.cn/ila/yovision/Sense/server/app/sense/quota"
|
|
)
|
|
|
|
func TestSpecsFromEnvironmentUsesDatabaseQuotaAndExternalFile(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = db.AutoMigrate("a.Setting{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = db.Create("a.Setting{ID: quota.SettingID, Limit: 23, Source: "database", Version: 1}).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
path := filepath.Join(t.TempDir(), "shards.json")
|
|
if err = os.WriteFile(path, []byte(`[{"id":"secondary","name":"备用分片","controlAPI":"http://127.0.0.1:19997","capacity":11}]`), 0600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Setenv("SENSE_MEDIAMTX_CAPACITY", "")
|
|
t.Setenv("SENSE_MEDIAMTX_SHARDS_FILE", path)
|
|
specs, err := SpecsFromEnvironment(db, "managed", "http://127.0.0.1:9997")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(specs) != 2 || specs[0].Capacity != 23 || specs[1].Capacity != 11 || specs[1].Mode != "external" {
|
|
t.Fatalf("unexpected specs: %+v", specs)
|
|
}
|
|
}
|
|
|
|
func TestSpecsRejectsNonLoopbackControlAPI(t *testing.T) {
|
|
t.Setenv("SENSE_MEDIAMTX_CAPACITY", "5")
|
|
t.Setenv("SENSE_MEDIAMTX_SHARDS_FILE", "")
|
|
if _, err := SpecsFromEnvironment(nil, "external", "http://192.0.2.1:9997"); err == nil {
|
|
t.Fatal("expected unsafe control API to be rejected")
|
|
}
|
|
}
|