Add integration tests

Author: ericdallo

integration-test/entrypoint.clj

@@ -14,7 +14,7 @@
     integration.chat.ollama-test
     integration.chat.custom-provider-test
     integration.chat.commands-test
-    ])
+    integration.rewrite.openai-test])
 
 (defn timeout [timeout-ms callback]
   (let [fut (future (callback))

integration-test/integration/fixture.clj

@@ -18,8 +18,8 @@
                      :key "foo-key"
                      :keyEnv "FOO"}
    "google" {:url (str base-llm-mock-url "/google")
-                     :key "foo-key"
-                     :keyEnv "FOO"}})
+             :key "foo-key"
+             :keyEnv "FOO"}})
 
 (def default-init-options {:pureConfig true
                            :env "test"
@@ -53,3 +53,6 @@
 
 (defn chat-query-commands-request [params]
   [:chat/queryCommands params])
+
+(defn rewrite-prompt-request [params]
+  [:rewrite/prompt params])

integration-test/integration/helper.clj

@@ -43,7 +43,7 @@
   ;; Do a better escape considering more chars
   (cond-> (string/replace uri "::" "%3A%3A")
     windows?
-    (string/replace  #"/([a-zA-Z]):/" "/$1%3A/")))
+    (string/replace #"/([a-zA-Z]):/" "/$1%3A/")))
 
 (defn json-escape-path
   "Escapes the filesystem PATH string for safe use in JSON and returns the
@@ -66,7 +66,13 @@
 
 (defn match-content [chat-id role content]
   (is (match?
-        {:chatId chat-id
-         :role role
-         :content content}
-        (eca/client-awaits-server-notification :chat/contentReceived))))
+       {:chatId chat-id
+        :role role
+        :content content}
+       (eca/client-awaits-server-notification :chat/contentReceived))))
+
+(defn match-rewrite-content [rewrite-id content]
+  (is (match?
+       {:rewriteId rewrite-id
+        :content content}
+       (eca/client-awaits-server-notification :rewrite/contentReceived))))

integration-test/integration/rewrite/openai_test.clj

@@ -0,0 +1,46 @@
+(ns integration.rewrite.openai-test
+  (:require
+   [clojure.test :refer [deftest is testing]]
+   [integration.eca :as eca]
+   [integration.fixture :as fixture]
+   [integration.helper :refer [match-rewrite-content]]
+   [llm-mock.mocks :as llm.mocks]
+   [matcher-combinators.matchers :as m]
+   [matcher-combinators.test :refer [match?]]))
+
+(eca/clean-after-test)
+
+(deftest basic-rewrite-openai
+  (eca/start-process!)
+
+  ;; initialize server with rewrite model set to openai to hit the openai mock
+  (let [init-opts (assoc-in fixture/default-init-options [:rewrite :model] "openai/gpt-4.1")]
+    (eca/request! (fixture/initialize-request {:initializationOptions init-opts
+                                               :capabilities {:codeAssistant {:chat {}}}})))
+  (eca/notify! (fixture/initialized-notification))
+
+  (testing "Rewrite streams text and finishes, and body contains user prompt input"
+    (llm.mocks/set-case! :simple-text-0)
+    (let [resp (eca/request! (fixture/rewrite-prompt-request
+                              {:id "rw-1"
+                               :prompt "Please rewrite"
+                               :text "Original text"
+                               :range {:start {:line 1 :character 0}
+                                       :end {:line 1 :character 13}}}))]
+      (is (match?
+           {:model "openai/gpt-4.1"
+            :status "prompting"}
+           resp))
+
+      ;; rewrite/contentReceived notifications
+      (match-rewrite-content "rw-1" {:type "started"})
+      (match-rewrite-content "rw-1" {:type "text" :text "Knock"})
+      (match-rewrite-content "rw-1" {:type "text" :text " knock!"})
+      (match-rewrite-content "rw-1" {:type "finished" :totalTimeMs (m/pred number?)})
+
+      ;; Verify request body sent to mock: user input contains the prompt we sent
+      (is (match?
+           {:input [{:role "user"
+                     :content [{:type "input_text" :text "Please rewrite"}]}]
+            :instructions (m/pred string?)}
+           (llm.mocks/get-req-body :simple-text-0))))))