如何在 rails 6 中为动作电缆连接类编写单元测试用例

Question

以下是我的 connection.rb

module ApplicationCable
  class Connection < ActionCable::Connection::Base
    identified_by :current_user

    def connect
      self.current_user = find_verified_user
    end

    private

    def find_verified_user
      verified_user = env['warden'].user
      return verified_user if verified_user
    end
  end
end

如何使用环境变量warden进行单元测试？ 我需要设置吗？

Answer 1

经过过多的搜索，我得到了connection_test.rb的解决方案

# frozen_string_literal: true

require "test_helper"

module ApplicationCable
  class ConnectionTest < ActionCable::Connection::TestCase
    include ActionCable::TestHelper

    test "connects with devise" do
      user = users(:admin)
      connect_with_user(user)
      assert_equal connection.current_user, user
    end

    private

    def connect_with_user(user)
      connect env: { 'warden' => FakeEnv.new(user) }
    end

    class FakeEnv
      attr_reader :user

      def initialize(user)
        @user = user
      end
    end
  end
end