spec/spec_helper.rb in shenandoah-0.0.0 vs spec/spec_helper.rb in shenandoah-0.1.0

@@ -9,26 +9,51 @@
 end
 
 module Shenandoah
   module Spec
     module Tmpfile
-      attr_accessor :tmpdir
-
+      attr_writer :tmpdir
+      
       def tmpfile(name, contents="contents not important")
         n = "#{tmpdir}/#{name}"
         FileUtils.mkdir_p File.dirname(n)
         File.open(n, 'w') { |f| f.write contents }
         n
       end
+      
+      def tmpdir(name=nil)
+        n = @tmpdir
+        if (name)
+          n = File.join(n, name)
+          FileUtils.mkdir_p(n)
+        end
+        n
+      end
 
       def self.included(klass)
         klass.class_eval do
           before do
             FileUtils.mkdir_p(self.tmpdir = File.dirname(__FILE__) + "/tmp")
           end
 
           after do
             FileUtils.rm_r self.tmpdir
+          end
+        end
+      end
+    end
+
+    module RailsRoot
+      def self.included(klass)
+        klass.class_eval do
+          include Shenandoah::Spec::Tmpfile
+
+          before do
+            Object.const_set(:RAILS_ROOT, tmpdir('rails-root'))
+          end
+
+          after do
+            Object.instance_eval { remove_const :RAILS_ROOT }
           end
         end
       end
     end
   end
\ No newline at end of file