rubygems-update 0.8.3

data/test/test_parse_commands.rb

@@ -0,0 +1,179 @@
+require 'test/unit'
+$:.unshift '../lib'
+require 'rubygems'
+require 'test/mockgemui'
+Gem::manage_gems
+
+class TestParseCommands < Test::Unit::TestCase
+  include Gem::DefaultUserInteraction
+
+  def setup
+    @cmd_manager = Gem::CommandManager.new
+  end
+
+  def test_parsing_bad_options
+    use_ui(MockGemUi.new) do
+      assert_raises(MockGemUi::TermError) {
+	@cmd_manager.process_args("--bad-arg")
+      }
+      assert_match /invalid option: --bad-arg/i, ui.error
+    end
+  end
+
+  def test_parsing_install_options
+    #capture all install options
+    use_ui(MockGemUi.new) do
+      check_options = nil
+      @cmd_manager['install'].when_invoked do |options|
+	check_options = options
+	true
+      end
+      
+      #check defaults
+      @cmd_manager.process_args("install")
+      assert_equal false, check_options[:test]
+      assert_equal true, check_options[:generate_rdoc]
+      assert_equal false, check_options[:force]
+      assert_equal :both, check_options[:domain]
+      assert_equal "> 0", check_options[:version]
+      assert_equal Gem.dir, check_options[:install_dir]
+      
+      #check settings
+      check_options = nil
+      @cmd_manager.process_args(
+	"install --force --test --local --rdoc --install-dir . --version 3.0")
+      assert_equal true, check_options[:test]
+      assert_equal true, check_options[:generate_rdoc]
+      assert_equal true, check_options[:force]
+      assert_equal :local, check_options[:domain]
+      assert_equal '3.0', check_options[:version]
+      assert_equal '.', check_options[:install_dir]
+      
+      #check remote domain
+      check_options = nil
+      @cmd_manager.process_args("install --remote")
+      assert_equal :remote, check_options[:domain]
+      
+      #check both domain
+      check_options = nil
+      @cmd_manager.process_args("install --both")
+      assert_equal :both, check_options[:domain]
+      
+      #check both domain
+      check_options = nil
+      @cmd_manager.process_args("install --both")
+      assert_equal :both, check_options[:domain]
+    end
+  end
+  
+  def test_parsing_uninstall_options
+    #capture all uninstall options
+    check_options = nil
+    @cmd_manager['uninstall'].when_invoked do |options|
+      check_options = options
+      true
+    end
+    
+    #check defaults
+    @cmd_manager.process_args("uninstall")
+    assert_equal "> 0", check_options[:version]
+
+    #check settings
+    check_options = nil
+    @cmd_manager.process_args("uninstall foobar --version 3.0")
+    assert_equal "foobar", check_options[:args].first
+    assert_equal "3.0", check_options[:version]
+  end
+
+  def test_parsing_check_options
+    #capture all check options
+    check_options = nil
+    @cmd_manager['check'].when_invoked do |options|
+      check_options = options
+      true
+    end
+    
+    #check defaults
+    @cmd_manager.process_args("check")
+    assert_equal false, check_options[:verify]
+    assert_equal false, check_options[:alien]
+
+    #check settings
+    check_options = nil
+    @cmd_manager.process_args("check --verify foobar --alien")
+    assert_equal "foobar", check_options[:verify]
+    assert_equal true, check_options[:alien]
+  end
+    
+  def test_parsing_build_options
+    #capture all build options
+    check_options = nil
+    @cmd_manager['build'].when_invoked do |options|
+      check_options = options
+      true
+    end
+    
+    #check defaults
+    @cmd_manager.process_args("build")
+    #NOTE: Currently no defaults
+    
+    #check settings
+    check_options = nil
+    @cmd_manager.process_args("build foobar.rb")
+    assert_equal 'foobar.rb', check_options[:args].first
+  end
+  
+  def test_parsing_query_options
+    #capture all query options
+    check_options = nil
+    @cmd_manager['query'].when_invoked do |options|
+      check_options = options
+      true
+    end
+    
+    #check defaults
+    @cmd_manager.process_args("query")
+    assert_equal /.*/, check_options[:name]
+    assert_equal :local, check_options[:domain]
+    assert_equal false, check_options[:details]
+    
+    #check settings
+    check_options = nil
+    @cmd_manager.process_args("query --name foobar --local --details")
+    assert_equal /foobar/i, check_options[:name]
+    assert_equal :local, check_options[:domain]
+    assert_equal true, check_options[:details]
+
+    #remote domain
+    check_options = nil
+    @cmd_manager.process_args("query --remote")
+    assert_equal :remote, check_options[:domain]
+
+    #both (local/remote) domains
+    check_options = nil
+    @cmd_manager.process_args("query --both")
+    assert_equal :both, check_options[:domain]
+  end  
+  
+  def test_parsing_update_options
+    #capture all update options
+    check_options = nil
+    @cmd_manager['update'].when_invoked do |options|
+      check_options = options
+      true
+    end
+    
+    #check defaults
+    @cmd_manager.process_args("update")
+    assert_equal true, check_options[:generate_rdoc]
+    
+    #check settings
+    check_options = nil
+    @cmd_manager.process_args("update --force --test --rdoc --install-dir .")
+    assert_equal true, check_options[:test]
+    assert_equal true, check_options[:generate_rdoc]
+    assert_equal true, check_options[:force]
+    assert_equal '.', check_options[:install_dir]
+  end 
+  
+end

