test/rubygems/test_gem_source_info_cache_entry.rb

   1 require 'test/unit'
   2 require File.join(File.expand_path(File.dirname(__FILE__)), 'gemutilities')
   3 require 'rubygems/source_info_cache_entry'
   4
   5 class TestGemSourceInfoCacheEntry < RubyGemTestCase
   6
   7   def setup
   8     super
   9
  10     util_setup_fake_fetcher
  11
  12     @si = Gem::SourceIndex.new
  13     @si.add_spec @a1
  14     @sic_e = Gem::SourceInfoCacheEntry.new @si, @si.dump.size
  15   end
  16
  17   def test_refresh
  18     @fetcher.data["#{@gem_repo}Marshal.#{@marshal_version}.Z"] =
  19       proc { raise }
  20     @fetcher.data["#{@gem_repo}Marshal.#{@marshal_version}"] = @si.dump
  21
  22     use_ui @ui do
  23       @sic_e.refresh @gem_repo, true
  24     end
  25   end
  26
  27   def test_refresh_all
  28     @si.add_spec @a2
  29
  30     a1_name = @a1.full_name
  31     a2_name = @a2.full_name
  32
  33     @fetcher.data["#{@gem_repo}quick/index.rz"] =
  34         util_zip [a1_name, a2_name].join("\n")
  35     @fetcher.data["#{@gem_repo}quick/latest_index.rz"] = util_zip a2_name
  36     @fetcher.data["#{@gem_repo}quick/Marshal.#{Gem.marshal_version}/#{a1_name}.gemspec.rz"] = util_zip Marshal.dump(@a1)
  37     @fetcher.data["#{@gem_repo}quick/Marshal.#{Gem.marshal_version}/#{a2_name}.gemspec.rz"] = util_zip Marshal.dump(@a2)
  38     @fetcher.data["#{@gem_repo}Marshal.#{Gem.marshal_version}"] =
  39       Marshal.dump @si
  40
  41     sic_e = Gem::SourceInfoCacheEntry.new Gem::SourceIndex.new, 0
  42
  43     assert_equal [], sic_e.source_index.map { |n,| n }
  44
  45     use_ui @ui do
  46       assert sic_e.refresh(@gem_repo, false)
  47     end
  48
  49     assert_equal [a2_name], sic_e.source_index.map { |n,| n }.sort
  50
  51     use_ui @ui do
  52       sic_e.refresh @gem_repo, true
  53     end
  54
  55     assert_equal [a1_name, a2_name], sic_e.source_index.map { |n,| n }.sort
  56   end
  57
  58   def test_refresh_bad_uri
  59     assert_raise URI::BadURIError do
  60       @sic_e.refresh 'gems.example.com', true
  61     end
  62   end
  63
  64   def test_refresh_update
  65     si = Gem::SourceIndex.new
  66     si.add_spec @a1
  67     si.add_spec @b2
  68     @fetcher.data["#{@gem_repo}Marshal.#{@marshal_version}"] = si.dump
  69
  70     use_ui @ui do
  71       @sic_e.refresh @gem_repo, true
  72     end
  73
  74     new_gem = @sic_e.source_index.specification(@b2.full_name)
  75     assert_equal @b2.full_name, new_gem.full_name
  76   end
  77
  78 end
  79