editions.rb

require_relative "../support/generic_edition"

FactoryBot.define do
  factory :edition, class: GenericEdition, traits: [:translated] do
    creator
    sequence(:title) { |index| "edition-title-#{index}" }
    body { "edition-body" }
    change_note { "change-note" }
    summary { "edition-summary" }
    previously_published { false }
    auth_bypass_id { SecureRandom.uuid }

    trait(:with_organisations) do
      transient do
        organisations { [] }
        create_default_organisation { true }
        supporting_organisations { [] }
        lead_organisations { organisations }
      end
      after :build do |edition, evaluator|
        if evaluator.lead_organisations.empty? && evaluator.create_default_organisation
          edition.edition_organisations.build(
            edition:,
            organisation: FactoryBot.build(:organisation),
            lead_ordering: 1,
            lead: true,
          )
        end
        Array.wrap(evaluator.lead_organisations).each.with_index do |org, idx|
          edition.edition_organisations.build(
            edition:,
            organisation: org,
            lead_ordering: idx + 1,
            lead: true,
          )
        end
        Array.wrap(evaluator.supporting_organisations).each do |org|
          edition.edition_organisations.build(
            edition:,
            organisation: org,
            lead: false,
          )
        end
      end
    end

    trait(:with_topical_events) do
      after :build do |edition, evaluator|
        if evaluator.topical_events.empty?
          edition.topical_event_memberships.build(
            edition:,
            topical_event: build(:topical_event),
          )
        end
      end
    end

    trait(:draft) { state { "draft" } }

    trait(:submitted) do
      transient do
        submitter { nil }
      end
      state { "submitted" }
      after :create do |edition, evaluator|
        edition.versions.first.update!(event: "create", state: "draft")
        submitter = evaluator.submitter.presence || edition.creator
        edition.versions.create! event: "update", whodunnit: submitter.id, state: "submitted"
      end
    end

    trait(:rejected) { state { "rejected" } }

    trait(:published) do
      state { "published" }
      first_published_at { 2.days.ago }
      major_change_published_at { 1.day.ago }
      force_published { false }
      published_major_version { 1 }
      published_minor_version { 0 }
    end

    trait(:non_english) { primary_locale { "cy" } }

    trait(:force_published) do
      state { "published" }
      first_published_at { 2.days.ago }
      major_change_published_at { 1.day.ago }
      force_published { true }
      published_major_version { 1 }
      published_minor_version { 0 }
    end

    trait(:deleted) { state { "deleted" } }

    trait(:superseded) do
      state { "superseded" }
      first_published_at { 2.days.ago }
    end

    trait(:superseded_with_published) do
      state { "superseded" }
      first_published_at { 2.days.ago }

      after(:create) do |edition|
        create(:published_edition, document: edition.document)
      end
    end

    trait(:featured) { featured { true } }

    trait(:scheduled) do
      state { "scheduled" }
      scheduled_publication { 7.days.from_now }
    end

    trait(:force_scheduled) do
      state { "scheduled" }
      force_published { true }
      scheduled_publication { 7.days.from_now }
    end

    trait(:access_limited) { access_limited { :organisations } }

    trait(:with_alternative_format_provider) do
      association :alternative_format_provider, factory: :organisation_with_alternative_format_contact_email
    end

    trait(:with_file_attachment) do
      association :alternative_format_provider, factory: :organisation_with_alternative_format_contact_email
      attachments { FactoryBot.build_list :file_attachment, 1 }
    end

    trait(:with_html_attachment) do
      association :alternative_format_provider, factory: :organisation_with_alternative_format_contact_email
      attachments { FactoryBot.build_list :html_attachment, 1 }
    end

    trait(:with_file_attachment_not_scanned) do
      association :alternative_format_provider, factory: :organisation_with_alternative_format_contact_email
      attachments { FactoryBot.build_list :file_attachment, 1 }
    end

    trait(:with_document) do
      document
    end

    trait(:unpublished) do
      state { :unpublished }
      unpublishing { build(:unpublishing) }
    end

    trait(:published_in_error_redirect) do
      state { :unpublished }
      unpublishing { build(:published_in_error_redirect_unpublishing) }
    end

    trait(:published_in_error_no_redirect) do
      state { :unpublished }
      unpublishing { build(:published_in_error_no_redirect_unpublishing) }
    end

    trait(:consolidated_redirect) do
      state { :unpublished }
      unpublishing { build(:consolidated_unpublishing) }
    end

    trait(:withdrawn) do
      state { "withdrawn" }
      unpublishing { build(:withdrawn_unpublishing) }
    end
  end

  factory :edition_with_document, parent: :edition, traits: [:with_document]
  factory :draft_edition, parent: :edition, traits: [:draft]
  factory :submitted_edition, parent: :edition, traits: [:submitted]
  factory :rejected_edition, parent: :edition, traits: [:rejected]
  factory :published_edition, parent: :edition, traits: [:published]
  factory :non_english_published_edition, parent: :edition, traits: %i[non_english published]
  factory :deleted_edition, parent: :edition, traits: [:deleted]
  factory :superseded_edition, parent: :edition, traits: [:superseded]
  factory :scheduled_edition, parent: :edition, traits: [:scheduled]
  factory :force_scheduled_edition, parent: :edition, traits: [:force_scheduled]
  factory :force_published_edition, parent: :edition, traits: [:force_published]
  factory :unpublished_edition, parent: :edition, traits: [:unpublished]
  factory :withdrawn_edition, parent: :edition, traits: [:withdrawn]
  factory :protected_edition, parent: :edition, traits: [:access_limited]
  factory :edition_with_organisations, parent: :edition, traits: [:with_organisations]
end