juanfont.headscale/hscontrol/db/suite_test.go

package db

import (
	"log"
	"os"
	"testing"

	"github.com/juanfont/headscale/hscontrol/types"
	"gopkg.in/check.v1"
)

func Test(t *testing.T) {
	check.TestingT(t)
}

var _ = check.Suite(&Suite{})

type Suite struct{}

var (
	tmpDir string
	db     *HSDatabase
)

func (s *Suite) SetUpTest(c *check.C) {
	s.ResetDB(c)
}

func (s *Suite) TearDownTest(c *check.C) {
	// os.RemoveAll(tmpDir)
}

func (s *Suite) ResetDB(c *check.C) {
	// if len(tmpDir) != 0 {
	// 	os.RemoveAll(tmpDir)
	// }

	var err error
	db, err = newTestDB()
	if err != nil {
		c.Fatal(err)
	}
}

func newTestDB() (*HSDatabase, error) {
	var err error
	tmpDir, err = os.MkdirTemp("", "headscale-db-test-*")
	if err != nil {
		return nil, err
	}

	log.Printf("database path: %s", tmpDir+"/headscale_test.db")

	db, err = NewHeadscaleDatabase(
		types.DatabaseConfig{
			Type: "sqlite3",
			Sqlite: types.SqliteConfig{
				Path: tmpDir + "/headscale_test.db",
			},
		},
		"",
	)
	if err != nil {
		return nil, err
	}

	return db, nil
}