Optionally specify branch for DAST on-demand scan

- Extend services to pass on dast_profile.branch_name
- Update specs

ee/app/graphql/mutations/dast/profiles/run.rb

@@ -60,6 +60,7 @@ module Mutations
             container: project,
             current_user: current_user,
             params: {
+              branch: dast_profile.branch_name,
               dast_site_profile: dast_profile.dast_site_profile,
               dast_scanner_profile: dast_profile.dast_scanner_profile
             }

ee/app/services/dast/profiles/create_service.rb

@@ -21,6 +21,7 @@ module Dast
           container: container,
           current_user: current_user,
           params: {
+            branch: dast_profile.branch_name,
             dast_site_profile: dast_site_profile,
             dast_scanner_profile: dast_scanner_profile
           }

ee/app/services/dast/profiles/update_service.rb

@@ -49,6 +49,7 @@ module Dast
 
       def create_scan(dast_profile)
         params = {
+          branch: dast_profile.branch_name,
           dast_site_profile: dast_profile.dast_site_profile,
           dast_scanner_profile: dast_profile.dast_scanner_profile
         }

ee/spec/graphql/mutations/dast/profiles/create_spec.rb

@@ -56,7 +56,7 @@ RSpec.describe Mutations::Dast::Profiles::Create do
             actual_url = subject[:pipeline_url]
             pipeline = Ci::Pipeline.find_by(
               project: project,
-              sha: project.repository.commit.sha,
+              sha: project.repository.commits('orphaned-branch', limit: 1)[0].id,
               source: :ondemand_dast_scan,
               config_source: :parameter_source
             )

ee/spec/graphql/mutations/dast/profiles/run_spec.rb

@@ -6,7 +6,7 @@ RSpec.describe Mutations::Dast::Profiles::Run do
   let_it_be_with_refind(:project) { create(:project, :repository) }
 
   let_it_be(:user) { create(:user) }
-  let_it_be(:dast_profile) { create(:dast_profile, project: project) }
+  let_it_be(:dast_profile) { create(:dast_profile, project: project, branch_name: 'orphaned-branch') }
 
   let(:full_path) { project.full_path }
   let(:dast_profile_id) { dast_profile.to_global_id }
@@ -60,6 +60,12 @@ RSpec.describe Mutations::Dast::Profiles::Run do
           project.add_developer(user)
         end
 
+        it_behaves_like 'it delegates scan creation to another service' do
+          let(:delegated_params) do
+            { branch: dast_profile.branch_name, dast_site_profile: dast_profile.dast_site_profile, dast_scanner_profile: dast_profile.dast_scanner_profile }
+          end
+        end
+
         it 'returns a pipeline_url containing the correct path' do
           actual_url = subject[:pipeline_url]
           pipeline = Ci::Pipeline.last

ee/spec/services/dast/profiles/create_service_spec.rb

@@ -7,8 +7,16 @@ RSpec.describe Dast::Profiles::CreateService do
   let_it_be(:developer) { create(:user, developer_projects: [project] ) }
   let_it_be(:dast_site_profile) { create(:dast_site_profile, project: project) }
   let_it_be(:dast_scanner_profile) { create(:dast_scanner_profile, project: project) }
+
   let_it_be(:default_params) do
-    { name: SecureRandom.hex, description: :description, dast_site_profile: dast_site_profile, dast_scanner_profile: dast_scanner_profile, run_after_create: false }
+    {
+      name: SecureRandom.hex,
+      description: :description,
+      branch_name: 'orphaned-branch',
+      dast_site_profile: dast_site_profile,
+      dast_scanner_profile: dast_scanner_profile,
+      run_after_create: false
+    }
   end
 
   let(:params) { default_params }
@@ -57,12 +65,10 @@ RSpec.describe Dast::Profiles::CreateService do
       context 'when param run_after_create: true' do
         let(:params) { default_params.merge(run_after_create: true) }
 
-        it 'calls DastOnDemandScans::CreateService' do
-          params = { dast_site_profile: dast_site_profile, dast_scanner_profile: dast_scanner_profile }
-
-          expect(DastOnDemandScans::CreateService).to receive(:new).with(hash_including(params: params)).and_call_original
-
-          subject
+        it_behaves_like 'it delegates scan creation to another service' do
+          let(:delegated_params) do
+            { branch: default_params[:branch_name], dast_site_profile: dast_site_profile, dast_scanner_profile: dast_scanner_profile }
+          end
         end
 
         it 'creates a ci_pipeline' do

ee/spec/services/dast/profiles/update_service_spec.rb

@@ -5,17 +5,18 @@ require 'spec_helper'
 RSpec.describe Dast::Profiles::UpdateService do
   let_it_be(:project) { create(:project, :repository) }
   let_it_be(:user) { create(:user) }
-  let_it_be(:dast_profile, reload: true) { create(:dast_profile, project: project) }
+  let_it_be(:dast_profile, reload: true) { create(:dast_profile, project: project, branch_name: 'orphaned-branch') }
   let_it_be(:dast_site_profile) { create(:dast_site_profile, project: project) }
   let_it_be(:dast_scanner_profile) { create(:dast_scanner_profile, project: project) }
 
   let(:default_params) do
     {
+      name: SecureRandom.hex,
+      description: SecureRandom.hex,
+      branch_name: 'orphaned-branch',
       dast_profile: dast_profile,
       dast_site_profile_id: dast_site_profile.id,
-      dast_scanner_profile_id: dast_scanner_profile.id,
-      name: SecureRandom.hex,
-      description: SecureRandom.hex
+      dast_scanner_profile_id: dast_scanner_profile.id
     }
   end
 
@@ -92,12 +93,10 @@ RSpec.describe Dast::Profiles::UpdateService do
         context 'when param run_after_update: true' do
           let(:params) { default_params.merge(run_after_update: true) }
 
-          it 'calls DastOnDemandScans::CreateService' do
-            params = { dast_site_profile: dast_site_profile, dast_scanner_profile: dast_scanner_profile }
-
-            expect(DastOnDemandScans::CreateService).to receive(:new).with(hash_including(params: params)).and_call_original
-
-            subject
+          it_behaves_like 'it delegates scan creation to another service' do
+            let(:delegated_params) do
+              { branch: dast_profile.branch_name, dast_site_profile: dast_site_profile, dast_scanner_profile: dast_scanner_profile }
+            end
           end
 
           it 'creates a ci_pipeline' do

ee/spec/support/shared_examples/services/dast_on_demand_scans_shared_examples.rb

+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.shared_examples 'it delegates scan creation to another service' do
+  it 'calls DastOnDemandScans::CreateService' do
+    expect(DastOnDemandScans::CreateService).to receive(:new).with(hash_including(params: delegated_params)).and_call_original
+
+    subject
+  end
+end