test/generators/test_mailer_generator.rb in minitest-rails-0.1.2 vs test/generators/test_mailer_generator.rb in minitest-rails-0.1.3

@@ -4,36 +4,45 @@
 require "rails"
 require "rails/generators"
 
 require "generators/mini_test/mailer/mailer_generator"
 
-require "fileutils"
+require "fakefs/safe"
 
+class FakeFS::File
+  def self.binread file
+    File.open(file, "rb") { |f| f.read }
+  end
+end
+
 class TestMailerGenerator < MiniTest::Unit::TestCase
-  Rails::Generators.no_color!
+  def setup
+    Rails::Generators.no_color!
+    FakeFS.activate!
+    FakeFS::FileSystem.clone "lib/generators"
+  end
 
+  def teardown
+    FakeFS::FileSystem.clear
+    FakeFS.deactivate!
+  end
+
   def test_mailer_generator
-    text = capture(:stdout) do
+    out, err = capture_io do
       MiniTest::Generators::MailerGenerator.start ["notification"]
     end
-    assert_match(/create  test\/mailers\/notification_test.rb/m, text)
+    assert_match(/create  test\/mailers\/notification_test.rb/m, out)
     assert File.exists? "test/mailers/notification_test.rb"
-    contents = open("test/mailers/notification_test.rb").read
+    contents = File.read "test/mailers/notification_test.rb"
     assert_match(/class NotificationTest/m, contents)
-  ensure
-    # TODO: Don"t write the files
-    # I agree, it would be better to mock the file getting written
-    FileUtils.rm_r "test/mailers"
   end
 
   def test_mailer_generator_spec
-    text = capture(:stdout) do
+    out, err = capture_io do
       MiniTest::Generators::MailerGenerator.start ["notification", "welcome", "--spec"]
     end
-    assert_match(/create  test\/mailers\/notification_test.rb/m, text)
+    assert_match(/create  test\/mailers\/notification_test.rb/m, out)
     assert File.exists? "test/mailers/notification_test.rb"
-    contents = open("test/mailers/notification_test.rb").read
+    contents = File.read "test/mailers/notification_test.rb"
     assert_match(/describe Notification do/m, contents)
-  ensure
-    FileUtils.rm_r "test/mailers"
   end
 end