Question

我有一个rake任务，使用FactoryGirl使用种子数据填充我的数据库。例如 -

def create_goals
  puts "Creating goals for each subject"
  Subject.all.each do |s|
    FactoryGirl.create_list :goal, 3, subject_id: s.id
  end
end

我有一个新模型 - Evaluations - belongs_to另外两个模型：Students和Goals。

目前我有这个：

def create_evaluations
  puts "Creating evaluation data for each goal (have you put the kettle on yet?)"
  Goal.all.each do |g|
    FactoryGirl.create_list :evaluation, 3, goal_id: g.id
  end    
end

但这显然只会提供具有goal_id的数据 - 我与Student父项做了同样的事情，但问题是相反的 - 所以如何为模型生成数据两个belongs_to个关联？

更新：

我试过了：

def create_evaluations
  puts "Creating evaluation data for each student and goal"
  Student.all.each do |s|
    Goal.all.each do |g|
      FactoryGirl.create_list :evaluation, 3, student_id: s.id, goal_id: g.id
    end
  end     
end

但未能测试它是否有效，因为我让它运行约1小时并且仍在运行。

UPDATE ：它运行这么久的原因是因为我忽略了改变它生成的用户数量。我把它改成只有2个用户，并且运行得足够快。这是我最终使用的代码（感谢Pierre-Louis Gottfrois的帮助！）：

require 'factory_girl_rails'

namespace :db do

  desc "Fill db with sample data"

  task :populate, :environment do

    # warning message
    puts "##############################################################"
    puts "# if this takes too long, reduce the number of created users #"
    puts "##############################################################"

    # reset the database
    reset_db

    # some messages
    puts "Database reset. Database population started"

    # create users
    create_users

    #create student_groups for each user  
    create_student_groups

    #create students for each student_group
    create_students

    #create subjects for each student_group
    create_subjects

    #create goals for each subject
    create_goals

    #create evaluation data for each goal
    create_evaluations

    # success message
    puts "The database has been populated successfully"

  end

###################
####  METHODS  ####
###################

  def reset_db
    puts "Resetting the database"
    Rake::Task['db:reset'].invoke
  end

  def create_users
    puts "Creating users"
    FactoryGirl.create_list :user, 2
  end

  def create_student_groups
    puts "Creating student groups for each user"
    User.all.each do |u|
      FactoryGirl.create_list :student_group, 3, user_id: u.id
    end
  end

  def create_students
    puts "Creating students for each student group"
    StudentGroup.all.each do |sg|
      FactoryGirl.create_list :student, 7, student_group_id: sg.id
    end
  end

  def create_subjects
    puts "Creating subjects for each student group"
    StudentGroup.all.each do |sg|
      FactoryGirl.create_list :subject, 4, student_group_id: sg.id
    end
  end  

  def create_goals
    puts "Creating goals for each subject"
    Subject.all.each do |s|
      FactoryGirl.create_list :goal, 3, subject_id: s.id
    end
  end  

  def create_evaluations
    puts "Creating evaluation data"
    Student.all.each_with_index do |s, index|
      @goals = Goal.all  
      @goals.count.times do |i|
        FactoryGirl.create_list :evaluation, 3, goal_id: @goals[i].id, student_id: s.id
      end
    # Counter to track progress
    puts "Created data for 'Student #{index}'"  
    end
  end

end

Answer 1

不确定在此处了解您的问题。这有帮助吗？

@students = Student.all

@students.count.times do |i|
  FactoryGirl.create_list :evaluation, 3, goal_id: g.id, student_id: @students[i].id
end

当模型belongs_to两个父项时，使用FactoryGirl填充数据库

1 个答案: