Function TestFallbackCooldownState

pkg/runtime/fallback_test.go:333–370

(t *testing.T)

}

func TestFallbackCooldownState(t *testing.T) {
	t.Parallel()

	synctest.Test(t, func(t *testing.T) {
		mockModel := &mockProvider{id: "test/model", stream: newStreamBuilder().AddContent("ok").AddStopWithUsage(1, 1).Build()}
		tm := team.New(team.WithAgents(
			agent.New("test-agent", "test instruction", agent.WithModel(mockModel)),
		))
		rt, err := NewLocalRuntime(t.Context(), tm, WithSessionCompaction(false), WithModelStore(mockModelStore{}))
		require.NoError(t, err)

		agentName := "test-agent"

		// Initially no cooldown
		assert.Nil(t, rt.fallback.cooldowns.Get(agentName), "should have no cooldown initially")

		// Fake clock so cooldown expiry needs no sleeping.
		current := time.Now()
		rt.fallback.cooldowns.now = func() time.Time { return current }

		// Set cooldown with short duration for testing
		rt.fallback.cooldowns.Set(agentName, 0, 100*time.Millisecond)
		state := rt.fallback.cooldowns.Get(agentName)
		require.NotNil(t, state, "should have cooldown state")
		assert.Equal(t, 0, state.fallbackIndex)

		// Advance fake time past the cooldown
		current = current.Add(101 * time.Millisecond)
		assert.Nil(t, rt.fallback.cooldowns.Get(agentName), "cooldown should have expired")

		// Set cooldown again and then clear it
		rt.fallback.cooldowns.Set(agentName, 1, 1*time.Hour)
		require.NotNil(t, rt.fallback.cooldowns.Get(agentName))

		rt.fallback.cooldowns.Clear(agentName)
		assert.Nil(t, rt.fallback.cooldowns.Get(agentName), "cooldown should be cleared")
	})
}

func TestGetEffectiveCooldown(t *testing.T) {
	t.Parallel()