active_record_shards 2.4.1 → 2.4.2

data/lib/active_record_shards/migration.rb

@@ -44,7 +44,8 @@ module ActiveRecordShards
 
     def migrate_with_forced_shard(direction)
       if migration_shard.blank?
-        raise "Can't run migrations with a shard spec: this may be :all, :none, or a specific shard (for data-fixups) please call shard(arg) in your migration"
+        raise "#{self.name}: Can't run migrations without a shard spec: this may be :all, :none,
+                 or a specific shard (for data-fixups).  please call shard(arg) in your migration."
       end
 
       shard = ActiveRecord::Base.current_shard_selection.shard

data/test/migrator_test.rb

@@ -58,6 +58,12 @@ class MigratorTest < ActiveSupport::TestCase
   end
 
   def test_bad_migration
+    begin
+      CowardlyMigration.up
+    rescue Exception => e
+      assert_match /CowardlyMigration/, e.message
+    end
+
     assert_raises(RuntimeError) do
       CowardlyMigration.up
     end