1 files changed, 128 insertions, 0 deletions
diff --git a/spec/ruby/library/zlib/deflate/deflate_spec.rb b/spec/ruby/library/zlib/deflate/deflate_spec.rb
new file mode 100644
index 0000000000..44b3389701
--- /dev/null
+++ b/spec/ruby/library/zlib/deflate/deflate_spec.rb
@@ -0,0 +1,128 @@
+require 'zlib'
+require File.expand_path('../../../../spec_helper', __FILE__)
+
+describe "Zlib::Deflate.deflate" do
+  it "deflates some data" do
+    data = Array.new(10,0).pack('C*')
+
+    zipped = Zlib::Deflate.deflate data
+
+    zipped.should == [120, 156, 99, 96, 128, 1, 0, 0, 10, 0, 1].pack('C*')
+  end
+
+  it "deflates lots of data" do
+    data = "\000" * 32 * 1024
+
+    zipped = Zlib::Deflate.deflate data
+
+    zipped.should == ([120, 156, 237, 193, 1, 1, 0, 0] +
+                      [0, 128, 144, 254, 175, 238, 8, 10] +
+                      Array.new(31, 0) +
+                      [24, 128, 0, 0, 1]).pack('C*')
+  end
+
+  it "deflates chunked data" do
+    random_generator = Random.new(0)
+    deflated         = ''
+
+    Zlib.deflate(random_generator.bytes(20000)) do |chunk|
+      deflated << chunk
+    end
+
+    deflated.length.should == 20016
+  end
+end
+
+describe "Zlib::Deflate#deflate" do
+  before :each do
+    @deflator = Zlib::Deflate.new
+  end
+
+  it "deflates some data" do
+    data = "\000" * 10
+
+    zipped = @deflator.deflate data, Zlib::FINISH
+    @deflator.finish
+
+    zipped.should == [120, 156, 99, 96, 128, 1, 0, 0, 10, 0, 1].pack('C*')
+  end
+
+  it "deflates lots of data" do
+    data = "\000" * 32 * 1024
+
+    zipped = @deflator.deflate data, Zlib::FINISH
+    @deflator.finish
+
+    zipped.should == ([120, 156, 237, 193, 1, 1, 0, 0] +
+                      [0, 128, 144, 254, 175, 238, 8, 10] +
+                      Array.new(31, 0) +
+                      [24, 128, 0, 0, 1]).pack('C*')
+  end
+end
+
+describe "Zlib::Deflate#deflate" do
+
+  before :each do
+    @deflator         = Zlib::Deflate.new
+    @random_generator = Random.new(0)
+    @original         = ''
+    @chunks           = []
+  end
+
+  describe "without break" do
+
+    before do
+      2.times do
+        @input = @random_generator.bytes(20000)
+        @original << @input
+        @deflator.deflate(@input) do |chunk|
+          @chunks << chunk
+        end
+      end
+    end
+
+    it "deflates chunked data" do
+      @deflator.finish
+      @chunks.map { |chunk| chunk.length }.should == [16384, 16384]
+    end
+
+    it "deflates chunked data with final chunk" do
+      final = @deflator.finish
+      final.length.should == 7253
+    end
+
+    it "deflates chunked data without errors" do
+      final = @deflator.finish
+      @chunks << final
+      @original.should == Zlib.inflate(@chunks.join)
+    end
+
+  end
+
+  describe "with break" do
+    before :each do
+      @input = @random_generator.bytes(20000)
+      @deflator.deflate(@input) do |chunk|
+        @chunks << chunk
+        break
+      end
+    end
+
+    it "deflates only first chunk" do
+      @deflator.finish
+      @chunks.map { |chunk| chunk.length }.should == [16384]
+    end
+
+    it "deflates chunked data with final chunk" do
+      final = @deflator.finish
+      final.length.should == 3632
+    end
+
+    it "deflates chunked data without errors" do
+      final = @deflator.finish
+      @chunks << final
+      @input.should == Zlib.inflate(@chunks.join)
+    end
+
+  end
+end