Merge branch 'fix-broken-duo-chat-spec' into 'master'

Fix broken duo chat spec after free access cutoff

See merge request https://gitlab.com/gitlab-org/gitlab/-/merge_requests/169528



Merged-by: Mark Chao <mchao@gitlab.com>
Approved-by: Mark Chao <mchao@gitlab.com>
Co-authored-by: Shinya Maeda <shinya@gitlab.com>

ee/spec/lib/gitlab/llm/chain/tools/build_reader/executor_spec.rb

@@ -50,6 +50,17 @@
     let_it_be(:user) { create(:user) }
     let_it_be_with_reload(:group) { create(:group_with_plan, plan: :ultimate_plan) }
     let_it_be_with_reload(:project) { create(:project, group: group) }
+    let_it_be(:add_on_purchase) do
+      create(:gitlab_subscription_add_on_purchase, :duo_enterprise, namespace: group)
+    end
+
+    let_it_be(:seat_assignment) do
+      create(
+        :gitlab_subscription_user_add_on_assignment,
+        user: user,
+        add_on_purchase: add_on_purchase
+      )
+    end
     before_all do
       project.add_developer(user)
@@ -235,7 +246,7 @@
         context 'with invalid build url' do
           let(:ai_response) { "url\", \"ResourceIdentifier\": \"#{url}\"}" }
-          let(:url) { "https://example.com/some/path/-/builds/1fd11" }
+          let(:url) { "https://example.com/some/path/-/builds/999999999999dfd11" }
           it "returns success response with an explanatory message" do
             allow(tool).to receive(:request).and_return(ai_response)

ee/spec/lib/gitlab/llm/completions/summarize_all_open_notes_spec.rb

@@ -115,6 +115,17 @@
     let_it_be(:user) { create(:user) }
     let_it_be_with_reload(:group) { create(:group_with_plan, plan: :ultimate_plan) }
     let_it_be(:project) { create(:project, group: group) }
+    let_it_be(:add_on_purchase) do
+      create(:gitlab_subscription_add_on_purchase, :duo_enterprise, namespace: group)
+    end
+
+    let_it_be(:seat_assignment) do
+      create(
+        :gitlab_subscription_user_add_on_assignment,
+        user: user,
+        add_on_purchase: add_on_purchase
+      )
+    end
     before_all do
       project.add_developer(user)

ee/spec/requests/api/graphql/mutations/projects/generate_commit_message_spec.rb

@@ -10,6 +10,17 @@
   let_it_be(:project) { create(:project, :public, group: group) }
   let_it_be(:current_user) { create(:user, developer_of: [project, group]) }
   let_it_be(:merge_request) { create(:merge_request, source_project: project) }
+  let_it_be(:add_on_purchase) do
+    create(:gitlab_subscription_add_on_purchase, :duo_enterprise, namespace: group)
+  end
+
+  let_it_be(:seat_assignment) do
+    create(
+      :gitlab_subscription_user_add_on_assignment,
+      user: current_user,
+      add_on_purchase: add_on_purchase
+    )
+  end
   let(:mutation) do
     params = { generate_commit_message: { resource_id: merge_request.to_gid } }

ee/spec/services/llm/git_command_service_spec.rb

@@ -8,6 +8,18 @@
   describe '#perform', :saas do
     let_it_be(:group) { create(:group_with_plan, plan: :ultimate_plan) }
     let_it_be(:user) { create(:user) }
+    let_it_be(:add_on_purchase) do
+      create(:gitlab_subscription_add_on_purchase, :duo_enterprise, namespace: group)
+    end
+
+    let_it_be(:seat_assignment) do
+      create(
+        :gitlab_subscription_user_add_on_assignment,
+        user: user,
+        add_on_purchase: add_on_purchase
+      )
+    end
+
     let(:current_user) { user }
     let(:options) { { prompt: 'list 10 commit titles' } }
@@ -72,13 +84,5 @@
         expect(response.payload).to be_nil
       end
     end
-
-    context 'when user is not a member of ultimate group' do
-      let(:current_user) { create(:user) }
-
-      it 'returns an error' do
-        expect(subject.execute).to be_error
-      end
-    end
   end
 end

ee/spec/services/llm/internal/completion_service_spec.rb

@@ -9,6 +9,18 @@
   let_it_be(:group) { create(:group_with_plan, plan: :ultimate_plan) }
   let_it_be(:project) { create(:project, group: group) }
   let_it_be(:resource) { create(:issue, project: project) }
+  let_it_be(:add_on_purchase) do
+    create(:gitlab_subscription_add_on_purchase, :duo_enterprise, namespace: group)
+  end
+
+  let_it_be(:seat_assignment) do
+    create(
+      :gitlab_subscription_user_add_on_assignment,
+      user: user,
+      add_on_purchase: add_on_purchase
+    )
+  end
+
   let(:options) { { 'key' => 'value' } }
   let(:ai_action_name) { :summarize_comments }
   let(:referer_url) { nil }