From d9a44e5482858d214da39c0c11fd0fbe1269153b Mon Sep 17 00:00:00 2001
From: Jeff Hodges <jeff@somethingsimilar.com>
Date: Sat, 30 May 2009 11:30:46 -0700
Subject: [PATCH] Dir.chdir support (but not without a block :()

---
 lib/fakefs.rb       | 13 +++++++++++++
 test/fakefs_test.rb | 43 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 56 insertions(+)

diff --git a/lib/fakefs.rb b/lib/fakefs.rb
index ae5d562..1792768 100644
--- a/lib/fakefs.rb
+++ b/lib/fakefs.rb
@@ -143,6 +143,10 @@ module FakeFS
     def self.[](pattern)
       glob(pattern)
     end
+
+    def self.chdir(dir, &blk)
+      FileSystem.chdir(dir, &blk)
+    end
   end
 
   module FileSystem
@@ -214,6 +218,15 @@ module FakeFS
       end
     end
 
+    def chdir(dir, &blk)
+      raise "you must pass in a block" unless blk
+      @old_fs = @fs
+      @fs = find(dir)
+      blk.call
+    ensure
+      @fs = @old_fs
+    end
+
     def path_parts(path)
       path.split(File::PATH_SEPARATOR).reject { |part| part.empty? }
     end
diff --git a/test/fakefs_test.rb b/test/fakefs_test.rb
index a298ade..af134bf 100644
--- a/test/fakefs_test.rb
+++ b/test/fakefs_test.rb
@@ -145,4 +145,47 @@ class FakeFSTest < Test::Unit::TestCase
     # Unsupported so far. More hackery than I want to work on right now
     # assert_equal ['/path'], Dir['/path*']
   end
+
+  def test_chdir_changes_directories_like_a_boss
+    # I know memes!
+    FileUtils.mkdir_p '/path'
+    assert_equal '.', FileSystem.fs.name
+    assert_equal({}, FileSystem.fs['path'])
+    Dir.chdir '/path' do
+      File.open('foo', 'w'){|f| f.write 'foo'}
+      File.open('foobar', 'w'){|f| f.write 'foo'}
+    end
+
+    assert_equal '.', FileSystem.fs.name
+    assert_equal(['foo', 'foobar'], FileSystem.fs['path'].keys.sort)
+
+    c = nil
+    Dir.chdir '/path' do
+      c = File.open('foo', 'r'){|f| f.read }
+    end
+
+    assert_equal 'foo', c
+
+    assert_raise(RuntimeError, 'no block chdir is not yet supported') do
+      Dir.chdir('/path')
+    end
+  end
+
+  def test_chdir_shouldnt_lose_state_because_of_errors
+    FileUtils.mkdir_p '/path'
+
+    Dir.chdir '/path' do
+      File.open('foo', 'w'){|f| f.write 'foo'}
+      File.open('foobar', 'w'){|f| f.write 'foo'}
+    end
+
+    begin
+      Dir.chdir('/path') do
+        raise Exception
+      end
+    rescue Exception # hardcore
+    end
+
+    assert_equal(['foo', 'foobar'], FileSystem.fs['path'].keys.sort)
+  end
 end