如何在Rails 5测试环境中读取开发数据库中的密钥？

嘿，手动复制密钥这事确实够折腾的！在Rails 5里，咱们可以通过两种简单的方式让测试会话启动时自动从开发数据库拉取最新密钥，彻底告别繁琐的手动操作：

Rails启动时会自动加载config/initializers目录下的脚本，咱们可以在这里写逻辑，让测试环境启动时主动从开发数据库取密钥，覆盖应用配置：

1. 创建初始化脚本
   在config/initializers下新建load_external_api_keys.rb文件，写入以下代码：

# 仅在测试环境执行该逻辑，避免影响其他环境
if Rails.env.test?
  # 连接到开发数据库
  development_db_config = Rails.application.config.database_configuration['development']
  ActiveRecord::Base.establish_connection(development_db_config)
  
  # 替换成你实际的密钥查询逻辑，比如取最新生成的密钥记录
  # 假设你用ApiKey模型存储密钥对
  latest_key_pair = ApiKey.order(created_at: :desc).first
  
  # 将密钥写入应用配置，方便后续调用
  Rails.application.config.external_api_access_key = latest_key_pair.access_key
  Rails.application.config.external_api_refresh_key = latest_key_pair.refresh_key
  
  # 切回测试数据库连接，避免影响测试执行
  ActiveRecord::Base.establish_connection(:test)
end

2. 在测试中调用配置
   之后在测试代码或外部API调用处，直接用Rails.application.config.external_api_access_key和Rails.application.config.external_api_refresh_key即可，无需再依赖application.yml里的硬编码值。

如果不想用初始化脚本，也可以在测试框架的全局前置钩子中加载密钥，以RSpec为例：

在spec/rails_helper.rb（或spec/spec_helper.rb）中添加如下配置：

RSpec.configure do |config|
  # 在整个测试套件启动前执行一次密钥加载逻辑
  config.before(:suite) do
    # 连接开发数据库
    development_db_config = Rails.application.config.database_configuration['development']
    ActiveRecord::Base.establish_connection(development_db_config)
    
    # 获取最新密钥对
    latest_key_pair = ApiKey.order(created_at: :desc).first
    
    # 将密钥存入环境变量，兼容现有代码中用ENV读取的逻辑
    ENV['EXTERNAL_API_ACCESS_KEY'] = latest_key_pair.access_key
    ENV['EXTERNAL_API_REFRESH_KEY'] = latest_key_pair.refresh_key
    
    # 切回测试数据库
    ActiveRecord::Base.establish_connection(:test)
  end
end

这样测试启动时会自动拉取一次最新密钥，存到环境变量里，现有代码如果是用ENV['EXTERNAL_API_ACCESS_KEY']读取的话，完全不用修改就能直接用。

• 确保测试环境的数据库schema和开发环境同步，要是测试库没有对应的密钥表，记得运行rails db:test:prepare同步结构。
• 如果你的密钥不是存在模型里，而是直接存在某个普通表中，把查询逻辑换成原生SQL即可，比如：

  key_data = ActiveRecord::Base.connection.execute("SELECT access_key, refresh_key FROM external_api_keys ORDER BY updated_at DESC LIMIT 1").first
  

• 一定要确保这些逻辑只在测试环境执行！上面的代码都加了环境判断或放在测试配置里，不会影响生产环境的运行。

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