module ActionCable::Connection::TestCase::Behavior::ClassMethods

def connection_class
  if connection = self._connection_class
    connection
  else
    tests determine_default_connection(name)
  end
end

def determine_default_connection(name)
  connection = determine_constant_from_test_name(name) do |constant|
    Class === constant && constant < ActionCable::Connection::Base
  end
  raise NonInferrableConnectionError.new(name) if connection.nil?
  connection
end

def tests(connection)
  case connection
  when String, Symbol
    self._connection_class = connection.to_s.camelize.constantize
  when Module
    self._connection_class = connection
  else
    raise NonInferrableConnectionError.new(connection)
  end
end