data/test/test_process_commands.rb

@@ -0,0 +1,21 @@
+require 'test/unit'
+$:.unshift '../lib'
+require 'rubygems'
+Gem::manage_gems
+
+require 'test/mockgemui'
+
+class TestProcessCommands < Test::Unit::TestCase
+  include Gem::DefaultUserInteraction
+
+  def setup
+    @cmd_manager = Gem::CommandManager.new
+  end
+
+  def test_query_command
+    use_ui(MockGemUi.new) do
+      @cmd_manager.process_args "query"
+      assert_match /LOCAL GEMS/, ui.output
+    end
+  end
+end

data/test/test_remote_fetcher.rb

@@ -0,0 +1,162 @@
+#!/usr/bin/env ruby
+
+require 'test/unit'
+require 'zlib'
+require 'webrick'
+require 'rubygems'
+
+require 'test/mockgemui'
+require 'test/yaml_data'
+
+Gem.manage_gems
+include WEBrick
+
+class TestRemoteFetcher < Test::Unit::TestCase
+  include Gem::DefaultUserInteraction
+
+  def setup
+    self.class.start_servers
+    self.class.enable_yaml = true
+    self.class.enable_zip = false
+    ENV['http_proxy'] = nil
+    ENV['HTTP_PROXY'] = nil
+  end
+
+
+  def test_no_proxy
+    use_ui(MockGemUi.new) do
+      fetcher = Gem::RemoteSourceFetcher.new("http://localhost:12344", nil)
+      assert_equal YAML_DATA, fetcher.fetch_path("/yaml")
+      assert_equal YAML_DATA.size, fetcher.size
+      source_index = fetcher.source_index
+      gems = source_index.search('rake', "= 0.4.11")
+      assert_equal 1, gems.size
+      assert_equal "rake", gems.first.name
+    end
+  end
+  
+  def test_explicit_proxy
+    use_ui(MockGemUi.new) do
+      fetcher = Gem::RemoteSourceFetcher.new("http://localhost:12344", "http://localhost:12345")
+      assert_equal PROXY_DATA.size, fetcher.size
+      assert_equal PROXY_DATA, fetcher.fetch_path("/yaml")
+      source_index = fetcher.source_index
+      gems = source_index.search('rake', "= 0.4.2")
+      assert_equal 1, gems.size
+      assert_equal "rake", gems.first.name
+    end
+  end
+  
+  def test_implicit_proxy
+    use_ui(MockGemUi.new) do
+      ENV['http_proxy'] = 'http://localhost:12345'
+      fetcher = Gem::RemoteSourceFetcher.new("http://localhost:12344", true)
+      assert_equal PROXY_DATA, fetcher.fetch_path("/yaml")
+    end
+  end
+  
+  def test_implicit_upper_case_proxy
+    use_ui(MockGemUi.new) do
+      ENV['HTTP_PROXY'] = 'http://localhost:12345'
+      fetcher = Gem::RemoteSourceFetcher.new("http://localhost:12344", true)
+      assert_equal PROXY_DATA, fetcher.fetch_path("/yaml")
+    end
+  end
+  
+  def test_implicit_proxy_no_env
+    use_ui(MockGemUi.new) do
+      fetcher = Gem::RemoteSourceFetcher.new("http://localhost:12344", true)
+      assert_equal YAML_DATA, fetcher.fetch_path("/yaml")
+    end
+  end
+  
+  def test_zip
+    use_ui(MockGemUi.new) do
+      self.class.enable_zip = true
+      fetcher = Gem::RemoteSourceFetcher.new("http://localhost:12344", nil)
+      assert_equal YAML_DATA.size, fetcher.size
+      zip_data = fetcher.fetch_path("/yaml.Z")
+      assert zip_data.size < YAML_DATA.size
+    end
+  end
+
+  def test_no_zip
+    use_ui(MockGemUi.new) do
+      fetcher = Gem::RemoteSourceFetcher.new("http://localhost:12344", nil)
+      assert_error { fetcher.fetch_path("/yaml.Z") }
+    end
+  end
+
+  def test_yaml_error_on_size
+    use_ui(MockGemUi.new) do
+      self.class.enable_yaml = false
+      fetcher = Gem::RemoteSourceFetcher.new("http://localhost:12344", nil)
+      assert_error { fetcher.size }
+    end
+  end
+
+  def assert_error(exception_class=Exception)
+    got_exception = false
+    begin
+      yield
+    rescue exception_class => ex
+      got_exception = true
+    end
+    assert got_exception, "Expected exception conforming to #{exception_class}" 
+  end
+
+  class << self
+    attr_reader :normal_server, :proxy_server
+    attr_accessor :enable_zip, :enable_yaml
+    
+    def start_servers
+      @normal_server ||= start_server(12344, YAML_DATA)
+      @proxy_server  ||= start_server(12345, PROXY_DATA)
+      @enable_yaml = true
+      @enable_zip = false
+      sleep 1
+    end
+    
+    private
+    
+    def start_server(port, data)
+      Thread.new do
+	begin
+	  null_logger = Log.new('/dev/null')
+	  s = HTTPServer.new(
+	    :Port            => port,
+	    :DocumentRoot    => ".",
+	    :Logger          => null_logger,
+	    :AccessLog       => null_logger
+	    )
+	  s.mount_proc("/kill") { |req, res| s.shutdown }
+	  s.mount_proc("/yaml") { |req, res|
+	    if @enable_yaml
+	      res.body = data
+	      res['Content-Type'] = 'text/plain'
+	      res['content-length'] = data.size
+	    else
+	      res.code = "404"
+	      res.body = "<h1>NOT FOUND</h1>"
+	      res['Content-Type'] = 'text/html'
+	    end
+	  }
+	  s.mount_proc("/yaml.Z") { |req, res|
+	    if @enable_zip
+	      res.body = Zlib::Deflate.deflate(data)
+	      res['Content-Type'] = 'text/plain'
+	    else
+	      res.code = "404"
+	      res.body = "<h1>NOT FOUND</h1>"
+	      res['Content-Type'] = 'text/html'
+	    end
+	  }
+	  s.start
+	rescue Exception => ex
+	  puts "ERROR during server thread: #{ex.message}"
+	end
+      end
+    end
+  end
+  
+end

