test/rubygems/test_gem_server.rb

   1 require 'test/unit'
   2 require File.join(File.expand_path(File.dirname(__FILE__)), 'gemutilities')
   3 require 'rubygems/server'
   4 require 'stringio'
   5
   6 class Gem::Server
   7   attr_accessor :source_index
   8   attr_reader :server
   9 end
  10
  11 class TestGemServer < RubyGemTestCase
  12
  13   def setup
  14     super
  15
  16     @a1 = quick_gem 'a', '1'
  17     @a2 = quick_gem 'a', '2'
  18
  19     @server = Gem::Server.new Gem.dir, process_based_port, false
  20     @req = WEBrick::HTTPRequest.new :Logger => nil
  21     @res = WEBrick::HTTPResponse.new :HTTPVersion => '1.0'
  22   end
  23
  24   def test_Marshal
  25     data = StringIO.new "GET /Marshal.#{Gem.marshal_version} HTTP/1.0\r\n\r\n"
  26     @req.parse data
  27
  28     @server.Marshal @req, @res
  29
  30     assert_equal 200, @res.status, @res.body
  31     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
  32     assert_equal 'application/octet-stream', @res['content-type']
  33
  34     si = Gem::SourceIndex.new
  35     si.add_specs @a1, @a2
  36
  37     assert_equal si, Marshal.load(@res.body)
  38   end
  39
  40   def test_Marshal_Z
  41     data = StringIO.new "GET /Marshal.#{Gem.marshal_version}.Z HTTP/1.0\r\n\r\n"
  42     @req.parse data
  43
  44     @server.Marshal @req, @res
  45
  46     assert_equal 200, @res.status, @res.body
  47     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
  48     assert_equal 'application/x-deflate', @res['content-type']
  49
  50     si = Gem::SourceIndex.new
  51     si.add_specs @a1, @a2
  52
  53     assert_equal si, Marshal.load(Gem.inflate(@res.body))
  54   end
  55
  56   def test_latest_specs
  57     data = StringIO.new "GET /latest_specs.#{Gem.marshal_version} HTTP/1.0\r\n\r\n"
  58     @req.parse data
  59
  60     @server.latest_specs @req, @res
  61
  62     assert_equal 200, @res.status, @res.body
  63     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
  64     assert_equal 'application/octet-stream', @res['content-type']
  65     assert_equal [['a', Gem::Version.new(2), Gem::Platform::RUBY]],
  66                  Marshal.load(@res.body)
  67   end
  68
  69   def test_latest_specs_gz
  70     data = StringIO.new "GET /latest_specs.#{Gem.marshal_version}.gz HTTP/1.0\r\n\r\n"
  71     @req.parse data
  72
  73     @server.latest_specs @req, @res
  74
  75     assert_equal 200, @res.status, @res.body
  76     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
  77     assert_equal 'application/x-gzip', @res['content-type']
  78     assert_equal [['a', Gem::Version.new(2), Gem::Platform::RUBY]],
  79                  Marshal.load(Gem.gunzip(@res.body))
  80   end
  81
  82   def test_quick_a_1_gemspec_rz
  83     data = StringIO.new "GET /quick/a-1.gemspec.rz HTTP/1.0\r\n\r\n"
  84     @req.parse data
  85
  86     @server.quick @req, @res
  87
  88     assert_equal 200, @res.status, @res.body
  89     assert @res['date']
  90     assert_equal 'application/x-deflate', @res['content-type']
  91
  92     spec = YAML.load Gem.inflate(@res.body)
  93     assert_equal 'a', spec.name
  94     assert_equal Gem::Version.new(1), spec.version
  95   end
  96
  97   def test_quick_a_1_mswin32_gemspec_rz
  98     a1_p = quick_gem 'a', '1' do |s| s.platform = Gem::Platform.local end
  99
 100     data = StringIO.new "GET /quick/a-1-#{Gem::Platform.local}.gemspec.rz HTTP/1.0\r\n\r\n"
 101     @req.parse data
 102
 103     @server.quick @req, @res
 104
 105     assert_equal 200, @res.status, @res.body
 106     assert @res['date']
 107     assert_equal 'application/x-deflate', @res['content-type']
 108
 109     spec = YAML.load Gem.inflate(@res.body)
 110     assert_equal 'a', spec.name
 111     assert_equal Gem::Version.new(1), spec.version
 112     assert_equal Gem::Platform.local, spec.platform
 113   end
 114
 115   def test_quick_common_substrings
 116     ab1 = quick_gem 'ab', '1'
 117
 118     data = StringIO.new "GET /quick/a-1.gemspec.rz HTTP/1.0\r\n\r\n"
 119     @req.parse data
 120
 121     @server.quick @req, @res
 122
 123     assert_equal 200, @res.status, @res.body
 124     assert @res['date']
 125     assert_equal 'application/x-deflate', @res['content-type']
 126
 127     spec = YAML.load Gem.inflate(@res.body)
 128     assert_equal 'a', spec.name
 129     assert_equal Gem::Version.new(1), spec.version
 130   end
 131
 132   def test_quick_index
 133     data = StringIO.new "GET /quick/index HTTP/1.0\r\n\r\n"
 134     @req.parse data
 135
 136     @server.quick @req, @res
 137
 138     assert_equal 200, @res.status, @res.body
 139     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
 140     assert_equal 'text/plain', @res['content-type']
 141     assert_equal "a-1\na-2", @res.body
 142   end
 143
 144   def test_quick_index_rz
 145     data = StringIO.new "GET /quick/index.rz HTTP/1.0\r\n\r\n"
 146     @req.parse data
 147
 148     @server.quick @req, @res
 149
 150     assert_equal 200, @res.status, @res.body
 151     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
 152     assert_equal 'application/x-deflate', @res['content-type']
 153     assert_equal "a-1\na-2", Gem.inflate(@res.body)
 154   end
 155
 156   def test_quick_latest_index
 157     data = StringIO.new "GET /quick/latest_index HTTP/1.0\r\n\r\n"
 158     @req.parse data
 159
 160     @server.quick @req, @res
 161
 162     assert_equal 200, @res.status, @res.body
 163     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
 164     assert_equal 'text/plain', @res['content-type']
 165     assert_equal 'a-2', @res.body
 166   end
 167
 168   def test_quick_latest_index_rz
 169     data = StringIO.new "GET /quick/latest_index.rz HTTP/1.0\r\n\r\n"
 170     @req.parse data
 171
 172     @server.quick @req, @res
 173
 174     assert_equal 200, @res.status, @res.body
 175     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
 176     assert_equal 'application/x-deflate', @res['content-type']
 177     assert_equal 'a-2', Gem.inflate(@res.body)
 178   end
 179
 180   def test_quick_missing
 181     data = StringIO.new "GET /quick/z-9.gemspec.rz HTTP/1.0\r\n\r\n"
 182     @req.parse data
 183
 184     @server.quick @req, @res
 185
 186     assert_equal 404, @res.status, @res.body
 187     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
 188     assert_equal 'text/plain', @res['content-type']
 189     assert_equal 'No gems found matching "z" "9" nil', @res.body
 190     assert_equal 404, @res.status
 191   end
 192
 193   def test_quick_marshal_a_1_gemspec_rz
 194     data = StringIO.new "GET /quick/Marshal.#{Gem.marshal_version}/a-1.gemspec.rz HTTP/1.0\r\n\r\n"
 195     @req.parse data
 196
 197     @server.quick @req, @res
 198
 199     assert_equal 200, @res.status, @res.body
 200     assert @res['date']
 201     assert_equal 'application/x-deflate', @res['content-type']
 202
 203     spec = Marshal.load Gem.inflate(@res.body)
 204     assert_equal 'a', spec.name
 205     assert_equal Gem::Version.new(1), spec.version
 206   end
 207
 208   def test_quick_marshal_a_1_mswin32_gemspec_rz
 209     a1_p = quick_gem 'a', '1' do |s| s.platform = Gem::Platform.local end
 210
 211     data = StringIO.new "GET /quick/Marshal.#{Gem.marshal_version}/a-1-#{Gem::Platform.local}.gemspec.rz HTTP/1.0\r\n\r\n"
 212     @req.parse data
 213
 214     @server.quick @req, @res
 215
 216     assert_equal 200, @res.status, @res.body
 217     assert @res['date']
 218     assert_equal 'application/x-deflate', @res['content-type']
 219
 220     spec = Marshal.load Gem.inflate(@res.body)
 221     assert_equal 'a', spec.name
 222     assert_equal Gem::Version.new(1), spec.version
 223     assert_equal Gem::Platform.local, spec.platform
 224   end
 225
 226
 227   def test_root
 228     data = StringIO.new "GET / HTTP/1.0\r\n\r\n"
 229     @req.parse data
 230
 231     @server.root @req, @res
 232
 233     assert_equal 200, @res.status, @res.body
 234     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
 235     assert_equal 'text/html', @res['content-type']
 236   end
 237
 238   def test_specs
 239     data = StringIO.new "GET /specs.#{Gem.marshal_version} HTTP/1.0\r\n\r\n"
 240     @req.parse data
 241
 242     @server.specs @req, @res
 243
 244     assert_equal 200, @res.status, @res.body
 245     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
 246     assert_equal 'application/octet-stream', @res['content-type']
 247
 248     assert_equal [['a', Gem::Version.new(1), Gem::Platform::RUBY],
 249                   ['a', Gem::Version.new(2), Gem::Platform::RUBY]],
 250                  Marshal.load(@res.body)
 251   end
 252
 253   def test_specs_gz
 254     data = StringIO.new "GET /specs.#{Gem.marshal_version}.gz HTTP/1.0\r\n\r\n"
 255     @req.parse data
 256
 257     @server.specs @req, @res
 258
 259     assert_equal 200, @res.status, @res.body
 260     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
 261     assert_equal 'application/x-gzip', @res['content-type']
 262
 263     assert_equal [['a', Gem::Version.new(1), Gem::Platform::RUBY],
 264                   ['a', Gem::Version.new(2), Gem::Platform::RUBY]],
 265                  Marshal.load(Gem.gunzip(@res.body))
 266   end
 267
 268   def test_yaml
 269     data = StringIO.new "GET /yaml.#{Gem.marshal_version} HTTP/1.0\r\n\r\n"
 270     @req.parse data
 271
 272     @server.yaml @req, @res
 273
 274     assert_equal 200, @res.status, @res.body
 275     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
 276     assert_equal 'text/plain', @res['content-type']
 277
 278     si = Gem::SourceIndex.new
 279     si.add_specs @a1, @a2
 280
 281     assert_equal si, YAML.load(@res.body)
 282   end
 283
 284   def test_yaml_Z
 285     data = StringIO.new "GET /yaml.#{Gem.marshal_version}.Z HTTP/1.0\r\n\r\n"
 286     @req.parse data
 287
 288     @server.yaml @req, @res
 289
 290     assert_equal 200, @res.status, @res.body
 291     assert_match %r| \d\d:\d\d:\d\d |, @res['date']
 292     assert_equal 'application/x-deflate', @res['content-type']
 293
 294     si = Gem::SourceIndex.new
 295     si.add_specs @a1, @a2
 296
 297     assert_equal si, YAML.load(Gem.inflate(@res.body))
 298   end
 299
 300 end
 301