Class: StarfleetSeeder

Inherits:
Object
  • Object
show all
Defined in:
lib/super/test_support/starfleet_seeder.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ StarfleetSeeder

Returns a new instance of StarfleetSeeder.



9
10
11
12
 
# File 'lib/super/test_support/starfleet_seeder.rb', line 9

def initialize
  @seeded = {}
  @association_reflections = {}
end

Class Method Details

.seed ⇒ Object 



2
3
4
5
6
7
 
# File 'lib/super/test_support/starfleet_seeder.rb', line 2

def self.seed
  seeder = StarfleetSeeder.new
  seeder.seed_from_fixture("ships", Ship)
  seeder.seed_from_fixture("members", Member)
  seeder.seed_from_fixture("favorite_things", FavoriteThing)
end

Instance Method Details

#seed(basename, key, klass, value) ⇒ Object 



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
 
# File 'lib/super/test_support/starfleet_seeder.rb', line 14

def seed(basename, key, klass, value)
  @seeded[basename] ||= {}
  puts "--> #{key} (#{value})"
  @association_reflections[klass] ||= klass.reflect_on_all_associations

  @association_reflections[klass].each do |ar|
    ar_name = ar.name.to_s
    ar_plural_name = ar.plural_name.to_s

    if value.key?(ar_name)
      value[ar_name] = @seeded.fetch(ar_plural_name).fetch(value[ar_name])
    end
  end

  @seeded[basename][key] = klass.find_or_create_by(value)
  puts "    success!"
rescue => e
  puts "    failure (#{e.message})"
end

#seed_from_fixture(basename, klass) ⇒ Object 



34
35
36
37
38
39
40
41
 
# File 'lib/super/test_support/starfleet_seeder.rb', line 34

def seed_from_fixture(basename, klass)
  puts "==> #{basename}"
  fixtures(basename).each do |key, value|
    seed(basename, key, klass, value)
  end

  nil
end