opentelemetry-erlang-contrib/instrumentation/opentelemetry_redix/test/opentelemetry_redix_test.exs

defmodule OpentelemetryRedixTest do
  use ExUnit.Case, async: false

  doctest OpentelemetryRedix

  require OpenTelemetry.Tracer
  require OpenTelemetry.Span
  require Record

  for {name, spec} <- Record.extract_all(from_lib: "opentelemetry/include/otel_span.hrl") do
    Record.defrecord(name, spec)
  end

  for {name, spec} <- Record.extract_all(from_lib: "opentelemetry_api/include/opentelemetry.hrl") do
    Record.defrecord(name, spec)
  end

  setup do
    :otel_simple_processor.set_exporter(:otel_exporter_pid, self())

    OpenTelemetry.Tracer.start_span("test")

    on_exit(fn ->
      OpenTelemetry.Tracer.end_span()
    end)
  end

  test "records span on commands" do
    OpentelemetryRedix.setup()

    conn = start_supervised!({Redix, []})

    {:ok, "OK"} = Redix.command(conn, ["SET", "foo", "bar"])

    assert_receive {:span,
                    span(
                      name: "SET",
                      kind: :client,
                      attributes: attributes
                    )}

    assert %{
             "db.operation": "SET",
             "db.statement": "SET foo ?",
             "db.system": "redis",
             "net.peer.name": "localhost",
             "net.peer.port": "6379"
           } = :otel_attributes.map(attributes)
  end

  test "records span on piplines" do
    OpentelemetryRedix.setup()

    conn = start_supervised!({Redix, []})

    {:ok, [_, _, _, "2"]} =
      Redix.pipeline(conn, [
        ["DEL", "counter"],
        ["INCR", "counter"],
        ["INCR", "counter"],
        ["GET", "counter"]
      ])

    assert_receive {:span,
                    span(
                      name: "pipeline",
                      kind: :client,
                      attributes: attributes
                    )}

    assert %{
             "db.operation": "pipeline",
             "db.statement": "DEL counter\nINCR counter\nINCR counter\nGET counter",
             "db.system": "redis",
             "net.peer.name": "localhost",
             "net.peer.port": "6379"
           } = :otel_attributes.map(attributes)
  end
end