add

Author: Watson1978

test/fluent/plugin/test_out_otlp.rb

@@ -0,0 +1,141 @@
+# frozen_string_literal: true
+
+require "helper"
+
+require "fluent/plugin/out_otlp"
+require "fluent/test/driver/output"
+
+require "webrick"
+
+class Fluent::Plugin::OtlpOutputTest < Test::Unit::TestCase
+  class HttpResult < Struct.new(:method, :content_type, :headers, :body); end
+
+  DEFAULT_LOGGER = ::WEBrick::Log.new(::STDOUT, ::WEBrick::BasicLog::FATAL)
+
+  def config
+    <<~CONFIG
+      <http>
+        endpoint "http://127.0.0.1:4318"
+      </http>
+    CONFIG
+  end
+
+  def server_config
+    config = { BindAddress: "127.0.0.1", Port: "4318" }
+    # Suppress webrick logs
+    config[:Logger] = DEFAULT_LOGGER
+    config[:AccessLog] = []
+    config
+  end
+
+  def run_http_server
+    server = ::WEBrick::HTTPServer.new(server_config)
+    server.mount_proc("/v1/metrics") do |req, res|
+      @server_request = req
+      @server_request_body = req.body
+      res.status = 200
+    end
+    server.mount_proc("/v1/traces") do |req, res|
+      @server_request = req
+      @server_request_body = req.body
+      res.status = 200
+    end
+    server.mount_proc("/v1/logs") do |req, res|
+      @server_request = req
+      @server_request_body = req.body
+      res.status = 200
+    end
+    server.start
+  ensure
+    begin
+      server.shutdown
+    rescue StandardError
+      nil
+    end
+  end
+
+  def setup
+    Fluent::Test.setup
+
+    @server_request = nil
+    @server_request_body = nil
+    @http_server_thread ||= Thread.new do
+      run_http_server
+    end
+  end
+
+  def teardown
+    @server_request = nil
+    @server_request_body = nil
+  end
+
+  def create_driver(conf = config)
+    Fluent::Test::Driver::Output.new(Fluent::Plugin::OtlpOutput).configure(conf)
+  end
+
+  def test_configure
+    d = create_driver
+    assert_equal "http://127.0.0.1:4318", d.instance.http_config.endpoint
+  end
+
+  def test_send_logs
+    event = { "type" => "otlp_logs", "message" => TestData::JSON::LOGS }
+
+    d = create_driver
+    d.run(default_tag: 'otlp.test') do
+      d.feed(event)
+    end
+
+    assert_equal("/v1/logs", @server_request.path)
+    assert_equal("POST", @server_request.request_method)
+    assert_equal(["application/x-protobuf"], @server_request.header["content-type"])
+    assert_equal(TestData::ProtocolBuffers::LOGS, @server_request_body)
+  end
+
+  def test_send_metrics
+    event = { "type" => "otlp_metrics", "message" => TestData::JSON::METRICS }
+
+    d = create_driver
+    d.run(default_tag: 'otlp.test') do
+      d.feed(event)
+    end
+
+    assert_equal("/v1/metrics", @server_request.path)
+    assert_equal("POST", @server_request.request_method)
+    assert_equal(["application/x-protobuf"], @server_request.header["content-type"])
+    assert_equal(TestData::ProtocolBuffers::METRICS, @server_request_body)
+  end
+
+  def test_send_traces
+    event = { "type" => "otlp_traces", "message" => TestData::JSON::TRACES }
+
+    d = create_driver
+    d.run(default_tag: 'otlp.test') do
+      d.feed(event)
+    end
+
+    assert_equal("/v1/traces", @server_request.path)
+    assert_equal("POST", @server_request.request_method)
+    assert_equal(["application/x-protobuf"], @server_request.header["content-type"])
+    assert_equal(TestData::ProtocolBuffers::TRACES, @server_request_body)
+  end
+
+  def test_send_compressed_message
+    event = { "type" => "otlp_logs", "message" => TestData::JSON::LOGS }
+
+    d = create_driver(config + "compress gzip")
+    d.run(default_tag: 'otlp.test') do
+      d.feed(event)
+    end
+
+    assert_equal("/v1/logs", @server_request.path)
+    assert_equal("POST", @server_request.request_method)
+    assert_equal(["application/x-protobuf"], @server_request.header["content-type"])
+    assert_equal(["gzip"], @server_request.header["content-encoding"])
+    assert_equal(TestData::ProtocolBuffers::LOGS, decompress(@server_request_body).force_encoding(Encoding::ASCII_8BIT))
+  end
+
+  def decompress(data)
+    Zlib::GzipReader.new(StringIO.new(data)).read
+  end
+end