Add sessions delegation command

app/lib/mavis_cli.rb

@@ -43,6 +43,7 @@ def self.progress_bar(total)
 require_relative "mavis_cli/schools/create"
 require_relative "mavis_cli/schools/move_patients"
 require_relative "mavis_cli/schools/remove_programme_year_group"
+require_relative "mavis_cli/sessions/delegation"
 require_relative "mavis_cli/stats/consents_by_school"
 require_relative "mavis_cli/stats/organisations"
 require_relative "mavis_cli/stats/vaccinations"

app/lib/mavis_cli/sessions/delegation.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+module MavisCLI
+  module Sessions
+    class Delegation < Dry::CLI::Command
+      desc "Configure delegation options in bulk"
+
+      argument :workgroup, required: true, desc: "Workgroup of the team"
+      argument :programme_type,
+               required: true,
+               desc: "Find sessions that administer this programme"
+
+      option :psd,
+             required: true,
+             type: :boolean,
+             desc: "Use patient specific direction (PSD)"
+      option :national_protocol,
+             required: true,
+             type: :boolean,
+             desc: "Use national protocol"
+
+      def call(workgroup:, programme_type:, psd:, national_protocol:)
+        MavisCLI.load_rails
+
+        team = Team.find_by!(workgroup:)
+        programme = Programme.find_by!(type: programme_type)
+
+        team
+          .sessions
+          .includes(:location)
+          .where(academic_year: AcademicYear.pending)
+          .has_programmes([programme])
+          .find_each do |session|
+            session.psd_enabled = psd
+            session.national_protocol_enabled = national_protocol
+
+            if session.changed?
+              session.save!
+              puts "Updated #{session.slug}: #{session.location.name}"
+            end
+          end
+      end
+    end
+  end
+
+  register "sessions" do |prefix|
+    prefix.register "delegation", Sessions::Delegation
+  end
+end

spec/features/cli_sessions_delegation_spec.rb

@@ -0,0 +1,92 @@
+# frozen_string_literal: true
+
+require_relative "../../app/lib/mavis_cli"
+
+describe "mavis sessions delegation" do
+  context "with valid arguments" do
+    it "runs successfully" do
+      given_a_team_exists
+      and_sessions_exist
+
+      when_i_run_the_command
+      then_the_team_sessions_are_updated
+      and_the_non_team_sessions_are_left
+    end
+  end
+
+  private
+
+  def command
+    Dry::CLI.new(MavisCLI).call(
+      arguments: %w[sessions delegation team flu --psd --national-protocol]
+    )
+  end
+
+  def given_a_team_exists
+    @programmes = [create(:programme, :hpv), create(:programme, :flu)]
+    @team = create(:team, workgroup: "team", programmes: @programmes)
+  end
+
+  def and_sessions_exist
+    academic_year = AcademicYear.pending
+
+    @team_hpv_session =
+      create(
+        :session,
+        :unscheduled,
+        team: @team,
+        programmes: [@programmes.first],
+        academic_year:
+      )
+    @non_team_hpv_session =
+      create(
+        :session,
+        :unscheduled,
+        programmes: [@programmes.first],
+        academic_year:
+      )
+    @team_flu_session =
+      create(
+        :session,
+        :unscheduled,
+        team: @team,
+        programmes: [@programmes.second],
+        academic_year:
+      )
+    @non_team_flu_session =
+      create(
+        :session,
+        :unscheduled,
+        programmes: [@programmes.second],
+        academic_year:
+      )
+  end
+
+  def when_i_run_the_command
+    @output = capture_output { command }
+  end
+
+  def then_the_team_sessions_are_updated
+    expect(@team_flu_session.reload).to have_attributes(
+      psd_enabled: true,
+      national_protocol_enabled: true
+    )
+
+    expect(@team_hpv_session.reload).to have_attributes(
+      psd_enabled: false,
+      national_protocol_enabled: false
+    )
+  end
+
+  def and_the_non_team_sessions_are_left
+    expect(@non_team_flu_session.reload).to have_attributes(
+      psd_enabled: false,
+      national_protocol_enabled: false
+    )
+
+    expect(@non_team_hpv_session.reload).to have_attributes(
+      psd_enabled: false,
+      national_protocol_enabled: false
+    )
+  end
+end