Add FixtureBot.require for loading fixtures from multiple files

Author: bradgessler

README.md

@@ -346,6 +346,40 @@ end
 
 In Rails, this is auto-detected from the database.
 
+### Multiple files
+
+For larger apps, split fixtures across multiple files using `FixtureBot.require`:
+
+```ruby
+# spec/fixtures.rb
+FixtureBot.require "spec/fixtures/**/*.rb"
+
+FixtureBot.define do
+  user.email { |fixture| "#{fixture.key}@example.com" }
+end
+```
+
+```ruby
+# spec/fixtures/users.rb
+FixtureBot.define do
+  user :brad do
+    name "Brad"
+  end
+end
+```
+
+```ruby
+# spec/fixtures/posts.rb
+FixtureBot.define do
+  post :hello do
+    title "Hello"
+    author :brad
+  end
+end
+```
+
+Each file calls `FixtureBot.define` with its own block. Files are loaded in alphabetical order. References across files work because everything is resolved after all files are loaded.
+
 ### Implicit vs explicit style
 
 By default, the block is evaluated implicitly. Table methods like `user` and `post` are available directly:

lib/fixturebot.rb

@@ -29,6 +29,10 @@ def self.define(schema = nil, &block)
     end
   end
 
+  def self.require(glob)
+    Dir.glob(glob).sort.each { |f| load f }
+  end
+
   def self.define_from_file(schema, fixtures_path)
     definition = Definition.new(schema)
     Thread.current[:fixturebot_definition] = definition

lib/generators/fixturebot/templates/fixtures.rb.tt

@@ -1,3 +1,5 @@
+FixtureBot.require "<%= rspec? ? "spec" : "test" %>/fixtures/**/*.rb"
+
 FixtureBot.define do
   # Define your fixtures here. Example:
   #

spec/fixturebot_spec.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require "tmpdir"
+
 RSpec.describe FixtureBot do
   it "has a version number" do
     expect(FixtureBot::VERSION).not_to be nil
@@ -146,6 +148,52 @@
 
   end
 
+  describe ".require" do
+    let(:schema) do
+      FixtureBot::Schema.define do
+        table :users, singular: :user, columns: [:name, :email]
+        table :posts, singular: :post, columns: [:title, :author_id] do
+          belongs_to :author, table: :users
+        end
+      end
+    end
+
+    it "loads fixture files from a glob" do
+      Dir.mktmpdir do |dir|
+        subdir = File.join(dir, "fixtures")
+        Dir.mkdir(subdir)
+
+        File.write("#{subdir}/users.rb", <<~RUBY)
+          FixtureBot.define do
+            user :alice do
+              name "Alice"
+              email "alice@example.com"
+            end
+          end
+        RUBY
+
+        File.write("#{subdir}/posts.rb", <<~RUBY)
+          FixtureBot.define do
+            post :hello do
+              title "Hello"
+              author :alice
+            end
+          end
+        RUBY
+
+        File.write("#{dir}/fixtures.rb", <<~RUBY)
+          FixtureBot.require "#{subdir}/*.rb"
+        RUBY
+
+        result = FixtureBot.define_from_file(schema, "#{dir}/fixtures.rb")
+
+        expect(result.tables[:users][:alice][:name]).to eq("Alice")
+        expect(result.tables[:posts][:hello][:title]).to eq("Hello")
+        expect(result.tables[:posts][:hello][:author_id]).to eq(result.tables[:users][:alice][:id])
+      end
+    end
+  end
+
   describe "unknown method errors" do
     let(:schema) do
       FixtureBot::Schema.define do