add specs

Author: SamSaffron

lib/ai_helper/assistant.rb

@@ -60,7 +60,7 @@ def available_prompts(user)
 
       def custom_locale_instructions(user = nil, force_default_locale)
         locale = SiteSetting.default_locale
-        locale = user.effective_locale if !force_default_locale
+        locale = user.effective_locale if !force_default_locale && user
         locale_hash = LocaleSiteSetting.language_names[locale]
 
         if locale != "en" && locale_hash
@@ -71,7 +71,7 @@ def custom_locale_instructions(user = nil, force_default_locale)
         end
       end
 
-      def localize_prompt!(prompt, user = nil, force_default_locale)
+      def localize_prompt!(prompt, user = nil, force_default_locale = false)
         locale_instructions = custom_locale_instructions(user, force_default_locale)
         if locale_instructions
           prompt.messages[0][:content] = prompt.messages[0][:content] + locale_instructions

spec/lib/modules/ai_helper/assistant_spec.rb

@@ -141,6 +141,45 @@
 
       expect(prompt.messages[0][:content].strip).to eq("This is a English (US) test")
     end
+
+    context "with temporal context" do
+      let(:prompt) do
+        CompletionPrompt.new(
+          messages: {
+            insts: "Current context: {{temporal_context}}",
+          },
+        ).messages_with_input("test")
+      end
+
+      it "replaces temporal context with timezone information" do
+        timezone = "America/New_York"
+        user.user_option.update!(timezone: timezone)
+        freeze_time "2024-01-01 12:00:00"
+
+        subject.localize_prompt!(prompt, user)
+
+        content = prompt.messages[0][:content]
+
+        expect(content).to include(%("timezone":"America/New_York"))
+      end
+
+      it "uses UTC as default timezone when user timezone is not set" do
+        user.user_option.update!(timezone: nil)
+
+        freeze_time "2024-01-01 12:00:00" do
+          subject.localize_prompt!(prompt, user)
+
+          parsed_context = JSON.parse(prompt.messages[0][:content].match(/context: (.+)$/)[1])
+          expect(parsed_context["user"]["timezone"]).to eq("UTC")
+        end
+      end
+
+      it "does not replace temporal context when user is nil" do
+        prompt_content = prompt.messages[0][:content].dup
+        subject.localize_prompt!(prompt, nil)
+        expect(prompt.messages[0][:content]).to eq(prompt_content)
+      end
+    end
   end
 
   describe "#generate_and_send_prompt" do