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,126 @@
+//go:build linux
+
+package scenario
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/nezhahq/nezha/integration/agentcompat/internal/contract"
+	"github.com/nezhahq/nezha/integration/agentcompat/internal/fixture"
+)
+
+func TestTransferEvidence_RejectsEmptyOrMismatchedSHA(t *testing.T) {
+	t.Parallel()
+
+	evidence := validTransferEvidence()
+	evidence.UploadSHA256 = ""
+	evidence.DownloadSHA256 = "different"
+
+	err := evidence.Validate()
+
+	require.ErrorIs(t, err, ErrTransferEvidenceHashMismatch)
+	require.NotErrorIs(t, err, ErrTransferEvidenceSizeMismatch)
+	require.NotErrorIs(t, err, ErrTransferEvidenceMeasurement)
+	require.NotErrorIs(t, err, ErrTransferEvidenceHeapBudget)
+	require.NotErrorIs(t, err, ErrTransferEvidenceContract)
+}
+
+func TestTransferEvidence_RejectsHeapBudgetBreach(t *testing.T) {
+	t.Parallel()
+
+	evidence := validTransferEvidence()
+	evidence.RetainedHeapBytes = contract.TransferHeapBytes + 1
+
+	err := evidence.Validate()
+
+	require.ErrorIs(t, err, ErrTransferEvidenceHeapBudget)
+	require.NotErrorIs(t, err, ErrTransferEvidenceSizeMismatch)
+	require.NotErrorIs(t, err, ErrTransferEvidenceHashMismatch)
+	require.NotErrorIs(t, err, ErrTransferEvidenceMeasurement)
+	require.NotErrorIs(t, err, ErrTransferEvidenceContract)
+}
+
+func TestTransferEvidence_AcceptsExactNonemptyEqualHashes(t *testing.T) {
+	t.Parallel()
+
+	require.NoError(t, validTransferEvidence().Validate())
+}
+
+func TestTransferEvidence_RejectsMissingStreamingMeasurements(t *testing.T) {
+	t.Parallel()
+
+	evidence := validTransferEvidence()
+	evidence.UploadChunks = 0
+	evidence.DownloadChunks = 0
+	evidence.UploadDuration = 0
+	evidence.DownloadDuration = 0
+
+	err := evidence.Validate()
+
+	require.ErrorIs(t, err, ErrTransferEvidenceMeasurement)
+	require.NotErrorIs(t, err, ErrTransferEvidenceSizeMismatch)
+	require.NotErrorIs(t, err, ErrTransferEvidenceHashMismatch)
+	require.NotErrorIs(t, err, ErrTransferEvidenceHeapBudget)
+	require.NotErrorIs(t, err, ErrTransferEvidenceContract)
+}
+
+func TestTransferEvidence_ReportsAllTypedValidationErrors(t *testing.T) {
+	t.Parallel()
+
+	err := (TransferEvidence{}).Validate()
+
+	require.ErrorIs(t, err, ErrTransferEvidenceSizeMismatch)
+	require.ErrorIs(t, err, ErrTransferEvidenceHashMismatch)
+	require.ErrorIs(t, err, ErrTransferEvidenceMeasurement)
+	require.ErrorIs(t, err, ErrTransferEvidenceContract)
+}
+
+func TestTransferUploadEvidence_RejectsMissingMode(t *testing.T) {
+	t.Parallel()
+
+	digest := fixture.PayloadDigest{Bytes: contract.TransferBytes, SHA256: [32]byte{1}}
+	evidence := transferPathEvidence{
+		bytes: contract.TransferBytes, sha256: digest.Hex(), chunks: 1, duration: time.Nanosecond,
+	}
+
+	require.False(t, evidence.validUpload(digest))
+	require.True(t, evidence.validDownload(digest))
+}
+
+func TestConfirmTransferQuiescence_RequiresDeadline(t *testing.T) {
+	t.Parallel()
+
+	_, err := confirmTransferQuiescence(context.Background(), transferResidueScope{})
+
+	require.EqualError(t, err, "transfer quiescence requires a context deadline")
+}
+
+func validTransferEvidence() TransferEvidence {
+	return TransferEvidence{
+		WarmupUploadBytes:             transferWarmupBytes,
+		WarmupDownloadBytes:           transferWarmupBytes,
+		WarmupSHA256:                  "bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb",
+		WarmupDuration:                time.Nanosecond,
+		WarmupDeadlineRemaining:       time.Second,
+		WarmupQuiescent:               true,
+		UploadBytes:                   contract.TransferBytes,
+		DownloadBytes:                 contract.TransferBytes,
+		UploadSHA256:                  "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
+		DownloadSHA256:                "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
+		UploadChunks:                  1,
+		DownloadChunks:                1,
+		UploadDuration:                time.Nanosecond,
+		DownloadDuration:              time.Nanosecond,
+		RetainedHeapBytes:             contract.TransferHeapBytes,
+		Mode:                          "0640",
+		CreateDirs:                    true,
+		UploadReplayRejected:          true,
+		DownloadReplayRejected:        true,
+		OversizeRejected:              true,
+		OutsideRootSentinelsUnchanged: true,
+	}
+}