add test

Author: Watson1978

Gemfile

@@ -9,3 +9,4 @@ gem "irb"
 gem "rake", "~> 13.0"
 gem "rubocop", "~> 1.75"
 gem "test-unit", "~> 3.0"
+gem "timecop"

lib/fluent/plugin/otlp/request.rb

@@ -10,10 +10,10 @@ class Fluent::Plugin::Otlp::Request
   class Logs
     def initialize(body)
       @request =
-        if body.encoding == Encoding::BINARY
-          Opentelemetry::Proto::Collector::Logs::V1::ExportLogsServiceRequest.decode(body)
-        else
+        if body.start_with?("{")
           Opentelemetry::Proto::Collector::Logs::V1::ExportLogsServiceRequest.decode_json(body)
+        else
+          Opentelemetry::Proto::Collector::Logs::V1::ExportLogsServiceRequest.decode(body)
         end
     end
 
@@ -29,10 +29,10 @@ def record
   class Metrics
     def initialize(body)
       @request =
-        if body.encoding == Encoding::BINARY
-          Opentelemetry::Proto::Collector::Metrics::V1::ExportMetricsServiceRequest.decode(body)
-        else
+        if body.start_with?("{")
           Opentelemetry::Proto::Collector::Metrics::V1::ExportMetricsServiceRequest.decode_json(body)
+        else
+          Opentelemetry::Proto::Collector::Metrics::V1::ExportMetricsServiceRequest.decode(body)
         end
     end
 
@@ -48,10 +48,10 @@ def record
   class Traces
     def initialize(body)
       @request =
-        if body.encoding == Encoding::BINARY
-          Opentelemetry::Proto::Collector::Trace::V1::ExportTraceServiceRequest.decode(body)
-        else
+        if body.start_with?("{")
           Opentelemetry::Proto::Collector::Trace::V1::ExportTraceServiceRequest.decode_json(body)
+        else
+          Opentelemetry::Proto::Collector::Trace::V1::ExportTraceServiceRequest.decode(body)
         end
     end
 

test/fluent/helper/metrics.json

@@ -0,0 +1,57 @@
+{
+  "resourceMetrics": [
+    {
+      "resource": {
+        "attributes": [
+          {
+            "key": "service.name",
+            "value": {
+              "stringValue": "unknown_service"
+            }
+          },
+          {
+            "key": "telemetry.sdk.version",
+            "value": {
+              "stringValue": "1.19.0"
+            }
+          },
+          {
+            "key": "telemetry.sdk.name",
+            "value": {
+              "stringValue": "opentelemetry"
+            }
+          },
+          {
+            "key": "telemetry.sdk.language",
+            "value": {
+              "stringValue": "cpp"
+            }
+          }
+        ]
+      },
+      "scopeMetrics": [
+        {
+          "scope": {
+            "name": "otlp_grpc_metric_example"
+          },
+          "metrics": [
+            {
+              "name": "otlp_grpc_metric_example_counter",
+              "sum": {
+                "dataPoints": [
+                  {
+                    "startTimeUnixNano": "1744167147880764228",
+                    "timeUnixNano": "1744167148881527028",
+                    "asDouble": 45
+                  }
+                ],
+                "aggregationTemporality": "AGGREGATION_TEMPORALITY_CUMULATIVE",
+                "isMonotonic": true
+              }
+            }
+          ]
+        }
+      ]
+    }
+  ]
+}

test/fluent/plugin/test_in_otel.rb

@@ -3,8 +3,10 @@
 require "helper"
 require 'fluent/test/driver/input'
 require 'fluent/plugin/in_otlp'
+require 'excon'
+require 'timecop'
 
-class Fluent::Plugin::OtelInputTest < Test::Unit::TestCase
+class Fluent::Plugin::OtlpInputTest < Test::Unit::TestCase
   def config
     <<~CONFIG
       tag otlp.test
@@ -15,6 +17,14 @@ def config
     CONFIG
   end
 
+  def setup
+    Fluent::Test.setup
+  end
+
+  def teardown
+    Timecop.return
+  end
+
   def create_driver(conf=config)
     Fluent::Test::Driver::Input.new(Fluent::Plugin::OtlpInput).configure(conf)
   end
@@ -26,4 +36,24 @@ def test_configure
     assert_equal 4318, d.instance.http_config.port
   end
 
+  def test_receive_json
+    d = create_driver
+    res = d.run(expect_records: 1) do
+      post_json("/v1/metrics", Fluent::Plugin::Otlp::JSON::METRICS)
+    end
+
+    assert_equal(200, res.status)
+    assert_equal("otlp.test", d.events[0][0])
+    assert_equal({ type: "otlp_metrics", message: Fluent::Plugin::Otlp::JSON::METRICS }, d.events[0][2])
+  end
+
+  def post_json(path, json)
+    headers = { "Content-Type" => "application/json" }
+    post(path, headers, json)
+  end
+
+  def post(path, headers, body)
+    connection = Excon.new("http://127.0.0.1:4318#{path}", body: body, headers: headers)
+    connection.post
+  end
 end

test/helper.rb

@@ -5,3 +5,13 @@
 require "test-unit"
 require 'fluent/test'
 require 'fluent/test/helpers'
+require 'json'
+
+include Fluent::Test::Helpers
+
+module Fluent::Plugin::Otlp
+  module JSON
+    # trim white spaces
+    METRICS = ::JSON.generate(::JSON.parse(File.read(File.join(__dir__, "./fluent/helper/metrics.json"))))
+  end
+end