Method StoreMocks

pkg/service/agent/agent.go:641–694

StoreMocks stores the filtered and unfiltered mocks for a client ID. Unification (Phase 1): every mock is run through DeriveLifetime on entry so TestModeInfo.Lifetime is populated before any matcher reads it. This is a second safety-net — the mockdb disk loader already derives at load time, but Sto

(ctx context.Context, filtered []*models.Mock, unfiltered []*models.Mock)

// deep copy here unless a concrete mutation-safety regression lands
// first.
func (a *Agent) StoreMocks(ctx context.Context, filtered []*models.Mock, unfiltered []*models.Mock) error {
	storage := &ClientMockStorage{
		filtered:   make([]*models.Mock, len(filtered)),
		unfiltered: make([]*models.Mock, len(unfiltered)),
	}

	// Shallow copy the slices — only the outer backing array is
	// duplicated, the *models.Mock pointers are shared with the
	// caller's slices. This is INTENTIONAL and load-bearing: matchers
	// look up mocks via pointer identity in MockManager's trees and
	// per-connID pools, and HitCount/Lifetime are bumped on the shared
	// Mock object so observability is consistent across the stack (see
	// the caveat block before this function for the full rationale).
	// Do NOT switch to a deep copy without coordinated updates at
	// every downstream site.
	copy(storage.filtered, filtered)
	copy(storage.unfiltered, unfiltered)

	// Derive Lifetime once per mock before they enter the runtime pool.
	// Idempotent: DeriveLifetime short-circuits when the Lifetime
	// field is already set (i.e. the disk loader has already run),
	// so re-deriving here is safe and never double-counts the
	// legacyKindFallbackFires telemetry.
	for _, m := range storage.filtered {
		if m != nil {
			m.DeriveLifetime()
		}
	}
	for _, m := range storage.unfiltered {
		if m != nil {
			m.DeriveLifetime()
		}
	}

	a.clientMocks.Store(uint64(0), storage)

	// Compute the freeze anchor as the earliest ReqTimestampMock across the
	// stored mocks and forward it to AgentHooks. The hook implementation
	// decides whether to apply it (a no-op when freezeTime is off). Doing
	// this here, alongside the StoreMocks write, guarantees the anchor is
	// known to the hook before BeforeTestRun fires — i.e. before the user
	// app's first datetime.now() call — which is what closes the bootstrap
	// gap that lets boto3 / JWT libs see "now > recorded Expiration" and
	// kill the worker.
	if anchor := earliestReqTimestamp(storage.filtered, storage.unfiltered); !anchor.IsZero() {
		if err := ActiveHooks.SetFreezeAnchor(ctx, anchor); err != nil {
			a.logger.Warn("SetFreezeAnchor hook returned error",
				zap.Error(err), zap.Time("anchor", anchor))
		}
	}

	a.logger.Debug("Successfully stored mocks for client")
	return nil
}

// earliestReqTimestamp returns the earliest non-zero ReqTimestampMock seen
// across the supplied mock slices, or a zero time.Time if none are set.