discourse/spec/support/uploads_helpers.rb

# frozen_string_literal: true

module UploadsHelpers
  def setup_s3
    SiteSetting.enable_s3_uploads = true

    SiteSetting.s3_region = 'us-west-1'
    SiteSetting.s3_upload_bucket = "s3-upload-bucket"

    SiteSetting.s3_access_key_id = "some key"
    SiteSetting.s3_secret_access_key = "some secrets3_region key"

    stub_request(:head, "https://#{SiteSetting.s3_upload_bucket}.s3.#{SiteSetting.s3_region}.amazonaws.com/")
  end

  # TODO (martin) Remove this alias once discourse-chat plugin has been
  # updated to use secure_uploads instead.
  def enable_secure_media
    enable_secure_uploads
    DiscourseEvent.trigger(:site_setting_changed, :secure_media, false, true)
  end

  def enable_secure_uploads
    setup_s3
    SiteSetting.secure_uploads = true
  end

  def stub_upload(upload)
    url = %r{https://#{SiteSetting.s3_upload_bucket}.s3.#{SiteSetting.s3_region}.amazonaws.com/original/\d+X.*#{upload.sha1}.#{upload.extension}\?acl}
    stub_request(:put, url)
  end

  def stub_s3_store
    store = FileStore::S3Store.new
    client = Aws::S3::Client.new(stub_responses: true)
    store.s3_helper.stubs(:s3_client).returns(client)
    Discourse.stubs(:store).returns(store)
  end
end