data/test/test_remote_installer.rb

@@ -0,0 +1,154 @@
+require 'test/unit'
+require 'rubygems'
+Gem::manage_gems
+require 'net/http'
+require 'yaml'
+require 'test/onegem'
+
+class MockFetcher
+  def initialize(uri, proxy)
+    @uri = uri
+  end
+
+  def size
+    1000
+  end
+  
+  def source_index
+    if @uri =~ /non.existent.url/
+      fail Gem::RemoteSourceException,
+	"Error fetching remote gem cache: Mock Socket Exception"
+    end
+    result = {
+      'foo-1.2.3' => Gem::Specification.new do |s|
+	s.name = 'foo'
+	s.version = "1.2.3"
+	s.summary = "This is a cool package"
+      end
+    }
+    result
+  end
+
+  def fetch_path(path)
+  end
+
+  def self.finish
+  end
+end
+
+class TestRemoteInstaller < Test::Unit::TestCase
+
+  PROPER_SOURCES = %w( http://gems.rubyforge.org )
+
+  def setup
+    Gem.clear_paths
+    @installer = Gem::RemoteInstaller.new
+    @installer.instance_variable_set("@fetcher_class", MockFetcher)
+  end
+
+  def test_create
+    assert_not_nil(@installer)
+  end
+  
+  # Make sure that the installer knows the proper list of places to go
+  # to get packages.
+  def test_source_list
+    assert_equal PROPER_SOURCES, @installer.sources
+  end
+
+  def test_source_index_hash
+    source_hash = @installer.source_index_hash
+    assert source_hash.has_key?("http://gems.rubyforge.org")
+    assert_equal 1, source_hash.size
+    gem_hash = source_hash['http://gems.rubyforge.org']
+    spec = gem_hash['foo-1.2.3']
+    assert_equal 'foo', spec.name
+    assert_equal '1.2.3', spec.version.to_s
+  end
+
+  def test_missing_source_exception
+    @installer.instance_variable_set("@sources", ["http://non.existent.url"])
+    assert_raise(Gem::RemoteSourceException) {
+      info = @installer.source_index_hash
+    }
+  end
+end
+
+# This test suite has a number of TODOs in the test cases.  The
+# TestRemoteInstaller test suite is a reworking of this class from
+# scratch.
+class RemoteInstallerTest < Test::Unit::TestCase
+  class RInst < Gem::RemoteInstaller
+    include Test::Unit::Assertions
+
+    attr_accessor :expected_destination_files
+    attr_accessor :expected_bodies
+    attr_accessor :caches
+    attr_accessor :responses
+
+    def source_index_hash
+      @caches
+    end
+
+    def fetch(uri)
+      @reponses ||= {}
+      @responses[uri]
+    end
+
+    def write_gem_to_file(body, destination_file)
+      expected_destination_file = expected_destination_files.pop
+      expected_body = expected_bodies.pop
+      assert_equal expected_body, body, "Unexpected body"
+      assert_equal expected_destination_file, destination_file, "Unexpected destination file"
+    end
+
+    def new_installer(gem)
+      return MockInstaller.new(gem)
+    end
+  end
+
+  CACHE_SOURCES = ["http://gems.rubyforge.org"]
+
+  def setup
+    Gem.clear_paths
+    @remote_installer = Gem::RemoteInstaller.new
+    @remote_installer.instance_eval { @fetcher_class = MockFetcher }
+  end
+
+  def test_sources
+    assert_equal CACHE_SOURCES, @remote_installer.sources
+  end
+
+  def test_source_index_hash
+    source_index_hash = @remote_installer.source_index_hash
+    assert_equal 1, source_index_hash.keys.size
+  end
+
+  def test_find_latest_valid_package_in_caches(cache)
+  end
+
+  def test_download_file
+  end
+
+  SAMPLE_SPEC = Gem::Specification.new do |s|
+    s.name = 'foo'
+    s.version = "1.2.3"
+    s.platform = Gem::Platform::RUBY
+    s.summary = "This is a cool package"
+    s.files = []
+  end
+  SAMPLE_CACHE = { 'foo-1.2.3' => SAMPLE_SPEC }
+  SAMPLE_CACHE_YAML = SAMPLE_CACHE.to_yaml
+
+  FOO_GEM = '' # TODO
+  CACHE_DIR = File.join(Gem.dir, 'cache')
+
+  # Disable this test for now.  We will come back to revisit this.
+  def disable_test_install
+    Gem.use_paths("test/data/gemhome")
+    result = @remote_installer.install('foo')
+    assert_equal [nil], result
+  end
+
+end
+