Source

amp / test / test_changegroup.rb

##################################################################
#                  Licensing Information                         #
#                                                                #
#  The following code is licensed, as standalone code, under     #
#  the Ruby License, unless otherwise directed within the code.  #
#                                                                #
#  For information on the license of this code when distributed  #
#  with and used in conjunction with the other modules in the    #
#  Amp project, please see the root-level LICENSE file.          #
#                                                                #
#  © Michael J. Edgar and Ari Brown, 2009-2010                   #
#                                                                #
##################################################################

require 'stringio'
require File.join(File.expand_path(File.dirname(__FILE__)), 'testutilities')
require 'zlib'
require File.expand_path(File.join(File.dirname(__FILE__), "../lib/amp"))

class TestChangegroup < AmpTestCase
  include Amp::Mercurial::RevlogSupport
  
  def test_compressor_uncompressed
    compressor = ChangeGroup.compressor_by_type("HG10UN")
    assert_kind_of StringIO, compressor
    assert_respond_to compressor, :flush
    compressor << "input string"
    assert_equal "input string", compressor.flush
  end
  
  def test_compressor_no_header
    assert_kind_of StringIO, ChangeGroup.compressor_by_type("")
  end
  
  def test_compressor_zlib
    compressor = ChangeGroup.compressor_by_type("HG10GZ")
    assert_kind_of Zlib::Deflate, compressor
  end
  
  def test_compressor_bz
    # this fails. bz2 is borked atm.
    # compressor = ChangeGroup.compressor_by_type("HG10BZ")
    # assert_kind_of BZ2::Writer, compressor
  end
  
  def test_unbundle_uncompressed
    input_io = StringIO.new("input data")
    decompressing_io = ChangeGroup.unbundle("HG10UN", input_io)
    assert_equal("input data", decompressing_io.read)
  end
  
  def test_unbundle_funny_header
    input_io = StringIO.new("input data")
    funny_header = "BUNDLE"
    decompressing_io = ChangeGroup.unbundle(funny_header, input_io)
    assert_equal("BUNDLEinput data", decompressing_io.read)
  end
  
  def test_unbundle_gzip
    input_io = StringIO.new("", "r+")
    gzip_maker = Zlib::Deflate.new()
    input_io << gzip_maker.deflate("input string")
    input_io << gzip_maker.flush
    input_io << gzip_maker.finish
    gzip_maker.close
    
    input_io.rewind
    decompressing_io = ChangeGroup.unbundle("HG10GZ", input_io)
    assert_equal "input string", decompressing_io.read
  end
    
  
end