1

我有一个模型，它使用attr_encryptedgem 来加密密码。 

class Credential < ApplicationRecord
  validates :user_name, presence: true
  enum credential_type: { windows: 1, linux: 2 }

  attr_encrypted :user_pass, key: :encryption_key

  def encryption_key
    # Some complex logic
  end
end

我正在学习编写测试用例，我的工厂如下所示：

FactoryBot.define do
  factory :credential do
    user_name { "rmishra" }
    user_pass { "secret" }
    credential_type { "linux" }
    encryption_key { "abcdefghijklmnopqrstuvw123456789" }
  end
end

我的规范文件如下所示：

RSpec.describe Credential, type: :model do
  let(:credential) { create(:credential) }
  ...
end

如何encryption_key在工厂定义中存根方法，当时正在使用create？

4

1 回答 1

1

由于encryption_key不是模型上的属性，因此您无法在工厂中对其进行配置。

encryption_key当您分配给对象时， attr_encryptedgem会自动调用。在这种情况下，这是由工厂完成的。user_passCredential

我会将您encryption_key方法中的逻辑移动到一个类中以方便测试：

class Credential < ApplicationRecord
  validates :user_name, presence: true
  enum credential_type: { windows: 1, linux: 2 }

  attr_encrypted :user_pass, key: :encryption_key

  def encryption_key
    EncryptionKeyGenerator.generate # or whatever name makes more sense
  end
end

然后，在我的测试中，我会存根EncryptionKeyGenerator：

RSpec.describe Credential, type: :model do
  let(:credential) { create(:credential) }
  let(:encryption_key) { "abcdefghijklmnopqrstuvw123456789" }

  before do
    allow(EncryptionKeyGenerator).to receive(:generate).and_return(encryption_key)
  end

  ...
end

将您的加密密钥生成逻辑封装到一个单独的对象中，可以将其与您的模型分离，让您无需创建Credential对象即可轻松测试该逻辑。

于 2019-03-31T01:04:54.360 回答