0

FactoryGirl を使用してデータベースにシード データを入力する rake タスクがあります。例えば ​​-

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

私は新しいモデルを持っています -Evaluationsbelongs_toの 2 つのモデル:StudentsGoals.

現時点で私はこれを持っています:

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親で同じことをしましたが、問題は逆です-では、2つの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時間実行してまだ実行していたため、機能するかどうかをテストできませんでした.

更新: 長時間実行されていた理由は、生成されたユーザーの数を変更するのを怠ったためです。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
4

1 に答える 1