headscale/app_test.go

62 lines
874 B
Go
Raw Normal View History

package headscale
import (
2022-09-02 15:16:19 +08:00
"net/netip"
"os"
"testing"
"gopkg.in/check.v1"
)
func Test(t *testing.T) {
check.TestingT(t)
}
var _ = check.Suite(&Suite{})
type Suite struct{}
2021-11-13 16:36:45 +08:00
var (
tmpDir string
2021-11-16 00:16:04 +08:00
app Headscale
2021-11-13 16:36:45 +08:00
)
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
2022-08-10 05:21:19 +08:00
tmpDir, err = os.MkdirTemp("", "autoygg-client-test")
if err != nil {
c.Fatal(err)
}
cfg := Config{
2022-09-02 15:16:19 +08:00
IPPrefixes: []netip.Prefix{
netip.MustParsePrefix("10.27.0.0/23"),
2022-01-16 21:16:59 +08:00
},
}
2021-11-16 00:16:04 +08:00
app = Headscale{
2022-06-05 23:47:26 +08:00
cfg: &cfg,
dbType: "sqlite3",
dbString: tmpDir + "/headscale_test.db",
}
2021-11-16 00:16:04 +08:00
err = app.initDB()
if err != nil {
c.Fatal(err)
}
2021-11-16 00:16:04 +08:00
db, err := app.openDB()
2021-07-05 03:40:46 +08:00
if err != nil {
c.Fatal(err)
}
2021-11-16 00:16:04 +08:00
app.db = db
}