RSpec设置ENV环境变量最佳方案：解决Gem加载时常量读取ENV问题

在封装为独立Gem的Cryption模块中，于模块单例类class << self作用域内定义了常量OFF_ENCRYPT，取值直接读取ENV['IS_OFF']，模块内的加解密类方法均调用该常量，实现代码如下：

module Cryption
 module Crypt
  class << self
    OFF_ENCRYPT = ENV['IS_OFF']
    
    def encrypt
       OFF_ENCRYPT
    end

    def decrypt
       OFF_ENCRYPT
    end
  end
 end
end

该写法的特性是常量会在模块被require加载的瞬间完成赋值，必须在加载模块前完成对应环境变量设置才能读取到正确值，正确的加载使用示例：

ENV["IS_OFF"] = "YES"
require "bundler/setup"
require "cryption"

Cryption::Crypt.encrypt
# 返回结果 => "YES"

编写RSpec测试用例时，spec_helper.rb中已经提前全局引入了相关依赖：

require "bundler/setup"
require "cryption"

模块加载完成后再修改ENV['IS_OFF']，不会改变已经完成赋值的常量值，因此编写的两组分别验证IS_OFF为YES、NO场景的测试用例无法得到预期结果，原有测试代码如下：

• crypt_off_spec.rb

RSpec.describe "IS_OFF=YES场景" do
  before(:context) do
    ENV['IS_OFF'] = "YES"
  end
  it "加密方法返回关闭标识" do
     expect(Cryption::Crypt.encrypt).to eq("YES")
  end
end

• crypt_on_spec.rb

RSpec.describe "IS_OFF=NO场景" do
  before(:context) do
    ENV['IS_OFF'] = "NO"
  end
  it "加密方法返回开启标识" do
     expect(Cryption::Crypt.encrypt).to eq("NO")
  end
end

核心诉求：在RSpec中是否存在规范的环境变量设置方式，适配常量在模块加载阶段就完成赋值的场景。

共有两种规范方案，可根据实际代码权限选择：

方案1：调整模块代码逻辑（长期推荐） #

从根源上避免加载顺序强依赖，不要在模块加载阶段把环境变量值固化到常量中，改为方法调用时动态读取环境变量：

module Cryption
  module Crypt
    class << self
      def encrypt
         ENV['IS_OFF']
      end

      def decrypt
         ENV['IS_OFF']
      end
    end
  end
end

如果需要保留常量调用的写法，可以额外增加常量重载入口，专门适配测试场景：

module Cryption
  module Crypt
    class << self
      # 提供常量重载入口
      def reload_config!
        const_set(:OFF_ENCRYPT, ENV['IS_OFF'])
      end

      def encrypt
        OFF_ENCRYPT
      end

      def decrypt
        OFF_ENCRYPT
      end
    end
  end
end

测试时在设置完环境变量后调用Cryption::Crypt.reload_config!即可刷新常量值，不需要改动spec_helper的全局加载逻辑。

方案2：无Gem修改权限时的测试适配 #

如果无法修改Gem源码，可通过RSpec按用例重载模块的方式实现，核心逻辑是每个需要不同环境变量的测试场景，在设置完ENV值后重新加载对应模块，覆盖之前已赋值的常量：

1. 首先修改spec_helper.rb，移除全局的require "cryption"语句，仅保留RSpec基础依赖：

   require "bundler/setup"
   # 保留原有RSpec基础配置即可
   

2. 调整测试用例，在每个场景的前置钩子中先设置环境变量，再加载模块，用例执行完成后卸载已加载的模块避免污染其他用例：
   • crypt_off_spec.rb

   RSpec.describe "IS_OFF=YES场景" do
     before(:context) do
       ENV['IS_OFF'] = "YES"
       require "cryption"
     end
   
     after(:context) do
       # 卸载已加载的Cryption模块
       Object.send(:remove_const, :Cryption)
       $LOADED_FEATURES.delete_if { |path| path.include?('cryption') }
     end
   
     it "encrypt返回YES，加密关闭" do
        expect(Cryption::Crypt.encrypt).to eq("YES")
     end
   end
   

   • crypt_on_spec.rb

   RSpec.describe "IS_OFF=NO场景" do
     before(:context) do
       ENV['IS_OFF'] = "NO"
       require "cryption"
     end
   
     after(:context) do
       Object.send(:remove_const, :Cryption)
       $LOADED_FEATURES.delete_if { |path| path.include?('cryption') }
     end
   
     it "encrypt返回NO，加密开启" do
        expect(Cryption::Crypt.encrypt).to eq("NO")
     end
   end
   

注意：如果Gem的依赖链路较复杂，模块重载可能出现依赖冲突，优先选择方案1调整代码逻辑稳定性更高。

内容的提问来源于stack exchange，提问作者itx