diff --git a/clients/macos/vellum-assistantTests/Features/Settings/InferenceProfileEditorTests.swift b/clients/macos/vellum-assistantTests/Features/Settings/InferenceProfileEditorTests.swift index 9839b57bc49..09fc6556aec 100644 --- a/clients/macos/vellum-assistantTests/Features/Settings/InferenceProfileEditorTests.swift +++ b/clients/macos/vellum-assistantTests/Features/Settings/InferenceProfileEditorTests.swift @@ -788,24 +788,4 @@ final class InferenceProfileEditorTests: XCTestCase { _ = editor.body XCTAssertEqual(cancelCalls, 0) } - - // MARK: - Provider dropdown filtering - - func testManagedModeShowsOnlyManagedCapableProviders() { - store.inferenceMode = "managed" - let (editor, _) = makeEditor(profile: InferenceProfile(name: "draft")) - XCTAssertEqual( - Set(editor.availableProviderIds), - Set(["anthropic", "openai", "gemini"]) - ) - XCTAssertFalse(editor.availableProviderIds.contains("openrouter")) - } - - func testYourOwnModeShowsAllCatalogProviders() { - store.inferenceMode = "your-own" - let (editor, _) = makeEditor(profile: InferenceProfile(name: "draft")) - XCTAssertTrue(editor.availableProviderIds.contains("openrouter"), - "your-own mode should expose all catalog providers including openrouter") - XCTAssertEqual(editor.availableProviderIds.count, store.dynamicProviderIds.count) - } } diff --git a/clients/macos/vellum-assistantTests/Network/ProviderConnectionClientTests.swift b/clients/macos/vellum-assistantTests/Network/ProviderConnectionClientTests.swift index cf55d0af20d..b5975dc8ec9 100644 --- a/clients/macos/vellum-assistantTests/Network/ProviderConnectionClientTests.swift +++ b/clients/macos/vellum-assistantTests/Network/ProviderConnectionClientTests.swift @@ -110,7 +110,7 @@ final class ProviderConnectionClientTests: XCTestCase { func testListReturnsEmptyWhenClientReturnsEmpty() async { mock.listResponse = [] let result = await mock.listProviderConnections(provider: nil) - XCTAssertEqual(result, []) + XCTAssertEqual(result?.count, 0) XCTAssertEqual(mock.listCallCount, 1) XCTAssertNil(mock.listProviderArg as Any? as? String) }