test/plugin/in_http.rb in fluentd-0.10.42 vs test/plugin/in_http.rb in fluentd-0.10.43

@@ -23,10 +23,11 @@
     d = create_driver
     assert_equal PORT, d.instance.port
     assert_equal '127.0.0.1', d.instance.bind
     assert_equal 10*1024*1024, d.instance.body_size_limit
     assert_equal 5, d.instance.keepalive_timeout
+    assert_equal false, d.instance.add_http_headers
   end
 
   def test_time
     d = create_driver
 
@@ -56,12 +57,36 @@
       d.expected_emits.each {|tag,time,record|
         res = post("/#{tag}", {"json"=>record.to_json, "time"=>time.to_s})
         assert_equal "200", res.code
       }
     end
+
+    d.emit_streams.each { |tag, es|
+      assert !include_http_header?(es.first[1])
+    }
   end
 
+  def test_json_with_add_http_headers
+    d = create_driver(CONFIG + "add_http_headers true")
+
+    time = Time.parse("2011-01-02 13:14:15 UTC").to_i
+
+    records = [["tag1", time, {"a"=>1}], ["tag2", time, {"a"=>2}]]
+
+    d.run do
+      records.each {|tag,time,record|
+        res = post("/#{tag}", {"json"=>record.to_json, "time"=>time.to_s})
+        assert_equal "200", res.code
+
+      }
+    end
+
+    d.emit_streams.each { |tag, es|
+      assert include_http_header?(es.first[1])
+    }
+  end
+
   def test_application_json
     d = create_driver
 
     time = Time.parse("2011-01-02 13:14:15 UTC").to_i
 
@@ -99,7 +124,10 @@
     http = Net::HTTP.new("127.0.0.1", PORT)
     req = Net::HTTP::Post.new(path, {})
     req.set_form_data(params)
     http.request(req)
   end
-end
 
+  def include_http_header?(record)
+    record.keys.find { |header| header.start_with?('HTTP_') }
+  end
+end