mirror of
https://github.com/Buriburizaem0n/nezha_domains.git
synced 2026-09-19 09:40:12 +00:00
60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
//go:build linux && agentcompat
|
|
|
|
package agent
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAgent_RejectsMalformedStartWithoutWorkspaceArtifact(t *testing.T) {
|
|
// Given
|
|
parent := t.TempDir()
|
|
t.Setenv("TMPDIR", parent)
|
|
|
|
// When
|
|
_, err := Start(t.Context(), AgentStartConfig{SourceDir: filepath.Join(parent, "missing"), Endpoint: "127.0.0.1:1", UUID: "bad"})
|
|
|
|
// Then
|
|
require.Error(t, err)
|
|
entries, readErr := os.ReadDir(parent)
|
|
require.NoError(t, readErr)
|
|
require.Empty(t, entries)
|
|
}
|
|
|
|
func TestAgent_ContextInterruptionCleansWorkspace(t *testing.T) {
|
|
// Given
|
|
processContext, interrupt := context.WithCancel(t.Context())
|
|
dashboardInstance := startTestDashboard(t, false)
|
|
agentInstance, err := Start(processContext, AgentStartConfig{
|
|
SourceDir: testAgentSourceDir(t),
|
|
Endpoint: dashboardInstance.Endpoint(),
|
|
Secret: dashboardInstance.AgentSecret(),
|
|
UUID: "00000000-0000-0000-0000-000000000085",
|
|
})
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() {
|
|
cleanupContext, cancel := context.WithTimeout(context.Background(), 15*time.Second)
|
|
defer cancel()
|
|
require.NoError(t, agentInstance.Stop(cleanupContext))
|
|
})
|
|
root := agentInstance.WorkspaceRoot()
|
|
|
|
// When
|
|
interrupt()
|
|
|
|
// Then
|
|
select {
|
|
case <-agentInstance.CleanupDone():
|
|
case <-time.After(30 * time.Second):
|
|
t.Fatal("agent cleanup did not complete")
|
|
}
|
|
_, err = os.Stat(root)
|
|
require.ErrorIs(t, err, os.ErrNotExist)
|
|
}
|