mirror of
https://github.com/Buriburizaem0n/nezha_domains.git
synced 2026-09-19 09:40:12 +00:00
84 lines
3.6 KiB
Go
84 lines
3.6 KiB
Go
//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)
|
|
}
|