3 files changed, 88 insertions, 0 deletions
diff --git a/spec/ruby/library/zlib/gzipwriter/append_spec.rb b/spec/ruby/library/zlib/gzipwriter/append_spec.rb
new file mode 100644
index 0000000000..ef9e3d3a6b
--- /dev/null
+++ b/spec/ruby/library/zlib/gzipwriter/append_spec.rb
@@ -0,0 +1,15 @@
+require_relative '../../../spec_helper'
+require 'stringio'
+require 'zlib'
+
+describe "Zlib::GzipWriter#<<" do
+  before :each do
+    @io = StringIO.new
+  end
+
+  it "returns self" do
+    Zlib::GzipWriter.wrap @io do |gzio|
+      (gzio << "test").should.equal?(gzio)
+    end
+  end
+end
diff --git a/spec/ruby/library/zlib/gzipwriter/mtime_spec.rb b/spec/ruby/library/zlib/gzipwriter/mtime_spec.rb
new file mode 100644
index 0000000000..a70fa68069
--- /dev/null
+++ b/spec/ruby/library/zlib/gzipwriter/mtime_spec.rb
@@ -0,0 +1,37 @@
+require_relative '../../../spec_helper'
+require 'stringio'
+require 'zlib'
+
+describe "Zlib::GzipWriter#mtime=" do
+  before :each do
+    @io = StringIO.new
+  end
+
+  it "sets mtime using Integer" do
+    Zlib::GzipWriter.wrap @io do |gzio|
+      gzio.mtime = 1
+
+      gzio.mtime.should == Time.at(1)
+    end
+
+    @io.string[4, 4].should == [1,0,0,0].pack('C*')
+  end
+
+  it "sets mtime using Time" do
+    Zlib::GzipWriter.wrap @io do |gzio|
+      gzio.mtime = Time.at 1
+
+      gzio.mtime.should == Time.at(1)
+    end
+
+    @io.string[4, 4].should == [1,0,0,0].pack('C*')
+  end
+
+  it "raises if the header was written" do
+    Zlib::GzipWriter.wrap @io do |gzio|
+      gzio.write ''
+
+      -> { gzio.mtime = nil }.should.raise(Zlib::GzipFile::Error, 'header is already written')
+    end
+  end
+end
diff --git a/spec/ruby/library/zlib/gzipwriter/write_spec.rb b/spec/ruby/library/zlib/gzipwriter/write_spec.rb
new file mode 100644
index 0000000000..522ae7f2aa
--- /dev/null
+++ b/spec/ruby/library/zlib/gzipwriter/write_spec.rb
@@ -0,0 +1,36 @@
+require_relative '../../../spec_helper'
+require 'stringio'
+require 'zlib'
+
+describe "Zlib::GzipWriter#write" do
+  before :each do
+    @data = '12345abcde'
+    @zip = [31, 139, 8, 0, 44, 220, 209, 71, 0, 3, 51, 52, 50, 54, 49, 77,
+            76, 74, 78, 73, 5, 0, 157, 5, 0, 36, 10, 0, 0, 0].pack('C*')
+    @io = StringIO.new "".b
+  end
+
+  it "writes some compressed data" do
+    Zlib::GzipWriter.wrap @io do |gzio|
+      gzio.write @data
+    end
+
+    # skip gzip header for now
+    @io.string.unpack('C*')[10..-1].should == @zip.unpack('C*')[10..-1]
+  end
+
+  it "returns the number of bytes in the input" do
+    Zlib::GzipWriter.wrap @io do |gzio|
+      gzio.write(@data).should == @data.size
+    end
+  end
+
+  it "handles inputs of 2^23 bytes" do
+    input = '.'.b * (2 ** 23)
+
+    Zlib::GzipWriter.wrap @io do |gzio|
+      gzio.write input
+    end
+    @io.string.size.should == 8176
+  end
+end