discourse/spec/lib/middleware/enforce_hostname_spec.rb

# frozen_string_literal: true

RSpec.describe Middleware::EnforceHostname do
  before do
    RailsMultisite::ConnectionManagement.stubs(:current_db_hostnames).returns(
      %w[primary.example.com secondary.example.com],
    )
    RailsMultisite::ConnectionManagement.stubs(:current_hostname).returns("primary.example.com")
  end

  def check_returned_host(input_host)
    resolved_host = nil

    app =
      described_class.new(
        lambda do |env|
          resolved_host = env["HTTP_HOST"]
          [200, {}, ["ok"]]
        end,
      )

    app.call({ "HTTP_HOST" => input_host })

    resolved_host
  end

  it "works for the primary domain" do
    expect(check_returned_host("primary.example.com")).to eq("primary.example.com")
  end

  it "works for the secondary domain" do
    expect(check_returned_host("secondary.example.com")).to eq("secondary.example.com")
  end

  it "returns primary domain otherwise" do
    expect(check_returned_host("other.example.com")).to eq("primary.example.com")
    expect(check_returned_host(nil)).to eq("primary.example.com")
  end
end