/test/rubygems/test_gem_server.rb
Ruby | 409 lines | 293 code | 115 blank | 1 comment | 0 complexity | 0638a97fc2d97a7ddb89101e271aceae MD5 | raw file
- # frozen_string_literal: true
- require 'rubygems/test_case'
- require 'rubygems/server'
- require 'stringio'
- class Gem::Server
- attr_reader :server
- end
- class TestGemServer < Gem::TestCase
- def process_based_port
- 0
- end
- def setup
- super
- @a1 = quick_gem 'a', '1'
- @a2 = quick_gem 'a', '2'
- @a3_p = quick_gem 'a', '3.a'
- @server = Gem::Server.new Gem.dir, process_based_port, false
- @req = WEBrick::HTTPRequest.new :Logger => nil
- @res = WEBrick::HTTPResponse.new :HTTPVersion => '1.0'
- end
- def test_doc_root_3
- orig_rdoc_version = Gem::RDoc.rdoc_version
- Gem::RDoc.instance_variable_set :@rdoc_version, Gem::Version.new('3.12')
- assert_equal '/doc_root/X-1/rdoc/index.html', @server.doc_root('X-1')
- ensure
- Gem::RDoc.instance_variable_set :@rdoc_version, orig_rdoc_version
- end
- def test_doc_root_4
- orig_rdoc_version = Gem::RDoc.rdoc_version
- Gem::RDoc.instance_variable_set :@rdoc_version, Gem::Version.new('4.0')
- assert_equal '/doc_root/X-1/', @server.doc_root('X-1')
- ensure
- Gem::RDoc.instance_variable_set :@rdoc_version, orig_rdoc_version
- end
- def test_have_rdoc_4_plus_eh
- orig_rdoc_version = Gem::RDoc.rdoc_version
- Gem::RDoc.instance_variable_set(:@rdoc_version, Gem::Version.new('4.0'))
- server = Gem::Server.new Gem.dir, 0, false
- assert server.have_rdoc_4_plus?
- Gem::RDoc.instance_variable_set :@rdoc_version, Gem::Version.new('3.12')
- server = Gem::Server.new Gem.dir, 0, false
- refute server.have_rdoc_4_plus?
- Gem::RDoc.instance_variable_set(:@rdoc_version,
- Gem::Version.new('4.0.0.preview2'))
- server = Gem::Server.new Gem.dir, 0, false
- assert server.have_rdoc_4_plus?
- ensure
- Gem::RDoc.instance_variable_set :@rdoc_version, orig_rdoc_version
- end
- def test_spec_dirs
- s = Gem::Server.new Gem.dir, process_based_port, false
- assert_equal [File.join(Gem.dir, 'specifications')], s.spec_dirs
- s = Gem::Server.new [Gem.dir, Gem.dir], process_based_port, false
- assert_equal [File.join(Gem.dir, 'specifications'),
- File.join(Gem.dir, 'specifications')], s.spec_dirs
- end
- def test_latest_specs
- data = StringIO.new "GET /latest_specs.#{Gem.marshal_version} HTTP/1.0\r\n\r\n"
- @req.parse data
- Gem::Deprecate.skip_during do
- @server.latest_specs @req, @res
- end
- assert_equal 200, @res.status, @res.body
- assert_match %r| \d\d:\d\d:\d\d |, @res['date']
- assert_equal 'application/octet-stream', @res['content-type']
- assert_equal [['a', Gem::Version.new(2), Gem::Platform::RUBY]],
- Marshal.load(@res.body)
- end
- def test_latest_specs_gemdirs
- data = StringIO.new "GET /latest_specs.#{Gem.marshal_version} HTTP/1.0\r\n\r\n"
- dir = "#{@gemhome}2"
- spec = util_spec 'z', 9
- specs_dir = File.join dir, 'specifications'
- FileUtils.mkdir_p specs_dir
- open File.join(specs_dir, spec.spec_name), 'w' do |io|
- io.write spec.to_ruby
- end
- server = Gem::Server.new dir, process_based_port, false
- @req.parse data
- server.latest_specs @req, @res
- assert_equal 200, @res.status
- assert_equal [['z', v(9), Gem::Platform::RUBY]], Marshal.load(@res.body)
- end
- def test_latest_specs_gz
- data = StringIO.new "GET /latest_specs.#{Gem.marshal_version}.gz HTTP/1.0\r\n\r\n"
- @req.parse data
- Gem::Deprecate.skip_during do
- @server.latest_specs @req, @res
- end
- assert_equal 200, @res.status, @res.body
- assert_match %r| \d\d:\d\d:\d\d |, @res['date']
- assert_equal 'application/x-gzip', @res['content-type']
- assert_equal [['a', Gem::Version.new(2), Gem::Platform::RUBY]],
- Marshal.load(Gem.gunzip(@res.body))
- end
- def test_listen
- util_listen
- capture_io do
- @server.listen
- end
- assert_equal 1, @server.server.listeners.length
- end
- def test_listen_addresses
- util_listen
- capture_io do
- @server.listen %w[a b]
- end
- assert_equal 2, @server.server.listeners.length
- end
- def test_prerelease_specs
- data = StringIO.new "GET /prerelease_specs.#{Gem.marshal_version} HTTP/1.0\r\n\r\n"
- @req.parse data
- Gem::Deprecate.skip_during do
- @server.prerelease_specs @req, @res
- end
- assert_equal 200, @res.status, @res.body
- assert_match %r| \d\d:\d\d:\d\d |, @res['date']
- assert_equal 'application/octet-stream', @res['content-type']
- assert_equal [['a', v('3.a'), Gem::Platform::RUBY]],
- Marshal.load(@res.body)
- end
- def test_prerelease_specs_gz
- data = StringIO.new "GET /prerelease_specs.#{Gem.marshal_version}.gz HTTP/1.0\r\n\r\n"
- @req.parse data
- Gem::Deprecate.skip_during do
- @server.prerelease_specs @req, @res
- end
- assert_equal 200, @res.status, @res.body
- assert_match %r| \d\d:\d\d:\d\d |, @res['date']
- assert_equal 'application/x-gzip', @res['content-type']
- assert_equal [['a', v('3.a'), Gem::Platform::RUBY]],
- Marshal.load(Gem.gunzip(@res.body))
- end
- def test_quick_gemdirs
- data = StringIO.new "GET /quick/Marshal.4.8/z-9.gemspec.rz HTTP/1.0\r\n\r\n"
- dir = "#{@gemhome}2"
- server = Gem::Server.new dir, process_based_port, false
- @req.parse data
- server.quick @req, @res
- assert_equal 404, @res.status
- spec = util_spec 'z', 9
- specs_dir = File.join dir, 'specifications'
- FileUtils.mkdir_p specs_dir
- open File.join(specs_dir, spec.spec_name), 'w' do |io|
- io.write spec.to_ruby
- end
- data.rewind
- req = WEBrick::HTTPRequest.new :Logger => nil
- res = WEBrick::HTTPResponse.new :HTTPVersion => '1.0'
- req.parse data
- server.quick req, res
- assert_equal 200, res.status
- end
- def test_quick_missing
- data = StringIO.new "GET /quick/Marshal.4.8/z-9.gemspec.rz HTTP/1.0\r\n\r\n"
- @req.parse data
- @server.quick @req, @res
- assert_equal 404, @res.status, @res.body
- assert_match %r| \d\d:\d\d:\d\d |, @res['date']
- assert_equal 'text/plain', @res['content-type']
- assert_equal 'No gems found matching "z" "9" nil', @res.body
- assert_equal 404, @res.status
- end
- def test_quick_marshal_a_1_gemspec_rz
- data = StringIO.new "GET /quick/Marshal.#{Gem.marshal_version}/a-1.gemspec.rz HTTP/1.0\r\n\r\n"
- @req.parse data
- @server.quick @req, @res
- assert_equal 200, @res.status, @res.body
- assert @res['date']
- assert_equal 'application/x-deflate', @res['content-type']
- spec = Marshal.load Gem.inflate(@res.body)
- assert_equal 'a', spec.name
- assert_equal Gem::Version.new(1), spec.version
- end
- def test_quick_marshal_a_1_mswin32_gemspec_rz
- quick_gem 'a', '1' do |s| s.platform = Gem::Platform.local end
- data = StringIO.new "GET /quick/Marshal.#{Gem.marshal_version}/a-1-#{Gem::Platform.local}.gemspec.rz HTTP/1.0\r\n\r\n"
- @req.parse data
- @server.quick @req, @res
- assert_equal 200, @res.status, @res.body
- assert @res['date']
- assert_equal 'application/x-deflate', @res['content-type']
- spec = Marshal.load Gem.inflate(@res.body)
- assert_equal 'a', spec.name
- assert_equal Gem::Version.new(1), spec.version
- assert_equal Gem::Platform.local, spec.platform
- end
- def test_quick_marshal_a_3_a_gemspec_rz
- data = StringIO.new "GET /quick/Marshal.#{Gem.marshal_version}/a-3.a.gemspec.rz HTTP/1.0\r\n\r\n"
- @req.parse data
- @server.quick @req, @res
- assert_equal 200, @res.status, @res.body
- assert @res['date']
- assert_equal 'application/x-deflate', @res['content-type']
- spec = Marshal.load Gem.inflate(@res.body)
- assert_equal 'a', spec.name
- assert_equal v('3.a'), spec.version
- end
- def test_quick_marshal_a_b_3_a_gemspec_rz
- quick_gem 'a-b', '3.a'
- data = StringIO.new "GET /quick/Marshal.#{Gem.marshal_version}/a-b-3.a.gemspec.rz HTTP/1.0\r\n\r\n"
- @req.parse data
- @server.quick @req, @res
- assert_equal 200, @res.status, @res.body
- assert @res['date']
- assert_equal 'application/x-deflate', @res['content-type']
- spec = Marshal.load Gem.inflate(@res.body)
- assert_equal 'a-b', spec.name
- assert_equal v('3.a'), spec.version
- end
- def test_rdoc
- data = StringIO.new "GET /rdoc?q=a HTTP/1.0\r\n\r\n"
- @req.parse data
- @server.rdoc @req, @res
- assert_equal 200, @res.status, @res.body
- assert_match %r|No documentation found|, @res.body
- assert_equal 'text/html', @res['content-type']
- end
- def test_root
- data = StringIO.new "GET / HTTP/1.0\r\n\r\n"
- @req.parse data
- @server.root @req, @res
- assert_equal 200, @res.status, @res.body
- assert_match %r| \d\d:\d\d:\d\d |, @res['date']
- assert_equal 'text/html', @res['content-type']
- end
- def test_root_gemdirs
- data = StringIO.new "GET / HTTP/1.0\r\n\r\n"
- dir = "#{@gemhome}2"
- spec = util_spec 'z', 9
- specs_dir = File.join dir, 'specifications'
- FileUtils.mkdir_p specs_dir
- open File.join(specs_dir, spec.spec_name), 'w' do |io|
- io.write spec.to_ruby
- end
- server = Gem::Server.new dir, process_based_port, false
- @req.parse data
- server.root @req, @res
- assert_equal 200, @res.status
- assert_match 'z 9', @res.body
- end
- def test_specs
- data = StringIO.new "GET /specs.#{Gem.marshal_version} HTTP/1.0\r\n\r\n"
- @req.parse data
- @server.specs @req, @res
- assert_equal 200, @res.status, @res.body
- assert_match %r| \d\d:\d\d:\d\d |, @res['date']
- assert_equal 'application/octet-stream', @res['content-type']
- assert_equal [['a', Gem::Version.new(1), Gem::Platform::RUBY],
- ['a', Gem::Version.new(2), Gem::Platform::RUBY],
- ['a', v('3.a'), Gem::Platform::RUBY]],
- Marshal.load(@res.body)
- end
- def test_specs_gemdirs
- data = StringIO.new "GET /specs.#{Gem.marshal_version} HTTP/1.0\r\n\r\n"
- dir = "#{@gemhome}2"
- spec = util_spec 'z', 9
- specs_dir = File.join dir, 'specifications'
- FileUtils.mkdir_p specs_dir
- open File.join(specs_dir, spec.spec_name), 'w' do |io|
- io.write spec.to_ruby
- end
- server = Gem::Server.new dir, process_based_port, false
- @req.parse data
- server.specs @req, @res
- assert_equal 200, @res.status
- assert_equal [['z', v(9), Gem::Platform::RUBY]], Marshal.load(@res.body)
- end
- def test_specs_gz
- data = StringIO.new "GET /specs.#{Gem.marshal_version}.gz HTTP/1.0\r\n\r\n"
- @req.parse data
- @server.specs @req, @res
- assert_equal 200, @res.status, @res.body
- assert_match %r| \d\d:\d\d:\d\d |, @res['date']
- assert_equal 'application/x-gzip', @res['content-type']
- assert_equal [['a', Gem::Version.new(1), Gem::Platform::RUBY],
- ['a', Gem::Version.new(2), Gem::Platform::RUBY],
- ['a', v('3.a'), Gem::Platform::RUBY]],
- Marshal.load(Gem.gunzip(@res.body))
- end
- def util_listen
- webrick = Object.new
- webrick.instance_variable_set :@listeners, []
- def webrick.listeners() @listeners end
- def webrick.listen(host, port)
- socket = Object.new
- socket.instance_variable_set :@host, host
- socket.instance_variable_set :@port, port
- def socket.addr() [nil, @port, @host] end
- @listeners << socket
- end
- @server.instance_variable_set :@server, webrick
- end
- end