Expose SpanKind in TestSpan and FinishedTestSpan

Author: slashmo

Sources/TracingTestKit/TestSpan.swift

@@ -18,18 +18,21 @@ import Tracing
 public struct TestSpan: Span {
     public let context: ServiceContext
     public let spanContext: TestSpanContext
+    public let kind: SpanKind
     public let startInstant: any TracerInstant
 
     init(
         operationName: String,
         context: ServiceContext,
         spanContext: TestSpanContext,
+        kind: SpanKind,
         startInstant: any TracerInstant,
         onEnd: @escaping @Sendable (FinishedTestSpan) -> Void
     ) {
         self._operationName = LockedValueBox(operationName)
         self.context = context
         self.spanContext = spanContext
+        self.kind = kind
         self.startInstant = startInstant
         self.onEnd = onEnd
     }
@@ -105,6 +108,7 @@ public struct TestSpan: Span {
         let finishedSpan = FinishedTestSpan(
             operationName: operationName,
             context: context,
+            kind: kind,
             spanContext: spanContext,
             startInstant: startInstant,
             endInstant: instant(),
@@ -149,6 +153,7 @@ public struct TestSpan: Span {
 public struct FinishedTestSpan: Sendable {
     public let operationName: String
     public let context: ServiceContext
+    public let kind: SpanKind
     public let spanContext: TestSpanContext
     public let startInstant: any TracerInstant
     public let endInstant: any TracerInstant

Sources/TracingTestKit/TestTracer.swift

@@ -59,6 +59,7 @@ public struct TestTracer: Tracer {
             operationName: operationName,
             context: context,
             spanContext: spanContext,
+            kind: kind,
             startInstant: instant()
         ) { finishedSpan in
             _activeSpans.withValue { $0[spanContext] = nil }

Tests/TracingTestKitTests/TestTracerTests.swift

@@ -272,6 +272,7 @@ struct TestTracerTests {
                 operationName: "stub",
                 context: .topLevel,
                 spanContext: spanContext,
+                kind: .internal,
                 startInstant: startInstant,
                 onEnd: { span in
                     _finishedSpan.withValue { $0 = span }
@@ -449,6 +450,7 @@ extension TestSpan {
             operationName: "stub",
             context: .topLevel,
             spanContext: TestSpanContext(traceID: "stub", spanID: "stub", parentSpanID: nil),
+            kind: .internal,
             startInstant: DefaultTracerClock().now,
             onEnd: { _ in }
         )