mirror of
https://github.com/cirruslabs/orchard.git
synced 2026-09-30 12:01:15 +02:00
91 lines
2.5 KiB
Go
91 lines
2.5 KiB
Go
//go:build darwin
|
|
|
|
package tests_test
|
|
|
|
import (
|
|
"net"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cirruslabs/orchard/internal/imageconstant"
|
|
"github.com/cirruslabs/orchard/internal/tests/devcontroller"
|
|
v1 "github.com/cirruslabs/orchard/pkg/resource/v1"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/crypto/ssh"
|
|
)
|
|
|
|
func TestEndpoint(t *testing.T) {
|
|
ctx := t.Context()
|
|
|
|
devClient, _, _ := devcontroller.StartIntegrationTestEnvironment(t)
|
|
|
|
// Create a VM that exposes its SSH service through the worker endpoint
|
|
const (
|
|
vmName = "endpoint-test-vm"
|
|
endpointName = "ssh"
|
|
)
|
|
|
|
require.NoError(t, devClient.VMs().Create(ctx, &v1.VM{
|
|
Name: vmName,
|
|
Image: imageconstant.DefaultMacosImage,
|
|
CPU: 4,
|
|
Memory: 8 * 1024,
|
|
Headless: true,
|
|
Endpoints: []v1.EndpointSpec{
|
|
{
|
|
Name: endpointName,
|
|
Protocol: v1.EndpointProtocolTCP,
|
|
Target: v1.ConnectionTarget{
|
|
VM: &v1.ConnectionTargetVM{Port: 22},
|
|
},
|
|
},
|
|
},
|
|
}))
|
|
|
|
// Wait for the Worker to expose a listening endpoint on the running VM
|
|
var (
|
|
vm *v1.VM
|
|
err error
|
|
)
|
|
|
|
require.Eventually(t, func() bool {
|
|
vm, err = devClient.VMs().Get(ctx, vmName)
|
|
|
|
return err == nil &&
|
|
vm.Status == v1.VMStatusRunning &&
|
|
len(vm.ObservedEndpoints) == 1 &&
|
|
vm.ObservedEndpoints[0].State == v1.EndpointStateListening
|
|
}, 2*time.Minute, time.Second, "failed to wait for the endpoint")
|
|
|
|
// Verify the endpoint uses the expected name and a dynamically assigned Worker port
|
|
endpointStatus := vm.ObservedEndpoints[0]
|
|
require.Equal(t, endpointName, endpointStatus.Name)
|
|
require.NotZero(t, endpointStatus.WorkerPort)
|
|
|
|
// Wait for SSH to accept connections through the worker port
|
|
var sshClient *ssh.Client
|
|
address := net.JoinHostPort("127.0.0.1", strconv.Itoa(int(endpointStatus.WorkerPort)))
|
|
sshConfig := &ssh.ClientConfig{
|
|
User: "admin",
|
|
Auth: []ssh.AuthMethod{ssh.Password("admin")},
|
|
HostKeyCallback: ssh.InsecureIgnoreHostKey(),
|
|
Timeout: 10 * time.Second,
|
|
}
|
|
|
|
require.Eventually(t, func() bool {
|
|
sshClient, err = ssh.Dial("tcp", address, sshConfig)
|
|
return err == nil
|
|
}, 2*time.Minute, time.Second, "failed to connect to the endpoint over SSH")
|
|
defer sshClient.Close()
|
|
|
|
// Run a command to verify the forwarded connection reaches the VM
|
|
sshSession, err := sshClient.NewSession()
|
|
require.NoError(t, err)
|
|
defer sshSession.Close()
|
|
|
|
unameOutput, err := sshSession.Output("uname -a")
|
|
require.NoError(t, err)
|
|
require.Contains(t, string(unameOutput), "Darwin")
|
|
}
|