Add GenericClinicFactory

This adds a new class which handles the creation of the generic clinic
factory and no longer creates these on the fly through the removal of
the `Organisation#generic_clinic` method.

I think this makes the code more explicit at describing how these
locations get created.

Author: thomasleese

app/lib/generic_clinic_factory.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+class GenericClinicFactory
+  def initialize(organisation:)
+    @organisation = organisation
+  end
+
+  def call
+    ActiveRecord::Base.transaction do
+      location.update!(year_groups:)
+      location.create_default_programme_year_groups!(programmes)
+      location
+    end
+  end
+
+  def self.call(...) = new(...).call
+
+  private_class_method :new
+
+  private
+
+  attr_reader :organisation
+
+  delegate :programmes, to: :organisation
+
+  def team
+    organisation
+      .teams
+      .create_with(
+        email: organisation.email,
+        phone: organisation.phone,
+        phone_instructions: organisation.phone_instructions
+      )
+      .find_or_create_by!(name: organisation.name)
+  end
+
+  def location
+    organisation.locations.find_by(
+      ods_code: organisation.ods_code,
+      type: :generic_clinic
+    ) ||
+      Location.create!(
+        name: "Community clinics",
+        ods_code: organisation.ods_code,
+        team:,
+        type: :generic_clinic
+      )
+  end
+
+  def year_groups
+    programmes.flat_map(&:default_year_groups).uniq.sort
+  end
+end

app/models/onboarding.rb

@@ -117,9 +117,10 @@ def errors
   def save!
     ActiveRecord::Base.transaction do
       models.each(&:save!)
-      organisation.generic_clinic.create_default_programme_year_groups!(
-        programmes.map(&:programme)
-      )
+
+      # Reload to ensure the programmes are loaded.
+      GenericClinicFactory.call(organisation: organisation.reload)
+
       @users.each { |user| user.organisations << organisation }
       UnscheduledSessionsFactory.new.call
     end

app/models/organisation.rb

@@ -80,30 +80,9 @@ def year_groups
     @year_groups ||= location_programme_year_groups.pluck_year_groups
   end
 
-  def generic_team
-    teams.create_with(email:, phone:, phone_instructions:).find_or_create_by!(
-      name:
-    )
-  end
-
-  def generic_clinic
-    locations.find_by(ods_code:, type: :generic_clinic) ||
-      ActiveRecord::Base.transaction do
-        Location
-          .create!(
-            name: "Community clinics",
-            team: generic_team,
-            ods_code:,
-            type: :generic_clinic,
-            year_groups: programmes.flat_map(&:default_year_groups).uniq.sort
-          )
-          .tap { it.create_default_programme_year_groups!(programmes) }
-      end
-  end
-
   def generic_clinic_session
     academic_year = AcademicYear.current
-    location = generic_clinic
+    location = locations.generic_clinic.first
 
     sessions
       .includes(:location, :programmes, :session_dates)

db/seeds.rb

@@ -65,11 +65,11 @@ def attach_sample_of_schools_to(organisation)
     .where(team_id: nil)
     .order("RANDOM()")
     .limit(50)
-    .update_all(team_id: organisation.generic_team.id)
+    .update_all(team_id: organisation.teams.first.id)
 end
 
 def attach_specific_school_to_organisation_if_present(organisation:, urn:)
-  Location.where(urn:).update_all(team_id: organisation.generic_team.id)
+  Location.where(urn:).update_all(team_id: organisation.teams.first.id)
 end
 
 def create_session(

lib/tasks/create_default_programme_year_groups.rake

@@ -7,11 +7,13 @@ task create_default_programme_year_groups: :environment do
     .find_each do |organisation|
       year_groups =
         organisation.programmes.flat_map(&:default_year_groups).uniq.sort
-      organisation.generic_clinic.update!(year_groups:)
 
-      organisation.generic_clinic.create_default_programme_year_groups!(
-        organisation.programmes
-      )
+      organisation.locations.generic_clinic.find_each do |generic_clinic|
+        generic_clinic.update_all(year_groups:)
+        generic_clinic.create_default_programme_year_groups!(
+          organisation.programmes
+        )
+      end
 
       organisation.schools.find_each do |school|
         school.create_default_programme_year_groups!(organisation.programmes)

lib/tasks/organisations.rake

@@ -7,5 +7,7 @@ namespace :organisations do
     programme = Programme.find_by!(type: args[:type])
 
     OrganisationProgramme.find_or_create_by!(organisation:, programme:)
+
+    GenericClinicFactory.call(organisation: organisation.reload)
   end
 end

spec/factories/organisations.rb

@@ -46,9 +46,7 @@
     end
 
     trait :with_generic_clinic do
-      after(:create) do |organisation|
-        create(:generic_clinic, team: organisation.generic_team)
-      end
+      after(:create) { |organisation| GenericClinicFactory.call(organisation:) }
     end
   end
 end

spec/features/td_ipv_already_had_spec.rb

@@ -65,13 +65,13 @@ def given_a_td_ipv_programme_with_a_session(clinic:)
     @programme = create(:programme, :td_ipv)
     programmes = [@programme]
 
-    organisation = create(:organisation, :with_generic_clinic, programmes:)
+    organisation = create(:organisation, programmes:)
     @nurse = create(:nurse, organisations: [organisation])
 
     location =
       (
         if clinic
-          organisation.generic_clinic
+          create(:generic_clinic, organisation:)
         else
           create(:school, :secondary, urn: 123_456, organisation:)
         end

spec/lib/generic_clinic_factory_spec.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+describe GenericClinicFactory do
+  describe "#call" do
+    subject(:call) { described_class.call(organisation:) }
+
+    let(:programmes) { [create(:programme, :hpv), create(:programme, :flu)] }
+
+    context "with a new organisation" do
+      let(:organisation) { create(:organisation, programmes:) }
+
+      it "creates a generic clinic location" do
+        expect { call }.to change(Location.generic_clinic, :count).by(1)
+
+        location = Location.generic_clinic.first
+        expect(location.organisation).to eq(organisation)
+        expect(location.year_groups).to contain_exactly(
+          0,
+          1,
+          2,
+          3,
+          4,
+          5,
+          6,
+          7,
+          8,
+          9,
+          10,
+          11
+        )
+      end
+    end
+
+    context "with an existing organisation" do
+      let(:organisation) do
+        create(:organisation, :with_generic_clinic, programmes:)
+      end
+
+      it "doesn't create a generic clinic location" do
+        organisation
+
+        expect { call }.not_to change(Location.generic_clinic, :count)
+      end
+    end
+  end
+end

spec/lib/reports/offline_session_exporter_spec.rb

@@ -651,7 +651,7 @@ def validation_formula(worksheet:, column_name:, row: 1)
     context "a clinic session" do
       subject(:workbook) { RubyXL::Parser.parse_buffer(call) }
 
-      let(:location) { organisation.generic_clinic }
+      let(:location) { organisation.locations.generic_clinic.first }
 
       it { should_not be_blank }