test/test_backend.rb

   1 require 'test/unit'
   2 require 'test/setup'
   3
   4 $TESTING = true
   5
   6 require 'mogilefs/backend'
   7
   8 class MogileFS::Backend
   9
  10   attr_accessor :hosts
  11   attr_reader :timeout, :dead
  12   attr_writer :lasterr, :lasterrstr, :socket
  13
  14 end
  15
  16 class TestBackend < Test::Unit::TestCase
  17
  18   def setup
  19     TCPSocket.connections = []
  20     TCPSocket.sockets = []
  21     @backend = MogileFS::Backend.new :hosts => ['localhost:1']
  22   end
  23
  24   def test_initialize
  25     assert_raises ArgumentError do MogileFS::Backend.new end
  26     assert_raises ArgumentError do MogileFS::Backend.new :hosts => [] end
  27     assert_raises ArgumentError do MogileFS::Backend.new :hosts => [''] end
  28
  29     assert_equal ['localhost:1'], @backend.hosts
  30     assert_equal 3, @backend.timeout
  31     assert_equal nil, @backend.lasterr
  32     assert_equal nil, @backend.lasterrstr
  33     assert_equal({}, @backend.dead)
  34
  35     @backend = MogileFS::Backend.new :hosts => ['localhost:6001'], :timeout => 1
  36     assert_equal 1, @backend.timeout
  37   end
  38
  39   def test_do_request
  40     socket_request = ''
  41     socket = Object.new
  42     def socket.closed?() false end
  43     def socket.send(request, flags) return request.length end
  44     def @backend.select(*args) return [true] end
  45     def socket.gets() return 'OK 1 you=win' end
  46
  47     @backend.instance_variable_set '@socket', socket
  48
  49     assert_equal({'you' => 'win'},
  50                  @backend.do_request('go!', { 'fight' => 'team fight!' }))
  51   end
  52
  53   def test_do_request_send_error
  54     socket_request = ''
  55     socket = Object.new
  56     def socket.closed?() false end
  57     def socket.send(request, flags) raise SystemCallError, 'dummy' end
  58
  59     @backend.instance_variable_set '@socket', socket
  60
  61     assert_raises RuntimeError do
  62       @backend.do_request 'go!', { 'fight' => 'team fight!' }
  63     end
  64
  65     assert_equal nil, @backend.instance_variable_get('@socket')
  66   end
  67
  68   def test_automatic_exception
  69     assert ! MogileFS::Backend.const_defined?('PebkacError')
  70     assert @backend.error('pebkac')
  71     assert_equal MogileFS::Error, @backend.error('PebkacError').superclass
  72     assert MogileFS::Backend.const_defined?('PebkacError')
  73
  74     assert ! MogileFS::Backend.const_defined?('PebKacError')
  75     assert @backend.error('peb_kac')
  76     assert_equal MogileFS::Error, @backend.error('PebKacError').superclass
  77     assert MogileFS::Backend.const_defined?('PebKacError')
  78   end
  79
  80   def test_do_request_truncated
  81     socket_request = ''
  82     socket = Object.new
  83     def socket.closed?() false end
  84     def socket.send(request, flags) return request.length - 1 end
  85
  86     @backend.instance_variable_set '@socket', socket
  87
  88     assert_raises RuntimeError do
  89       @backend.do_request 'go!', { 'fight' => 'team fight!' }
  90     end
  91   end
  92
  93   def test_make_request
  94     assert_equal "go! fight=team+fight%21\r\n",
  95                  @backend.make_request('go!', { 'fight' => 'team fight!' })
  96   end
  97
  98   def test_parse_response
  99     assert_equal({'foo' => 'bar', 'baz' => 'hoge'},
 100                  @backend.parse_response('OK 1 foo=bar&baz=hoge'))
 101
 102     err = nil
 103     begin
 104       @backend.parse_response('ERR you totally suck')
 105     rescue MogileFS::Error => err
 106       assert_equal 'MogileFS::Backend::YouError', err.class.to_s
 107     end
 108     assert_equal 'MogileFS::Backend::YouError', err.class.to_s
 109
 110     assert_equal 'you', @backend.lasterr
 111     assert_equal 'totally suck', @backend.lasterrstr
 112
 113     assert_raises RuntimeError do
 114       @backend.parse_response 'garbage'
 115     end
 116   end
 117
 118   def test_readable_eh_readable
 119     socket = Object.new
 120     def socket.closed?() false end
 121     def @backend.select(*args) return [true] end
 122     @backend.instance_variable_set '@socket', socket
 123
 124     assert_equal true, @backend.readable?
 125   end
 126
 127   def test_readable_eh_not_readable
 128     socket = FakeSocket.new
 129     def socket.closed?() false end
 130     def @backend.select(*args) return [] end
 131     @backend.instance_variable_set '@socket', socket
 132
 133     begin
 134       @backend.readable?
 135     rescue MogileFS::UnreadableSocketError => e
 136       assert_equal '127.0.0.1:6001 never became readable', e.message
 137     rescue Exception
 138       flunk "MogileFS::UnreadableSocketError not raised"
 139     else
 140       flunk "MogileFS::UnreadableSocketError not raised"
 141     end
 142   end
 143
 144   def test_socket
 145     assert_equal({}, @backend.dead)
 146     assert_raises RuntimeError do @backend.socket end
 147     assert_equal(['localhost:1'], @backend.dead.keys)
 148   end
 149
 150   def test_socket_robust
 151     @backend.hosts = ['localhost:6001', 'localhost:6002']
 152     def @backend.connect_to(host, port)
 153       @first = (defined? @first) ? false : true
 154       raise Errno::ECONNREFUSED if @first
 155     end
 156
 157     assert_equal({}, @backend.dead)
 158     @backend.socket
 159     assert_equal false, @backend.dead.keys.empty?
 160   end
 161
 162   def test_shutdown
 163     fake_socket = FakeSocket.new
 164     @backend.socket = fake_socket
 165     assert_equal fake_socket, @backend.socket
 166     @backend.shutdown
 167     assert_equal nil, @backend.instance_variable_get(:@socket)
 168   end
 169
 170   def test_url_decode
 171     assert_equal({"\272z" => "\360opy", "f\000" => "\272r"},
 172                  @backend.url_decode("%baz=%f0opy&f%00=%bar"))
 173   end
 174
 175   def test_url_encode
 176     params = [["f\000", "\272r"], ["\272z", "\360opy"]]
 177     assert_equal "f%00=%bar&%baz=%f0opy", @backend.url_encode(params)
 178   end
 179
 180   def test_url_escape # \n for unit_diff
 181     actual = (0..255).map { |c| @backend.url_escape c.chr }.join "\n"
 182
 183     expected = []
 184     expected.push(*(0..0x1f).map { |c| "%%%0.2x" % c })
 185     expected << '+'
 186     expected.push(*(0x21..0x2b).map { |c| "%%%0.2x" % c })
 187     expected.push(*%w[, - . /])
 188     expected.push(*('0'..'9'))
 189     expected.push(*%w[: %3b %3c %3d %3e %3f %40])
 190     expected.push(*('A'..'Z'))
 191     expected.push(*%w[%5b \\ %5d %5e _ %60])
 192     expected.push(*('a'..'z'))
 193     expected.push(*(0x7b..0xff).map { |c| "%%%0.2x" % c })
 194
 195     expected = expected.join "\n"
 196
 197     assert_equal expected, actual
 198   end
 199
 200   def test_url_unescape
 201     input = []
 202     input.push(*(0..0x1f).map { |c| "%%%0.2x" % c })
 203     input << '+'
 204     input.push(*(0x21..0x2b).map { |c| "%%%0.2x" % c })
 205     input.push(*%w[, - . /])
 206     input.push(*('0'..'9'))
 207     input.push(*%w[: %3b %3c %3d %3e %3f %40])
 208     input.push(*('A'..'Z'))
 209     input.push(*%w[%5b \\ %5d %5e _ %60])
 210     input.push(*('a'..'z'))
 211     input.push(*(0x7b..0xff).map { |c| "%%%0.2x" % c })
 212
 213     actual = input.map { |c| @backend.url_unescape c }.join "\n"
 214
 215     expected = (0..255).map { |c| c.chr }.join "\n"
 216     expected.sub! '+', ' '
 217
 218     assert_equal expected, actual
 219   end
 220
 221 end
 222