mirror of
https://github.com/balzack/databag.git
synced 2025-02-12 03:29:16 +00:00
90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package databag
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"databag/internal/store"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
|
|
// SetHideLog(true)
|
|
SetKeySize(2048)
|
|
os.Remove("databag.db")
|
|
os.RemoveAll("testdata")
|
|
os.RemoveAll("testscripts")
|
|
|
|
store.SetPath("databag.db")
|
|
if err := os.Mkdir("testdata", os.ModePerm); err != nil {
|
|
panic("failed to create testdata path")
|
|
}
|
|
if err := os.Mkdir("testscripts", os.ModePerm); err != nil {
|
|
panic("failed to create testscripts path")
|
|
}
|
|
P01 := []byte("#!/bin/bash\n echo \"P01 $1 $2 $3\"\n")
|
|
if err := os.WriteFile("testscripts/P01.sh", P01, 0555); err != nil {
|
|
panic("failed to create P01 script")
|
|
}
|
|
|
|
r, w, _ := NewRequest("GET", "/admin/status", nil)
|
|
GetNodeStatus(w, r)
|
|
var available bool
|
|
if ReadResponse(w, &available) != nil {
|
|
panic("server not claimable")
|
|
}
|
|
|
|
// claim server
|
|
r, w, _ = NewRequest("PUT", "/admin/status", nil)
|
|
SetCredentials(r, "admin:pass");
|
|
SetNodeStatus(w, r)
|
|
if ReadResponse(w, nil) != nil {
|
|
panic("failed to claim server")
|
|
}
|
|
|
|
// config data path
|
|
scripts := &store.Config{ ConfigId: CONFIG_SCRIPTPATH, StrValue: "./testscripts" }
|
|
if err := store.DB.Save(scripts).Error; err != nil {
|
|
panic("failed to configure scripts path")
|
|
}
|
|
|
|
// config data path
|
|
path := &store.Config{ ConfigId: CONFIG_ASSETPATH, StrValue: "./testdata" }
|
|
if err := store.DB.Save(path).Error; err != nil {
|
|
panic("failed to configure data path")
|
|
}
|
|
|
|
// config server
|
|
config := NodeConfig{Domain: "example.com", PublicLimit: 1024, AccountStorage: 4096}
|
|
r, w, _ = NewRequest("PUT", "/admin/config", &config)
|
|
SetBasicAuth(r, "admin:pass")
|
|
SetNodeConfig(w, r)
|
|
if ReadResponse(w, nil) != nil {
|
|
panic("failed to set config")
|
|
}
|
|
|
|
// check config
|
|
r, w, _ = NewRequest("GET", "/admin/config", nil)
|
|
SetBasicAuth(r, "admin:pass")
|
|
GetNodeConfig(w, r)
|
|
var check NodeConfig
|
|
if ReadResponse(w, &check) != nil {
|
|
panic("failed to get node config")
|
|
}
|
|
if check.Domain != "example.com" {
|
|
panic("failed to set config domain");
|
|
}
|
|
if check.PublicLimit != 1024 {
|
|
panic("failed to set public limit");
|
|
}
|
|
if check.AccountStorage != 4096 {
|
|
panic("failed to set account storage");
|
|
}
|
|
|
|
go SendNotifications()
|
|
|
|
m.Run()
|
|
|
|
ExitNotifications()
|
|
}
|
|
|