bzip2-ruby 0.2.4 → 0.2.5

data/ext/extconf.rb

@@ -7,7 +7,12 @@ if have_library("bz2", "BZ2_bzWriteOpen")
   if enable_config("shared", true)
      $static = nil
   end
-  create_makefile("bzip2")
+  
+  if RUBY_VERSION =~ /1.9/
+    $CFLAGS << ' -DRUBY_19_COMPATIBILITY'
+  end
+  
+  create_makefile("bzip2_ext")
 else
   puts "libbz2 not found, maybe try manually specifying --with-bz2-dir to find it?"
 end

data/lib/bzip2.rb

@@ -0,0 +1,5 @@
+require 'bzip2_ext'
+
+module Bzip2
+  VERSION = "0.2.5"
+end

data/spec/reader_spec.rb

@@ -0,0 +1,357 @@
+# encoding: UTF-8
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper.rb')
+
+describe "Bzip2::Writer" do
+  before(:all) do
+    @sample = "08: This is a line\n"
+    @file = "_10lines_"
+    @data = [
+      "00: This is a line\n",
+      "01: This is a line\n",
+      "02: This is a line\n",
+      "03: This is a line\n",
+      "04: This is a line\n",
+      "05: This is a line\n",
+      "06: This is a line\n",
+      "07: This is a line\n",
+      "08: This is a line\n",
+      "09: This is a line\n" 
+    ]
+
+    open("|bzip2 > #{@file}", "w") do |f|
+      @data.each { |l| f.puts l }
+    end
+  end
+  
+  after(:all) do
+    File.unlink(@file)
+  end
+
+  it "should test_f_s_foreach" do
+    count = 0
+    Bzip2::Reader.foreach(@file) do |line|
+      num = line[0..1].to_i
+      count.should == num
+      count += 1
+    end
+    @data.size.should == count
+
+    count = 0
+    Bzip2::Reader.foreach(@file, nil) do |file|
+      file.split(/\n/).each do |line|
+        num = line[0..1].to_i
+        count.should == num
+        count += 1
+      end
+    end
+    @data.size.should == count
+
+    count = 0
+    Bzip2::Reader.foreach(@file, ' ') do |thing|
+      count += 1
+    end
+    41.should == count
+  end
+  
+  it "should test_f_s_readlines" do
+    lines = Bzip2::Reader.readlines(@file)
+    @data.size.should == lines.size
+
+    lines = Bzip2::Reader.readlines(@file, nil)
+    1.should == lines.size
+    (@sample.length * @data.size).should == lines[0].size
+  end
+
+  it "should test_f_closed?" do
+    f = Bzip2::Reader.open(@file)
+    f.closed?.should be_false
+    f.close
+    f.closed?.should be_true
+  end
+
+  it "should test_f_each" do
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      file.each do |line|
+        num = line[0..1].to_i
+        count.should == num
+        count += 1
+      end
+      @data.size.should == count
+    end
+
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      file.each(nil) do |contents|
+        contents.split(/\n/).each do |line|
+          num = line[0..1].to_i
+          count.should == num
+          count += 1
+        end
+      end
+    end
+    @data.size.should == count
+
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      file.each(' ') do |thing|
+        count += 1
+      end
+    end
+    41.should == count
+  end
+
+  it "should test_f_each_byte" do
+    count = 0
+    data = @data.join
+
+    Bzip2::Reader.open(@file) do |file|
+      file.each_byte do |b|
+        data.getbyte(count).should == b
+        count += 1
+      end
+    end
+    (@sample.length * @data.size).should == count
+  end
+
+  it "should test_f_each_line" do
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      file.each_line do |line|
+        num = line[0..1].to_i
+        count.should == num
+        count += 1
+      end
+      @data.size.should == count
+    end
+
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      file.each_line(nil) do |contents|
+        contents.split(/\n/).each do |line|
+          num = line[0..1].to_i
+          count.should == num
+          count += 1
+        end
+      end
+    end
+    @data.size.should == count
+
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      file.each_line(' ') do |thing|
+        count += 1
+      end
+    end
+    41.should == count
+  end
+
+  it "should test_f_eof" do
+    Bzip2::Reader.open(@file) do |file|
+      @data.size.times do
+        (!file.eof).should be_true
+        (!file.eof?).should be_true
+        file.gets
+      end
+      file.eof.should be_true
+      file.eof?.should be_true
+    end
+  end
+
+  it "should test_f_getc" do
+    count = 0
+    data = @data.join
+
+    Bzip2::Reader.open(@file) do |file|
+      while (ch = file.getc)
+        data.getbyte(count).should == ch
+        count += 1
+      end
+      file.getc.should be_nil
+    end
+    (@sample.length * @data.size).should == count
+  end
+
+  it "should test_f_gets" do
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      while (line = file.gets)
+        num = line[0..1].to_i
+        count.should == num
+        count += 1
+      end
+      file.gets.should be_nil
+      @data.size.should == count
+    end
+
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      while (line = file.gets("line\n"))
+        @data[count].should == line
+        num = line[0..1].to_i
+        count.should == num
+        count += 1
+      end
+      file.gets.should be_nil
+      @data.size.should == count
+    end
+
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      while (contents = file.gets(nil))
+        contents.split(/\n/).each do |line|
+          num = line[0..1].to_i
+          count.should == num
+          count += 1
+        end
+      end
+    end
+    @data.size.should == count
+
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      while (thing = file.gets(' '))
+        count += 1
+      end
+    end
+    41.should == count
+  end
+
+  it "should test_f_read" do
+    Bzip2::Reader.open(@file) do |file|
+      content = file.read
+      (@sample.length * @data.size).should == content.length
+      count = 0
+      content.split(/\n/).each do |line|
+        num = line[0..1].to_i
+        count.should == num
+        count += 1
+      end
+    end
+
+    Bzip2::Reader.open(@file) do |file|
+      "00: This is ".should == file.read(12)
+      "a line\n01: T".should == file.read(12)
+    end
+  end
+
+  it "should test_f_readchar" do
+    count = 0
+    data = @data.join
+    Bzip2::Reader.open(@file) do |file|
+      190.times do |count|
+        ch = file.readchar
+        data.getbyte(count).should == ch
+        count += 1
+      end
+      lambda { file.readchar }.should raise_error(Bzip2::EOZError)
+    end
+  end
+
+  it "should test_f_readline" do
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      @data.size.times do |count|
+        line = file.readline
+        num = line[0..1].to_i
+        count.should == num
+        count += 1
+      end
+      lambda { file.readline }.should raise_error(Bzip2::EOZError)
+    end
+
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      contents = file.readline(nil)
+      contents.split(/\n/).each do |line|
+        num = line[0..1].to_i
+        count.should == num
+        count += 1
+      end
+      lambda { file.readline }.should raise_error(Bzip2::EOZError)
+    end
+    @data.size.should == count
+
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      41.times do |count|
+        thing = file.readline(' ')
+        count += 1
+      end
+      lambda { file.readline }.should raise_error(Bzip2::EOZError)
+    end
+  end
+
+  it "should test_f_readlines" do
+    Bzip2::Reader.open(@file) do |file|
+      lines = file.readlines
+      @data.size.should == lines.size
+    end
+
+    Bzip2::Reader.open(@file) do |file|
+      lines = file.readlines(nil)
+      1.should == lines.size
+      (@sample.length * @data.size).should == lines[0].size
+    end
+  end
+
+  # it "should test_f_ungetc" do
+  #   Bzip2::Reader.open(@file) do |file|
+  #     ?0.getbyte(0).should == file.getc
+  #     ?0.getbyte(0).should == file.getc
+  #     ?:.getbyte(0).should == file.getc
+  #     ?\s.getbyte(0).should == file.getc
+  #     file.ungetc(?:.to_i).should be_nil
+  #     ?:.getbyte(0).should == file.getc
+  #     1 while file.getc
+  #     file.ungetc(?A).should be_nil
+  #     ?A.should == file.getc
+  #   end
+  # end
+
+  it "should test_f_ungets" do
+    count = 0
+    Bzip2::Reader.open(@file) do |file|
+      @data[count].should == file.gets
+      (count + 1).should == file.lineno
+      file.ungets(@data[count]).should be_nil
+      @data[count].should == file.gets
+      count += 1
+    end
+  end
+
+  it "should test_s_readline" do
+    count = 0
+    string = IO.readlines(@file, nil)[0]
+    file = Bzip2::Reader.new(string)
+    @data.size.times do |count|
+      line = file.readline
+      num = line[0..1].to_i
+      count.should == num
+      count += 1
+    end
+    lambda { file.readline }.should raise_error(Bzip2::EOZError)
+    file.close
+
+    count = 0
+    file = Bzip2::Reader.new(string)
+    contents = file.readline(nil)
+    contents.split(/\n/).each do |line|
+      num = line[0..1].to_i
+      count.should == num
+      count += 1
+    end
+    lambda { file.readline }.should raise_error(Bzip2::EOZError)
+    @data.size.should == count
+    file.close
+
+    count = 0
+    file = Bzip2::Reader.new(string)
+    41.times do |count|
+      thing = file.readline(' ')
+      count += 1
+    end
+    lambda { file.readline }.should raise_error(Bzip2::EOZError)
+    file.close
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,11 @@
+# encoding: UTF-8
+require 'bzip2_ext'
+
+# back-port 1.9 method so the tests will pass in 1.8 as well
+if RUBY_VERSION.include?("1.8")
+  class String
+    def getbyte(idx)
+      self[idx]
+    end
+  end
+end

