test(macos): boost Clawdis coverage to 40%
This commit is contained in:
@@ -0,0 +1,54 @@
|
||||
import Foundation
|
||||
import Testing
|
||||
@testable import Clawdis
|
||||
|
||||
@Suite
|
||||
struct ModelCatalogLoaderTests {
|
||||
@Test
|
||||
func loadParsesModelsFromTypeScriptAndSorts() async throws {
|
||||
let src = """
|
||||
export const MODELS = {
|
||||
openai: {
|
||||
"gpt-4o-mini": { name: "GPT-4o mini", contextWindow: 128000 } satisfies any,
|
||||
"gpt-4o": { name: "GPT-4o", contextWindow: 128000 } as any,
|
||||
"gpt-3.5": { contextWindow: 16000 },
|
||||
},
|
||||
anthropic: {
|
||||
"claude-3": { name: "Claude 3", contextWindow: 200000 },
|
||||
},
|
||||
};
|
||||
"""
|
||||
|
||||
let tmp = FileManager.default.temporaryDirectory
|
||||
.appendingPathComponent("models-\(UUID().uuidString).ts")
|
||||
defer { try? FileManager.default.removeItem(at: tmp) }
|
||||
try src.write(to: tmp, atomically: true, encoding: .utf8)
|
||||
|
||||
let choices = try await ModelCatalogLoader.load(from: tmp.path)
|
||||
#expect(choices.count == 4)
|
||||
#expect(choices.first?.provider == "anthropic")
|
||||
#expect(choices.first?.id == "claude-3")
|
||||
|
||||
let ids = Set(choices.map(\.id))
|
||||
#expect(ids == Set(["claude-3", "gpt-4o", "gpt-4o-mini", "gpt-3.5"]))
|
||||
|
||||
let openai = choices.filter { $0.provider == "openai" }
|
||||
let openaiNames = openai.map(\.name)
|
||||
#expect(openaiNames == openaiNames.sorted { a, b in
|
||||
a.localizedCaseInsensitiveCompare(b) == .orderedAscending
|
||||
})
|
||||
}
|
||||
|
||||
@Test
|
||||
func loadWithNoExportReturnsEmptyChoices() async throws {
|
||||
let src = "const NOPE = 1;"
|
||||
let tmp = FileManager.default.temporaryDirectory
|
||||
.appendingPathComponent("models-\(UUID().uuidString).ts")
|
||||
defer { try? FileManager.default.removeItem(at: tmp) }
|
||||
try src.write(to: tmp, atomically: true, encoding: .utf8)
|
||||
|
||||
let choices = try await ModelCatalogLoader.load(from: tmp.path)
|
||||
#expect(choices.isEmpty)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user