opentelemetry-dotnet-instru.../test/IntegrationTests/EntityFrameworkCoreTests.cs

55 lines
1.6 KiB
C#

// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0
#if NET
using IntegrationTests.Helpers;
using Xunit.Abstractions;
namespace IntegrationTests;
public class EntityFrameworkCoreTests : TestHelper
{
public EntityFrameworkCoreTests(ITestOutputHelper output)
: base("EntityFrameworkCore", output)
{
}
public static TheoryData<string, bool> GetData()
{
var theoryData = new TheoryData<string, bool>();
foreach (var version in LibraryVersion.EntityFrameworkCore)
{
theoryData.Add(version, true);
theoryData.Add(version, false);
}
return theoryData;
}
[Theory]
[Trait("Category", "EndToEnd")]
[MemberData(nameof(GetData))]
public void SubmitTraces(string packageVersion, bool dbStatementForText)
{
SetEnvironmentVariable("OTEL_DOTNET_AUTO_ENTITYFRAMEWORKCORE_SET_DBSTATEMENT_FOR_TEXT", dbStatementForText.ToString());
using var collector = new MockSpansCollector(Output);
SetExporter(collector);
if (dbStatementForText)
{
collector.Expect("OpenTelemetry.Instrumentation.EntityFrameworkCore", span => span.Attributes.Any(attr => attr.Key == "db.statement" && !string.IsNullOrWhiteSpace(attr.Value?.StringValue)));
}
else
{
collector.Expect("OpenTelemetry.Instrumentation.EntityFrameworkCore", span => span.Attributes.All(attr => attr.Key != "db.statement"));
}
RunTestApplication(new TestSettings { PackageVersion = packageVersion });
collector.AssertExpectations();
}
}
#endif