databag/net/server/internal/main_test.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()
}