Tests for SpanStorage

Author: solnic

test/sentry/opentelemetry/span_storage_test.exs

@@ -0,0 +1,183 @@
+defmodule Sentry.OpenTelemetry.SpanStorageTest do
+  use ExUnit.Case, async: false
+
+  alias Sentry.OpenTelemetry.{SpanStorage, SpanRecord}
+
+  setup do
+    if :ets.whereis(:span_storage) != :undefined do
+      :ets.delete_all_objects(:span_storage)
+    else
+      start_supervised!(SpanStorage)
+    end
+
+    on_exit(fn ->
+      if :ets.whereis(:span_storage) != :undefined do
+        :ets.delete_all_objects(:span_storage)
+      end
+    end)
+
+    :ok
+  end
+
+  describe "root spans" do
+    test "stores and retrieves a root span" do
+      root_span = %SpanRecord{
+        span_id: "abc123",
+        parent_span_id: nil,
+        trace_id: "trace123",
+        name: "root_span"
+      }
+
+      SpanStorage.store_span(root_span)
+
+      assert ^root_span = SpanStorage.get_root_span("abc123")
+    end
+
+    test "updates an existing root span" do
+      root_span = %SpanRecord{
+        span_id: "abc123",
+        parent_span_id: nil,
+        trace_id: "trace123",
+        name: "root_span"
+      }
+
+      updated_root_span = %SpanRecord{
+        span_id: "abc123",
+        parent_span_id: nil,
+        trace_id: "trace123",
+        name: "updated_root_span"
+      }
+
+      SpanStorage.store_span(root_span)
+      SpanStorage.update_span(updated_root_span)
+
+      assert ^updated_root_span = SpanStorage.get_root_span("abc123")
+    end
+
+    test "removes a root span" do
+      root_span = %SpanRecord{
+        span_id: "abc123",
+        parent_span_id: nil,
+        trace_id: "trace123",
+        name: "root_span"
+      }
+
+      SpanStorage.store_span(root_span)
+      assert root_span == SpanStorage.get_root_span("abc123")
+
+      SpanStorage.remove_span("abc123")
+      assert nil == SpanStorage.get_root_span("abc123")
+    end
+  end
+
+  describe "child spans" do
+    test "stores and retrieves child spans" do
+      child_span1 = %SpanRecord{
+        span_id: "child1",
+        parent_span_id: "parent123",
+        trace_id: "trace123",
+        name: "child_span_1"
+      }
+
+      child_span2 = %SpanRecord{
+        span_id: "child2",
+        parent_span_id: "parent123",
+        trace_id: "trace123",
+        name: "child_span_2"
+      }
+
+      SpanStorage.store_span(child_span1)
+      SpanStorage.store_span(child_span2)
+
+      children = SpanStorage.get_child_spans("parent123")
+      assert length(children) == 2
+      assert child_span1 in children
+      assert child_span2 in children
+    end
+
+    test "updates an existing child span" do
+      child_span = %SpanRecord{
+        span_id: "child1",
+        parent_span_id: "parent123",
+        trace_id: "trace123",
+        name: "child_span"
+      }
+
+      updated_child_span = %SpanRecord{
+        span_id: "child1",
+        parent_span_id: "parent123",
+        trace_id: "trace123",
+        name: "updated_child_span"
+      }
+
+      SpanStorage.store_span(child_span)
+      SpanStorage.update_span(updated_child_span)
+
+      children = SpanStorage.get_child_spans("parent123")
+      assert [^updated_child_span] = children
+    end
+
+    test "removes child spans" do
+      child_span1 = %SpanRecord{
+        span_id: "child1",
+        parent_span_id: "parent123",
+        trace_id: "trace123",
+        name: "child_span_1"
+      }
+
+      child_span2 = %SpanRecord{
+        span_id: "child2",
+        parent_span_id: "parent123",
+        trace_id: "trace123",
+        name: "child_span_2"
+      }
+
+      SpanStorage.store_span(child_span1)
+      SpanStorage.store_span(child_span2)
+      assert length(SpanStorage.get_child_spans("parent123")) == 2
+
+      SpanStorage.remove_child_spans("parent123")
+      assert [] == SpanStorage.get_child_spans("parent123")
+    end
+  end
+
+  test "handles complete span hierarchy" do
+    root_span = %SpanRecord{
+      span_id: "root123",
+      parent_span_id: nil,
+      trace_id: "trace123",
+      name: "root_span"
+    }
+
+    child_span1 = %SpanRecord{
+      span_id: "child1",
+      parent_span_id: "root123",
+      trace_id: "trace123",
+      name: "child_span_1"
+    }
+
+    child_span2 = %SpanRecord{
+      span_id: "child2",
+      parent_span_id: "root123",
+      trace_id: "trace123",
+      name: "child_span_2"
+    }
+
+    SpanStorage.store_span(root_span)
+    SpanStorage.store_span(child_span1)
+    SpanStorage.store_span(child_span2)
+
+    assert ^root_span = SpanStorage.get_root_span("root123")
+
+    children = SpanStorage.get_child_spans("root123")
+    assert length(children) == 2
+    assert child_span1 in children
+    assert child_span2 in children
+
+    SpanStorage.remove_span("root123")
+    SpanStorage.remove_child_spans("root123")
+
+    assert nil == SpanStorage.get_root_span("root123")
+    assert [] == SpanStorage.get_child_spans("root123")
+  end
+end