module RSpec::Rails::ChannelExampleGroup::ClassMethods

def connection_class
  (_connection_class || described_class).tap do |klass|
    next if klass <= ::ActionCable::Connection::Base
    raise "Described class is not a connection class.\n" \
          "Specify the connection class in the `describe` statement " \
          "or set it manually using `tests MyConnectionClass`"
  end
end