discourse/spec/lib/middleware/enforce_hostname_spec.rb

38 lines
1.0 KiB
Ruby

# frozen_string_literal: true
describe Middleware::EnforceHostname do
before do
RailsMultisite::ConnectionManagement.stubs(:current_db_hostnames).returns(['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