test/file_writer_test.rb in iostreams-0.20.2 vs test/file_writer_test.rb in iostreams-0.20.3

@@ -19,9 +19,24 @@
         end
         result = File.read(file_name)
         assert_equal raw, result
       end
 
+      it 'cleans up failed upload' do
+        temp_file = Tempfile.new('rocket_job')
+        file_name = temp_file.to_path
+        begin
+          IOStreams::File::Writer.open(file_name) do |io|
+            io.write("hello world")
+            assert File.exist?(file_name)
+            raise(ArgumentError, "Oh no")
+          end
+        rescue ArgumentError
+        end
+
+        refute File.exist?(file_name)
+      end
+
       it 'does not support streams' do
         io_string = StringIO.new
         assert_raises ArgumentError do
           IOStreams::File::Writer.open(io_string) do |io|
             io.write(raw)