Buriburizaemon/nezha_domains
1
0
Fork 0
mirror of https://github.com/Buriburizaem0n/nezha_domains.git synced 2026-09-22 03:00:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(agentcompat): add integration scenarios

Browse Source 
Co-authored-by: naiba/CloudCode <hi+cloudcode@nai.ba>
This commit is contained in:
naiba
2026-07-20 04:51:20 +00:00
co-authored by naiba/CloudCode
parent b3b3d92895
commit 9af3720bf0
96 changed files with 13091 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
integration/agentcompat/internal/scenario/held_terminal_real_test.go
+56
View File
@@ -0,0 +1,56 @@
//go:build linux && agentcompat
package scenario
import (
"context"
"fmt"
"os"
"testing"
"time"
"github.com/stretchr/testify/require"
"github.com/nezhahq/nezha/integration/agentcompat/internal/client"
"github.com/nezhahq/nezha/integration/agentcompat/internal/contract"
)
func TestHeldTerminalSessionUsesExistingDashboardAndAgent(t *testing.T) {
requireHeldRealSources(t)
paths, err := contract.NewPaths(os.Getenv("AGENTCOMPAT_NEZHA_SOURCE"), os.Getenv("AGENTCOMPAT_AGENT_SOURCE"), t.TempDir())
require.NoError(t, err)
ctx, cancel := context.WithTimeout(t.Context(), 10*time.Minute)
defer cancel()
realFixture, readiness, patClient := startHeldRealFixture(t, ctx, paths, "00000000-0000-0000-0000-000000000118", "held-terminal")
dashboardInstance, agentInstance := realFixture.dashboard, realFixture.agent
t.Cleanup(func() { _, _ = realFixture.Close(context.Background(), false, false, false) })
plan := heldTestPlan(t)
plan.ID, err = NewStressSessionID(fmt.Sprintf("held-terminal-real-%d", time.Now().UnixNano()))
require.NoError(t, err)
baseline, err := patClient.IOStreamState(ctx)
require.NoError(t, err)
dashboardPID, agentPID := dashboardInstance.PID(), agentInstance.PID()
sessionCtx, sessionCancel := context.WithTimeout(ctx, 45*time.Second)
defer sessionCancel()
session, err := newHeldTerminalSession(sessionCtx, heldTerminalInput{Dashboard: dashboardInstance, PATClient: patClient, Agent: agentInstance, Readiness: readiness, Plan: plan})
require.NoError(t, err)
require.NoError(t, session.WaitLive(ctx))
streamID, present := session.IOStreamID()
require.True(t, present)
live, err := patClient.WaitForIOStreamState(ctx, client.IOStreamStateExpectation{ExpectedCount: client.ExpectedIOStreamCount(baseline.Count + 1), PresentStreamID: streamID})
require.NoError(t, err)
require.Equal(t, baseline.Count+1, live.Count)
require.Equal(t, dashboardPID, dashboardInstance.PID())
require.Equal(t, agentPID, agentInstance.PID())
require.NoError(t, session.Close(ctx))
require.NoError(t, session.WaitClosed(ctx))
closed, err := patClient.WaitForIOStreamState(ctx, client.IOStreamStateExpectation{ExpectedCount: client.ExpectedIOStreamCount(baseline.Count), AbsentStreamID: streamID})
require.NoError(t, err)
require.Equal(t, baseline.Count, closed.Count)
cleanup, cleanupErr := realFixture.Close(ctx, true, closed.Count == baseline.Count, true)
require.NoError(t, cleanupErr)
require.True(t, heldRealCleanupOK(cleanup))
require.NoError(t, writeHeldRealEvidence("terminal", heldRealEvidence{Kind: "terminal", BaselineCount: baseline.Count, LiveCount: live.Count, ClosedCount: closed.Count, ExactIDPresent: true, ExactIDAbsent: true, ProtocolProved: session.ProtocolProved(), DashboardPIDUnchanged: dashboardPID == dashboardInstance.PID(), AgentPIDUnchanged: agentPID == agentInstance.PID(), CleanupOK: heldRealCleanupOK(cleanup)}))
}