From 7999e66c9c78ada3666eb7cea7c22a352bb4cbf0 Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Wed, 24 Jun 2026 02:30:51 +0800
Subject: [PATCH] Add cancellable model downloads

---
 privatevoice.src/internal/engine/engine_darwin_test.go |   89 ++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 89 insertions(+), 0 deletions(-)

diff --git a/privatevoice.src/internal/engine/engine_darwin_test.go b/privatevoice.src/internal/engine/engine_darwin_test.go
index fdfbf8e..85191e7 100644
--- a/privatevoice.src/internal/engine/engine_darwin_test.go
+++ b/privatevoice.src/internal/engine/engine_darwin_test.go
@@ -6,6 +6,7 @@
 	"os"
 	"path/filepath"
 	"testing"
+	"time"
 
 	"voicesnap/internal/model"
 )
@@ -48,6 +49,47 @@
 	}
 	if config.ModelConfig.ModelType != model.BackendNemoTransducer {
 		t.Fatalf("model type = %q, want %q", config.ModelConfig.ModelType, model.BackendNemoTransducer)
+	}
+}
+
+func TestOfflineConfigForZipformerUsesPlainTransducer(t *testing.T) {
+	root := t.TempDir()
+	files := map[string]string{
+		"encoder": filepath.Join(root, "encoder-epoch-99-avg-1.int8.onnx"),
+		"decoder": filepath.Join(root, "decoder-epoch-99-avg-1.onnx"),
+		"joiner":  filepath.Join(root, "joiner-epoch-99-avg-1.int8.onnx"),
+		"tokens":  filepath.Join(root, "tokens.txt"),
+	}
+	resolved := model.ResolvedModel{
+		ModelID:     model.JapaneseModelID,
+		BackendKind: model.BackendTransducer,
+		Profile: model.ModelProfile{
+			DisplayName:   "Zipformer Japanese",
+			BackendKind:   model.BackendTransducer,
+			ProviderOrder: []string{"cpu"},
+			NumThreads:    1,
+		},
+		Files: files,
+	}
+
+	config, err := offlineConfigForResolvedModel(resolved, "cpu")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if config.ModelConfig.Transducer.Encoder != files["encoder"] {
+		t.Fatalf("encoder = %q, want %q", config.ModelConfig.Transducer.Encoder, files["encoder"])
+	}
+	if config.ModelConfig.Transducer.Decoder != files["decoder"] {
+		t.Fatalf("decoder = %q, want %q", config.ModelConfig.Transducer.Decoder, files["decoder"])
+	}
+	if config.ModelConfig.Transducer.Joiner != files["joiner"] {
+		t.Fatalf("joiner = %q, want %q", config.ModelConfig.Transducer.Joiner, files["joiner"])
+	}
+	if config.ModelConfig.Tokens != files["tokens"] {
+		t.Fatalf("tokens = %q, want %q", config.ModelConfig.Tokens, files["tokens"])
+	}
+	if config.ModelConfig.ModelType != "" {
+		t.Fatalf("model type = %q, want empty for plain transducer", config.ModelConfig.ModelType)
 	}
 }
 
@@ -163,6 +205,53 @@
 	}
 }
 
+func TestXASRHeadPaddingPreparesInitialContext(t *testing.T) {
+	if xasrHeadPaddingSamples != 4000 {
+		t.Fatalf("head padding samples = %d, want 4000", xasrHeadPaddingSamples)
+	}
+	input := []float32{0.1, -0.2, 0.3}
+	got := xasrSamplesWithHeadPadding(input)
+
+	if len(got) != xasrHeadPaddingSamples+len(input) {
+		t.Fatalf("padded len = %d, want %d", len(got), xasrHeadPaddingSamples+len(input))
+	}
+	for i := 0; i < xasrHeadPaddingSamples; i++ {
+		if got[i] != 0 {
+			t.Fatalf("head padding sample %d = %v, want 0", i, got[i])
+		}
+	}
+	for i, want := range input {
+		if got[xasrHeadPaddingSamples+i] != want {
+			t.Fatalf("payload sample %d = %v, want %v", i, got[xasrHeadPaddingSamples+i], want)
+		}
+	}
+}
+
+func TestXASRRequestsReleaseTailCaptureDelay(t *testing.T) {
+	got := (&xasrStreamingEngine{}).ReleaseTailCaptureDelay()
+	if got != 300*time.Millisecond {
+		t.Fatalf("ReleaseTailCaptureDelay() = %v, want 300ms", got)
+	}
+}
+
+func TestXASREnablesHoldPreCapture(t *testing.T) {
+	if !(&xasrStreamingEngine{}).HoldPreCaptureEnabled() {
+		t.Fatal("X-ASR should enable hold pre-capture")
+	}
+}
+
+func TestOfflineSherpaDoesNotRequestReleaseTailCapture(t *testing.T) {
+	if _, ok := any(&sherpaEngine{}).(ReleaseTailCaptureEngine); ok {
+		t.Fatal("offline sherpa engine should not request release tail capture")
+	}
+}
+
+func TestOfflineSherpaDoesNotEnableHoldPreCapture(t *testing.T) {
+	if _, ok := any(&sherpaEngine{}).(HoldPreCaptureEngine); ok {
+		t.Fatal("offline sherpa engine should not enable hold pre-capture")
+	}
+}
+
 func TestXASRRealModelSmoke(t *testing.T) {
 	if os.Getenv("PRIVATEVOICE_XASR_SMOKE") != "1" {
 		t.Skip("set PRIVATEVOICE_XASR_SMOKE=1 to run the real X-ASR model smoke test")

--
Gitblit v1.9.3