package integration import ( "testing" "github.com/juanfont/headscale/integration/dockertestutil" ) // This file is intended to "test the test framework", by proxy it will also test // some Headcsale/Tailscale stuff, but mostly in very simple ways. func IntegrationSkip(t *testing.T) { t.Helper() if !dockertestutil.IsRunningInContainer() { t.Skip("not running in docker, skipping") } if testing.Short() { t.Skip("skipping integration tests due to short flag") } } // If subtests are parallel, then they will start before setup is run. // This might mean we approach setup slightly wrong, but for now, ignore // the linter // nolint:tparallel func TestHeadscale(t *testing.T) { IntegrationSkip(t) t.Parallel() var err error namespace := "test-space" scenario, err := NewScenario() if err != nil { t.Errorf("failed to create scenario: %s", err) } t.Run("start-headscale", func(t *testing.T) { headscale, err := scenario.Headscale() if err != nil { t.Errorf("failed to create start headcale: %s", err) } err = headscale.WaitForReady() if err != nil { t.Errorf("headscale failed to become ready: %s", err) } }) t.Run("create-namespace", func(t *testing.T) { err := scenario.CreateNamespace(namespace) if err != nil { t.Errorf("failed to create namespace: %s", err) } if _, ok := scenario.namespaces[namespace]; !ok { t.Errorf("namespace is not in scenario") } }) t.Run("create-auth-key", func(t *testing.T) { _, err := scenario.CreatePreAuthKey(namespace, true, false) if err != nil { t.Errorf("failed to create preauthkey: %s", err) } }) err = scenario.Shutdown() if err != nil { t.Errorf("failed to tear down scenario: %s", err) } } // If subtests are parallel, then they will start before setup is run. // This might mean we approach setup slightly wrong, but for now, ignore // the linter // nolint:tparallel func TestCreateTailscale(t *testing.T) { IntegrationSkip(t) t.Parallel() namespace := "only-create-containers" scenario, err := NewScenario() if err != nil { t.Errorf("failed to create scenario: %s", err) } scenario.namespaces[namespace] = &Namespace{ Clients: make(map[string]TailscaleClient), } t.Run("create-tailscale", func(t *testing.T) { err := scenario.CreateTailscaleNodesInNamespace(namespace, "all", 3) if err != nil { t.Errorf("failed to add tailscale nodes: %s", err) } if clients := len(scenario.namespaces[namespace].Clients); clients != 3 { t.Errorf("wrong number of tailscale clients: %d != %d", clients, 3) } // TODO(kradalby): Test "all" version logic }) err = scenario.Shutdown() if err != nil { t.Errorf("failed to tear down scenario: %s", err) } } // If subtests are parallel, then they will start before setup is run. // This might mean we approach setup slightly wrong, but for now, ignore // the linter // nolint:tparallel func TestTailscaleNodesJoiningHeadcale(t *testing.T) { IntegrationSkip(t) t.Parallel() var err error namespace := "join-node-test" count := 1 scenario, err := NewScenario() if err != nil { t.Errorf("failed to create scenario: %s", err) } t.Run("start-headscale", func(t *testing.T) { headscale, err := scenario.Headscale() if err != nil { t.Errorf("failed to create start headcale: %s", err) } err = headscale.WaitForReady() if err != nil { t.Errorf("headscale failed to become ready: %s", err) } }) t.Run("create-namespace", func(t *testing.T) { err := scenario.CreateNamespace(namespace) if err != nil { t.Errorf("failed to create namespace: %s", err) } if _, ok := scenario.namespaces[namespace]; !ok { t.Errorf("namespace is not in scenario") } }) t.Run("create-tailscale", func(t *testing.T) { err := scenario.CreateTailscaleNodesInNamespace(namespace, "1.30.2", count) if err != nil { t.Errorf("failed to add tailscale nodes: %s", err) } if clients := len(scenario.namespaces[namespace].Clients); clients != count { t.Errorf("wrong number of tailscale clients: %d != %d", clients, count) } }) t.Run("join-headscale", func(t *testing.T) { key, err := scenario.CreatePreAuthKey(namespace, true, false) if err != nil { t.Errorf("failed to create preauthkey: %s", err) } headscale, err := scenario.Headscale() if err != nil { t.Errorf("failed to create start headcale: %s", err) } err = scenario.RunTailscaleUp( namespace, headscale.GetEndpoint(), key.GetKey(), ) if err != nil { t.Errorf("failed to login: %s", err) } }) t.Run("get-ips", func(t *testing.T) { ips, err := scenario.GetIPs(namespace) if err != nil { t.Errorf("failed to get tailscale ips: %s", err) } if len(ips) != count*2 { t.Errorf("got the wrong amount of tailscale ips, %d != %d", len(ips), count*2) } }) err = scenario.Shutdown() if err != nil { t.Errorf("failed to tear down scenario: %s", err) } }