Only re-define BSON::ObjectId's <=> operator for mongoid2.

Joseph Chen
1 parent 2556919e
Showing 1 changed file with 1 additions and 1 deletions Show diff stats
test/test_helper.rb
@@ -19,7 +19,7 @@ if defined?(Mongoid)
   end
  
   # Mongoid2 uses BSON gem. We need to re-define <=> in order for TestSql.test_order_id to pass.
-  if defined?(BSON) && defined?(BSON::ObjectId)
+  if Mongoid.mongoid2? && defined?(BSON) && defined?(BSON::ObjectId)
     module BSON
       class ObjectId
         def <=>(other)
...	...	@@ -19,7 +19,7 @@ if defined?(Mongoid)
19	19	end
20	20
21	21	# Mongoid2 uses BSON gem. We need to re-define <=> in order for TestSql.test_order_id to pass.
22		- if defined?(BSON) && defined?(BSON::ObjectId)
	22	+ if Mongoid.mongoid2? && defined?(BSON) && defined?(BSON::ObjectId)
23	23	module BSON
24	24	class ObjectId
25	25	def <=>(other)
...	...