fluentd 1.13.0 → 1.13.1

data/test/plugin/test_in_http.rb

@@ -19,29 +19,32 @@ class HttpInputTest < Test::Unit::TestCase
 
   def setup
     Fluent::Test.setup
+    @port = unused_port
   end
 
   def teardown
     Timecop.return
+    @port = nil
   end
 
-  PORT = unused_port
-  CONFIG = %[
-    port #{PORT}
-    bind "127.0.0.1"
-    body_size_limit 10m
-    keepalive_timeout 5
-    respond_with_empty_img true
-    use_204_response false
-  ]
-
-  def create_driver(conf=CONFIG)
+  def config
+    %[
+      port #{@port}
+      bind "127.0.0.1"
+      body_size_limit 10m
+      keepalive_timeout 5
+      respond_with_empty_img true
+     use_204_response false
+    ]
+  end
+
+  def create_driver(conf=config)
     Fluent::Test::Driver::Input.new(Fluent::Plugin::HttpInput).configure(conf)
   end
 
   def test_configure
     d = create_driver
-    assert_equal PORT, d.instance.port
+    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
@@ -121,7 +124,7 @@ class HttpInputTest < Test::Unit::TestCase
        'msgpack' => ['msgpack', :to_msgpack])
   def test_default_with_time_format(data)
     param, method_name = data
