Merge pull request #5960 from nhsuk/draft-vaccination-record-uuid

Set a UUID on draft vaccination records

Author: misaka

app/forms/vaccinate_form.rb

@@ -151,6 +151,7 @@ def save(draft_vaccination_record:)
     draft_vaccination_record.source = "service"
     draft_vaccination_record.supplied_by_user_id =
       supplied_by_user_id || psd_created_by_user_id
+    draft_vaccination_record.uuid = generate_uuid
 
     draft_vaccination_record.save # rubocop:disable Rails/SaveBang
   end
@@ -211,4 +212,12 @@ def pre_screening
         programme:
       )
   end
+
+  def generate_uuid
+    10
+      .times
+      .map { SecureRandom.uuid }
+      .find { !VaccinationRecord.exists?(uuid: it) } or
+      raise "Unable to generate unique UUID"
+  end
 end

app/models/draft_vaccination_record.rb

@@ -37,6 +37,7 @@ class DraftVaccinationRecord
   attribute :session_id, :integer
   attribute :source, :string
   attribute :supplied_by_user_id, :integer
+  attribute :uuid, :string
   attribute :vaccine_id, :integer
 
   MMR_OR_MMRV_INTRODUCTION_DATE = Date.new(2020, 1, 1).freeze
@@ -417,6 +418,7 @@ def writable_attribute_names
       session_id
       source
       supplied_by_user_id
+      uuid
       vaccine_id
     ]
   end

spec/forms/vaccinate_form_spec.rb

@@ -91,4 +91,41 @@
       end
     end
   end
+
+  describe "#save" do
+    subject(:form) do
+      described_class.new(
+        programme:,
+        current_user:,
+        session:,
+        patient: create(:patient),
+        identity_check_confirmed_by_patient: true,
+        pre_screening_notes: "",
+        pre_screening_confirmed: true,
+        vaccine_method: "injection",
+        delivery_site: "left_arm_upper_position"
+      )
+    end
+
+    describe "a uuid" do
+      let(:draft_vaccination_record) do
+        DraftVaccinationRecord.new(request_session: {}, current_user:)
+      end
+
+      it "is generated" do
+        form.save(draft_vaccination_record:) # rubocop:disable Rails/SaveBang
+        expect(draft_vaccination_record.uuid).to be_present
+      end
+
+      it "tries 10 times to find a uuid that's not already in use" do
+        allow(VaccinationRecord).to receive(:exists?).and_return(true)
+
+        expect { form.save(draft_vaccination_record:) }.to raise_error(
+          RuntimeError,
+          "Unable to generate unique UUID"
+        )
+        expect(VaccinationRecord).to have_received(:exists?).exactly(10).times
+      end
+    end
+  end
 end