150 lines
3.4 KiB
Go
150 lines
3.4 KiB
Go
package integration
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/juanfont/headscale/integration/tsic"
|
|
)
|
|
|
|
func TestHeadscale(t *testing.T) {
|
|
var err error
|
|
|
|
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("test-space")
|
|
if err != nil {
|
|
t.Errorf("failed to create namespace: %s", err)
|
|
}
|
|
|
|
if _, ok := scenario.namespaces["test-space"]; !ok {
|
|
t.Errorf("namespace is not in scenario")
|
|
}
|
|
})
|
|
|
|
t.Run("create-auth-key", func(t *testing.T) {
|
|
_, err := scenario.CreatePreAuthKey("test-space")
|
|
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) {
|
|
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) {
|
|
var err error
|
|
|
|
namespace := "join-node-test"
|
|
|
|
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.32.0", 2)
|
|
if err != nil {
|
|
t.Errorf("failed to add tailscale nodes: %s", err)
|
|
}
|
|
|
|
if clients := len(scenario.namespaces[namespace].Clients); clients != 2 {
|
|
t.Errorf("wrong number of tailscale clients: %d != %d", clients, 2)
|
|
}
|
|
})
|
|
|
|
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)
|
|
}
|
|
|
|
})
|
|
|
|
err = scenario.Shutdown()
|
|
if err != nil {
|
|
t.Errorf("failed to tear down scenario: %s", err)
|
|
}
|
|
}
|