From d348ea338468e58b1ada0f20af5a76c69a33ae86 Mon Sep 17 00:00:00 2001
From: Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>
Date: Sat, 5 Sep 2015 01:24:43 +0900
Subject: [PATCH] integration: Fix test failures

---
 integration-tests/req-set-header.rb  | 4 ++--
 integration-tests/resp-set-header.rb | 4 ++--
 integration-tests/return.rb          | 4 +++-
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/integration-tests/req-set-header.rb b/integration-tests/req-set-header.rb
index 6f63e357..27e9dc34 100644
--- a/integration-tests/req-set-header.rb
+++ b/integration-tests/req-set-header.rb
@@ -1,3 +1,3 @@
-Nghttpx.run do |resp, req|
-  req.set_header "User-Agent", "mruby"
+Nghttpx.run do |env|
+  env.req.set_header "User-Agent", "mruby"
 end
diff --git a/integration-tests/resp-set-header.rb b/integration-tests/resp-set-header.rb
index a1f2d806..b947ce3b 100644
--- a/integration-tests/resp-set-header.rb
+++ b/integration-tests/resp-set-header.rb
@@ -1,3 +1,3 @@
-Nghttpx.run do |resp, req|
-  resp.set_header "Alpha", "bravo"
+Nghttpx.run do |env|
+  env.resp.set_header "Alpha", "bravo"
 end
diff --git a/integration-tests/return.rb b/integration-tests/return.rb
index 42b1e345..907d837f 100644
--- a/integration-tests/return.rb
+++ b/integration-tests/return.rb
@@ -1,4 +1,6 @@
-Nghttpx.run do |resp, req|
+Nghttpx.run do |env|
+  resp = env.resp
+
   resp.clear_headers
   resp.status = 404
   resp.add_header "from", "mruby"