-    d = create_driver(CONFIG + %[
+    d = create_driver(config + %[
       <parse>
         keep_time_key
         time_format %iso8601
@@ -198,7 +201,7 @@ class HttpInputTest < Test::Unit::TestCase
        'msgpack' => ['msgpack', :to_msgpack])
   def test_default_multi_with_time_format(data)
     param, method_name = data
-    d = create_driver(CONFIG + %[
+    d = create_driver(config + %[
       <parse>
         keep_time_key
         time_format %iso8601
@@ -222,7 +225,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_multi_json_with_nonexistent_time_key
-    d = create_driver(CONFIG + %[
+    d = create_driver(config + %[
       <parse>
         time_key missing
       </parse>
@@ -245,7 +248,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_json_with_add_remote_addr
-    d = create_driver(CONFIG + "add_remote_addr true")
+    d = create_driver(config + "add_remote_addr true")
     time = event_time("2011-01-02 13:14:15 UTC")
     time_i = time.to_i
 
@@ -267,7 +270,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_exact_match_for_expect
-    d = create_driver(CONFIG)
+    d = create_driver(config)
     records = [{ "a" => 1}, { "a" => 2 }]
     tag = "tag1"
     res_codes = []
@@ -280,7 +283,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_exact_match_for_expect_with_other_header
-    d = create_driver(CONFIG)
+    d = create_driver(config)
 
     records = [{ "a" => 1}, { "a" => 2 }]
     tag = "tag1"
@@ -299,7 +302,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_multi_json_with_add_remote_addr
-    d = create_driver(CONFIG + "add_remote_addr true")
+    d = create_driver(config + "add_remote_addr true")
     time = event_time("2011-01-02 13:14:15 UTC")
     time_i = time.to_i
 
@@ -324,7 +327,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_json_with_add_remote_addr_given_x_forwarded_for
-    d = create_driver(CONFIG + "add_remote_addr true")
+    d = create_driver(config + "add_remote_addr true")
     time = event_time("2011-01-02 13:14:15 UTC")
     time_i = time.to_i
 
@@ -352,7 +355,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_multi_json_with_add_remote_addr_given_x_forwarded_for
-    d = create_driver(CONFIG + "add_remote_addr true")
+    d = create_driver(config + "add_remote_addr true")
 
     tag = "tag1"
     time = event_time("2011-01-02 13:14:15 UTC")
@@ -375,7 +378,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_add_remote_addr_given_multi_x_forwarded_for
-    d = create_driver(CONFIG + "add_remote_addr true")
+    d = create_driver(config + "add_remote_addr true")
 
     tag = "tag1"
     time = event_time("2011-01-02 13:14:15 UTC")
@@ -406,7 +409,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_multi_json_with_add_http_headers
-    d = create_driver(CONFIG + "add_http_headers true")
+    d = create_driver(config + "add_http_headers true")
     time = event_time("2011-01-02 13:14:15 UTC")
     time_i = time.to_i
     records = [{"a"=>1},{"a"=>2}]
@@ -432,7 +435,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_json_with_add_http_headers
-    d = create_driver(CONFIG + "add_http_headers true")
+    d = create_driver(config + "add_http_headers true")
     time = event_time("2011-01-02 13:14:15 UTC")
     time_i = time.to_i
     events = [
@@ -462,7 +465,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_multi_json_with_custom_parser
-    d = create_driver(CONFIG + %[
+    d = create_driver(config + %[
         <parse>
           @type json
           keep_time_key true
@@ -584,7 +587,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_with_regexp
-    d = create_driver(CONFIG + %[
+    d = create_driver(config + %[
       format /^(?<field_1>\\d+):(?<field_2>\\w+)$/
       types field_1:integer
     ])
@@ -615,7 +618,7 @@ class HttpInputTest < Test::Unit::TestCase
   def test_with_csv
     require 'csv'
 
-    d = create_driver(CONFIG + %[
+    d = create_driver(config + %[
       format csv
       keys foo,bar
     ])
@@ -641,7 +644,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_response_with_empty_img
-    d = create_driver(CONFIG)
+    d = create_driver(config)
     assert_equal true, d.instance.respond_with_empty_img
 
     time = event_time("2011-01-02 13:14:15 UTC")
@@ -669,7 +672,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_response_without_empty_img
-    d = create_driver(CONFIG + "respond_with_empty_img false")
+    d = create_driver(config + "respond_with_empty_img false")
     assert_equal false, d.instance.respond_with_empty_img
 
     time = event_time("2011-01-02 13:14:15 UTC")
@@ -695,7 +698,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_response_use_204_response
-    d = create_driver(CONFIG + %[
+    d = create_driver(config + %[
           respond_with_empty_img false
           use_204_response true
         ])
@@ -724,7 +727,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_cors_allowed
-    d = create_driver(CONFIG + "cors_allow_origins [\"http://foo.com\"]")
+    d = create_driver(config + "cors_allow_origins [\"http://foo.com\"]")
     assert_equal ["http://foo.com"], d.instance.cors_allow_origins
 
     time = event_time("2011-01-02 13:14:15 UTC")
@@ -751,7 +754,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_cors_allowed_wildcard
-    d = create_driver(CONFIG + 'cors_allow_origins ["*"]')
+    d = create_driver(config + 'cors_allow_origins ["*"]')
 
     time = event_time("2011-01-02 13:14:15 UTC")
     events = [
@@ -771,7 +774,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_get_request
-    d = create_driver(CONFIG)
+    d = create_driver(config)
 
     d.run do
       res = get("/cors.test", {}, {})
@@ -780,7 +783,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_cors_preflight
-    d = create_driver(CONFIG + 'cors_allow_origins ["*"]')
+    d = create_driver(config + 'cors_allow_origins ["*"]')
 
     d.run do
       header = {
@@ -797,7 +800,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_cors_allowed_wildcard_for_subdomain
-    d = create_driver(CONFIG + 'cors_allow_origins ["http://*.foo.com"]')
+    d = create_driver(config + 'cors_allow_origins ["http://*.foo.com"]')
 
     time = event_time("2011-01-02 13:14:15 UTC")
     events = [
@@ -817,7 +820,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_cors_allowed_exclude_empty_string
-    d = create_driver(CONFIG + 'cors_allow_origins ["", "http://*.foo.com"]')
+    d = create_driver(config + 'cors_allow_origins ["", "http://*.foo.com"]')
 
     time = event_time("2011-01-02 13:14:15 UTC")
     events = [
@@ -837,7 +840,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_cors_allowed_wildcard_preflight_for_subdomain
-    d = create_driver(CONFIG + 'cors_allow_origins ["http://*.foo.com"]')
+    d = create_driver(config + 'cors_allow_origins ["http://*.foo.com"]')
 
     d.run do
       header = {
@@ -900,7 +903,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_cors_disallowed
-    d = create_driver(CONFIG + "cors_allow_origins [\"http://foo.com\"]")
+    d = create_driver(config + "cors_allow_origins [\"http://foo.com\"]")
     assert_equal ["http://foo.com"], d.instance.cors_allow_origins
 
     time = event_time("2011-01-02 13:14:15 UTC")
@@ -918,7 +921,7 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def test_add_query_params 
-    d = create_driver(CONFIG + "add_query_params true")
+    d = create_driver(config + "add_query_params true")
     assert_equal true, d.instance.add_query_params
 
     time = event_time("2011-01-02 13:14:15 UTC")
@@ -978,7 +981,7 @@ class HttpInputTest < Test::Unit::TestCase
       $test_in_http_content_types_flag = true
       d.run do
         # Send two requests the second one has no Content-Type in Keep-Alive
-        Net::HTTP.start("127.0.0.1", PORT) do |http|
+        Net::HTTP.start("127.0.0.1", @port) do |http|
           req = Net::HTTP::Post.new("/foodb/bartbl", {"connection" => "keepalive", "Content-Type" => "application/json"})
           http.request(req)
           req = Net::HTTP::Get.new("/foodb/bartbl", {"connection" => "keepalive"})
@@ -995,19 +998,19 @@ class HttpInputTest < Test::Unit::TestCase
   end
 
   def get(path, params, header = {})
-    http = Net::HTTP.new("127.0.0.1", PORT)
+    http = Net::HTTP.new("127.0.0.1", @port)
     req = Net::HTTP::Get.new(path, header)
     http.request(req)
   end
 
   def options(path, params, header = {})
-    http = Net::HTTP.new("127.0.0.1", PORT)
+    http = Net::HTTP.new("127.0.0.1", @port)
     req = Net::HTTP::Options.new(path, header)
     http.request(req)
   end
 
   def post(path, params, header = {}, &block)
-    http = Net::HTTP.new("127.0.0.1", PORT)
+    http = Net::HTTP.new("127.0.0.1", @port)
     req = Net::HTTP::Post.new(path, header)
     block.call(http, req) if block
     if params.is_a?(String)

data/test/plugin/test_in_syslog.rb

@@ -5,42 +5,51 @@ require 'fluent/plugin/in_syslog'
 class SyslogInputTest < Test::Unit::TestCase
   def setup
     Fluent::Test.setup
+    @port = unused_port
   end
 
-  PORT = unused_port
-  CONFIG = %[
-    port #{PORT}
-    bind 127.0.0.1
-    tag syslog
-  ]
+  def teardown
+    @port = nil
+  end
+
+  def ipv4_config
+    %[
+      port #{@port}
+      bind 127.0.0.1
+      tag syslog
+    ]
+  end
 
-  IPv6_CONFIG = %[
-    port #{PORT}
-    bind ::1
-    tag syslog
-  ]
+  def ipv6_config
+    %[
+      port #{@port}
+      bind ::1
+      tag syslog
+    ]
+  end
 
-  def create_driver(conf=CONFIG)
+  def create_driver(conf=ipv4_config)
     Fluent::Test::Driver::Input.new(Fluent::Plugin::SyslogInput).configure(conf)
   end
 
   data(
-    ipv4: ['127.0.0.1', CONFIG, ::Socket::AF_INET],
-    ipv6: ['::1', IPv6_CONFIG, ::Socket::AF_INET6],
+    ipv4: ['127.0.0.1', :ipv4, ::Socket::AF_INET],
+    ipv6: ['::1', :ipv6, ::Socket::AF_INET6],
   )
   def test_configure(data)
-    bind_addr, config, family = data
+    bind_addr, protocol, family = data
+    config = send("#{protocol}_config")
     omit "IPv6 unavailable" if family == ::Socket::AF_INET6 && !ipv6_enabled?
 
     d = create_driver(config)
-    assert_equal PORT, d.instance.port
+    assert_equal @port, d.instance.port
     assert_equal bind_addr, d.instance.bind
   end
 
   sub_test_case 'source_hostname_key and source_address_key features' do
     test 'resolve_hostname must be true with source_hostname_key' do
       assert_raise(Fluent::ConfigError) {
-        create_driver(CONFIG + <<EOS)
+        create_driver(ipv4_config + <<EOS)
 resolve_hostname false
 source_hostname_key hostname
 EOS
@@ -50,7 +59,7 @@ EOS
     data('resolve_hostname' => 'resolve_hostname true',
          'source_hostname_key' => 'source_hostname_key source_host')
     def test_configure_resolve_hostname(param)
-      d = create_driver([CONFIG, param].join("\n"))
+      d = create_driver([ipv4_config, param].join("\n"))
       assert_true d.instance.resolve_hostname
     end
   end
@@ -60,7 +69,7 @@ EOS
        'Use transport and protocol' => ["protocol_type udp\n<transport tcp>\n </transport>", :udp, :tcp])
   def test_configure_protocol(param)
     conf, proto_type, transport_proto_type = *param
-    d = create_driver([CONFIG, conf].join("\n"))
+    d = create_driver([ipv4_config, conf].join("\n"))
 
     assert_equal(d.instance.protocol_type, proto_type)
     assert_equal(d.instance.transport_config.protocol, transport_proto_type)
@@ -68,12 +77,12 @@ EOS
 
   # For backward compat
   def test_respect_protocol_type_than_transport
-    d = create_driver([CONFIG, "<transport tcp> \n</transport>", "protocol_type udp"].join("\n"))
+    d = create_driver([ipv4_config, "<transport tcp> \n</transport>", "protocol_type udp"].join("\n"))
     tests = create_test_case
 
     d.run(expect_emits: 2) do
       u = UDPSocket.new
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       tests.each {|test|
         u.send(test['msg'], 0)
       }
@@ -85,11 +94,12 @@ EOS
 
 
   data(
-    ipv4: ['127.0.0.1', CONFIG, ::Socket::AF_INET],
-    ipv6: ['::1', IPv6_CONFIG, ::Socket::AF_INET6],
+    ipv4: ['127.0.0.1', :ipv4, ::Socket::AF_INET],
+    ipv6: ['::1', :ipv6, ::Socket::AF_INET6],
   )
   def test_time_format(data)
-    bind_addr, config, family = data
+    bind_addr, protocol, family = data
+    config = send("#{protocol}_config")
     omit "IPv6 unavailable" if family == ::Socket::AF_INET6 && !ipv6_enabled?
 
     d = create_driver(config)
@@ -100,7 +110,7 @@ EOS
     ]
     d.run(expect_emits: 2) do
       u = UDPSocket.new(family)
-      u.connect(bind_addr, PORT)
+      u.connect(bind_addr, @port)
       tests.each {|test|
         u.send(test['msg'], 0)
       }
@@ -119,7 +129,7 @@ EOS
 
     d.run(expect_emits: 2) do
       u = UDPSocket.new
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       tests.each {|test|
         u.send(test['msg'], 0)
       }
@@ -130,14 +140,14 @@ EOS
   end
 
   def test_msg_size_udp_for_large_msg
-    d = create_driver(CONFIG + %[
+    d = create_driver(ipv4_config + %[
       message_length_limit 5k
     ])
     tests = create_test_case(large_message: true)
 
     d.run(expect_emits: 3) do
       u = UDPSocket.new
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       tests.each {|test|
         u.send(test['msg'], 0)
       }
@@ -148,12 +158,12 @@ EOS
   end
 
   def test_msg_size_with_tcp
-    d = create_driver([CONFIG, "<transport tcp> \n</transport>"].join("\n"))
+    d = create_driver([ipv4_config, "<transport tcp> \n</transport>"].join("\n"))
     tests = create_test_case
 
     d.run(expect_emits: 2) do
       tests.each {|test|
-        TCPSocket.open('127.0.0.1', PORT) do |s|
+        TCPSocket.open('127.0.0.1', @port) do |s|
           s.send(test['msg'], 0)
         end
       }
@@ -164,12 +174,12 @@ EOS
   end
 
   def test_emit_rfc5452
-    d = create_driver([CONFIG, "facility_key pri\n<parse>\n message_format rfc5424\nwith_priority true\n</parse>"].join("\n"))
+    d = create_driver([ipv4_config, "facility_key pri\n<parse>\n message_format rfc5424\nwith_priority true\n</parse>"].join("\n"))
     msg = '<1>1 2017-02-06T13:14:15.003Z myhostname 02abaf0687f5 10339 02abaf0687f5 - method=POST db=0.00'
 
     d.run(expect_emits: 1, timeout: 2) do
       u = UDPSocket.new
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       u.send(msg, 0)
     end
 
@@ -179,11 +189,11 @@ EOS
   end
 
   def test_msg_size_with_same_tcp_connection
-    d = create_driver([CONFIG, "<transport tcp> \n</transport>"].join("\n"))
+    d = create_driver([ipv4_config, "<transport tcp> \n</transport>"].join("\n"))
     tests = create_test_case
 
     d.run(expect_emits: 2) do
-      TCPSocket.open('127.0.0.1', PORT) do |s|
+      TCPSocket.open('127.0.0.1', @port) do |s|
         tests.each {|test|
           s.send(test['msg'], 0)
         }
@@ -195,7 +205,7 @@ EOS
   end
 
   def test_msg_size_with_json_format
-    d = create_driver([CONFIG, 'format json'].join("\n"))
+    d = create_driver([ipv4_config, 'format json'].join("\n"))
     time = Time.parse('2013-09-18 12:00:00 +0900').to_i
     tests = ['Hello!', 'Syslog!'].map { |msg|
       event = {'time' => time, 'message' => msg}
@@ -204,7 +214,7 @@ EOS
 
     d.run(expect_emits: 2) do
       u = UDPSocket.new
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       tests.each {|test|
         u.send(test['msg'], 0)
       }
@@ -215,13 +225,13 @@ EOS
   end
 
   def test_msg_size_with_include_source_host
-    d = create_driver([CONFIG, 'include_source_host true'].join("\n"))
+    d = create_driver([ipv4_config, 'include_source_host true'].join("\n"))
     tests = create_test_case
 
     host = nil
     d.run(expect_emits: 2) do
       u = UDPSocket.new
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       host = u.peeraddr[2]
       tests.each {|test|
         u.send(test['msg'], 0)
@@ -237,13 +247,13 @@ EOS
     priority_key: 'priority_key',
   )
   def test_msg_size_with_severity_key(param_name)
-    d = create_driver([CONFIG, "#{param_name} severity"].join("\n"))
+    d = create_driver([ipv4_config, "#{param_name} severity"].join("\n"))
     tests = create_test_case
 
     severity = 'info'
     d.run(expect_emits: 2) do
       u = UDPSocket.new
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       tests.each {|test|
         u.send(test['msg'], 0)
       }
@@ -254,13 +264,13 @@ EOS
   end
 
   def test_msg_size_with_facility_key
-    d = create_driver([CONFIG, 'facility_key facility'].join("\n"))
+    d = create_driver([ipv4_config, 'facility_key facility'].join("\n"))
     tests = create_test_case
 
     facility = 'kern'
     d.run(expect_emits: 2) do
       u = UDPSocket.new
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       tests.each {|test|
         u.send(test['msg'], 0)
       }
@@ -271,13 +281,13 @@ EOS
   end
 
   def test_msg_size_with_source_address_key
-    d = create_driver([CONFIG, 'source_address_key source_address'].join("\n"))
+    d = create_driver([ipv4_config, 'source_address_key source_address'].join("\n"))
     tests = create_test_case
 
     address = nil
     d.run(expect_emits: 2) do
       u = UDPSocket.new
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       address = u.peeraddr[3]
       tests.each {|test|
         u.send(test['msg'], 0)
@@ -289,14 +299,14 @@ EOS
   end
 
   def test_msg_size_with_source_hostname_key
-    d = create_driver([CONFIG, 'source_hostname_key source_hostname'].join("\n"))
+    d = create_driver([ipv4_config, 'source_hostname_key source_hostname'].join("\n"))
     tests = create_test_case
 
     hostname = nil
     d.run(expect_emits: 2) do
       u = UDPSocket.new
       u.do_not_reverse_lookup = false
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       hostname = u.peeraddr[2]
       tests.each {|test|
         u.send(test['msg'], 0)
@@ -337,12 +347,12 @@ EOS
 
   sub_test_case 'octet counting frame' do
     def test_msg_size_with_tcp
-      d = create_driver([CONFIG, "<transport tcp> \n</transport>", 'frame_type octet_count'].join("\n"))
+      d = create_driver([ipv4_config, "<transport tcp> \n</transport>", 'frame_type octet_count'].join("\n"))
       tests = create_test_case
 
       d.run(expect_emits: 2) do
         tests.each {|test|
-          TCPSocket.open('127.0.0.1', PORT) do |s|
+          TCPSocket.open('127.0.0.1', @port) do |s|
             s.send(test['msg'], 0)
           end
         }
@@ -353,11 +363,11 @@ EOS
     end
 
     def test_msg_size_with_same_tcp_connection
-      d = create_driver([CONFIG, "<transport tcp> \n</transport>", 'frame_type octet_count'].join("\n"))
+      d = create_driver([ipv4_config, "<transport tcp> \n</transport>", 'frame_type octet_count'].join("\n"))
       tests = create_test_case
 
       d.run(expect_emits: 2) do
-        TCPSocket.open('127.0.0.1', PORT) do |s|
+        TCPSocket.open('127.0.0.1', @port) do |s|
           tests.each {|test|
             s.send(test['msg'], 0)
           }
@@ -404,13 +414,13 @@ EOS
   end
 
   def test_emit_unmatched_lines
-    d = create_driver([CONFIG, 'emit_unmatched_lines true'].join("\n"))
+    d = create_driver([ipv4_config, 'emit_unmatched_lines true'].join("\n"))
     tests = create_unmatched_lines_test_case
 
     d.run(expect_emits: 3) do
       u = UDPSocket.new
       u.do_not_reverse_lookup = false
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       tests.each {|test|
         u.send(test['msg'], 0)
       }
@@ -421,14 +431,14 @@ EOS
   end
 
   def test_emit_unmatched_lines_with_hostname
-    d = create_driver([CONFIG, 'emit_unmatched_lines true', 'source_hostname_key source_hostname'].join("\n"))
+    d = create_driver([ipv4_config, 'emit_unmatched_lines true', 'source_hostname_key source_hostname'].join("\n"))
     tests = create_unmatched_lines_test_case
 
     hostname = nil
     d.run(expect_emits: 3) do
       u = UDPSocket.new
       u.do_not_reverse_lookup = false
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       hostname = u.peeraddr[2]
       tests.each {|test|
         u.send(test['msg'], 0)
@@ -440,14 +450,14 @@ EOS
   end
 
   def test_emit_unmatched_lines_with_address
-    d = create_driver([CONFIG, 'emit_unmatched_lines true', 'source_address_key source_address'].join("\n"))
+    d = create_driver([ipv4_config, 'emit_unmatched_lines true', 'source_address_key source_address'].join("\n"))
     tests = create_unmatched_lines_test_case
 
     address = nil
     d.run(expect_emits: 3) do
       u = UDPSocket.new
       u.do_not_reverse_lookup = false
-      u.connect('127.0.0.1', PORT)
+      u.connect('127.0.0.1', @port)
       address = u.peeraddr[3]
       tests.each {|test|
         u.send(test['msg'], 0)