38 lines
1.2 KiB
Go
38 lines
1.2 KiB
Go
package session_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/mathiasbq/supervisor/internal/exec"
|
|
"github.com/mathiasbq/supervisor/internal/session"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAttemptsFromEmpty(t *testing.T) {
|
|
result := session.AttemptsFrom(nil)
|
|
assert.Empty(t, result)
|
|
}
|
|
|
|
func TestAttemptsFromSetsIndex(t *testing.T) {
|
|
records := []exec.AttemptRecord{
|
|
{Model: "ollama/phi4", Tier: "local", DurationMs: 1200, WarmStart: true, Verdict: "escalate", Feedback: "too vague"},
|
|
{Model: "claude-sonnet-4-6", Tier: "subagent", DurationMs: 3400, WarmStart: false, Verdict: "accept"},
|
|
}
|
|
result := session.AttemptsFrom(records)
|
|
require.Len(t, result, 2)
|
|
|
|
assert.Equal(t, 1, result[0].Attempt)
|
|
assert.Equal(t, "ollama/phi4", result[0].Model)
|
|
assert.Equal(t, "local", result[0].Tier)
|
|
assert.Equal(t, int64(1200), result[0].DurationMs)
|
|
assert.True(t, result[0].WarmStart)
|
|
assert.Equal(t, "escalate", result[0].Verdict)
|
|
assert.Equal(t, "too vague", result[0].Feedback)
|
|
assert.False(t, result[0].Verified)
|
|
|
|
assert.Equal(t, 2, result[1].Attempt)
|
|
assert.Equal(t, "claude-sonnet-4-6", result[1].Model)
|
|
assert.True(t, result[1].Verified)
|
|
}
|