Spec for choosing persona and llm

Author: markvanlan

spec/system/ai_bot/homepage_spec.rb

@@ -18,10 +18,20 @@
       provider: "anthropic",
       url: "https://api.anthropic.com/v1/messages",
       name: "claude-2",
+      display_name: "Claude 2",
+    )
+  end
+  fab!(:claude_2_dup) do
+    Fabricate(
+      :llm_model,
+      provider: "anthropic",
+      url: "https://api.anthropic.com/v1/messages",
+      name: "claude-2",
+      display_name: "Duplicate",
     )
   end
   fab!(:bot_user) do
-    toggle_enabled_bots(bots: [claude_2])
+    toggle_enabled_bots(bots: [claude_2, claude_2_dup])
     SiteSetting.ai_bot_enabled = true
     claude_2.reload.user
   end
@@ -216,6 +226,18 @@
       expect(ai_pm_homepage).to have_homepage
       expect(sidebar).to have_no_section_link(pm.title)
     end
+
+    it "Allows choosing persona and LLM" do
+      ai_pm_homepage.visit
+
+      ai_pm_homepage.persona_selector.expand
+      ai_pm_homepage.persona_selector.select_row_by_name(persona.name)
+      ai_pm_homepage.persona_selector.collapse
+
+      ai_pm_homepage.llm_selector.expand
+      ai_pm_homepage.llm_selector.select_row_by_name(claude_2_dup.display_name)
+      ai_pm_homepage.llm_selector.collapse
+    end
   end
 
   context "when `ai_enable_experimental_bot_ux` is disabled" do

spec/system/page_objects/components/ai_pm_homepage.rb

@@ -5,6 +5,10 @@ module Components
     class AiPmHomepage < PageObjects::Components::Base
       HOMEPAGE_WRAPPER_CLASS = ".ai-bot-conversations__content-wrapper"
 
+      def visit
+        page.visit("/discourse-ai/ai-bot/conversations")
+      end
+
       def input
         page.find("#ai-bot-conversations-input")
       end
@@ -27,6 +31,14 @@ def has_homepage?
       def has_no_homepage?
         page.has_no_css?(HOMEPAGE_WRAPPER_CLASS)
       end
+
+      def persona_selector
+        PageObjects::Components::SelectKit.new(".persona-llm-selector__persona-dropdown")
+      end
+
+      def llm_selector
+        PageObjects::Components::SelectKit.new(".persona-llm-selector__llm-dropdown")
+      end
     end
   end
 end