test/gzip_writer_test.rb in iostreams-0.14.0 vs test/gzip_writer_test.rb in iostreams-0.15.0

@@ -1,39 +1,45 @@
 require_relative 'test_helper'
 
-# Unit Test for IOStreams::Gzip
-module Streams
-  class GzipWriterTest < Minitest::Test
-    describe IOStreams::Gzip::Writer do
-      before do
-        @file_name = File.join(File.dirname(__FILE__), 'files', 'text.txt.gz')
-        @data      = Zlib::GzipReader.open(@file_name) { |gz| gz.read }
-      end
+class GzipWriterTest < Minitest::Test
+  describe IOStreams::Gzip::Writer do
+    let :temp_file do
+      Tempfile.new('iostreams')
+    end
 
-      describe '.open' do
-        it 'file' do
-          temp_file = Tempfile.new('rocket_job')
-          file_name = temp_file.to_path
-          IOStreams::Gzip::Writer.open(file_name) do |io|
-            io.write(@data)
-          end
-          result = Zlib::GzipReader.open(file_name) { |gz| gz.read }
-          temp_file.delete
-          assert_equal @data, result
-        end
+    let :file_name do
+      temp_file.path
+    end
 
-        it 'stream' do
-          io_string = StringIO.new(''.force_encoding('ASCII-8BIT'))
-          IOStreams::Gzip::Writer.open(io_string) do |io|
-            io.write(@data)
-          end
-          io   = StringIO.new(io_string.string)
-          gz   = Zlib::GzipReader.new(io)
-          data = gz.read
-          gz.close
-          assert_equal @data, data
+    let :decompressed do
+      File.read(File.join(File.dirname(__FILE__), 'files', 'text.txt'))
+    end
+
+    after do
+      temp_file.delete
+    end
+
+    describe '.open' do
+      it 'file' do
+        IOStreams::Gzip::Writer.open(file_name) do |io|
+          io.write(decompressed)
         end
+        result = Zlib::GzipReader.open(file_name) { |gz| gz.read }
+        temp_file.delete
+        assert_equal decompressed, result
       end
 
+      it 'stream' do
+        io_string = StringIO.new(''.b)
+        IOStreams::Gzip::Writer.open(io_string) do |io|
+          io.write(decompressed)
+        end
+        io   = StringIO.new(io_string.string)
+        gz   = Zlib::GzipReader.new(io)
+        data = gz.read
+        gz.close
+        assert_equal decompressed, data
+      end
     end
+
   end
 end