fixup! Update unit tests to work with new APIs

Author: andrewlock

tracer/test/Datadog.Trace.ClrProfiler.IntegrationTests/CI/Agent/ApmAgentWriterTests.cs

@@ -33,7 +33,7 @@ public ApmAgentWriterTests()
         [Fact]
         public async Task WriteTrace_2Traces_SendToApi()
         {
-            var spans1 = new ArraySegment<Span>(new[] { new Span(new SpanContext(1, 1), DateTimeOffset.UtcNow) });
+            var spans1 = new SpanCollection(new Span(new SpanContext(1, 1), DateTimeOffset.UtcNow));
             var traceChunk1 = new TraceChunkModel(spans1);
             var expectedData1 = Vendors.MessagePack.MessagePackSerializer.Serialize(traceChunk1, SpanFormatterResolver.Instance);
 
@@ -43,7 +43,7 @@ public async Task WriteTrace_2Traces_SendToApi()
             _api.Verify(x => x.SendTracesAsync(It.Is<ArraySegment<byte>>(y => Equals(y, expectedData1)), It.Is<int>(i => i == 1), It.IsAny<bool>(), It.IsAny<long>(), It.IsAny<long>(), It.IsAny<bool>()), Times.Once);
             _api.Invocations.Clear();
 
-            var spans2 = new ArraySegment<Span>(new[] { new Span(new SpanContext(2, 2), DateTimeOffset.UtcNow) });
+            var spans2 = new SpanCollection(new Span(new SpanContext(2, 2), DateTimeOffset.UtcNow));
             var traceChunk2 = new TraceChunkModel(spans2);
             var expectedData2 = Vendors.MessagePack.MessagePackSerializer.Serialize(traceChunk2, SpanFormatterResolver.Instance);
 

tracer/test/Datadog.Trace.ClrProfiler.IntegrationTests/CI/Agent/CiVisibilityProtocolWriterTests.cs

@@ -59,8 +59,7 @@ public async Task AgentlessTestEventTest()
                            return Task.CompletedTask;
                        });
 
-            var trace = new[] { span };
-            agentlessWriter.WriteTrace(new ArraySegment<Span>(trace));
+            agentlessWriter.WriteTrace(new SpanCollection(span));
             await agentlessWriter.FlushTracesAsync(); // Force a flush to make sure the trace is written to the API
 
             Assert.True(finalPayload.SequenceEqual(expectedBytes));
@@ -92,8 +91,7 @@ public async Task AgentlessStreamTestEventTest()
                            return Task.CompletedTask;
                        });
 
-            var trace = new[] { span };
-            agentlessWriter.WriteTrace(new ArraySegment<Span>(trace));
+            agentlessWriter.WriteTrace(new SpanCollection(span));
             await agentlessWriter.FlushTracesAsync(); // Force a flush to make sure the trace is written to the API
 
             Assert.True(finalPayload.SequenceEqual(expectedBytes));

tracer/test/Datadog.Trace.IntegrationTests/SpanTagTests.cs

@@ -213,9 +213,7 @@ public void SpanSampler_ShouldNotTag_WhenSamplingPriority_IsNull()
 
             var spanContext = new SpanContext(4, 5, samplingPriority: null, serviceName: "serviceName");
             var span = new Span(spanContext, DateTimeOffset.Now) { OperationName = "test" };
-            var spans = new Span[1];
-            spans[0] = span;
-            _writer.WriteTrace(new ArraySegment<Span>(spans));
+            _writer.WriteTrace(new SpanCollection(span));
             var trace = _testApi.Wait();
             trace.Should().HaveCount(1);
             trace[0].Should().HaveCount(1);