mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-11 17:26:57 +02:00
Rebaptise AJEAN en Loki (fork, lignée conservée)
- module github.com/R0m1k3/Loki, cmd/loki, internal/loki (package loki) - LOKI_HOME, LOKI_MODEL_DIRS, LOKI_SERVICE, LOKI_DL_CONNS ; /etc/loki ; units loki-engine / loki-ui ; binaire et aide CLI - updateRepo pointe sur R0m1k3/Loki (l'auto-update ne tirera plus les binaires AJEAN amont) Conservé à l'identique : le domaine ajean.link (service de tunnel amont), les littéraux de migration 0.7.x (migrate_07.go), RELEASE_NOTES.md et LICENSE (historique et licence de l'amont). go build/vet/test : verts.
This commit is contained in:
157 files changed
+928
-928
No files matched your search
@@ -0,0 +1,89 @@
|
||||
package loki
|
||||
|
||||
// Destination des téléchargements : le dossier doit être choisi parmi les
|
||||
// dossiers de modèles déclarés (sinon l'UI web écrirait n'importe où), et
|
||||
// l'espace libre doit être vérifié AVANT de transférer 40 Go.
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestDownloadDestPathDir(t *testing.T) {
|
||||
testHome(t)
|
||||
extra := t.TempDir()
|
||||
t.Setenv("LOKI_MODEL_DIRS", extra)
|
||||
|
||||
// Dossier vide → models/.
|
||||
got, err := downloadDestPath("m.gguf", "")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if want := filepath.Join(modelsDir(), "m.gguf"); got != want {
|
||||
t.Fatalf("défaut = %s, attendu %s", got, want)
|
||||
}
|
||||
|
||||
// Dossier déclaré → accepté.
|
||||
got, err = downloadDestPath("m.gguf", extra)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if want := filepath.Join(extra, "m.gguf"); got != want {
|
||||
t.Fatalf("dossier déclaré = %s, attendu %s", got, want)
|
||||
}
|
||||
|
||||
// Dossier non déclaré → refusé.
|
||||
if _, err := downloadDestPath("m.gguf", t.TempDir()); err == nil {
|
||||
t.Fatal("un dossier non déclaré a été accepté")
|
||||
}
|
||||
|
||||
// Le nom reste un simple fichier .gguf : pas de remontée de dossier.
|
||||
if _, err := downloadDestPath(filepath.Join("..", "evil.gguf"), extra); err != nil {
|
||||
t.Fatal(err)
|
||||
} else if p, _ := downloadDestPath(filepath.Join("..", "evil.gguf"), extra); filepath.Dir(p) != extra {
|
||||
t.Fatalf("path traversal : %s", p)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDiskFreeAndCheck(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
free := diskFree(dir)
|
||||
if free <= 0 {
|
||||
t.Skipf("espace libre non mesurable ici (%d)", free)
|
||||
}
|
||||
// Un dossier pas encore créé doit être mesuré via son parent.
|
||||
if diskFree(filepath.Join(dir, "pas", "encore")) <= 0 {
|
||||
t.Fatal("diskFree ne remonte pas aux parents existants")
|
||||
}
|
||||
if err := checkDiskSpace(dir, 1<<10); err != nil {
|
||||
t.Fatalf("1 Ko refusé alors que %d octets sont libres : %v", free, err)
|
||||
}
|
||||
err := checkDiskSpace(dir, free)
|
||||
if err == nil {
|
||||
t.Fatal("téléchargement accepté alors qu'il remplit le disque")
|
||||
}
|
||||
if !strings.Contains(err.Error(), "espace insuffisant") {
|
||||
t.Fatalf("message inattendu : %v", err)
|
||||
}
|
||||
// Taille inconnue (serveur sans Content-Length) : on laisse passer.
|
||||
if err := checkDiskSpace(dir, 0); err != nil {
|
||||
t.Fatalf("taille inconnue refusée : %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Les .part orphelins doivent aussi être nettoyés dans les dossiers ajoutés.
|
||||
func TestCleanStalePartFilesExtraDirs(t *testing.T) {
|
||||
testHome(t)
|
||||
extra := t.TempDir()
|
||||
t.Setenv("LOKI_MODEL_DIRS", extra)
|
||||
stale := filepath.Join(extra, "coupe.gguf.part")
|
||||
if err := os.WriteFile(stale, []byte("x"), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
cleanStalePartFiles()
|
||||
if _, err := os.Stat(stale); !os.IsNotExist(err) {
|
||||
t.Fatal(".part orphelin d'un dossier ajouté non supprimé")
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user