40 lines
1.1 KiB
Ruby
40 lines
1.1 KiB
Ruby
# 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
|