mirror of
https://github.com/balzack/databag.git
synced 2025-02-11 19:19:16 +00:00
92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package databag
|
|
|
|
import (
|
|
"databag/internal/store"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
|
|
os.RemoveAll("testdata")
|
|
os.RemoveAll("testscripts")
|
|
|
|
if err := os.Mkdir("testdata", os.ModePerm); err != nil {
|
|
panic("failed to create testdata path")
|
|
}
|
|
if err := os.Mkdir("testdata/assets", os.ModePerm); err != nil {
|
|
panic("failed to create assets path")
|
|
}
|
|
if err := os.Mkdir("testscripts", os.ModePerm); err != nil {
|
|
panic("failed to create testscripts path")
|
|
}
|
|
script := []byte("#!/bin/bash\n\ncp $1 $2\n")
|
|
if err := os.WriteFile("testscripts/transform_copy.sh", script, 0555); err != nil {
|
|
panic("failed to create P01 script")
|
|
}
|
|
|
|
store.SetPath("./testdata", "./testscripts")
|
|
|
|
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?token=pass", nil)
|
|
SetNodeStatus(w, r)
|
|
if ReadResponse(w, nil) != nil {
|
|
panic("failed to claim server")
|
|
}
|
|
|
|
// config open access
|
|
access := &store.Config{ConfigID: CNFEnableOpenAccess, BoolValue: true}
|
|
if err := store.DB.Save(access).Error; err != nil {
|
|
panic("failed to configure open access")
|
|
}
|
|
|
|
// config account limit
|
|
limit := &store.Config{ConfigID: CNFOpenAccessLimit, NumValue: 1024}
|
|
if err := store.DB.Save(limit).Error; err != nil {
|
|
panic("failed to configure account limit")
|
|
}
|
|
|
|
// admin login
|
|
r, w, _ = NewRequest("PUT", "/admin/access?token=pass", nil);
|
|
SetAdminAccess(w, r)
|
|
var session string
|
|
if ReadResponse(w, &session) != nil {
|
|
panic("failed to login as admin")
|
|
}
|
|
|
|
// config server
|
|
config := NodeConfig{Domain: "databag.coredb.org", AccountStorage: 4096, KeyType: "RSA2048"}
|
|
r, w, _ = NewRequest("PUT", "/admin/config?token=" + session, &config)
|
|
SetNodeConfig(w, r)
|
|
if ReadResponse(w, nil) != nil {
|
|
panic("failed to set config")
|
|
}
|
|
|
|
// check config
|
|
r, w, _ = NewRequest("GET", "/admin/config?token=" + session, nil)
|
|
GetNodeConfig(w, r)
|
|
var check NodeConfig
|
|
if ReadResponse(w, &check) != nil {
|
|
panic("failed to get node config")
|
|
}
|
|
if check.Domain != "databag.coredb.org" {
|
|
panic("failed to set config domain")
|
|
}
|
|
if check.AccountStorage != 4096 {
|
|
panic("failed to set account storage")
|
|
}
|
|
|
|
go SendNotifications()
|
|
|
|
m.Run()
|
|
|
|
ExitNotifications()
|
|
}
|