Record Round 4 candidate install verification

Ariver

2026-06-03 94992b563fdfddce683fc07ef5bb70e18ff910fe

 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