FactoryGirl关于多态关联

时间:2014-12-05 16:25:22

标签: ruby rspec factory-bot

我需要找到构建模型对象的最佳方法。 我继承了代码库,它不是很漂亮。

class Notification < ActiveRecord::Base
   attr_accessible :author_id, :notifable_id, :notifable_type, :seen, :user_id
   belongs_to :notifiable, :polymorphic => true

因此author_id和user_id是用户模型的隐式指针

#A user 
has_many :notifications

多态关系通知允许具有例如可通知的类似模型。 如果你喜欢帖子,你可以创建一个类似的通知和一个有作者和接收者的通知(user_id)

我无法为有通知的用户创建工厂(即通知的接收方)

factory :notification do |f|
  n = FactoryGirl.generate(:number_notification)
  user = create(:user)
  #author_id
  f.author_id user.id
  f.user_id :user_dest
  f.association :user, user
end

factory :notification_like, parent: :notification  do |f|
  f.notifiable_type "Like"
  f.notifiable_id 1
end 

此代码不起作用。它抱怨user.id,说它不存在。

我的功能测试如下所示:

let(:max) { create(:user) }
let(:notification) { create(:notification_like, :user_dest => max.id) }
let(:login_page) { HomeLoginPage.new }
let(:main_page) { MainPage.new }

scenario "login" do
  notification
  login_page.visit_page.login(max)
  main_page
  sleep(60)
end

1 个答案:

答案 0 :(得分:0)

您需要使用callbacks来使您的关联有效。

factory :notification do |f|
  FactoryGirl.generate(:number_notification)
  user

  after(:build) do |notification|
    notification.author_id = notification.user.id
    notification.user_id = notification.user.id
  end
end