mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-11 17:26:57 +02:00
Loki
de6551a153
- 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.
90 lines
2.7 KiB
Go
90 lines
2.7 KiB
Go
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é")
|
|
}
|
|
}
|