From 94992b563fdfddce683fc07ef5bb70e18ff910fe Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Wed, 03 Jun 2026 16:25:00 +0800
Subject: [PATCH] Record Round 4 candidate install verification
---
privatevoice.src/internal/model/model_test.go | 157 ++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 157 insertions(+), 0 deletions(-)
diff --git a/privatevoice.src/internal/model/model_test.go b/privatevoice.src/internal/model/model_test.go
index 90a325e..f0e822c 100644
--- a/privatevoice.src/internal/model/model_test.go
+++ b/privatevoice.src/internal/model/model_test.go
@@ -23,6 +23,54 @@
}
}
+func TestRegistryReturnsMoonshineEnglish(t *testing.T) {
+ profile, err := GetModelProfile(MoonshineModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if profile.BackendKind != BackendMoonshine {
+ t.Fatalf("backend kind = %q, want %q", profile.BackendKind, BackendMoonshine)
+ }
+ if profile.InstallDirName != "moonshine-en" {
+ t.Fatalf("install dir = %q, want moonshine-en", profile.InstallDirName)
+ }
+ if len(profile.DownloadURLs) == 0 {
+ t.Fatal("expected moonshine download URL")
+ }
+}
+
+func TestRegistryReturnsParakeetEnglish(t *testing.T) {
+ profile, err := GetModelProfile(ParakeetModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if profile.BackendKind != BackendNemoTransducer {
+ t.Fatalf("backend kind = %q, want %q", profile.BackendKind, BackendNemoTransducer)
+ }
+ if profile.InstallDirName != "parakeet-en" {
+ t.Fatalf("install dir = %q, want parakeet-en", profile.InstallDirName)
+ }
+ if profile.Tier != "advanced" {
+ t.Fatalf("tier = %q, want advanced", profile.Tier)
+ }
+ if len(profile.DownloadURLs) == 0 {
+ t.Fatal("expected parakeet download URL")
+ }
+}
+
+func TestEnglishLanguageDefaultsToMoonshine(t *testing.T) {
+ profile, err := GetLanguageProfile(EnglishLanguageID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if profile.DefaultModelID != MoonshineModelID {
+ t.Fatalf("English default model = %q, want %q", profile.DefaultModelID, MoonshineModelID)
+ }
+ if len(profile.UpgradeModelIDs) != 1 || profile.UpgradeModelIDs[0] != ParakeetModelID {
+ t.Fatalf("English upgrade models = %v, want [%s]", profile.UpgradeModelIDs, ParakeetModelID)
+ }
+}
+
func TestValidateSenseVoiceAnyOfAndAllOf(t *testing.T) {
dir := t.TempDir()
writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens")
@@ -70,6 +118,78 @@
result := ValidateModelDir(DefaultModelProfile(), dir)
if result.Valid {
t.Fatal("expected zero-byte model file to fail validation")
+ }
+}
+
+func TestValidateMoonshineRequiredFiles(t *testing.T) {
+ profile, err := GetModelProfile(MoonshineModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ dir := t.TempDir()
+ createValidMoonshine(t, dir)
+
+ result := ValidateModelDir(profile, dir)
+ if !result.Valid {
+ t.Fatalf("expected valid moonshine dir, missing=%v problems=%v", result.Missing, result.Problems)
+ }
+ for _, role := range []string{"preprocessor", "encoder", "uncached_decoder", "cached_decoder", "tokens"} {
+ if result.Files[role] == "" {
+ t.Fatalf("missing resolved file role %q", role)
+ }
+ }
+}
+
+func TestValidateMoonshineRejectsMissingFile(t *testing.T) {
+ profile, err := GetModelProfile(MoonshineModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ dir := t.TempDir()
+ createValidMoonshine(t, dir)
+ if err := os.Remove(filepath.Join(dir, "cached_decode.int8.onnx")); err != nil {
+ t.Fatal(err)
+ }
+
+ result := ValidateModelDir(profile, dir)
+ if result.Valid {
+ t.Fatal("expected missing cached decoder to fail validation")
+ }
+}
+
+func TestValidateParakeetRequiredFiles(t *testing.T) {
+ profile, err := GetModelProfile(ParakeetModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ dir := t.TempDir()
+ createValidParakeet(t, dir)
+
+ result := ValidateModelDir(profile, dir)
+ if !result.Valid {
+ t.Fatalf("expected valid parakeet dir, missing=%v problems=%v", result.Missing, result.Problems)
+ }
+ for _, role := range []string{"encoder", "decoder", "joiner", "tokens"} {
+ if result.Files[role] == "" {
+ t.Fatalf("missing resolved file role %q", role)
+ }
+ }
+}
+
+func TestValidateParakeetRejectsMissingFile(t *testing.T) {
+ profile, err := GetModelProfile(ParakeetModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ dir := t.TempDir()
+ createValidParakeet(t, dir)
+ if err := os.Remove(filepath.Join(dir, "joiner.int8.onnx")); err != nil {
+ t.Fatal(err)
+ }
+
+ result := ValidateModelDir(profile, dir)
+ if result.Valid {
+ t.Fatal("expected missing joiner to fail validation")
}
}
@@ -257,6 +377,20 @@
}
}
+func TestDownloadProfileRemovesStalePartialDownload(t *testing.T) {
+ root := t.TempDir()
+ stalePath := filepath.Join(root, ".downloads", DefaultModelID, "old-run", "model_package")
+ writeTestFile(t, stalePath, "partial")
+
+ err := DownloadProfile(DefaultModelProfile(), []string{"://bad-url"}, root, nil)
+ if err == nil {
+ t.Fatal("expected download failure")
+ }
+ if _, statErr := os.Stat(filepath.Dir(stalePath)); !os.IsNotExist(statErr) {
+ t.Fatalf("stale download run still exists or stat failed: %v", statErr)
+ }
+}
+
func TestDownloadProfileFallsBackWhenPrimaryArchiveInvalid(t *testing.T) {
root := t.TempDir()
validArchive := tarArchive(t, map[string]string{
@@ -354,6 +488,29 @@
writeTestFile(t, filepath.Join(dir, "model.int8.onnx"), "model")
}
+func createValidMoonshine(t *testing.T, dir string) {
+ t.Helper()
+ if err := os.MkdirAll(dir, 0755); err != nil {
+ t.Fatal(err)
+ }
+ writeTestFile(t, filepath.Join(dir, "preprocess.onnx"), "preprocessor")
+ writeTestFile(t, filepath.Join(dir, "encode.int8.onnx"), "encoder")
+ writeTestFile(t, filepath.Join(dir, "uncached_decode.int8.onnx"), "uncached")
+ writeTestFile(t, filepath.Join(dir, "cached_decode.int8.onnx"), "cached")
+ writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens")
+}
+
+func createValidParakeet(t *testing.T, dir string) {
+ t.Helper()
+ if err := os.MkdirAll(dir, 0755); err != nil {
+ t.Fatal(err)
+ }
+ writeTestFile(t, filepath.Join(dir, "encoder.int8.onnx"), "encoder")
+ writeTestFile(t, filepath.Join(dir, "decoder.int8.onnx"), "decoder")
+ writeTestFile(t, filepath.Join(dir, "joiner.int8.onnx"), "joiner")
+ writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens")
+}
+
func tarArchive(t *testing.T, files map[string]string) []byte {
t.Helper()
var buf bytes.Buffer
--
Gitblit v1.9.3