mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-22 06:28:11 +02:00
API interne du bail du moteur pour la conversation tenue par Rust
Le verrou de génération (conv.Generating) est partagé par le chat, les tâches planifiées, le bench et l'optimiseur. Une conversation tenue par la façade doit prendre ce verrou-là : sinon une tâche partirait vers le même moteur en plein tour. /internal/v1/engine/lease prend le verrou (409 avec la raison du refus et la tâche en cours), /renew le prolonge et remonte un stop passé par le Go, /release le rend (seulement s'il le tient encore), /engine/stop arrête un détenteur Go. Sans renouvellement le bail expire : une façade tuée en plein tour ne garde pas le moteur. Reset et les bascules de discussion rompent le bail comme ils libèrent un tour Go. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_016tcGfJWefW5kUGoC2wL8cv
This commit is contained in:
4 files changed
+363
-1
No files matched your search
@@ -0,0 +1,153 @@
|
||||
package loki
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func leaseJSON(t *testing.T, body string) map[string]any {
|
||||
t.Helper()
|
||||
var m map[string]any
|
||||
if err := json.Unmarshal([]byte(body), &m); err != nil {
|
||||
t.Fatalf("réponse illisible %q : %v", body, err)
|
||||
}
|
||||
return m
|
||||
}
|
||||
|
||||
// libérerVerrou rend le verrou de génération quoi qu'un test ait laissé.
|
||||
func libérerVerrou(t *testing.T) {
|
||||
t.Cleanup(func() {
|
||||
conv.mu.Lock()
|
||||
conv.extLeaseClearLocked()
|
||||
conv.Generating, conv.benching, conv.cancel, conv.leaseWhy = false, false, nil, nil
|
||||
conv.mu.Unlock()
|
||||
})
|
||||
}
|
||||
|
||||
// Le bail de la façade EST le verrou de génération : tant qu'il tient, une
|
||||
// mesure (bench, optimiseur) ou un second bail est refusé.
|
||||
func TestBailInterneVerrouPartage(t *testing.T) {
|
||||
testHome(t)
|
||||
t.Setenv(internalTokenEnv, jetonTest)
|
||||
libérerVerrou(t)
|
||||
|
||||
w := appelOAIInterne(t, "POST", "/internal/v1/engine/lease", `{"ttl_ms":60000}`)
|
||||
if w.Code != 200 {
|
||||
t.Fatalf("bail : %d %s", w.Code, w.Body.String())
|
||||
}
|
||||
m := leaseJSON(t, w.Body.String())
|
||||
id, _ := m["lease"].(string)
|
||||
if len(id) != 32 || m["ttl_ms"] != float64(60000) {
|
||||
t.Fatalf("bail : %v", m)
|
||||
}
|
||||
if !conv.isGenerating() {
|
||||
t.Fatal("le bail ne tient pas le verrou de génération")
|
||||
}
|
||||
if _, err := conv.measureLease(func() {}, errBenchBusy); err != ErrBusy {
|
||||
t.Fatalf("bench accepté pendant le bail : %v", err)
|
||||
}
|
||||
if w := appelOAIInterne(t, "POST", "/internal/v1/engine/lease", ""); w.Code != 409 ||
|
||||
!strings.Contains(w.Body.String(), ErrBusy.Error()) {
|
||||
t.Fatalf("second bail : %d %s", w.Code, w.Body.String())
|
||||
}
|
||||
|
||||
// Renouvelé ; le stop passé par le Go remonte à la façade.
|
||||
w = appelOAIInterne(t, "POST", "/internal/v1/engine/lease/renew", `{"lease":"`+id+`"}`)
|
||||
if w.Code != 200 || leaseJSON(t, w.Body.String())["stop"] != false {
|
||||
t.Fatalf("renouvellement : %d %s", w.Code, w.Body.String())
|
||||
}
|
||||
conv.Stop()
|
||||
w = appelOAIInterne(t, "POST", "/internal/v1/engine/lease/renew", `{"lease":"`+id+`"}`)
|
||||
if w.Code != 200 || leaseJSON(t, w.Body.String())["stop"] != true {
|
||||
t.Fatalf("stop non transmis : %d %s", w.Code, w.Body.String())
|
||||
}
|
||||
if w := appelOAIInterne(t, "POST", "/internal/v1/engine/lease/renew", `{"lease":"autre"}`); w.Code != 410 {
|
||||
t.Fatalf("bail inconnu renouvelé : %d", w.Code)
|
||||
}
|
||||
|
||||
// Rendu : le verrou est libre, le bail est mort, rendre deux fois est sans effet.
|
||||
for i := 0; i < 2; i++ {
|
||||
if w := appelOAIInterne(t, "POST", "/internal/v1/engine/lease/release", `{"lease":"`+id+`"}`); w.Code != 204 {
|
||||
t.Fatalf("libération : %d", w.Code)
|
||||
}
|
||||
}
|
||||
if conv.isGenerating() {
|
||||
t.Fatal("verrou toujours tenu après la libération")
|
||||
}
|
||||
if w := appelOAIInterne(t, "POST", "/internal/v1/engine/lease/renew", `{"lease":"`+id+`"}`); w.Code != 410 {
|
||||
t.Fatalf("bail rendu renouvelé : %d", w.Code)
|
||||
}
|
||||
}
|
||||
|
||||
// Un bail rendu après coup ne libère pas le verrou d'un autre détenteur.
|
||||
func TestBailInterneNeLibereQueLeSien(t *testing.T) {
|
||||
testHome(t)
|
||||
libérerVerrou(t)
|
||||
id, err := conv.extLeaseAcquire(time.Minute)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
conv.Reset() // rompt le bail, comme un tour Go
|
||||
if _, ok := conv.extLeaseRenew(id); ok {
|
||||
t.Fatal("bail encore valide après Reset")
|
||||
}
|
||||
release, err := conv.measureLease(func() {}, errBenchBusy)
|
||||
if err != nil {
|
||||
t.Fatalf("verrou non rendu par Reset : %v", err)
|
||||
}
|
||||
defer release()
|
||||
conv.extLeaseRelease(id)
|
||||
if !conv.isGenerating() {
|
||||
t.Fatal("un bail périmé a libéré le verrou du bench")
|
||||
}
|
||||
}
|
||||
|
||||
// Une façade tuée en plein tour ne garde pas le moteur : sans renouvellement,
|
||||
// le bail expire et rend le verrou.
|
||||
func TestBailInterneExpire(t *testing.T) {
|
||||
testHome(t)
|
||||
libérerVerrou(t)
|
||||
id, err := conv.extLeaseAcquire(50 * time.Millisecond)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
deadline := time.Now().Add(5 * time.Second)
|
||||
for conv.isGenerating() {
|
||||
if time.Now().After(deadline) {
|
||||
t.Fatal("bail jamais expiré")
|
||||
}
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
}
|
||||
if _, ok := conv.extLeaseRenew(id); ok {
|
||||
t.Fatal("bail expiré encore renouvelable")
|
||||
}
|
||||
}
|
||||
|
||||
// Moteur tenu par le Go : le refus dit qui le tient, et /engine/stop l'arrête.
|
||||
func TestBailInterneRefusEtStop(t *testing.T) {
|
||||
testHome(t)
|
||||
t.Setenv(internalTokenEnv, jetonTest)
|
||||
libérerVerrou(t)
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
release, err := conv.measureLease(cancel, errBenchBusy)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer release()
|
||||
w := appelOAIInterne(t, "POST", "/internal/v1/engine/lease", "")
|
||||
if w.Code != 409 || leaseJSON(t, w.Body.String())["error"] != errBenchBusy.Error() {
|
||||
t.Fatalf("refus : %d %s", w.Code, w.Body.String())
|
||||
}
|
||||
if w := appelOAIInterne(t, "POST", "/internal/v1/engine/stop", ""); w.Code != 204 {
|
||||
t.Fatalf("stop : %d", w.Code)
|
||||
}
|
||||
if ctx.Err() == nil {
|
||||
t.Fatal("le bench n'a pas été arrêté")
|
||||
}
|
||||
if w := appelOAIInterne(t, "GET", "/internal/v1/engine/stop", ""); w.Code != 405 {
|
||||
t.Fatalf("GET accepté : %d", w.Code)
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user