Add basic test for heartbeat channel

Author: rossta

app/channels/application_cable/connection.rb

@@ -12,11 +12,9 @@ def connect
     private
 
     def find_verified_admin_user
-      env["warden"].user(scope: :admin_user) || reject_unauthorized_connection
+      env["warden"]&.user(scope: :admin_user) || reject_unauthorized_connection
     end
 
-    private
-
     def report_error(e)
       Rails.logger.error(e)
       Honeybadger.notify(e)

app/channels/heartbeat_channel.rb

@@ -2,7 +2,4 @@ class HeartbeatChannel < ApplicationCable::Channel
   def subscribed
     stream_for current_admin_user
   end
-
-  def unsubscribed
-  end
 end

spec/channels/connection_spec.rb

@@ -0,0 +1,13 @@
+require "rails_helper"
+
+RSpec.describe ApplicationCable::Connection, type: :channel do
+  it "successfully connects" do
+    admin = instance_double(AdminUser, id: "323")
+    connect "/cable", env: {"warden" => double(user: admin)}
+    expect(connection.current_admin_user).to eq admin
+  end
+
+  it "rejects connection" do
+    expect { connect "/cable", env: {} }.to have_rejected_connection
+  end
+end

spec/channels/heartbeat_channel_spec.rb

@@ -0,0 +1,10 @@
+require "rails_helper"
+
+RSpec.describe HeartbeatChannel, type: :channel do
+  it "successfully subscribes" do
+    stub_connection(current_admin_user: FactoryBot.create(:admin_user))
+
+    subscribe room_id: 42
+    expect(subscription).to be_confirmed
+  end
+end

spec/support/channels.rb

@@ -0,0 +1,3 @@
+RSpec.configure do |config|
+  config.include Warden::Test::Helpers, type: :channel
+end

spec/support/warden_helpers.rb

@@ -13,6 +13,7 @@ def login_user(user = FactoryBot.create(:user))
 end
 
 RSpec.configure do |config|
+  config.include WardenHelpers, type: :channel
   config.include WardenHelpers, type: :request
   config.include WardenHelpers, type: :system