discourse/spec/integration/multisite_spec.rb

# frozen_string_literal: true

require 'rails_helper'

describe 'multisite', type: :multisite do
  class DBNameMiddleware
    def initialize(app, config = {})
      @app = app
    end

    def call(env)
      # note current_db is already being ruined on boot cause its not multisite
      [200, {}, [RailsMultisite::ConnectionManagement.current_hostname]]
    end
  end

  let :session do
    stack = ActionDispatch::MiddlewareStack.new
    stack.use RailsMultisite::Middleware, RailsMultisite::DiscoursePatches.config
    stack.use DBNameMiddleware

    routes = ActionDispatch::Routing::RouteSet.new
    stack.build(routes)
  end

  it "should always allow /srv/status through" do
    headers = {
      "HTTP_HOST" => "unknown.com",
      "REQUEST_METHOD" => "GET",
      "PATH_INFO" => "/srv/status",
      "rack.input" => StringIO.new
    }

    code, _, body = session.call(headers)
    expect(code).to eq(200)
    expect(body.join).to eq("test.localhost")
  end

  it "should 404 on unknown routes" do
    headers = {
      "HTTP_HOST" => "unknown.com",
      "REQUEST_METHOD" => "GET",
      "PATH_INFO" => "/topics",
      "rack.input" => StringIO.new
    }

    code, _ = session.call(headers)
    expect(code).to eq(404)
  end

  it "should hit correct site elsewise" do
    headers = {
      "HTTP_HOST" => "test2.localhost",
      "REQUEST_METHOD" => "GET",
      "PATH_INFO" => "/topics",
      "rack.input" => StringIO.new
    }

    code, _, body = session.call(headers)
    expect(code).to eq(200)
    expect(body.join).to eq("test2.localhost")
  end
end