Rspec2 support

midu · Apr 30, 2015 · 108fe47 · 108fe47
1 parent 3fbda79
commit 108fe47
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/spec/support/matchers/have_column.rb b/spec/support/matchers/have_column.rb
@@ -8,7 +8,13 @@
     column && column.default.to_s == @default.to_s
   end
 
-  failure_message do |columns|
-    "expected to find '#{column_name}', default '#{@default}' in #{columns.map{|column| [column.name, column.default] }}"
+  if RSpec::Version::STRING.to_i >= 3
+    failure_message do |columns|
+      "expected to find '#{column_name}', default '#{@default}' in #{columns.map{|column| [column.name, column.default] }}"
+    end
+  else
+    failure_message_for_should do |columns|
+      "expected to find '#{column_name}', default '#{@default}' in #{columns.map{|column| [column.name, column.default] }}"
+    end
   end
 end