actioncable/test/stubs/test_server.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

require "ostruct"

class TestServer
  include ActionCable::Server::Connections
  include ActionCable::Server::Broadcasting

  attr_reader :logger, :config, :mutex

  def initialize(subscription_adapter: SuccessAdapter)
    @logger = ActiveSupport::TaggedLogging.new ActiveSupport::Logger.new(StringIO.new)

    @config = OpenStruct.new(log_tags: [], subscription_adapter: subscription_adapter)
    @config.use_faye = ENV["FAYE"].present?
    @config.client_socket_class = if @config.use_faye
      ActionCable::Connection::FayeClientSocket
    else
      ActionCable::Connection::ClientSocket
    end

    @mutex = Monitor.new
  end

  def pubsub
    @pubsub ||= @config.subscription_adapter.new(self)
  end

  def event_loop
    @event_loop ||= if @config.use_faye
      ActionCable::Connection::FayeEventLoop.new
    else
      ActionCable::Connection::StreamEventLoop.new.tap do |loop|
        loop.instance_variable_set(:@executor, Concurrent.global_io_executor)
      end
    end
  end

  def worker_pool
    @worker_pool ||= ActionCable::Server::Worker.new(max_size: 5)
  end
end