lib/ammeter/rspec/generator/matchers/be_a_migration.rb in ammeter-0.2.3 vs lib/ammeter/rspec/generator/matchers/be_a_migration.rb in ammeter-0.2.4

@@ -1,7 +1,11 @@
 RSpec::Matchers.define :be_a_migration do
   match do |file_path|
-    dirname, file_name = File.dirname(file_path), File.basename(file_path).sub(/\.rb$/, '')
-    migration_file_path = Dir.glob("#{dirname}/[0-9]*_*.rb").grep(/\d+_#{file_name}.rb$/).first
+    dirname, file_name = File.dirname(file_path), File.basename(file_path)
+    if file_name.match(/\d+_.*\.rb/)
+      migration_file_path = file_path
+    else
+      migration_file_path = Dir.glob("#{dirname}/[0-9]*_*.rb").grep(/\d+_#{file_name}$/).first
+    end
     File.exist?(migration_file_path)
   end
 end