fix more spec warnings

rs-pro · Nov 17, 2013 · a907025 · a907025
1 parent 842d948
commit a907025
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/spec/integration/integration_spec.rb b/spec/integration/integration_spec.rb
@@ -2,10 +2,6 @@
 
 describe Mongoid::Audit do
   before :all do
-    class HistoryTracker
-      include Mongoid::Audit::Tracker
-    end
-
     class Post
       include Mongoid::Document
       include Mongoid::Timestamps

diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
@@ -23,3 +23,6 @@ def self.add_extension(*args); end
 end
 
 require 'mongoid-audit'
+class HistoryTracker
+  include Mongoid::Audit::Tracker
+end
diff --git a/spec/trackable_spec.rb b/spec/trackable_spec.rb
@@ -29,6 +29,8 @@ def compare( option )
 
   describe "#track_history" do
     before :each do
+      Object.send(:remove_const, :MyModel) if Object.constants.include?(:MyModel)
+
       class MyModel
         include Mongoid::Document
         include Mongoid::Audit::Trackable
@@ -73,6 +75,14 @@ class MyModel
 
     context "sub-model" do
       before :each do
+        Object.send(:remove_const, :MyModel) if Object.constants.include?(:MyModel)
+        Object.send(:remove_const, :MySubModel) if Object.constants.include?(:MySubModel)
+
+        class MyModel
+          include Mongoid::Document
+          include Mongoid::Audit::Trackable
+          track_history
+        end
         class MySubModel < MyModel
         end
       end