test(agentcompat): add integration scenarios

Co-authored-by: naiba/CloudCode <hi+cloudcode@nai.ba>
2026-09-22 03:00:12 +00:00 · 2026-07-20 04:51:20 +00:00
parent b3b3d92895
commit 9af3720bf0
96 changed files with 13091 additions and 0 deletions
@@ -0,0 +1,83 @@
+//go:build linux && agentcompat
+
+package scenario
+
+import (
+	"context"
+	"encoding/json"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+
+	"github.com/nezhahq/nezha/integration/agentcompat/internal/contract"
+	"github.com/stretchr/testify/require"
+)
+
+func TestReconnectScenario_RealDashboardAndAgentProcessRestarts(t *testing.T) {
+	// Given
+	nezhaSource := os.Getenv("AGENTCOMPAT_NEZHA_SOURCE")
+	agentSource := os.Getenv("AGENTCOMPAT_AGENT_SOURCE")
+	if nezhaSource == "" || agentSource == "" {
+		t.Skip("set AGENTCOMPAT_NEZHA_SOURCE and AGENTCOMPAT_AGENT_SOURCE")
+	}
+	evidenceDirectory := os.Getenv("AGENTCOMPAT_RECONNECT_EVIDENCE_DIR")
+	if evidenceDirectory == "" {
+		evidenceDirectory = t.TempDir()
+	}
+	require.NoError(t, os.MkdirAll(evidenceDirectory, 0o700))
+	paths, err := contract.NewPaths(nezhaSource, agentSource, evidenceDirectory)
+	require.NoError(t, err)
+	testContext, cancel := context.WithTimeout(t.Context(), 10*time.Minute)
+	defer cancel()
+
+	// When
+	result, reconnectEvidence, err := (Reconnect{}).RunWithEvidence(testContext, ReconnectInput{Paths: paths})
+
+	// Then
+	for _, assertion := range result.Assertions {
+		t.Logf("assertion=%q passed=%t details=%q", assertion.Name, assertion.Passed, assertion.Details)
+	}
+	t.Logf("agent cleanup=%+v dashboard cleanup=%+v", reconnectEvidence.AgentCleanup, reconnectEvidence.DashboardCleanup)
+	require.NoError(t, err)
+	require.True(t, result.Passed)
+	require.True(t, result.CleanupOK)
+	require.NoError(t, reconnectEvidence.Validate())
+	require.True(t, reconnectEvidence.AgentCleanup.Passed)
+	require.False(t, reconnectEvidence.AgentCleanup.Forced)
+	require.Len(t, reconnectEvidence.AgentCleanup.Processes, 2)
+	require.True(t, reconnectEvidence.DashboardCleanup.Passed)
+	require.False(t, reconnectEvidence.DashboardCleanup.Forced)
+	require.Len(t, reconnectEvidence.DashboardCleanup.Processes, 2)
+	require.True(t, reconnectEvidence.Identity.DashboardFixtureUnchanged)
+	require.NotZero(t, reconnectEvidence.Fixture.Dashboard.HTTP.Inode)
+	require.NotZero(t, reconnectEvidence.Fixture.Dashboard.Receipt.Inode)
+	require.Greater(t, reconnectEvidence.Runtime.DashboardAfter.Generation, reconnectEvidence.Runtime.DashboardBefore.Generation)
+	require.NotEqual(t, reconnectEvidence.Runtime.DashboardBefore.PID, reconnectEvidence.Runtime.DashboardAfter.PID)
+	require.Greater(t, reconnectEvidence.Runtime.AgentAfter.Generation, reconnectEvidence.Runtime.AgentBefore.Generation)
+	require.NotEqual(t, reconnectEvidence.Runtime.AgentBefore.PID, reconnectEvidence.Runtime.AgentAfter.PID)
+	require.Positive(t, reconnectEvidence.Lifecycle.ReconnectInterval)
+	require.Zero(t, reconnectEvidence.Lifecycle.StaleGenerationReceipts)
+	require.Zero(t, reconnectEvidence.Lifecycle.DuplicateTaskIDs)
+	require.Zero(t, reconnectEvidence.Lifecycle.LostResultIDs)
+	require.Len(t, reconnectEvidence.Observation.TaskIDs, 5)
+	require.Equal(t, reconnectEvidence.Observation.TaskIDs, reconnectEvidence.Observation.ResultIDs)
+
+	artifactPath := filepath.Join(evidenceDirectory, "reconnect-real-process.json")
+	artifact, err := json.MarshalIndent(struct {
+		Result   Result            `json:"result"`
+		Evidence ReconnectEvidence `json:"evidence"`
+	}{Result: result, Evidence: reconnectEvidence}, "", "  ")
+	require.NoError(t, err)
+	require.NoError(t, os.WriteFile(artifactPath, append(artifact, '\n'), 0o600))
+	readArtifact, err := os.ReadFile(artifactPath)
+	require.NoError(t, err)
+	var recorded struct {
+		Result   Result            `json:"result"`
+		Evidence ReconnectEvidence `json:"evidence"`
+	}
+	require.NoError(t, json.Unmarshal(readArtifact, &recorded))
+	require.Equal(t, result, recorded.Result)
+	require.Equal(t, reconnectEvidence, recorded.Evidence)
+	t.Logf("reconnect evidence artifact: %s", artifactPath)
+}