+ tempdir_for

lib/toolshack.rb

@@ -5,6 +5,10 @@
 # place and that I just don't want to write again. 
 #
 module Toolshack
+  def self.included(by)
+    by.extend(ClassMethods)
+  end
+
   # Creates a temporary directory and returns its path to the caller. Note
   # that this can also be used in block style: 
   #
@@ -37,26 +41,27 @@ def temporary_directory(base)
   end
   module_function :temporary_directory
 
-  # Use this in rspec to create and manage a tempdir at the same time. Roughly
-  # aequivalent to this: 
-  #
-  #   attr_reader :tempdir
-  #   around(scope) { |example|
-  #     temporary_directory do |path|
-  #       @tempdir = path
-  #       example.run
-  #     end
-  #   end
-  # 
-  def tempdir_for(scope, name=:tempdir)
-    attr_reader name
+  module ClassMethods
+    # Use this in rspec to create and manage a tempdir at the same time. Roughly
+    # aequivalent to this: 
+    #
+    #   attr_reader :tempdir
+    #   around(scope) { |example|
+    #     temporary_directory do |path|
+    #       @tempdir = path
+    #       example.run
+    #     end
+    #   end
+    # 
+    def tempdir_for(scope, base, name=:tempdir)
+      attr_reader name
 
-    around(scope) { |example|
-      temporary_directory do |path|
-        instance_variable_set("@#{name}", path)
-        example.run
-      end
-    }
+      around(scope) { |example|
+        temporary_directory(base) do |path|
+          instance_variable_set("@#{name}", path)
+          example.run
+        end
+      }
+    end
   end
-  module_function :tempdir_for
 end

spec/unit/toolshack_spec.rb

@@ -28,4 +28,32 @@ def call
       call.should_not == call
     end  
   end
+  describe "<- .tempdir_for" do
+    tempdir_for(:each, 'tempdir_for')
+    subject { tempdir }
+
+    it { should_not be_nil }
+    it "should be a directory" do
+      File.directory?(tempdir).should == true
+    end 
+
+    2.times do |i|
+      # Create and clean a tempdir twice. There should be no junk left lying 
+      # around.
+      let(:tempdirs) { [] }
+
+      context "when creating more than one temporary directory (#{i+1})" do
+        tempdir_for(:each, 'tempdir_for')
+
+        before(:each) { tempdirs << tempdir }
+
+        it "should have at most one directory alive" do
+          (tempdirs - Array(tempdir)).each do |path|
+            File.exist?(path).should == false
+          end
+        end 
+      end
+    end
+
+  end
 end