test/test_output_file.rb in sstephenson-sprockets-0.1.1 vs test/test_output_file.rb in sstephenson-sprockets-0.3.0

@@ -1,20 +1,28 @@
 require "test_helper"
 
 class OutputFileTest < Test::Unit::TestCase
   def setup
     @output_file = Sprockets::OutputFile.new
+    @environment = environment_for_fixtures
   end
   
   def test_record
     assert_equal [], @output_file.source_lines
-    assert_equal "hello\n", @output_file.record("hello\n")
-    assert_equal "world\n", @output_file.record("world\n")
-    assert_equal ["hello\n", "world\n"], @output_file.source_lines
+    assert_equal "hello\n", @output_file.record(source_line("hello\n")).to_s
+    assert_equal "world\n", @output_file.record(source_line("world\n")).to_s
+    assert_equal ["hello\n", "world\n"], @output_file.source_lines.map { |source_line| source_line.to_s }
   end
   
   def test_to_s
     @output_file.record(source_line("hello\n"))
     @output_file.record(source_line("world\n"))
     assert_equal "hello\nworld\n", @output_file.to_s
+  end
+  
+  def test_save_to
+    filename = File.join(FIXTURES_PATH, "output.js")
+    @output_file.save_to(filename)
+    assert_equal @output_file.to_s, IO.read(filename)
+    File.unlink(filename)
   end
 end