Introduce Programme#all_as_variants

This is similar to `Programme.all`, but returns all
`Programme::Variant`s where
available, in place of the related full `Programme.`

Jira-Issue: MAV-3195

Author: alistair-white-horne-tng

app/jobs/search_vaccination_records_in_nhs_job.rb

@@ -17,7 +17,7 @@ def perform(patient_id)
     SemanticLogger.tagged(tx_id:, job_id:) do
       Sentry.set_tags(tx_id:, job_id:)
 
-      programmes = Programme.all.to_a
+      programmes = Programme.all_as_variants
 
       feature_flag_enabled =
         programmes.any? do |programme|

app/models/programme.rb

@@ -88,6 +88,10 @@ def find_all(types, disease_types: nil, patient: nil)
       types.map { |type| find(type, patient:, disease_types:) }
     end
 
+    def all_as_variants
+      all.flat_map(&:variants)
+    end
+
     def find(type, disease_types: nil, patient: nil)
       validate_type!(type)
 

spec/models/programme_spec.rb

@@ -17,6 +17,41 @@
 describe Programme do
   subject(:programme) { described_class.sample }
 
+  describe "#all_as_variants" do
+    subject(:all_as_variants) { described_class.all_as_variants }
+
+    context "with the mmrv feature flag off" do
+      it "returns all the full programme objects" do
+        expect(all_as_variants).to contain_exactly(
+          described_class.flu,
+          described_class.hpv,
+          described_class.td_ipv,
+          described_class.menacwy,
+          described_class.mmr
+        )
+      end
+    end
+
+    context "with the mmrv feature flag on" do
+      before { Flipper.enable(:mmrv) }
+
+      it "returns all variants where applicable" do
+        expect(all_as_variants).to contain_exactly(
+          described_class.flu,
+          described_class.hpv,
+          described_class.td_ipv,
+          described_class.menacwy,
+          described_class.mmr.variant_for(
+            disease_types: Programme::Variant::DISEASE_TYPES.fetch("mmr")
+          ),
+          described_class.mmr.variant_for(
+            disease_types: Programme::Variant::DISEASE_TYPES.fetch("mmrv")
+          )
+        )
+      end
+    end
+  end
+
   describe "#find" do
     subject(:find) { described_class.find(type) }