Return dose sequence in StatusGenerator

Author: jf-x-dev

app/lib/status_generator/vaccination.rb

@@ -59,8 +59,6 @@ def disease_types
 
   def dose_sequence
     # TODO: Implement this for multi-dose HPV and Td/IPV in a more generic way.
-    return unless programme.mmr?
-
     return unless status_should_be_due? || status_should_be_eligible?
 
     valid_vaccination_records.count + 1

spec/lib/status_generator/vaccination_spec.rb

@@ -606,17 +606,16 @@
   describe "#dose_sequence" do
     subject(:dose_sequence) { generator.dose_sequence }
 
-    it { should be_nil }
+    let(:patient) do
+      create(:patient, :consent_given_triage_not_needed, session:)
+    end
+    let(:session) { create(:session, programmes: [programme]) }
 
-    context "for MMR programme" do
-      let(:programme) { Programme.mmr }
-      let(:session) { create(:session, programmes: [programme]) }
-      let(:patient) do
-        create(:patient, :consent_given_triage_not_needed, session:)
+    shared_examples "with 0 and 1 vaccination records" do
+      context "with no existing vaccination record" do
+        it { should eq(1) }
       end
 
-      it { should eq(1) }
-
       context "with an existing vaccination record" do
         before do
           create(:vaccination_record, patient:, programme:, dose_sequence: 1)
@@ -625,6 +624,36 @@
         it { should eq(2) }
       end
     end
+
+    context "for MMR programme" do
+      let(:programme) { Programme.mmr }
+
+      include_examples "with 0 and 1 vaccination records"
+    end
+
+    context "for HPV programme" do
+      let(:programme) { Programme.hpv }
+
+      include_examples "with 0 and 1 vaccination records"
+    end
+
+    context "for MenACWY programme" do
+      let(:programme) { Programme.menacwy }
+
+      include_examples "with 0 and 1 vaccination records"
+    end
+
+    context "for Td/IPV programme" do
+      let(:programme) { Programme.td_ipv }
+
+      include_examples "with 0 and 1 vaccination records"
+    end
+
+    context "for flu programme" do
+      let(:programme) { Programme.flu }
+
+      include_examples "with 0 and 1 vaccination records"
+    end
   end
 
   describe "#latest_date" do