feat: add ModelStorage and ModelCatalog; transcriber uses ModelStorage

Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
2026-10-01 15:33:22 +02:00
parent fbc90cbd12
commit ae20008679
3 changed files with 149 additions and 34 deletions
@@ -0,0 +1,63 @@
+import Foundation
+import Testing
+@testable import SstWhisper
+
+struct ModelStorageTests {
+    func makeStorage() -> ModelStorage {
+        ModelStorage(base: FileManager.default.temporaryDirectory.appending(path: "sst-models-\(UUID().uuidString)"))
+    }
+
+    /// Creates `<repo>/<folder>/<model>.mlmodelc/coremldata.bin` for the given models.
+    func install(_ storage: ModelStorage, folder: String, models: [String] = ModelStorage.requiredModels) throws {
+        for model in models {
+            let dir = storage.repoDirectory.appending(path: "\(folder)/\(model).mlmodelc", directoryHint: .isDirectory)
+            try FileManager.default.createDirectory(at: dir, withIntermediateDirectories: true)
+            try Data([1]).write(to: dir.appending(path: "coremldata.bin"))
+        }
+    }
+
+    @Test func emptyStorageHasNothingInstalled() {
+        let storage = makeStorage()
+        #expect(storage.folder(for: "small") == nil)
+        #expect(!storage.isInstalled("small"))
+    }
+
+    @Test func completeInstallIsFound() throws {
+        let storage = makeStorage()
+        try install(storage, folder: "openai_whisper-large-v3-v20240930_turbo")
+        #expect(storage.isInstalled("large-v3-v20240930_turbo"))
+        #expect(storage.folder(for: "large-v3-v20240930_turbo")?.lastPathComponent == "openai_whisper-large-v3-v20240930_turbo")
+    }
+
+    @Test func incompleteInstallDoesNotCount() throws {
+        let storage = makeStorage()
+        try install(storage, folder: "openai_whisper-small", models: ["MelSpectrogram", "AudioEncoder"])
+        #expect(!storage.isInstalled("small"))
+    }
+
+    @Test func variantsAreMatchedExactly() throws {
+        let storage = makeStorage()
+        try install(storage, folder: "openai_whisper-large-v3-v20240930_turbo_632MB")
+        #expect(!storage.isInstalled("large-v3-v20240930_turbo"))
+        #expect(storage.isInstalled("large-v3-v20240930_turbo_632MB"))
+    }
+
+    @Test func deleteRemovesOnlyThatVariant() throws {
+        let storage = makeStorage()
+        try install(storage, folder: "openai_whisper-small")
+        try install(storage, folder: "openai_whisper-base")
+        try storage.delete("small")
+        #expect(!storage.isInstalled("small"))
+        #expect(storage.isInstalled("base"))
+    }
+
+    @Test func deletingSomethingNotInstalledIsHarmless() throws {
+        try makeStorage().delete("small")
+    }
+
+    @Test func catalogStartsWithTheDefaultAndHasUniqueIds() {
+        #expect(ModelCatalog.all.first?.id == ModelCatalog.defaultVariant)
+        #expect(Set(ModelCatalog.all.map(\.id)).count == ModelCatalog.all.count)
+        #expect(ModelCatalog.all.allSatisfy { $0.approxSizeMB > 0 })
+    }
+}