mirror of
https://github.com/cirruslabs/orchard.git
synced 2026-09-30 12:01:15 +02:00
edi-oai
9ac6ca6326
* Support Tart Guest Agent transport for VM startup scripts * TestTartGuestAgentStartupScript: make ScriptContent more readable * Align Tart Guest Agent's startup script execution with SSH * Use mvdan.cc/sh/v3/syntax to validate "export" shell commands * Use Bash instead of Zsh on Linux * Invert logic when validating vm.StartupScript.Transport for simplicity * Document startup_script's transport field in OpenAPI specification * Prevent unbounded buffering of startup script output * grpc.WaitForReady(true) → retry.DoWithData()
52 lines
1.8 KiB
Go
52 lines
1.8 KiB
Go
//go:build darwin
|
|
|
|
package tests_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cirruslabs/orchard/internal/tests/devcontroller"
|
|
"github.com/cirruslabs/orchard/internal/tests/platformdependent"
|
|
v1 "github.com/cirruslabs/orchard/pkg/resource/v1"
|
|
"github.com/google/uuid"
|
|
"github.com/lithammer/dedent"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestTartGuestAgentStartupScript(t *testing.T) {
|
|
devClient, _, _ := devcontroller.StartIntegrationTestEnvironment(t)
|
|
|
|
// Create a VM whose startup script runs through Tart Guest Agent
|
|
vm := platformdependent.VM("test-agent-startup-" + uuid.NewString())
|
|
|
|
// Agent execution must work without valid SSH credentials.
|
|
vm.Username = "invalid-ssh-user"
|
|
vm.Password = "invalid-ssh-password"
|
|
vm.StartupScript = &v1.VMScript{
|
|
Transport: v1.VMScriptTransportTartGuestAgent,
|
|
ScriptContent: dedent.Dedent(`
|
|
printf 'Hello, %s!\n' "$FOO"
|
|
printf 'startup stderr\n' >&2
|
|
exit 123
|
|
`),
|
|
Env: map[string]string{"FOO": "Tart Guest Agent"},
|
|
}
|
|
require.NoError(t, devClient.VMs().Create(t.Context(), vm))
|
|
|
|
// Ensure that the script's non-zero exit status is reported as a VM failure
|
|
require.EventuallyWithT(t, func(collect *assert.CollectT) {
|
|
current, err := devClient.VMs().Get(t.Context(), vm.Name)
|
|
require.NoError(collect, err)
|
|
require.Equal(collect, v1.VMStatusFailed, current.Status)
|
|
require.Contains(collect, current.StatusMessage, "failed to run startup script: process exited with status 123")
|
|
|
|
// Ensure that the script received FOO and both stdout and stderr were captured
|
|
logs, err := devClient.VMs().Logs(t.Context(), vm.Name)
|
|
require.NoError(collect, err)
|
|
require.Contains(collect, logs, "Hello, Tart Guest Agent!")
|
|
require.Contains(collect, logs, "startup stderr")
|
|
}, 2*time.Minute, time.Second)
|
|
}
|