Group seeds by top level model + plus data tables

lib/oaken.rb

@@ -76,7 +76,7 @@ def update(id, **attributes)
       attributes = super
 
       if record = @type.find_by(id: identify(id))
-        record.update!(**attributes)
+        record.tap { _1.update!(**attributes) }
       else
         @type.create!(id: identify(id), **attributes)
       end

test/oaken_test.rb

@@ -29,36 +29,25 @@ def test_accessing_fixture
     assert_equal [users.kasper, users.coworker], accounts.business.users
   end
 
-  def test_default_attributes_override
-    assert_equal "Big Business Co.", accounts.business.name
-  end
-
-  def test_default_attributes_last_into_test
-    users.update :homer
-    assert_equal [accounts.business], users.homer.accounts
-  end
-
-  def test_default_attributes_block
-    users.with accounts: [accounts.update(:home_co)] do
+  def test_default_attributes
+    users.with name: -> { id.to_s.humanize }, accounts: [accounts.update(:home_co, name: "Home Co.")] do
       users.update :homer
     end
+    assert_equal "Homer", users.homer.name
     assert_equal [accounts.home_co], users.homer.accounts
-
-    users.update :homer
-    assert_equal [accounts.business], users.homer.accounts
   end
 
   def test_updating_fixture
-    users.update :kasper, name: "Kasper2"
-    assert_equal "Kasper2", users.kasper.name
+    kasper = users.update :kasper, name: "Kasper2"
+    assert_equal "Kasper2", kasper.name
   end
 
   def test_upserting_vs_updating
-    assert_equal "Nice!", comments.praise.title
+    assert_equal "Basic", plans.basic.title
 
     error = assert_raises RuntimeError do
-      comments.update :salty, title: "foo"
+      plans.update :salty, title: "foo", price_cents: 0
     end
-    assert_equal "after_create", error.message
+    assert_equal "after_save", error.message
   end
 end

test/seeds.rb

@@ -1,8 +1 @@
-accounts.with name: -> { id.to_s.humanize }
-accounts.update :business, name: "Big Business Co."
 
-users.with name: -> { id.to_s.capitalize }, accounts: [accounts.business]
-users.update :kasper
-users.update :coworker
-
-comments.upsert :praise, title: "Nice!"

test/seeds/accounts/business.rb

@@ -0,0 +1,4 @@
+business = accounts.update :business, name: "Big Business Co."
+
+users.update :kasper,   name: "Kasper",   accounts: [business]
+users.update :coworker, name: "Coworker", accounts: [business]

test/seeds/data/plans.rb

@@ -0,0 +1 @@
+plans.upsert :basic, title: "Basic", price_cents: 1000

test/test_helper.rb

@@ -26,8 +26,9 @@
     t.timestamps
   end
 
-  create_table :comments, force: true do |t|
+  create_table :plans, force: true do |t|
     t.string :title, null: false
+    t.integer :price_cents, null: false
     t.timestamps
   end
 
@@ -52,8 +53,8 @@ class User < ActiveRecord::Base
   has_many :accounts, through: :memberships
 end
 
-class Comment < ActiveRecord::Base
-  after_create { raise "after_create" }
+class Plan < ActiveRecord::Base
+  after_save { raise "after_save" }
 end
 
 class YamlRecord < ActiveRecord::Base