write spec

Author: SamSaffron

lib/ai_bot/tools/create_artifact.rb

@@ -93,7 +93,7 @@ def update_custom_html(artifact = nil)
           js = parameters[:js].to_s
 
           artifact_div =
-            "<div class=\"ai-artifact\" data-ai-artifact-id=#{artifact.id}></div>" if artifact
+            "<div class=\"ai-artifact\" data-ai-artifact-id=\"#{artifact.id}\"></div>" if artifact
 
           content = []
 

spec/lib/modules/ai_bot/tools/create_artifact_spec.rb

@@ -4,10 +4,46 @@
   fab!(:llm_model)
   let(:bot_user) { DiscourseAi::AiBot::EntryPoint.find_user_from_model(llm_model.name) }
   let(:llm) { DiscourseAi::Completions::Llm.proxy("custom:#{llm_model.id}") }
+  fab!(:post)
 
   before { SiteSetting.ai_bot_enabled = true }
 
   describe "#process" do
+    it "correctly adds details block on final invoke" do
+      tool =
+        described_class.new(
+          { html_body: "hello" },
+          bot_user: Fabricate(:user),
+          llm: llm,
+          context: {
+            post_id: post.id,
+          },
+        )
+
+      tool.parameters = { html_body: "hello" }
+
+      tool.invoke {}
+
+      artifact_id = AiArtifact.order("id desc").limit(1).pluck(:id).first
+
+      expected = <<~MD
+        [details='View Source']
+
+        ### HTML
+
+        ```html
+        hello
+        ```
+
+        [/details]
+
+        ### Preview
+
+        <div class="ai-artifact" data-ai-artifact-id="#{artifact_id}"></div>
+      MD
+      expect(tool.custom_raw.strip).to eq(expected.strip)
+    end
+
     it "can correctly handle partial updates" do
       tool = described_class.new({}, bot_user: bot_user, llm: llm)