From 4e7591ebb1294f611e7b85082be55313dd53820f Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Wed, 03 Jun 2026 13:25:19 +0800
Subject: [PATCH] Implement Round 3 Moonshine English model

---
 privatevoice.src/internal/model/model_test.go |   74 +++++++++++++++++++++++++++++++++++++
 1 files changed, 74 insertions(+), 0 deletions(-)

diff --git a/privatevoice.src/internal/model/model_test.go b/privatevoice.src/internal/model/model_test.go
index 90a325e..b5273c2 100644
--- a/privatevoice.src/internal/model/model_test.go
+++ b/privatevoice.src/internal/model/model_test.go
@@ -23,6 +23,32 @@
 	}
 }
 
+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 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)
+	}
+}
+
 func TestValidateSenseVoiceAnyOfAndAllOf(t *testing.T) {
 	dir := t.TempDir()
 	writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens")
@@ -70,6 +96,42 @@
 	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")
 	}
 }
 
@@ -354,6 +416,18 @@
 	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 tarArchive(t *testing.T, files map[string]string) []byte {
 	t.Helper()
 	var buf bytes.Buffer

--
Gitblit v1.9.3