mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-11 17:26:57 +02:00
nathaninline
62ad4155a2
- suppression des alias de commandes herites (skills, machine, tools, mem, upgrade...) - aide reecrite et regroupee - helper de test testHome + fermeture de la base (verrou bbolt sous Windows) - tests des dossiers de modeles alignes sur models/
90 lines
2.7 KiB
Go
90 lines
2.7 KiB
Go
package ajean
|
|
|
|
// 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("AJEAN_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("AJEAN_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é")
|
|
}
|
|
}
|