Add asset_manager_id to file attachments

app/models/attachment_data.rb

@@ -9,7 +9,7 @@ class AttachmentData < ApplicationRecord
            as: :assetable,
            inverse_of: :assetable
 
-  delegate :url, :path, to: :file, allow_nil: true
+  delegate :url, :path, :asset_manager_id, to: :file, allow_nil: true
 
   before_save :update_file_attributes
 

app/models/file_attachment.rb

@@ -55,6 +55,7 @@ def publishing_api_details_for_format
       filename:,
       number_of_pages:,
       preview_url:,
+      asset_manager_id:,
     }
   end
 
@@ -66,6 +67,12 @@ def alternative_format_contact_email
     nil
   end
 
+  def asset_manager_id
+    return unless csv? && attachable.is_a?(Edition) && attachment_data.all_asset_variants_uploaded?
+
+    attachment_data.asset_manager_id
+  end
+
   def preview_url
     return unless csv? && attachable.is_a?(Edition) && attachment_data.all_asset_variants_uploaded?
 

app/uploaders/attachment_uploader.rb

@@ -26,6 +26,10 @@ def extension_allowlist
     EXTENSION_ALLOW_LIST
   end
 
+  def asset_manager_id
+    file.try(:asset_manager_id)
+  end
+
   class ZipFile
     class NonUTF8ContentsError < RuntimeError; end
 

lib/whitehall/asset_manager_storage.rb

@@ -34,6 +34,13 @@ def url
       end
     end
 
+    def asset_manager_id
+      asset = get_asset
+      if asset
+        asset.asset_manager_id
+      end
+    end
+
     def path
       # We keep this because carrierwave needs this in after commit hook
       # to look for previous files to delete

test/unit/app/models/attachment_data_test.rb

@@ -452,4 +452,10 @@ class AttachmentDataTest < ActiveSupport::TestCase
 
     assert_not attachment_data.all_asset_variants_uploaded?
   end
+
+  test "#asset_manager_id returns asset manager id" do
+    attachment_data = create(:attachment_data_for_csv, attachable: create(:draft_edition, id: 1))
+
+    assert_equal "asset_manager_id", attachment_data.asset_manager_id
+  end
 end

test/unit/app/models/file_attachment_test.rb

@@ -68,4 +68,10 @@ def assert_delegated(attachment, method)
     attachment = create(:csv_attachment, attachable: create(:edition))
     assert_equal Plek.asset_root + "/media/#{attachment.attachment_data.id}/sample.csv/preview", attachment.publishing_api_details_for_format[:preview_url]
   end
+
+  test "#asset_manager_id returns asset manager id if all_asset_variants_uploaded?" do
+    attachment = create(:csv_attachment, attachable: create(:edition))
+    assert_not_nil attachment.attachment_data.asset_manager_id
+    assert_equal attachment.attachment_data.asset_manager_id, attachment.publishing_api_details_for_format[:asset_manager_id]
+  end
 end

test/unit/app/uploaders/attachment_uploader_test.rb

@@ -138,6 +138,13 @@ class AttachmentUploaderTest < ActiveSupport::TestCase
     end
   end
 
+  test "#asset_manager_id should return asset manager id" do
+    uploader = AttachmentUploader.new(@attachment_data, "mounted-as")
+    uploader.store!(file_fixture("sample_attachment.zip"))
+
+    assert_equal "asset_manager_id", uploader.asset_manager_id
+  end
+
   def required_arcgis_file_list
     %w[london.shp london.shx london.dbf]
   end

test/unit/lib/whitehall/asset_manager_storage_test.rb

@@ -68,4 +68,16 @@ class Whitehall::AssetManagerStorage::FileTest < ActiveSupport::TestCase
 
     assert_equal model.file.path, model.file.url
   end
+
+  test "returns nil when asset is not available" do
+    model = create(:attachment_data_with_no_assets, attachable: create(:draft_edition, id: 1))
+
+    assert_nil model.file.file.asset_manager_id
+  end
+
+  test "returns asset_manager_id when asset is available" do
+    model = create(:attachment_data_for_csv, attachable: create(:draft_edition, id: 1))
+
+    assert_equal "asset_manager_id", model.file.file.asset_manager_id
+  end
 end