Ilker ManapandClaude Opus 5.5
76746715ec
CI / test (push) Failing after 6m51s
Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
71 lines
2.5 KiB
Go
71 lines
2.5 KiB
Go
package reconciler
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"gitea.mynodes.xyz/ilker/llmgw/internal/agent/supervisor"
|
|
)
|
|
|
|
func TestProgressIsVisibleWithoutBeingATransition(t *testing.T) {
|
|
// The failure this replaces: a replica pulling a hundred gigabytes and one
|
|
// that is wedged look identical from the control plane — both are a shard
|
|
// that has been starting for ten minutes.
|
|
//
|
|
// And progress must not count as a state change. Treating each chunk as
|
|
// one would reset the timer that decides when a slow start has become a
|
|
// failed one, so a download would keep a replica alive forever.
|
|
h := newHarness(t, testModel(1), gpuNode("n1", 8))
|
|
h.r.Reconcile(t.Context())
|
|
|
|
reps := h.r.Replicas()
|
|
if len(reps) != 1 || len(reps[0].Shards) == 0 {
|
|
t.Fatalf("expected a placed replica with shards, got %+v", reps)
|
|
}
|
|
rep, shard := reps[0].ID, reps[0].Shards[0].ID
|
|
|
|
h.r.OnFetchProgress(rep, shard, "hf:org/repo/model.gguf", "", 3<<30, 12<<30)
|
|
|
|
got := h.r.Replicas()[0].Shards[0]
|
|
if got.Fetch == nil {
|
|
t.Fatal("the download is not visible on the shard")
|
|
}
|
|
if got.Fetch.BytesDone != 3<<30 || got.Fetch.BytesTotal != 12<<30 {
|
|
t.Errorf("progress came back as %+v", got.Fetch)
|
|
}
|
|
if p, ok := got.Fetch.Percent(); !ok || p != 25 {
|
|
t.Errorf("percent = %d (known %v), want 25", p, ok)
|
|
}
|
|
}
|
|
|
|
func TestAnUnknownTotalHasNoPercentage(t *testing.T) {
|
|
// A server that sends no Content-Length is common. An invented percentage
|
|
// that stops moving is worse than a byte count that keeps rising.
|
|
f := FetchProgress{Source: "https://x/y.gguf", BytesDone: 500}
|
|
if _, ok := f.Percent(); ok {
|
|
t.Error("a percentage was invented from an unknown total")
|
|
}
|
|
}
|
|
|
|
func TestATransitionClearsTheDownload(t *testing.T) {
|
|
// Left in place it would sit in the listing at ninety-nine percent under
|
|
// a shard that has been serving for an hour.
|
|
h := newHarness(t, testModel(1), gpuNode("n1", 8))
|
|
h.r.Reconcile(t.Context())
|
|
reps := h.r.Replicas()
|
|
rep, shard := reps[0].ID, reps[0].Shards[0].ID
|
|
|
|
h.r.OnFetchProgress(rep, shard, "hf:org/repo/model.gguf", "", 1, 2)
|
|
h.r.OnShardStatus(rep, shard, supervisor.Ready, "")
|
|
|
|
if got := h.r.Replicas()[0].Shards[0]; got.Fetch != nil {
|
|
t.Errorf("a finished shard still shows a download: %+v", got.Fetch)
|
|
}
|
|
}
|
|
|
|
func TestProgressForAReplicaThatIsGoneIsIgnored(t *testing.T) {
|
|
// A download that finishes after its replica was torn down should not
|
|
// resurrect anything or panic.
|
|
h := newHarness(t, testModel(1), gpuNode("n1", 8))
|
|
h.r.OnFetchProgress("no-such-replica", "no-such-shard", "x", "", 1, 2)
|
|
}
|