WIP

alphagov · Oct 24, 2023 · ae82340 · ae82340
1 parent ff23073
commit ae82340
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/app/models/featured_image_data.rb b/app/models/featured_image_data.rb
@@ -4,6 +4,10 @@ class FeaturedImageData < ApplicationRecord
   validates :file, presence: true, if: :image_changed?
   validates_with ImageValidator, size: [960, 640], if: :image_changed?
 
+  has_many :assets,
+           as: :assetable,
+           inverse_of: :assetable
+
 private
 
   def image_changed?

diff --git a/app/models/organisation.rb b/app/models/organisation.rb
@@ -10,8 +10,8 @@ class Organisation < ApplicationRecord
   DEFAULT_JOBS_URL = "https://www.civilservicejobs.service.gov.uk/csr".freeze
   FEATURED_DOCUMENTS_DISPLAY_LIMIT = 6
 
-  belongs_to :default_news_image_legacy, class_name: "DefaultNewsOrganisationImageData", foreign_key: :default_news_organisation_image_data_id
-  belongs_to :default_news_image, class_name: "FeaturedImageData", foreign_key: :featured_image_data_id
+  belongs_to :default_news_image, class_name: "DefaultNewsOrganisationImageData", foreign_key: :default_news_organisation_image_data_id
+  belongs_to :default_news_image_new, class_name: "FeaturedImageData", foreign_key: :featured_image_data_id
 
   has_many :assets,
            as: :assetable,