Enable DefaultNewsOrganisationImageData to have assets and use non-le…

…gacy-url

Co-authored-by: Iris Lau <[email protected]>

app/models/default_news_organisation_image_data.rb

@@ -1,4 +1,8 @@
 class DefaultNewsOrganisationImageData < ApplicationRecord
+  has_many :assets,
+           as: :assetable,
+           inverse_of: :assetable
+
   mount_uploader :file, FeaturedImageUploader, mount_on: :carrierwave_image
   validates :file, presence: true
 end

lib/whitehall/asset_manager_storage.rb

@@ -102,7 +102,7 @@ def get_asset
   def self.use_non_legacy_behaviour?(model)
     return unless model
 
-    return true if model.instance_of?(AttachmentData) || model.instance_of?(ImageData) || model.instance_of?(Organisation)
+    return true if model.instance_of?(AttachmentData) || model.instance_of?(ImageData) || model.instance_of?(Organisation) || model.instance_of?(DefaultNewsOrganisationImageData)
 
     model.respond_to?("use_non_legacy_endpoints") && model.use_non_legacy_endpoints
   end

test/unit/lib/whitehall/asset_manager_storage_test.rb

@@ -164,6 +164,14 @@ def assets_protected?
       @uploader.store!(@file)
     end
   end
+
+  test "should call AssetManagerCreateAssetWorker if uploader model is DefaultNewsOrganisationImageData" do
+    model = build(:default_news_organisation_image_data)
+    @uploader.stubs(:model).returns(model)
+
+    AssetManagerCreateAssetWorker.expects(:perform_async).with(anything, anything, anything, anything, anything, anything)
+    @uploader.store!(@file)
+  end
 end
 
 class Whitehall::AssetManagerStorage::FileTest < ActiveSupport::TestCase