101 lines
3.2 KiB
C#
101 lines
3.2 KiB
C#
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
using IntegrationTests.Helpers;
|
|
using OpenTelemetry.AutoInstrumentation.Configurations;
|
|
using Xunit.Abstractions;
|
|
|
|
namespace IntegrationTests;
|
|
|
|
[Collection(SqlServerCollection.Name)]
|
|
public class SqlClientSystemTests : TestHelper
|
|
{
|
|
private readonly SqlServerFixture _sqlServerFixture;
|
|
|
|
public SqlClientSystemTests(ITestOutputHelper output, SqlServerFixture sqlServerFixture)
|
|
: base("SqlClient.System", output)
|
|
{
|
|
_sqlServerFixture = sqlServerFixture;
|
|
}
|
|
|
|
public static TheoryData<string, bool> GetData()
|
|
{
|
|
var theoryData = new TheoryData<string, bool>();
|
|
|
|
foreach (var version in LibraryVersion.SqlClientSystem)
|
|
{
|
|
#if NET
|
|
theoryData.Add(version, true);
|
|
#endif
|
|
theoryData.Add(version, false);
|
|
}
|
|
|
|
return theoryData;
|
|
}
|
|
|
|
[SkippableTheory]
|
|
[Trait("Category", "EndToEnd")]
|
|
[Trait("Containers", "Linux")]
|
|
[MemberData(nameof(GetData))]
|
|
public void SubmitTraces(string packageVersion, bool dbStatementForText)
|
|
{
|
|
// Skip the test if fixture does not support current platform
|
|
_sqlServerFixture.SkipIfUnsupportedPlatform();
|
|
|
|
SetEnvironmentVariable("OTEL_DOTNET_AUTO_SQLCLIENT_SET_DBSTATEMENT_FOR_TEXT", dbStatementForText.ToString());
|
|
using var collector = new MockSpansCollector(Output);
|
|
SetExporter(collector);
|
|
|
|
if (dbStatementForText)
|
|
{
|
|
collector.Expect("OpenTelemetry.Instrumentation.SqlClient", span => span.Attributes.Any(attr => attr.Key == "db.statement" && !string.IsNullOrWhiteSpace(attr.Value?.StringValue)));
|
|
}
|
|
else
|
|
{
|
|
collector.Expect("OpenTelemetry.Instrumentation.SqlClient", span => span.Attributes.All(attr => attr.Key != "db.statement"));
|
|
}
|
|
|
|
RunTestApplication(new()
|
|
{
|
|
Arguments = $"{_sqlServerFixture.Password} {_sqlServerFixture.Port}",
|
|
PackageVersion = packageVersion
|
|
});
|
|
|
|
collector.AssertExpectations();
|
|
}
|
|
|
|
[SkippableTheory]
|
|
[Trait("Category", "EndToEnd")]
|
|
[Trait("Containers", "Linux")]
|
|
[MemberData(nameof(LibraryVersion.SqlClientSystem), MemberType = typeof(LibraryVersion))]
|
|
public void SubmitMetrics(string packageVersion)
|
|
{
|
|
// Skip the test if fixture does not support current platform
|
|
_sqlServerFixture.SkipIfUnsupportedPlatform();
|
|
|
|
using var collector = new MockMetricsCollector(Output);
|
|
SetExporter(collector);
|
|
|
|
collector.Expect("OpenTelemetry.Instrumentation.SqlClient");
|
|
|
|
SetEnvironmentVariable("LONG_RUNNING", "true");
|
|
SetEnvironmentVariable("OTEL_METRIC_EXPORT_INTERVAL", "100");
|
|
SetEnvironmentVariable(ConfigurationKeys.Traces.TracesEnabled, bool.FalseString); // make sure that traces instrumentation is not needed
|
|
|
|
using var process = StartTestApplication(new TestSettings
|
|
{
|
|
Arguments = $"{_sqlServerFixture.Password} {_sqlServerFixture.Port}",
|
|
PackageVersion = packageVersion
|
|
});
|
|
|
|
try
|
|
{
|
|
collector.AssertExpectations();
|
|
}
|
|
finally
|
|
{
|
|
process?.Kill();
|
|
}
|
|
}
|
|
}
|