test/test_network.rb

   1 # -*- encoding: binary -*-
   2 require './test/setup'
   3 require 'mogilefs'
   4 require 'mogilefs/network'
   5
   6 class TestNetwork < Test::Unit::TestCase
   7   include MogileFS::Network
   8
   9   def test_verify_uris
  10     good = TempServer.new(Proc.new do |serv,port|
  11       client,client_addr = serv.accept
  12       client.readpartial(4096)
  13       client.syswrite("HTTP/1.0 200 OK\r\nContent-Length: 0\r\n\r\n")
  14     end)
  15     bad = TempServer.new(Proc.new do |serv,port|
  16       client, client_addr = serv.accept
  17       client.close rescue nil
  18     end)
  19
  20     good_uri = URI.parse("http://127.0.0.1:#{good.port}/")
  21     bad_uri = URI.parse("http://127.0.0.1:#{bad.port}/")
  22     ok = verify_uris([ good_uri, bad_uri ])
  23     assert_equal [ good_uri ], ok
  24     ensure
  25       TempServer.destroy_all!
  26   end
  27
  28   def test_verify_non_existent
  29     good = TempServer.new(Proc.new do |serv,port|
  30       client,client_addr = serv.accept
  31       client.readpartial(4096)
  32       client.syswrite("HTTP/1.0 200 OK\r\nContent-Length: 0\r\n\r\n")
  33     end)
  34     bad = TempServer.new(Proc.new { |serv,port| serv.close })
  35
  36     good_uri = URI.parse("http://127.0.0.1:#{good.port}/")
  37     bad_uri = URI.parse("http://127.0.0.1:#{bad.port}/")
  38     ok = verify_uris([ good_uri, bad_uri ])
  39     assert_equal [ good_uri ], ok
  40     ensure
  41       TempServer.destroy_all!
  42   end
  43
  44   def test_verify_all_bad
  45     good = TempServer.new(Proc.new { |serv,port| serv.close })
  46     bad = TempServer.new(Proc.new { |serv,port| serv.close })
  47
  48     good_uri = URI.parse("http://127.0.0.1:#{good.port}/")
  49     bad_uri = URI.parse("http://127.0.0.1:#{bad.port}/")
  50     ok = verify_uris([ good_uri, bad_uri ], '200', 1.0)
  51     assert ok.empty?, "nothing returned"
  52     ensure
  53       TempServer.destroy_all!
  54   end
  55
  56 end