test/test_network.rb

   1 require 'test/setup'
   2 require 'mogilefs'
   3 require 'mogilefs/network'
   4
   5 class TestNetwork < Test::Unit::TestCase
   6   include MogileFS::Network
   7
   8   def test_verify_uris
   9     good = TempServer.new(Proc.new do |serv,port|
  10       client,client_addr = serv.accept
  11       client.readpartial(4096)
  12       client.syswrite("HTTP/1.0 200 OK\r\nContent-Length: 0\r\n\r\n")
  13     end)
  14     bad = TempServer.new(Proc.new do |serv,port|
  15       client, client_addr = serv.accept
  16       client.close rescue nil
  17     end)
  18
  19     good_uri = URI.parse("http://127.0.0.1:#{good.port}/")
  20     bad_uri = URI.parse("http://127.0.0.1:#{bad.port}/")
  21     ok = verify_uris([ good_uri, bad_uri ])
  22     assert_equal [ good_uri ], ok
  23     ensure
  24       TempServer.destroy_all!
  25   end
  26
  27 end