data/spec/writer_spec.rb

@@ -0,0 +1,139 @@
+# encoding: UTF-8
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper.rb')
+
+describe "Bzip2::Writer" do
+  class Dummy
+    def to_s
+      "dummy"
+    end
+  end
+
+  before(:all) do
+    @file = "_10lines_"
+  end
+
+  after(:all) do
+    File.unlink(@file)
+  end
+
+  def test_f_LSHIFT # '<<'
+    Bzip2::Writer.open($file, "w") do |file|
+      file << 1 << "\n" << Dummy.new << "\n" << "cat\n"
+    end
+    expected = [ "1\n", "dummy\n", "cat\n"]
+    Bzip2::Reader.foreach($file) do |line|
+      expected.shift.should == line
+    end
+    [].should == expected
+  end
+
+  def test_f_error
+    io = File.new($file, "w")
+    bz2 = Bzip2::Writer.new(io)
+    bz2 << 1 << "\n" << Dummy.new << "\n" << "cat\n"
+    bz = Bzip2::Reader.new($file)
+    lambda { bz.gets }.should raise_error(Bzip2::EOZError)
+    bz = Bzip2::Reader.open($file)
+    lambda { bz.gets }.should raise_error(Bzip2::EOZError)
+    io.close
+    lambda { Bzip2::Reader.new(io) }.should raise_error(IOError)
+  end
+
+  def test_f_gets_para
+    Bzip2::Writer.open($file) do |file|
+      file.print "foo\n"*4096, "\n"*4096, "bar"*4096, "\n"*4096, "zot\n"*1024
+    end
+    Bzip2::Reader.open($file) do |file|
+      ("foo\n"*4096+"\n").should == file.gets("")
+      ("bar"*4096+"\n\n").should == file.gets("")
+      ("zot\n"*1024).should == file.gets("")
+    end
+  end
+
+  def test_f_print
+    Bzip2::Writer.open($file) do |file|
+      file.print "hello"
+      file.print 1,2
+      $_ = "wombat\n"
+      file.print
+      $\ = ":"
+      $, = ","
+      file.print 3, 4
+      file.print 5, 6
+      $\ = nil
+      file.print "\n"
+      $, = nil
+    end
+
+    Bzip2::Reader.open($file) do |file|
+      content = file.gets(nil)
+      "hello12wombat\n3,4:5,6:\n".should == content
+    end
+  end
+
+  def test_f_putc
+    Bzip2::Writer.open($file, "wb") do |file|
+      file.putc "A"
+      0.upto(255) { |ch| file.putc ch }
+    end
+
+    Bzip2::Reader.open($file, "rb") do |file|
+      ?A.should == file.getc
+      0.upto(255) { |ch| ch.should == file.getc }
+    end
+  end
+
+  def test_f_puts
+    Bzip2::Writer.open($file, "w") do |file|
+      file.puts "line 1", "line 2"
+      file.puts [ Dummy.new, 4 ]
+    end
+
+    Bzip2::Reader.open($file) do |file|
+      "line 1\n".should == file.gets
+      "line 2\n".should == file.gets
+      "dummy\n".should == file.gets
+      "4\n".should == file.gets
+    end
+  end
+
+  def test_f_write
+    Bzip2::Writer.open($file, "w") do |file|
+      10.should == file.write('*' * 10)
+      5.should == file.write('!' * 5)
+      0.should == file.write('')
+      1.should == file.write(1)
+      3.should == file.write(2.30000)
+      1.should == file.write("\n")
+    end
+
+    Bzip2::Reader.open($file) do |file|
+      "**********!!!!!12.3\n".should == file.gets
+    end
+  end
+
+  def test_s_string
+    file = Bzip2::Writer.new
+    10.should == file.write('*' * 10)
+    5.should == file.write('!' * 5)
+    0.should == file.write('')
+    1.should == file.write(1)
+    3.should == file.write(2.30000)
+    1.should == file.write("\n")
+    line = Bzip2::bunzip2(file.flush)
+    "**********!!!!!12.3\n".should == line
+
+    line = Bzip2::bunzip2(Bzip2::bzip2("**********!!!!!12.3\n"))
+    "**********!!!!!12.3\n".should == line
+
+    test = "foo\n"*4096 + "\n"*4096 + "bar"*4096 + "\n"*4096 + "zot\n"*1024
+    line = Bzip2::bunzip2(Bzip2::bzip2(test))
+    test.should == line
+
+    Bzip2::bzip2("aaaaaaaaa".taint).tainted?.should be_true
+    Bzip2::bunzip2(Bzip2::bzip2("aaaaaaaaa".taint)).tainted?.should be_true
+    (!Bzip2::bzip2("aaaaaaaaa").tainted?).should be_false
+    (!Bzip2::bunzip2(Bzip2::bzip2("aaaaaaaaa")).tainted?).should be_false
+
+  end
+end