package integration import ( "testing" "github.com/juanfont/headscale/integration/dockertestutil" "github.com/juanfont/headscale/integration/tsic" ) // This file is intendet 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") } } func TestHeadscale(t *testing.T) { IntegrationSkip(t) 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) { err = scenario.StartHeadscale() if err != nil { t.Errorf("failed to create start headcale: %s", err) } err = scenario.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) 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) } } func TestCreateTailscale(t *testing.T) { IntegrationSkip(t) var scenario *Scenario var err error 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]*tsic.TailscaleInContainer), } t.Run("create-tailscale", func(t *testing.T) { err := scenario.CreateTailscaleNodesInNamespace(namespace, "1.32.0", 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) } }) err = scenario.Shutdown() if err != nil { t.Errorf("failed to tear down scenario: %s", err) } } func TestTailscaleNodesJoiningHeadcale(t *testing.T) { IntegrationSkip(t) 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) { err = scenario.StartHeadscale() if err != nil { t.Errorf("failed to create start headcale: %s", err) } headscale := scenario.Headscale() 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) if err != nil { t.Errorf("failed to create preauthkey: %s", err) } err = scenario.RunTailscaleUp(namespace, scenario.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) } }