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
私は新しいモデルを持っています -Evaluations
他belongs_to
の 2 つのモデル: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
親で同じことをしましたが、問題は逆です-では、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