test/rubygems/test_gem_package_tar_input.rb

   1 #--
   2 # Copyright 2006 by Chad Fowler, Rich Kilmer, Jim Weirich and others.
   3 # All rights reserved.
   4 # See LICENSE.txt for permissions.
   5 #++
   6
   7 require File.join(File.expand_path(File.dirname(__FILE__)),
   8                   'gem_package_tar_test_case')
   9 require 'rubygems/package/tar_input'
  10
  11 class TestGemPackageTarInput < TarTestCase
  12
  13   # Sometimes the setgid bit doesn't take.  Don't know if this is a problem on
  14   # all systems, or just some.  But for now, we will ignore it in the tests.
  15   SETGID_BIT = 02000
  16
  17   def setup
  18     super
  19
  20     inner_tar = tar_file_header("bla", "", 0612, 10)
  21     inner_tar += "0123456789" + "\0" * 502
  22     inner_tar += tar_file_header("foo", "", 0636, 5)
  23     inner_tar += "01234" + "\0" * 507
  24     inner_tar += tar_dir_header("__dir__", "", 0600)
  25     inner_tar += "\0" * 1024
  26     str = TempIO.new
  27
  28     begin
  29       os = Zlib::GzipWriter.new str
  30       os.write inner_tar
  31     ensure
  32       os.finish
  33     end
  34
  35     str.rewind
  36
  37     @file = File.join @tempdir, 'bla.tar'
  38
  39     File.open @file, 'wb' do |f|
  40       f.write tar_file_header("data.tar.gz", "", 0644, str.string.size)
  41       f.write str.string
  42       f.write "\0" * ((512 - (str.string.size % 512)) % 512 )
  43
  44       @spec = Gem::Specification.new do |spec|
  45         spec.author = "Mauricio :)"
  46       end
  47
  48       meta = @spec.to_yaml
  49
  50       f.write tar_file_header("metadata", "", 0644, meta.size)
  51       f.write meta + "\0" * (1024 - meta.size)
  52       f.write "\0" * 1024
  53     end
  54
  55     @entry_names = %w{bla foo __dir__}
  56     @entry_sizes = [10, 5, 0]
  57     #FIXME: are these modes system dependent?
  58     @entry_modes = [0100612, 0100636, 040600]
  59     @entry_files = %W[#{@tempdir}/bla #{@tempdir}/foo]
  60     @entry_contents = %w[0123456789 01234]
  61   end
  62
  63   def test_each_works
  64     open @file, 'rb' do |io|
  65       Gem::Package::TarInput.open io do |tar_input|
  66         count = 0
  67
  68         tar_input.each_with_index do |entry, i|
  69           count = i
  70
  71           assert_kind_of Gem::Package::TarReader::Entry, entry
  72           assert_equal @entry_names[i], entry.header.name
  73           assert_equal @entry_sizes[i], entry.header.size
  74         end
  75
  76         assert_equal 2, count
  77
  78         assert_equal @spec, tar_input.metadata
  79       end
  80     end
  81   end
  82
  83   def test_extract_entry_works
  84     open @file, 'rb' do |io|
  85       Gem::Package::TarInput.open io do |tar_input|
  86         assert_equal @spec, tar_input.metadata
  87
  88         count = 0
  89
  90         tar_input.each_with_index do |entry, i|
  91           count = i
  92           tar_input.extract_entry @tempdir, entry
  93           name = File.join @tempdir, entry.header.name
  94
  95           if entry.directory?
  96             assert File.dir?(name)
  97           else
  98             assert File.file?(name)
  99             assert_equal @entry_sizes[i], File.stat(name).size
 100             #FIXME: win32? !!
 101           end
 102
 103           unless Gem.win_platform? then
 104             assert_equal @entry_modes[i], File.stat(name).mode & (~SETGID_BIT)
 105           end
 106         end
 107
 108         assert_equal 2, count
 109       end
 110     end
 111
 112     @entry_files.each_with_index do |x, i|
 113       assert File.file?(x)
 114       assert_equal @entry_contents[i], File.read_b(x)
 115     end
 116   end
 117
 118 end
 119