le2 ex1 test

ex1_test.rb

port = 1234
port = ARGV[0].to_i if ARGV[0]

puts "Test #1"
hoge_txt = IO.read('hoge.txt')
server_pid = spawn("./server #{port}")
begin
  result = `./client localhost #{port} hoge.txt`
  result.gsub("\r\n", "\n")
  if result.index(hoge_txt)
    puts "ooo OK ooo"
  else
    puts "!!! Failed !!!"
  end
ensure
  Process.kill(:SIGTERM, server_pid)
end

puts "Test #2"
server_pid = spawn("./server #{port + 1}")
begin
  result = `./client localhost #{port + 1} fuga.txt`
  if /404 Not Found/ =~ result
    puts "ooo OK ooo"
  else
    puts "!!! Failed !!!"
  end
ensure
  Process.kill(:SIGTERM, server_pid)
end

puts "Test #3"
huge_txt = IO.read('huge.txt')
server_pid = spawn("./server #{port + 2}")
begin
  result = `./client localhost #{port + 2} huge.txt`
  result.gsub("\r\n", "\n")
  if result.index(huge_txt)
    puts "ooo OK ooo"
  else
    puts "!!! Failed !!!"
  end
ensure
  Process.kill(:SIGTERM, server_pid)
end

puts "Test #4"
server_pid = spawn("./server #{port + 3}")
begin
  ok = true

  result = `./client localhost #{port + 3} hoge.txt`
  result.gsub("\r\n", "\n")
  ok = false unless result.index(hoge_txt)

  result = `./client localhost #{port + 3} hoge.txt`
  result.gsub("\r\n", "\n")
  ok = false unless result.index(hoge_txt)

  result = `./client localhost #{port + 3} foobar`
  result.gsub("\r\n", "\n")
  ok = false unless /404 Not Found/ =~ result

  result = `./client localhost #{port + 3} huge.txt`
  result.gsub("\r\n", "\n")
  ok = false unless result.index(huge_txt)

  if ok
    puts "ooo OK ooo"
  else
    puts "!!! Failed !!!"
  end
ensure
  Process.kill(:SIGTERM, server_pid)
end