Fix broken tests

Author: intitni

Core/Package.swift

@@ -269,6 +269,7 @@ let package = Package(
                 "LanguageClient",
                 "SuggestionModel",
                 "XPCShared",
+                .product(name: "Logger", package: "Tool"),
                 .product(name: "Preferences", package: "Tool"),
                 .product(name: "Terminal", package: "Tool"),
             ]

Core/Sources/SuggestionModel/LanguageIdentifierFromFilePath.swift

@@ -16,7 +16,7 @@ public enum CodeLanguage: RawRepresentable, Codable, CaseIterable, Hashable {
             return language
         }
     }
-    
+
     public var hashValue: Int {
         rawValue.hashValue
     }
@@ -30,15 +30,15 @@ public enum CodeLanguage: RawRepresentable, Codable, CaseIterable, Hashable {
             self = .other(rawValue)
         }
     }
-    
+
     public static var allCases: [CodeLanguage] {
         var all = LanguageIdentifier.allCases.map(CodeLanguage.builtIn)
         all.append(.plaintext)
         return all
     }
 }
 
-extension LanguageIdentifier {
+public extension LanguageIdentifier {
     /// Copied from https://github.com/github/linguist/blob/master/lib/linguist/languages.yml [MIT]
     var fileExtensions: [String] {
         switch self {
@@ -263,3 +263,4 @@ public func languageIdentifierFromFileURL(_ fileURL: URL) -> CodeLanguage {
     }
     return .init(rawValue: fileExtension) ?? .plaintext
 }
+

Core/Tests/GitHubCopilotServiceTests/FetchSuggestionsTests.swift

@@ -103,6 +103,10 @@ final class FetchSuggestionTests: XCTestCase {
         }
 
         class TestServer: GitHubCopilotLSP {
+            func sendNotification(_ notif: LanguageServerProtocol.ClientNotification) async throws {
+                // unimplemented
+            }
+            
             func sendRequest<E>(_ r: E) async throws -> E.Response where E: GitHubCopilotRequestType {
                 return GitHubCopilotRequest.GetCompletionsCycling.Response(completions: [
                     .init(

TestPlan.xctestplan

@@ -25,50 +25,57 @@
     {
       "target" : {
         "containerPath" : "container:Core",
-        "identifier" : "CopilotModelTests",
-        "name" : "CopilotModelTests"
+        "identifier" : "ServiceTests",
+        "name" : "ServiceTests"
       }
     },
     {
       "target" : {
         "containerPath" : "container:Core",
-        "identifier" : "CopilotServiceTests",
-        "name" : "CopilotServiceTests"
+        "identifier" : "SuggestionInjectorTests",
+        "name" : "SuggestionInjectorTests"
       }
     },
     {
       "target" : {
         "containerPath" : "container:Core",
-        "identifier" : "ServiceTests",
-        "name" : "ServiceTests"
+        "identifier" : "SuggestionWidgetTests",
+        "name" : "SuggestionWidgetTests"
       }
     },
     {
       "target" : {
         "containerPath" : "container:Core",
-        "identifier" : "SuggestionInjectorTests",
-        "name" : "SuggestionInjectorTests"
+        "identifier" : "PromptToCodeServiceTests",
+        "name" : "PromptToCodeServiceTests"
       }
     },
     {
       "target" : {
         "containerPath" : "container:Core",
-        "identifier" : "OpenAIServiceTests",
-        "name" : "OpenAIServiceTests"
+        "identifier" : "GitHubCopilotServiceTests",
+        "name" : "GitHubCopilotServiceTests"
       }
     },
     {
       "target" : {
         "containerPath" : "container:Core",
-        "identifier" : "SuggestionWidgetTests",
-        "name" : "SuggestionWidgetTests"
+        "identifier" : "SuggestionModelTests",
+        "name" : "SuggestionModelTests"
       }
     },
     {
       "target" : {
-        "containerPath" : "container:Core",
-        "identifier" : "PromptToCodeServiceTests",
-        "name" : "PromptToCodeServiceTests"
+        "containerPath" : "container:Tool",
+        "identifier" : "LangChainTests",
+        "name" : "LangChainTests"
+      }
+    },
+    {
+      "target" : {
+        "containerPath" : "container:Tool",
+        "identifier" : "OpenAIServiceTests",
+        "name" : "OpenAIServiceTests"
       }
     }
   ],

Tool/Tests/OpenAIServiceTests/ChatGPTServiceTests.swift

@@ -45,6 +45,7 @@ final class ChatGPTServiceTests: XCTestCase {
             requestBody = _requestBody
             return MockCompletionStreamAPI_Success()
         }
+        await service.mutateSystemPrompt("system")
         let stream = try await service.send(content: "Hello")
         var all = [String]()
         for try await text in stream {
@@ -58,7 +59,7 @@ final class ChatGPTServiceTests: XCTestCase {
         }
 
         XCTAssertEqual(requestBody?.messages, [
-            .init(role: .system, content: ""),
+            .init(role: .system, content: "system"),
             .init(role: .user, content: "Hello"),
         ], "System prompt is included")
         XCTAssertEqual(all, ["hello", "my", "friends"], "Text stream is correct")