test: cover CliDbSeedTestWidget and CliDbSeedTest

test/test_cli_db_seed.rb

@@ -0,0 +1,106 @@
+require 'minitest/autorun'
+require 'prism'
+require 'stringio'
+require 'tmpdir'
+require 'fileutils'
+
+class CliDbSeedTestWidget < Prism::Orm::Model
+  table_name 'cli_db_seed_test_widgets'
+end
+
+class CliDbSeedTest < Minitest::Test
+  def setup
+    @workdir = Dir.mktmpdir('prism-cli-db-seed')
+    @cli = Prism::Cli.new(@workdir)
+    @database = File.join(@workdir, 'db', 'test.pstore')
+    FileUtils.mkdir_p(File.join(@workdir, 'db'))
+    Prism::Orm.connect(@database)
+    store.create_table('cli_db_seed_test_widgets', %w[id name])
+  end
+
+  def teardown
+    FileUtils.remove_entry(@workdir)
+    Object.send(:remove_const, :CliDbSeedTestRequiredApp) if Object.const_defined?(:CliDbSeedTestRequiredApp)
+  end
+
+  def store
+    Prism::Orm::Store.new(@database)
+  end
+
+  def write_seeds(content, filename = 'db/seeds.rb')
+    path = File.join(@workdir, filename)
+    FileUtils.mkdir_p(File.dirname(path))
+    File.write(path, content)
+    filename
+  end
+
+  def test_db_seed_runs_the_seeds_file_against_the_already_loaded_environment
+    write_seeds("CliDbSeedTestWidget.create!(name: 'sprocket')\n")
+
+    out = StringIO.new
+    result = @cli.db_seed([], out: out)
+
+    assert_equal true, result
+    assert_includes out.string, 'seeded from db/seeds.rb'
+    assert_equal ['sprocket'], store.all('cli_db_seed_test_widgets').map { |row| row['name'] }
+  end
+
+  def test_db_seed_with_no_seeds_file_reports_that_instead_of_erroring
+    out = StringIO.new
+    result = @cli.db_seed([], out: out)
+
+    assert_equal false, result
+    assert_includes out.string, 'no seeds file at db/seeds.rb'
+  end
+
+  def test_db_seed_dispatches_through_run
+    write_seeds("CliDbSeedTestWidget.create!(name: 'via-run')\n")
+
+    capture_io { @cli.run(['db:seed']) }
+
+    assert_equal ['via-run'], store.all('cli_db_seed_test_widgets').map { |row| row['name'] }
+  end
+
+  def test_db_seed_honors_a_custom_file_path
+    write_seeds("CliDbSeedTestWidget.create!(name: 'custom')\n", 'db/custom_seeds.rb')
+
+    out = StringIO.new
+    result = @cli.db_seed(['--file=db/custom_seeds.rb'], out: out)
+
+    assert_equal true, result
+    assert_includes out.string, 'seeded from db/custom_seeds.rb'
+    assert_equal ['custom'], store.all('cli_db_seed_test_widgets').map { |row| row['name'] }
+  end
+
+  def test_db_seed_requires_the_environment_before_running_the_seeds_file
+    File.write(File.join(@workdir, 'app.rb'), <<~RUBY)
+      class CliDbSeedTestRequiredApp
+        def self.slogan
+          'from --require'
+        end
+      end
+    RUBY
+    write_seeds("CliDbSeedTestWidget.create!(name: CliDbSeedTestRequiredApp.slogan)\n")
+
+    @cli.db_seed(['--require=app.rb'], out: StringIO.new)
+
+    assert_equal ['from --require'], store.all('cli_db_seed_test_widgets').map { |row| row['name'] }
+  end
+
+  # `load`, not `require`: a seeds file is a sequence of side effects, not a
+  # class/module definition with meaningful "already loaded" state, so a
+  # second db:seed run must re-execute it rather than silently no-op the way
+  # `require` would once that exact path had already been loaded once.
+  def test_db_seed_reruns_the_file_on_every_call_instead_of_requiring_it_once
+    write_seeds("CliDbSeedTestWidget.create!(name: 'again')\n")
+
+    @cli.db_seed([], out: StringIO.new)
+    @cli.db_seed([], out: StringIO.new)
+
+    assert_equal %w[again again], store.all('cli_db_seed_test_widgets').map { |row| row['name'] }
+  end
+
+  def test_db_seed_rejects_unknown_options
+    assert_raises(Prism::Cli::UsageError) { @cli.db_seed(['--bogus=1']) }
+  end
+end