From 5df683eba7c103010a7c110d97a5fd808b39fe46 Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Thu, 02 Jul 2026 01:47:41 +0800
Subject: [PATCH] Fix local universal macOS package assembly
---
privatevoice.src/internal/model/model_test.go | 222 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 222 insertions(+), 0 deletions(-)
diff --git a/privatevoice.src/internal/model/model_test.go b/privatevoice.src/internal/model/model_test.go
index 1e01c14..d3f025e 100644
--- a/privatevoice.src/internal/model/model_test.go
+++ b/privatevoice.src/internal/model/model_test.go
@@ -3,6 +3,8 @@
import (
"archive/tar"
"bytes"
+ "context"
+ "errors"
"fmt"
"net/http"
"net/http/httptest"
@@ -10,7 +12,10 @@
"path/filepath"
"strconv"
"strings"
+ "sync"
+ "sync/atomic"
"testing"
+ "time"
)
func TestRegistryReturnsDefaultSenseVoice(t *testing.T) {
@@ -58,6 +63,54 @@
}
if len(profile.DownloadURLs) == 0 {
t.Fatal("expected parakeet download URL")
+ }
+}
+
+func TestRegistryReturnsParakeetV3ForEuropeanLanguages(t *testing.T) {
+ profile, err := GetModelProfile(ParakeetV3ModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if profile.BackendKind != BackendNemoTransducer {
+ t.Fatalf("backend kind = %q, want %q", profile.BackendKind, BackendNemoTransducer)
+ }
+ for _, languageID := range []string{FrenchLanguageID, GermanLanguageID, SpanishLanguageID, ItalianLanguageID, PortugueseLanguageID} {
+ if !stringSliceContains(profile.SupportedLanguageIDs, languageID) {
+ t.Fatalf("parakeet v3 languages = %v, want %s", profile.SupportedLanguageIDs, languageID)
+ }
+ languageProfile, err := GetLanguageProfile(languageID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if languageProfile.DefaultModelID != ParakeetV3ModelID {
+ t.Fatalf("%s default model = %q, want %q", languageID, languageProfile.DefaultModelID, ParakeetV3ModelID)
+ }
+ }
+}
+
+func TestRegistryReturnsJapaneseAndKoreanDefaults(t *testing.T) {
+ tests := []struct {
+ languageID string
+ modelID string
+ }{
+ {JapaneseLanguageID, JapaneseModelID},
+ {KoreanLanguageID, KoreanModelID},
+ }
+ for _, tt := range tests {
+ languageProfile, err := GetLanguageProfile(tt.languageID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if languageProfile.DefaultModelID != tt.modelID {
+ t.Fatalf("%s default model = %q, want %q", tt.languageID, languageProfile.DefaultModelID, tt.modelID)
+ }
+ modelProfile, err := GetModelProfile(tt.modelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if modelProfile.BackendKind != BackendTransducer {
+ t.Fatalf("%s backend = %q, want %q", tt.modelID, modelProfile.BackendKind, BackendTransducer)
+ }
}
}
@@ -238,6 +291,61 @@
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")
+ }
+}
+
+func TestValidateJapaneseZipformerRequiredFiles(t *testing.T) {
+ profile, err := GetModelProfile(JapaneseModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ dir := t.TempDir()
+ createValidZipformerSpecialized(t, dir)
+
+ result := ValidateModelDir(profile, dir)
+ if !result.Valid {
+ t.Fatalf("expected valid japanese zipformer 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 TestValidateKoreanZipformerRequiredFiles(t *testing.T) {
+ profile, err := GetModelProfile(KoreanModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ dir := t.TempDir()
+ createValidZipformerSpecialized(t, dir)
+
+ result := ValidateModelDir(profile, dir)
+ if !result.Valid {
+ t.Fatalf("expected valid korean zipformer 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 TestValidateZipformerSpecializedRejectsMissingJoiner(t *testing.T) {
+ profile, err := GetModelProfile(JapaneseModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ dir := t.TempDir()
+ createValidZipformerSpecialized(t, dir)
+ if err := os.Remove(filepath.Join(dir, "joiner-epoch-99-avg-1.int8.onnx")); err != nil {
t.Fatal(err)
}
@@ -489,6 +597,18 @@
}
}
+func TestHasAnyUsableModelInRoot(t *testing.T) {
+ root := t.TempDir()
+ if HasAnyUsableModelInRoot(root) {
+ t.Fatal("expected no usable model in empty root")
+ }
+
+ createValidSenseVoice(t, filepath.Join(root, "sensevoice-zh"))
+ if !HasAnyUsableModelInRoot(root) {
+ t.Fatal("expected installed sensevoice to count as a usable model")
+ }
+}
+
func TestInstallStateRoundTrip(t *testing.T) {
root := t.TempDir()
if err := UpdateInstalledModelState(root, DefaultModelProfile(), "sensevoice-zh", "new"); err != nil {
@@ -646,6 +766,97 @@
}
}
+func TestDownloadProfileCancelStopsWithoutFallbackOrState(t *testing.T) {
+ root := t.TempDir()
+ ctx, cancel := context.WithCancel(context.Background())
+ defer cancel()
+
+ var once sync.Once
+ var fallbackHits int32
+ started := make(chan struct{})
+ payload := bytes.Repeat([]byte("a"), 512*1024)
+ fallbackArchive := tarArchive(t, map[string]string{
+ "sensevoice/tokens.txt": "tokens",
+ "sensevoice/model.int8.onnx": "fallback-model",
+ })
+ server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+ switch r.URL.Path {
+ case "/slow":
+ once.Do(func() { close(started) })
+ w.Header().Set("Content-Length", strconv.Itoa(len(payload)))
+ w.WriteHeader(http.StatusOK)
+ flusher, _ := w.(http.Flusher)
+ for offset := 0; offset < len(payload); offset += 32 * 1024 {
+ select {
+ case <-r.Context().Done():
+ return
+ default:
+ }
+ end := offset + 32*1024
+ if end > len(payload) {
+ end = len(payload)
+ }
+ if _, err := w.Write(payload[offset:end]); err != nil {
+ return
+ }
+ if flusher != nil {
+ flusher.Flush()
+ }
+ time.Sleep(5 * time.Millisecond)
+ }
+ case "/fallback":
+ atomic.AddInt32(&fallbackHits, 1)
+ w.WriteHeader(http.StatusOK)
+ w.Write(fallbackArchive)
+ default:
+ http.NotFound(w, r)
+ }
+ }))
+ defer server.Close()
+
+ errCh := make(chan error, 1)
+ go func() {
+ errCh <- DownloadProfileWithContext(ctx, DefaultModelProfile(), []string{server.URL + "/slow", server.URL + "/fallback"}, root, func(percent float64, _, _ int64) {
+ if percent > 0 {
+ cancel()
+ }
+ })
+ }()
+
+ select {
+ case <-started:
+ case <-time.After(2 * time.Second):
+ t.Fatal("slow download did not start")
+ }
+
+ var err error
+ select {
+ case err = <-errCh:
+ case <-time.After(2 * time.Second):
+ t.Fatal("cancelled download did not return")
+ }
+ if !errors.Is(err, context.Canceled) {
+ t.Fatalf("download error = %v, want context.Canceled", err)
+ }
+ if got := atomic.LoadInt32(&fallbackHits); got != 0 {
+ t.Fatalf("fallback hits = %d, want 0", got)
+ }
+ state, err := LoadInstallStateFromRoot(root)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if _, ok := state.InstalledModels[DefaultModelID]; ok {
+ t.Fatal("cancelled download should not write installed model state")
+ }
+ resolved, err := ResolveModelInRoot(DefaultModelID, root)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if resolved.IsUsable() {
+ t.Fatal("cancelled download should not leave a usable model")
+ }
+}
+
func TestDownloadFileResumesExistingPartialFile(t *testing.T) {
payload := []byte("0123456789abcdefghijklmnopqrstuvwxyz")
@@ -724,6 +935,17 @@
writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens")
}
+func createValidZipformerSpecialized(t *testing.T, dir string) {
+ t.Helper()
+ if err := os.MkdirAll(dir, 0755); err != nil {
+ t.Fatal(err)
+ }
+ writeTestFile(t, filepath.Join(dir, "encoder-epoch-99-avg-1.int8.onnx"), "encoder")
+ writeTestFile(t, filepath.Join(dir, "decoder-epoch-99-avg-1.onnx"), "decoder")
+ writeTestFile(t, filepath.Join(dir, "joiner-epoch-99-avg-1.int8.onnx"), "joiner")
+ writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens")
+}
+
func createValidQwen3ASR(t *testing.T, dir string) {
t.Helper()
if err := os.MkdirAll(filepath.Join(dir, "tokenizer"), 0755); err != nil {
--
Gitblit v1.9.3