From d0177f0d29f8fb0931634b3e4fc7cc9192c87d7c Mon Sep 17 00:00:00 2001 From: Nev <54870357+MSNev@users.noreply.github.com> Date: Mon, 23 Feb 2026 17:10:05 -0800 Subject: [PATCH] [Main] Merge Trace API Features from Beta This commit merges all beta branch features into main with clean linear history. Key changes from beta: - Merge AppInsightsCommon into AppInsightsCore (Phase 1-4) - Add W3C Trace State support/handling - Add startSpan, withSpan, Route Strategy features - Add tree-shaking annotations - Update build dependencies and test layouts - Sync with OTel-SDK file structure Squash merge of origin/beta (3.4.0-beta) with conflict resolution: - Updated Rush version to 5.169.3 (from main) - Removed beta-specific publishConfig --- .aiAutoMinify.json | 46 +- .github/copilot-instructions.md | 119 +- AISKU/README.md | 1 + AISKU/Tests/Manual/README.md | 262 ++ AISKU/Tests/Manual/span-e2e-manual-test.html | 818 ++++ AISKU/Tests/Perf/src/AISKUPerf.Tests.ts | 6 +- AISKU/Tests/Unit/src/AISKUSize.Tests.ts | 18 +- AISKU/Tests/Unit/src/CdnPackaging.tests.ts | 2 +- AISKU/Tests/Unit/src/CdnThrottle.tests.ts | 2 +- .../Tests/Unit/src/IAnalyticsConfig.Tests.ts | 2 +- .../Tests/Unit/src/NonRecordingSpan.Tests.ts | 773 ++++ AISKU/Tests/Unit/src/OTelInit.Tests.ts | 112 + .../Unit/src/SnippetInitialization.Tests.ts | 25 +- .../Unit/src/SpanContextPropagation.Tests.ts | 733 ++++ AISKU/Tests/Unit/src/SpanE2E.Tests.ts | 751 ++++ .../Tests/Unit/src/SpanErrorHandling.Tests.ts | 768 ++++ AISKU/Tests/Unit/src/SpanHelperUtils.Tests.ts | 992 +++++ AISKU/Tests/Unit/src/SpanLifeCycle.Tests.ts | 655 ++++ .../Unit/src/SpanPluginIntegration.Tests.ts | 1020 +++++ AISKU/Tests/Unit/src/SpanUtils.Tests.ts | 1971 ++++++++++ AISKU/Tests/Unit/src/StartSpan.Tests.ts | 301 ++ .../Unit/src/TelemetryItemGeneration.Tests.ts | 831 ++++ .../Unit/src/ThrottleSentMessage.tests.ts | 2 +- AISKU/Tests/Unit/src/TraceContext.Tests.ts | 735 ++++ AISKU/Tests/Unit/src/TraceProvider.Tests.ts | 716 ++++ .../Tests/Unit/src/TraceSuppression.Tests.ts | 706 ++++ AISKU/Tests/Unit/src/UseSpan.Tests.ts | 1164 ++++++ AISKU/Tests/Unit/src/WithSpan.Tests.ts | 1114 ++++++ AISKU/Tests/Unit/src/aiskuunittests.ts | 51 +- .../applicationinsights.e2e.fetch.tests.ts | 2 +- .../Unit/src/applicationinsights.e2e.tests.ts | 443 ++- AISKU/Tests/Unit/src/sanitizer.e2e.tests.ts | 2 +- AISKU/Tests/Unit/src/sender.e2e.tests.ts | 2 +- AISKU/Tests/Unit/src/validate.e2e.tests.ts | 2 +- .../Tests/es6-module-type-check/package.json | 2 +- AISKU/Tests/es6-module-type-check/src/main.ts | 2 +- AISKU/examples/span-usage-example.ts | 68 + AISKU/package.json | 1 - AISKU/scripts/publishAzReleaseToCdn.ps1 | 2 +- AISKU/scripts/setAzActiveCdnVersion.ps1 | 2 +- AISKU/src/AISku.ts | 243 +- AISKU/src/IApplicationInsights.ts | 24 +- AISKU/src/Init.ts | 5 +- AISKU/src/InternalConstants.ts | 2 + AISKU/src/Snippet.ts | 3 +- AISKU/src/applicationinsights-web.ts | 12 +- AISKU/src/internal/trace/spanUtils.ts | 506 +++ .../Tests/Unit/src/AISKULightSize.Tests.ts | 8 +- .../Tests/Unit/src/aiskuliteunittests.ts | 2 + AISKULight/Tests/Unit/src/config.tests.ts | 2 +- .../Tests/Unit/src/dynamicconfig.tests.ts | 2 +- .../Tests/Unit/src/otelNegative.tests.ts | 189 + AISKULight/Tests/UnitTests.html | 6 +- AISKULight/package.json | 1 - AISKULight/src/index.ts | 15 +- README.md | 22 + RELEASES.md | 118 +- channels/1ds-post-js/src/ClockSkewManager.ts | 176 +- channels/1ds-post-js/src/HttpManager.ts | 12 +- channels/1ds-post-js/src/KillSwitch.ts | 114 +- .../test/Unit/src/HttpManagerTest.ts | 7 +- .../test/Unit/src/KillSwitchTest.ts | 18 +- .../test/Unit/src/PostChannelTest.ts | 5 + channels/1ds-post-js/test/UnitTests.html | 6 +- .../Tests/Unit/src/Sample.tests.ts | 23 +- .../Tests/Unit/src/Sender.tests.ts | 319 +- .../Tests/Unit/src/StatsBeat.tests.ts | 4 +- .../package.json | 1 - .../src/EnvelopeCreator.ts | 87 +- .../src/Interfaces.ts | 3 +- .../src/Interfaces/Contracts/IRequestData.ts | 60 + .../src/SendBuffer.ts | 3 +- .../src/Sender.ts | 63 +- .../src/Serializer.ts | 290 +- .../src/Telemetry/Common/Data.ts | 15 + .../src/Telemetry/RemoteDependencyData.ts | 91 + .../src/Telemetry/RequestData.ts | 40 + .../src/TelemetryProcessors/Sample.ts | 67 +- .../HashCodeScoreGenerator.ts | 24 +- .../SamplingScoreGenerator.ts | 30 +- .../Tests/Unit/src/TestHelper.ts | 2 +- .../Tests/Unit/src/channel.tests.ts | 2 +- .../Tests/Unit/src/customprovider.tests.ts | 2 +- .../Tests/Unit/src/dbprovider.tests.ts | 2 +- .../Unit/src/offlinebatchhandler.tests.ts | 2 +- .../Tests/Unit/src/offlinetimer.tests.ts | 2 +- .../Tests/Unit/src/webprovider.tests.ts | 2 +- channels/offline-channel-js/package.json | 1 - .../offline-channel-js/src/Helpers/Utils.ts | 3 +- .../src/Interfaces/IInMemoryBatch.ts | 3 +- .../src/Interfaces/IOfflineBatch.ts | 3 +- .../src/Interfaces/IOfflineProvider.ts | 3 +- .../offline-channel-js/src/OfflineChannel.ts | 14 +- channels/offline-channel-js/src/Sender.ts | 10 +- channels/tee-channel-js/package.json | 1 - channels/tee-channel-js/src/TeeChannel.ts | 3 +- common/Tests/Framework/src/AITestClass.ts | 30 +- common/Tests/Framework/src/Assert.ts | 10 + common/Tests/Framework/src/TestHelper.ts | 2 +- common/config/rush/npm-shrinkwrap.json | 117 +- docs/OTel/README.md | 434 +++ docs/OTel/examples.md | 1289 +++++++ docs/OTel/otelApi.md | 513 +++ docs/OTel/startActiveSpan.md | 962 +++++ docs/OTel/traceApi.md | 809 ++++ docs/OTel/useSpan.md | 804 ++++ docs/OTel/withSpan.md | 725 ++++ docs/README.md | 1 + docs/upgrade/MergeCommonToCore.md | 114 + examples/AISKU/src/utils.ts | 3 +- examples/dependency/src/appinsights-init.ts | 198 +- .../src/dependencies-example-index.ts | 4 +- examples/dependency/src/startSpan-example.ts | 377 ++ examples/startSpan/package.json | 62 + examples/startSpan/rollup.config.js | 100 + examples/startSpan/src/startSpanExample.ts | 196 + .../startSpan/src/startspan-example-index.ts | 8 + examples/startSpan/tsconfig.json | 29 + .../Unit/src/AnalyticsExtensionSize.tests.ts | 2 +- .../Tests/Unit/src/AnalyticsPlugin.tests.ts | 38 +- .../Unit/src/TelemetryItemCreator.tests.ts | 50 +- .../package.json | 1 - .../src/JavaScriptSDK/AnalyticsPlugin.ts | 189 +- .../Interfaces/IAnalyticsConfig.ts | 32 + .../Telemetry/PageViewManager.ts | 468 +-- .../Telemetry/PageViewPerformanceManager.ts | 239 +- .../Telemetry/PageVisitTimeManager.ts | 215 +- .../src/JavaScriptSDK/Timing.ts | 45 +- .../Tests/Unit/src/cfgsynchelper.tests.ts | 27 +- .../Tests/Unit/src/cfgsyncplugin.tests.ts | 4 +- .../package.json | 1 - .../src/CfgSyncHelperFuncs.ts | 3 +- .../src/CfgSyncPlugin.ts | 3 +- .../src/Interfaces/ICfgSyncConfig.ts | 3 +- .../src/Interfaces/ICfgSyncPlugin.ts | 5 +- .../Tests/Unit/src/ClickEventTest.ts | 2 +- .../package.json | 1 - .../scripts/publishAzReleaseToCdn.ps1 | 2 +- .../scripts/setAzActiveCdnVersion.ps1 | 2 +- .../src/ClickAnalyticsPlugin.ts | 10 +- .../src/Interfaces/Datamodel.ts | 3 +- .../src/events/PageAction.ts | 3 +- .../package.json | 1 - .../scripts/publishAzReleaseToCdn.ps1 | 2 +- .../scripts/setAzActiveCdnVersion.ps1 | 2 +- .../Tests/Unit/src/TestChannelPlugin.ts | 54 + .../Unit/src/W3CTraceStateDependency.tests.ts | 989 +++++ .../Tests/Unit/src/ajax.tests.ts | 413 +- .../Tests/Unit/src/dependencies.tests.ts | 2 + .../package.json | 1 - .../src/DependencyInitializer.ts | 2 +- .../src/DependencyListener.ts | 9 +- .../src/InternalConstants.ts | 1 + .../src/ajax.ts | 435 ++- .../src/ajaxRecord.ts | 410 +- .../src/ajaxUtils.ts | 23 - .../applicationinsights-dependencies-js.ts | 5 +- .../package.json | 1 - .../src/OsPlugin.ts | 8 +- .../scripts/publishAzReleaseToCdn.ps1 | 2 +- .../scripts/setAzActiveCdnVersion.ps1 | 2 +- .../Tests/Unit/src/TelemetryContext.Tests.ts | 9 +- .../Tests/Unit/src/properties.tests.ts | 31 +- .../package.json | 1 - .../src/Context/Application.ts | 2 +- .../src/Context/Device.ts | 2 +- .../src/Context/Internal.ts | 5 +- .../src/Context/Location.ts | 2 +- .../src/Context/Session.ts | 8 +- .../src/Context/TelemetryTrace.ts | 29 - .../src/Context/User.ts | 5 +- .../src/Interfaces/IPropTelemetryContext.ts | 3 +- .../src/PropertiesPlugin.ts | 35 +- .../src/TelemetryContext.ts | 172 +- .../src/applicationinsights-properties-js.ts | 5 +- gruntfile.js | 13 +- package.json | 2 +- rush.json | 5 + shared/1ds-core-js/src/AppInsightsCore.ts | 5 +- shared/1ds-core-js/src/DataModels.ts | 12 +- shared/1ds-core-js/src/Index.ts | 6 +- shared/1ds-core-js/src/InternalConstants.ts | 2 + shared/1ds-core-js/src/extSpanUtils.ts | 503 +++ .../test/Unit/src/FileSizeCheckTest.ts | 8 +- .../test/Unit/src/SpanUtilsTests.ts | 765 ++++ .../test/Unit/src/core.unittests.ts | 2 + shared/1ds-core-js/test/UnitTests.html | 6 +- shared/AppInsightsCommon/README.md | 23 +- .../Unit/src/appinsights-common.tests.ts | 19 - shared/AppInsightsCommon/Tests/UnitTests.html | 44 - shared/AppInsightsCommon/Tests/tsconfig.json | 13 - shared/AppInsightsCommon/package.json | 5 +- shared/AppInsightsCommon/src/Enums.ts | 79 - shared/AppInsightsCommon/src/HelperFuncs.ts | 55 - .../src/Interfaces/Contracts/IEnvelope.ts | 50 - .../src/Interfaces/Contracts/RequestData.ts | 50 - .../src/Interfaces/ICorrelationConfig.ts | 100 - .../src/Interfaces/PartAExtensions.ts | 14 - .../src/applicationinsights-common.ts | 195 +- shared/AppInsightsCore/README.md | 11 + .../SpanImplementationSummary.md | 100 + ...Span_Implementation_Refactoring_Summary.md | 126 + .../Tests/Unit/src/GlobalTestHooks.Test.ts | 13 - .../src/OpenTelemetry/commonUtils.Tests.ts | 1031 +++++ .../Unit/src/OpenTelemetry/errors.Tests.ts | 680 ++++ .../src/OpenTelemetry/otelNegative.Tests.ts | 690 ++++ .../Unit/src/ai}/AppInsightsCommon.tests.ts | 5 +- .../src/{ => ai}/AppInsightsCoreSize.Tests.ts | 14 +- .../{ => ai}/ApplicationInsightsCore.Tests.ts | 19 +- .../src/ai}/ConnectionStringParser.tests.ts | 4 +- .../Unit/src/{ => ai}/CookieManager.Tests.ts | 8 +- .../Unit/src/{ => ai}/EventHelper.Tests.ts | 18 +- .../{ => ai}/EventsDiscardedReason.Tests.ts | 2 +- .../Tests/Unit/src/ai}/Exception.tests.ts | 13 +- .../Unit/src/ai}/GlobalTestHooks.Test.ts | 2 +- .../Unit/src/{ => ai}/HelperFunc.Tests.ts | 10 +- .../Unit/src/{ => ai}/LoggingEnum.Tests.ts | 4 +- .../Unit/src/ai}/RequestHeaders.tests.ts | 4 +- .../src/{ => ai}/SendPostManager.Tests.ts | 8 +- .../Tests/Unit/src/ai}/SeverityLevel.tests.ts | 4 +- .../Unit/src/{ => ai}/StatsBeat.Tests.ts | 22 +- .../Tests/Unit/src/{ => ai}/TestPlugins.ts | 24 +- .../Tests/Unit/src/ai}/ThrottleMgr.tests.ts | 16 +- .../Unit/src/{ => ai}/UpdateConfig.Tests.ts | 8 +- .../Tests/Unit/src/ai}/Util.tests.ts | 24 +- .../Tests/Unit/src/aiunittests.ts | 66 +- .../src/attribute/attributeContainer.Tests.ts | 3412 +++++++++++++++++ .../Unit/src/{ => config}/Dynamic.Tests.ts | 22 +- .../src/{ => config}/DynamicConfig.Tests.ts | 32 +- .../src/trace/W3CTraceStateModes.tests.ts | 224 ++ .../src/{ => trace}/W3cTraceParentTests.ts | 6 +- .../Unit/src/trace/W3cTraceState.Tests.ts | 1606 ++++++++ .../Tests/Unit/src/trace/span.Tests.ts | 1970 ++++++++++ .../Tests/Unit/src/trace/traceState.Tests.ts | 448 +++ .../Tests/Unit/src/trace/traceUtils.Tests.ts | 1355 +++++++ shared/AppInsightsCore/Tests/tsconfig.json | 2 +- shared/AppInsightsCore/api-extractor.json | 2 +- shared/AppInsightsCore/package.json | 4 +- shared/AppInsightsCore/rollup.config.js | 43 +- .../IDistributedTraceContext.ts | 54 - .../src/JavaScriptSDK/Constants.ts | 14 - .../src/JavaScriptSDK/TelemetryHelpers.ts | 183 - .../src/applicationinsights-core-js.ts | 112 - .../ConfigDefaultHelpers.ts | 8 +- .../src/{Config => config}/ConfigDefaults.ts | 4 +- .../src/{Config => config}/DynamicConfig.ts | 20 +- .../src/{Config => config}/DynamicProperty.ts | 10 +- .../src/{Config => config}/DynamicState.ts | 10 +- .../src/{Config => config}/DynamicSupport.ts | 9 +- .../src/constants}/Constants.ts | 4 + .../InternalConstants.ts | 0 .../AggregationError.ts | 0 .../AppInsightsCore.ts | 420 +- .../src/{JavaScriptSDK => core}/AsyncUtils.ts | 0 .../BaseTelemetryPlugin.ts | 34 +- .../src/{JavaScriptSDK => core}/CookieMgr.ts | 36 +- .../DbgExtensionUtils.ts | 8 +- .../InstrumentHooks.ts | 4 +- .../NotificationManager.ts | 17 +- .../{JavaScriptSDK => core}/PerfManager.ts | 11 +- .../ProcessTelemetryContext.ts | 172 +- .../ResponseHelpers.ts | 13 +- .../SenderPostManager.ts | 40 +- .../src/{JavaScriptSDK => core}/StatsBeat.ts | 26 +- .../src/core/TelemetryHelpers.ts | 374 ++ .../TelemetryInitializerPlugin.ts | 18 +- .../UnloadHandlerContainer.ts | 9 +- .../UnloadHookContainer.ts | 8 +- .../DiagnosticLogger.ts | 30 +- .../src/diagnostics}/ThrottleMgr.ts | 28 +- .../EnumHelperFuncs.ts | 0 .../src/enums/W3CTraceFlags.ts | 18 + .../src/enums/ai/DependencyTypes.ts | 50 + shared/AppInsightsCore/src/enums/ai/Enums.ts | 149 + .../ai}/EventsDiscardedReason.ts | 2 +- .../ai}/FeatureOptInEnums.ts | 3 + .../ai}/InitActiveStatusEnum.ts | 5 +- .../ai}/LoggingEnums.ts | 7 +- .../ai}/SendRequestReason.ts | 0 .../ai}/StatsType.ts | 0 .../ai}/TelemetryUnloadReason.ts | 0 .../ai}/TelemetryUpdateReason.ts | 0 .../src/enums/ai/TraceHeadersMode.ts | 29 + .../src/enums/otel/OTelSamplingDecision.ts | 33 + .../src/enums/otel/OTelSpanKind.ts | 53 + .../src/enums/otel/OTelSpanStatus.ts | 34 + .../src/enums/otel/eAttributeChangeOp.ts | 46 + shared/AppInsightsCore/src/index.ts | 370 ++ .../src/interfaces/IException.ts | 37 + .../src/interfaces/IOTelHrTime.ts | 81 + .../src/interfaces/ai}/ConnectionString.ts | 3 + .../src/interfaces/ai}/IAppInsights.ts | 3 +- .../ai}/IAppInsightsCore.ts | 33 +- .../ai}/IChannelControls.ts | 2 +- .../ai}/IChannelControlsHost.ts | 0 .../src/interfaces/ai}/IConfig.ts | 6 +- .../ai}/IConfiguration.ts | 14 +- .../ai}/ICookieMgr.ts | 0 .../src/interfaces/ai/ICorrelationConfig.ts | 192 + .../ai}/IDbgExtension.ts | 0 .../interfaces/ai}/IDependencyTelemetry.ts | 0 .../ai}/IDiagnosticLogger.ts | 19 +- .../interfaces/ai/IDistributedTraceContext.ts | 334 ++ .../src/interfaces/ai}/IEventTelemetry.ts | 4 +- .../ai}/IExceptionConfig.ts | 0 .../src/interfaces/ai}/IExceptionTelemetry.ts | 2 +- .../ai}/IFeatureOptIn.ts | 9 +- .../ai}/IInstrumentHooks.ts | 0 .../src/interfaces/ai}/IMetricTelemetry.ts | 3 + .../ai}/INetworkStatsbeat.ts | 0 .../ai}/INotificationListener.ts | 0 .../ai}/INotificationManager.ts | 4 +- .../ai}/IPageViewPerformanceTelemetry.ts | 4 +- .../src/interfaces/ai}/IPageViewTelemetry.ts | 3 + .../src/interfaces/ai}/IPartC.ts | 0 .../ai}/IPerfEvent.ts | 0 .../ai}/IPerfManager.ts | 2 +- .../ai}/IProcessTelemetryContext.ts | 9 +- .../src/interfaces/ai}/IPropertiesPlugin.ts | 0 .../src/interfaces/ai}/IRequestContext.ts | 3 + .../src/interfaces/ai/IRequestTelemetry.ts | 44 + .../ai}/ISenderPostManager.ts | 5 +- .../ai}/IStatsBeat.ts | 2 +- .../ai}/IStatsEventData.ts | 0 .../ai}/IStatsMgr.ts | 0 .../src/interfaces/ai}/IStorageBuffer.ts | 5 +- .../src/interfaces/ai}/ITelemetryContext.ts | 22 +- .../ai}/ITelemetryInitializers.ts | 3 + .../ai}/ITelemetryItem.ts | 0 .../ai}/ITelemetryPlugin.ts | 0 .../ai}/ITelemetryPluginChain.ts | 0 .../ai}/ITelemetryUnloadState.ts | 2 +- .../ai}/ITelemetryUpdateState.ts | 2 +- .../src/interfaces/ai}/IThrottleMgr.ts | 2 + .../ai}/ITraceParent.ts | 0 .../src/interfaces/ai/ITraceProvider.ts | 157 + .../src/interfaces/ai}/ITraceTelemetry.ts | 2 +- .../ai}/IUnloadHook.ts | 0 .../ai}/IUnloadableComponent.ts | 0 .../src/interfaces/ai/IW3cTraceState.ts | 78 + .../ai}/IXDomainRequest.ts | 3 + .../ai}/IXHROverride.ts | 12 +- .../src/interfaces/ai/PartAExtensions.ts | 17 + .../interfaces/ai/context}/IApplication.ts | 0 .../src/interfaces/ai/context}/IDevice.ts | 0 .../src/interfaces/ai/context}/IInternal.ts | 0 .../src/interfaces/ai/context}/ILocation.ts | 0 .../ai/context}/IOperatingSystem.ts | 0 .../src/interfaces/ai/context}/ISample.ts | 2 +- .../src/interfaces/ai/context}/ISession.ts | 0 .../interfaces/ai/context}/ISessionManager.ts | 0 .../interfaces/ai/context}/ITelemetryTrace.ts | 14 +- .../src/interfaces/ai/context}/IUser.ts | 0 .../src/interfaces/ai/context}/IWeb.ts | 0 .../ai/contracts}/AvailabilityData.ts | 4 +- .../ai/contracts}/ContextTagKeys.ts | 2 +- .../interfaces/ai/contracts}/DataPointType.ts | 0 .../ai/contracts}/DependencyKind.ts | 0 .../ai/contracts}/DependencySourceType.ts | 0 .../src/interfaces/ai/contracts}/IBase.ts | 0 .../src/interfaces/ai/contracts}/IData.ts | 0 .../interfaces/ai/contracts}/IDataPoint.ts | 0 .../src/interfaces/ai/contracts}/IDomain.ts | 0 .../interfaces/ai/contracts}/IEventData.ts | 0 .../ai/contracts}/IExceptionData.ts | 0 .../ai/contracts}/IExceptionDetails.ts | 0 .../interfaces/ai/contracts}/IMessageData.ts | 0 .../interfaces/ai/contracts}/IMetricData.ts | 0 .../interfaces/ai/contracts}/IPageViewData.ts | 0 .../ai/contracts}/IPageViewPerfData.ts | 0 .../ai/contracts}/IRemoteDependencyData.ts | 0 .../interfaces/ai/contracts}/IStackFrame.ts | 0 .../interfaces/ai/contracts}/SeverityLevel.ts | 2 +- .../src/interfaces/ai/telemetry}/IEnvelope.ts | 0 .../interfaces/ai/telemetry}/ISerializable.ts | 4 +- .../config}/IConfigDefaults.ts | 2 +- .../config}/IDynamicConfigHandler.ts | 6 +- .../config}/IDynamicPropertyHandler.ts | 4 +- .../config}/IDynamicWatcher.ts | 4 +- .../config}/_IDynamicConfigHandlerState.ts | 3 + .../src/interfaces/otel/IOTelApi.ts | 58 + .../src/interfaces/otel/IOTelApiCtx.ts | 18 + .../src/interfaces/otel/IOTelAttributes.ts | 36 + .../otel/attribute/IAttributeContainer.ts | 191 + .../otel/config/IOTelAttributeLimits.ts | 72 + .../src/interfaces/otel/config/IOTelConfig.ts | 55 + .../otel/config/IOTelErrorHandlers.ts | 196 + .../interfaces/otel/config/IOTelSpanLimits.ts | 131 + .../interfaces/otel/config/IOTelTraceCfg.ts | 85 + .../src/interfaces/otel/trace/IOTelSpan.ts | 488 +++ .../interfaces/otel/trace/IOTelSpanContext.ts | 23 + .../src/interfaces/otel/trace/IOTelSpanCtx.ts | 75 + .../interfaces/otel/trace/IOTelSpanOptions.ts | 34 + .../interfaces/otel/trace/IOTelSpanStatus.ts | 16 + .../interfaces/otel/trace/IOTelTraceApi.ts | 51 + .../interfaces/otel/trace/IOTelTraceState.ts | 62 + .../src/interfaces/otel/trace/IOTelTracer.ts | 147 + .../otel/trace/IOTelTracerOptions.ts | 9 + .../otel/trace/IOTelTracerProvider.ts | 39 + .../interfaces/otel/trace/IReadableSpan.ts | 56 + .../EventHelpers.ts | 69 +- .../src/internal/attributeHelpers.ts | 93 + .../src/internal/commonUtils.ts | 329 ++ .../src/internal/handleErrors.ts | 107 + .../src/internal/noopHelpers.ts | 16 + .../src/internal/timeHelpers.ts | 371 ++ .../AppInsightsCore/src/otel/api/OTelApi.ts | 25 + .../src/otel/api/errors/OTelError.ts | 32 + .../api/errors/OTelInvalidAttributeError.ts | 34 + .../src/otel/api/errors/OTelSpanError.ts | 31 + .../src/otel/api/trace/span.ts | 290 ++ .../src/otel/api/trace/spanContext.ts | 65 + .../src/otel/api/trace/traceApi.ts | 46 + .../src/otel/api/trace/traceProvider.ts | 76 + .../src/otel/api/trace/traceState.ts | 109 + .../src/otel/api/trace/tracer.ts | 52 + .../src/otel/api/trace/tracerProvider.ts | 40 + .../src/otel/api/trace/utils.ts | 383 ++ .../src/otel/attribute/SemanticConventions.ts | 106 + .../src/otel/attribute/attributeContainer.ts | 1136 ++++++ .../src/telemetry}/ConnectionStringParser.ts | 6 +- .../src/telemetry}/RequestResponseHeaders.ts | 10 +- .../src/telemetry}/TelemetryItemCreator.ts | 17 +- .../src/telemetry/W3cTraceState.ts | 458 +++ .../src/telemetry/ai}/Common/Data.ts | 10 +- .../src/telemetry/ai}/Common/DataPoint.ts | 8 +- .../src/telemetry/ai}/Common/DataSanitizer.ts | 26 +- .../src/telemetry/ai}/Common/Envelope.ts | 11 +- .../src/telemetry/ai/DataTypes.ts | 11 + .../src/telemetry/ai/EnvelopeTypes.ts | 16 + .../src/telemetry/ai}/Event.ts | 23 +- .../src/telemetry/ai}/Exception.ts | 49 +- .../src/telemetry/ai}/Metric.ts | 22 +- .../src/telemetry/ai}/PageView.ts | 24 +- .../src/telemetry/ai}/PageViewPerformance.ts | 24 +- .../src/telemetry/ai}/RemoteDependencyData.ts | 30 +- .../src/telemetry/ai}/Trace.ts | 24 +- .../src/{JavaScriptSDK => utils}/CoreUtils.ts | 2 +- .../DataCacheHelper.ts | 2 +- .../src/utils}/DomHelperFuncs.ts | 3 +- .../src/{JavaScriptSDK => utils}/EnvUtils.ts | 165 +- .../{JavaScriptSDK => utils}/HelperFuncs.ts | 225 +- .../src/utils}/Offline.ts | 13 +- .../{JavaScriptSDK => utils}/RandomHelper.ts | 12 +- .../src/utils}/StorageHelperFuncs.ts | 16 +- .../TraceParent.ts} | 36 +- .../src/utils}/UrlHelperFuncs.ts | 17 +- .../src => AppInsightsCore/src/utils}/Util.ts | 65 +- shared/AppInsightsCore/tsconfig.json | 10 +- shared/AppInsightsCore/typedoc.json | 2 +- .../package.json | 2 +- .../src/aiSupport.ts | 2 +- .../src/snippet.ts | 2 +- .../src/type.ts | 2 +- tools/chrome-debug-extension/package.json | 1 - .../scripts/publishAzReleaseToCdn.ps1 | 2 +- .../scripts/setAzActiveCdnVersion.ps1 | 2 +- .../src/dataSources/dataSources.ts | 2 +- .../config/scripts/publishAzReleaseToCdn.ps1 | 2 +- .../config/scripts/setAzActiveCdnVersion.ps1 | 2 +- tools/grunt-tasks/minifyNames.js | 9 +- tools/grunt-tasks/qunit.js | 52 +- tools/release-tools/setVersion.js | 2 +- tools/rollup-es5/Tests/UnitTests.html | 36 +- tools/rollup-es5/src/ImportCheck.ts | 5 +- tools/shims/Tests/UnitTests.html | 2 +- tsdoc.json | 10 + version.json | 6 +- 468 files changed, 49174 insertions(+), 3932 deletions(-) create mode 100644 AISKU/Tests/Manual/README.md create mode 100644 AISKU/Tests/Manual/span-e2e-manual-test.html create mode 100644 AISKU/Tests/Unit/src/NonRecordingSpan.Tests.ts create mode 100644 AISKU/Tests/Unit/src/OTelInit.Tests.ts create mode 100644 AISKU/Tests/Unit/src/SpanContextPropagation.Tests.ts create mode 100644 AISKU/Tests/Unit/src/SpanE2E.Tests.ts create mode 100644 AISKU/Tests/Unit/src/SpanErrorHandling.Tests.ts create mode 100644 AISKU/Tests/Unit/src/SpanHelperUtils.Tests.ts create mode 100644 AISKU/Tests/Unit/src/SpanLifeCycle.Tests.ts create mode 100644 AISKU/Tests/Unit/src/SpanPluginIntegration.Tests.ts create mode 100644 AISKU/Tests/Unit/src/SpanUtils.Tests.ts create mode 100644 AISKU/Tests/Unit/src/StartSpan.Tests.ts create mode 100644 AISKU/Tests/Unit/src/TelemetryItemGeneration.Tests.ts create mode 100644 AISKU/Tests/Unit/src/TraceContext.Tests.ts create mode 100644 AISKU/Tests/Unit/src/TraceProvider.Tests.ts create mode 100644 AISKU/Tests/Unit/src/TraceSuppression.Tests.ts create mode 100644 AISKU/Tests/Unit/src/UseSpan.Tests.ts create mode 100644 AISKU/Tests/Unit/src/WithSpan.Tests.ts create mode 100644 AISKU/examples/span-usage-example.ts create mode 100644 AISKU/src/internal/trace/spanUtils.ts create mode 100644 AISKULight/Tests/Unit/src/otelNegative.tests.ts create mode 100644 channels/applicationinsights-channel-js/src/Interfaces/Contracts/IRequestData.ts create mode 100644 channels/applicationinsights-channel-js/src/Telemetry/Common/Data.ts create mode 100644 channels/applicationinsights-channel-js/src/Telemetry/RemoteDependencyData.ts create mode 100644 channels/applicationinsights-channel-js/src/Telemetry/RequestData.ts create mode 100644 docs/OTel/README.md create mode 100644 docs/OTel/examples.md create mode 100644 docs/OTel/otelApi.md create mode 100644 docs/OTel/startActiveSpan.md create mode 100644 docs/OTel/traceApi.md create mode 100644 docs/OTel/useSpan.md create mode 100644 docs/OTel/withSpan.md create mode 100644 docs/upgrade/MergeCommonToCore.md create mode 100644 examples/dependency/src/startSpan-example.ts create mode 100644 examples/startSpan/package.json create mode 100644 examples/startSpan/rollup.config.js create mode 100644 examples/startSpan/src/startSpanExample.ts create mode 100644 examples/startSpan/src/startspan-example-index.ts create mode 100644 examples/startSpan/tsconfig.json create mode 100644 extensions/applicationinsights-dependencies-js/Tests/Unit/src/TestChannelPlugin.ts create mode 100644 extensions/applicationinsights-dependencies-js/Tests/Unit/src/W3CTraceStateDependency.tests.ts delete mode 100644 extensions/applicationinsights-dependencies-js/src/ajaxUtils.ts delete mode 100644 extensions/applicationinsights-properties-js/src/Context/TelemetryTrace.ts create mode 100644 shared/1ds-core-js/src/extSpanUtils.ts create mode 100644 shared/1ds-core-js/test/Unit/src/SpanUtilsTests.ts delete mode 100644 shared/AppInsightsCommon/Tests/Unit/src/appinsights-common.tests.ts delete mode 100644 shared/AppInsightsCommon/Tests/UnitTests.html delete mode 100644 shared/AppInsightsCommon/Tests/tsconfig.json delete mode 100644 shared/AppInsightsCommon/src/Enums.ts delete mode 100644 shared/AppInsightsCommon/src/HelperFuncs.ts delete mode 100644 shared/AppInsightsCommon/src/Interfaces/Contracts/IEnvelope.ts delete mode 100644 shared/AppInsightsCommon/src/Interfaces/Contracts/RequestData.ts delete mode 100644 shared/AppInsightsCommon/src/Interfaces/ICorrelationConfig.ts delete mode 100644 shared/AppInsightsCommon/src/Interfaces/PartAExtensions.ts create mode 100644 shared/AppInsightsCore/SpanImplementationSummary.md create mode 100644 shared/AppInsightsCore/Span_Implementation_Refactoring_Summary.md delete mode 100644 shared/AppInsightsCore/Tests/Unit/src/GlobalTestHooks.Test.ts create mode 100644 shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/commonUtils.Tests.ts create mode 100644 shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/errors.Tests.ts create mode 100644 shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/otelNegative.Tests.ts rename shared/{AppInsightsCommon/Tests/Unit/src => AppInsightsCore/Tests/Unit/src/ai}/AppInsightsCommon.tests.ts (98%) rename shared/AppInsightsCore/Tests/Unit/src/{ => ai}/AppInsightsCoreSize.Tests.ts (94%) rename shared/AppInsightsCore/Tests/Unit/src/{ => ai}/ApplicationInsightsCore.Tests.ts (99%) rename shared/{AppInsightsCommon/Tests/Unit/src => AppInsightsCore/Tests/Unit/src/ai}/ConnectionStringParser.tests.ts (97%) rename shared/AppInsightsCore/Tests/Unit/src/{ => ai}/CookieManager.Tests.ts (99%) rename shared/AppInsightsCore/Tests/Unit/src/{ => ai}/EventHelper.Tests.ts (96%) rename shared/AppInsightsCore/Tests/Unit/src/{ => ai}/EventsDiscardedReason.Tests.ts (99%) rename shared/{AppInsightsCommon/Tests/Unit/src => AppInsightsCore/Tests/Unit/src/ai}/Exception.tests.ts (98%) rename shared/{AppInsightsCommon/Tests/Unit/src => AppInsightsCore/Tests/Unit/src/ai}/GlobalTestHooks.Test.ts (82%) rename shared/AppInsightsCore/Tests/Unit/src/{ => ai}/HelperFunc.Tests.ts (98%) rename shared/AppInsightsCore/Tests/Unit/src/{ => ai}/LoggingEnum.Tests.ts (98%) rename shared/{AppInsightsCommon/Tests/Unit/src => AppInsightsCore/Tests/Unit/src/ai}/RequestHeaders.tests.ts (97%) rename shared/AppInsightsCore/Tests/Unit/src/{ => ai}/SendPostManager.Tests.ts (97%) rename shared/{AppInsightsCommon/Tests/Unit/src => AppInsightsCore/Tests/Unit/src/ai}/SeverityLevel.tests.ts (97%) rename shared/AppInsightsCore/Tests/Unit/src/{ => ai}/StatsBeat.Tests.ts (94%) rename shared/AppInsightsCore/Tests/Unit/src/{ => ai}/TestPlugins.ts (84%) rename shared/{AppInsightsCommon/Tests/Unit/src => AppInsightsCore/Tests/Unit/src/ai}/ThrottleMgr.tests.ts (99%) rename shared/AppInsightsCore/Tests/Unit/src/{ => ai}/UpdateConfig.Tests.ts (98%) rename shared/{AppInsightsCommon/Tests/Unit/src => AppInsightsCore/Tests/Unit/src/ai}/Util.tests.ts (98%) create mode 100644 shared/AppInsightsCore/Tests/Unit/src/attribute/attributeContainer.Tests.ts rename shared/AppInsightsCore/Tests/Unit/src/{ => config}/Dynamic.Tests.ts (98%) rename shared/AppInsightsCore/Tests/Unit/src/{ => config}/DynamicConfig.Tests.ts (98%) create mode 100644 shared/AppInsightsCore/Tests/Unit/src/trace/W3CTraceStateModes.tests.ts rename shared/AppInsightsCore/Tests/Unit/src/{ => trace}/W3cTraceParentTests.ts (98%) create mode 100644 shared/AppInsightsCore/Tests/Unit/src/trace/W3cTraceState.Tests.ts create mode 100644 shared/AppInsightsCore/Tests/Unit/src/trace/span.Tests.ts create mode 100644 shared/AppInsightsCore/Tests/Unit/src/trace/traceState.Tests.ts create mode 100644 shared/AppInsightsCore/Tests/Unit/src/trace/traceUtils.Tests.ts delete mode 100644 shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IDistributedTraceContext.ts delete mode 100644 shared/AppInsightsCore/src/JavaScriptSDK/Constants.ts delete mode 100644 shared/AppInsightsCore/src/JavaScriptSDK/TelemetryHelpers.ts delete mode 100644 shared/AppInsightsCore/src/applicationinsights-core-js.ts rename shared/AppInsightsCore/src/{Config => config}/ConfigDefaultHelpers.ts (93%) rename shared/AppInsightsCore/src/{Config => config}/ConfigDefaults.ts (98%) rename shared/AppInsightsCore/src/{Config => config}/DynamicConfig.ts (91%) rename shared/AppInsightsCore/src/{Config => config}/DynamicProperty.ts (96%) rename shared/AppInsightsCore/src/{Config => config}/DynamicState.ts (94%) rename shared/AppInsightsCore/src/{Config => config}/DynamicSupport.ts (94%) rename shared/{AppInsightsCommon/src => AppInsightsCore/src/constants}/Constants.ts (95%) rename shared/AppInsightsCore/src/{JavaScriptSDK => constants}/InternalConstants.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/AggregationError.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/AppInsightsCore.ts (79%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/AsyncUtils.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/BaseTelemetryPlugin.ts (92%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/CookieMgr.ts (95%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/DbgExtensionUtils.ts (84%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/InstrumentHooks.ts (99%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/NotificationManager.ts (95%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/PerfManager.ts (96%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/ProcessTelemetryContext.ts (78%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/ResponseHelpers.ts (68%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/SenderPostManager.ts (96%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/StatsBeat.ts (94%) create mode 100644 shared/AppInsightsCore/src/core/TelemetryHelpers.ts rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/TelemetryInitializerPlugin.ts (87%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/UnloadHandlerContainer.ts (79%) rename shared/AppInsightsCore/src/{JavaScriptSDK => core}/UnloadHookContainer.ts (90%) rename shared/AppInsightsCore/src/{JavaScriptSDK => diagnostics}/DiagnosticLogger.ts (93%) rename shared/{AppInsightsCommon/src => AppInsightsCore/src/diagnostics}/ThrottleMgr.ts (94%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Enums => enums}/EnumHelperFuncs.ts (100%) create mode 100644 shared/AppInsightsCore/src/enums/W3CTraceFlags.ts create mode 100644 shared/AppInsightsCore/src/enums/ai/DependencyTypes.ts create mode 100644 shared/AppInsightsCore/src/enums/ai/Enums.ts rename shared/AppInsightsCore/src/{JavaScriptSDK.Enums => enums/ai}/EventsDiscardedReason.ts (97%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Enums => enums/ai}/FeatureOptInEnums.ts (86%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Enums => enums/ai}/InitActiveStatusEnum.ts (84%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Enums => enums/ai}/LoggingEnums.ts (95%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Enums => enums/ai}/SendRequestReason.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Enums => enums/ai}/StatsType.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Enums => enums/ai}/TelemetryUnloadReason.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Enums => enums/ai}/TelemetryUpdateReason.ts (100%) create mode 100644 shared/AppInsightsCore/src/enums/ai/TraceHeadersMode.ts create mode 100644 shared/AppInsightsCore/src/enums/otel/OTelSamplingDecision.ts create mode 100644 shared/AppInsightsCore/src/enums/otel/OTelSpanKind.ts create mode 100644 shared/AppInsightsCore/src/enums/otel/OTelSpanStatus.ts create mode 100644 shared/AppInsightsCore/src/enums/otel/eAttributeChangeOp.ts create mode 100644 shared/AppInsightsCore/src/index.ts create mode 100644 shared/AppInsightsCore/src/interfaces/IException.ts create mode 100644 shared/AppInsightsCore/src/interfaces/IOTelHrTime.ts rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/ConnectionString.ts (67%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IAppInsights.ts (94%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IAppInsightsCore.ts (92%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IChannelControls.ts (98%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IChannelControlsHost.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IConfig.ts (98%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IConfiguration.ts (95%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/ICookieMgr.ts (100%) create mode 100644 shared/AppInsightsCore/src/interfaces/ai/ICorrelationConfig.ts rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IDbgExtension.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IDependencyTelemetry.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IDiagnosticLogger.ts (85%) create mode 100644 shared/AppInsightsCore/src/interfaces/ai/IDistributedTraceContext.ts rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IEventTelemetry.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IExceptionConfig.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IExceptionTelemetry.ts (97%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IFeatureOptIn.ts (85%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IInstrumentHooks.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IMetricTelemetry.ts (91%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/INetworkStatsbeat.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/INotificationListener.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/INotificationManager.ts (95%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IPageViewPerformanceTelemetry.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IPageViewTelemetry.ts (94%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IPartC.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IPerfEvent.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IPerfManager.ts (95%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IProcessTelemetryContext.ts (94%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IPropertiesPlugin.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IRequestContext.ts (64%) create mode 100644 shared/AppInsightsCore/src/interfaces/ai/IRequestTelemetry.ts rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/ISenderPostManager.ts (97%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IStatsBeat.ts (98%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IStatsEventData.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IStatsMgr.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IStorageBuffer.ts (77%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/ITelemetryContext.ts (68%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/ITelemetryInitializers.ts (88%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/ITelemetryItem.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/ITelemetryPlugin.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/ITelemetryPluginChain.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/ITelemetryUnloadState.ts (72%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/ITelemetryUpdateState.ts (93%) rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/IThrottleMgr.ts (96%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/ITraceParent.ts (100%) create mode 100644 shared/AppInsightsCore/src/interfaces/ai/ITraceProvider.ts rename shared/{AppInsightsCommon/src/Interfaces => AppInsightsCore/src/interfaces/ai}/ITraceTelemetry.ts (89%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IUnloadHook.ts (100%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IUnloadableComponent.ts (100%) create mode 100644 shared/AppInsightsCore/src/interfaces/ai/IW3cTraceState.ts rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IXDomainRequest.ts (92%) rename shared/AppInsightsCore/src/{JavaScriptSDK.Interfaces => interfaces/ai}/IXHROverride.ts (82%) create mode 100644 shared/AppInsightsCore/src/interfaces/ai/PartAExtensions.ts rename shared/{AppInsightsCommon/src/Interfaces/Context => AppInsightsCore/src/interfaces/ai/context}/IApplication.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Context => AppInsightsCore/src/interfaces/ai/context}/IDevice.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Context => AppInsightsCore/src/interfaces/ai/context}/IInternal.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Context => AppInsightsCore/src/interfaces/ai/context}/ILocation.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Context => AppInsightsCore/src/interfaces/ai/context}/IOperatingSystem.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Context => AppInsightsCore/src/interfaces/ai/context}/ISample.ts (76%) rename shared/{AppInsightsCommon/src/Interfaces/Context => AppInsightsCore/src/interfaces/ai/context}/ISession.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Context => AppInsightsCore/src/interfaces/ai/context}/ISessionManager.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Context => AppInsightsCore/src/interfaces/ai/context}/ITelemetryTrace.ts (64%) rename shared/{AppInsightsCommon/src/Interfaces/Context => AppInsightsCore/src/interfaces/ai/context}/IUser.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Context => AppInsightsCore/src/interfaces/ai/context}/IWeb.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/AvailabilityData.ts (92%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/ContextTagKeys.ts (99%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/DataPointType.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/DependencyKind.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/DependencySourceType.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IBase.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IData.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IDataPoint.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IDomain.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IEventData.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IExceptionData.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IExceptionDetails.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IMessageData.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IMetricData.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IPageViewData.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IPageViewPerfData.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IRemoteDependencyData.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/IStackFrame.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Contracts => AppInsightsCore/src/interfaces/ai/contracts}/SeverityLevel.ts (90%) rename shared/{AppInsightsCommon/src/Interfaces/Telemetry => AppInsightsCore/src/interfaces/ai/telemetry}/IEnvelope.ts (100%) rename shared/{AppInsightsCommon/src/Interfaces/Telemetry => AppInsightsCore/src/interfaces/ai/telemetry}/ISerializable.ts (73%) rename shared/AppInsightsCore/src/{Config => interfaces/config}/IConfigDefaults.ts (97%) rename shared/AppInsightsCore/src/{Config => interfaces/config}/IDynamicConfigHandler.ts (96%) rename shared/AppInsightsCore/src/{Config => interfaces/config}/IDynamicPropertyHandler.ts (85%) rename shared/AppInsightsCore/src/{Config => interfaces/config}/IDynamicWatcher.ts (94%) rename shared/AppInsightsCore/src/{Config => interfaces/config}/_IDynamicConfigHandlerState.ts (95%) create mode 100644 shared/AppInsightsCore/src/interfaces/otel/IOTelApi.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/IOTelApiCtx.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/IOTelAttributes.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/attribute/IAttributeContainer.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/config/IOTelAttributeLimits.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/config/IOTelConfig.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/config/IOTelErrorHandlers.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/config/IOTelSpanLimits.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/config/IOTelTraceCfg.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpan.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanContext.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanCtx.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanOptions.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanStatus.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTraceApi.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTraceState.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracer.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracerOptions.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracerProvider.ts create mode 100644 shared/AppInsightsCore/src/interfaces/otel/trace/IReadableSpan.ts rename shared/AppInsightsCore/src/{JavaScriptSDK => internal}/EventHelpers.ts (89%) create mode 100644 shared/AppInsightsCore/src/internal/attributeHelpers.ts create mode 100644 shared/AppInsightsCore/src/internal/commonUtils.ts create mode 100644 shared/AppInsightsCore/src/internal/handleErrors.ts create mode 100644 shared/AppInsightsCore/src/internal/noopHelpers.ts create mode 100644 shared/AppInsightsCore/src/internal/timeHelpers.ts create mode 100644 shared/AppInsightsCore/src/otel/api/OTelApi.ts create mode 100644 shared/AppInsightsCore/src/otel/api/errors/OTelError.ts create mode 100644 shared/AppInsightsCore/src/otel/api/errors/OTelInvalidAttributeError.ts create mode 100644 shared/AppInsightsCore/src/otel/api/errors/OTelSpanError.ts create mode 100644 shared/AppInsightsCore/src/otel/api/trace/span.ts create mode 100644 shared/AppInsightsCore/src/otel/api/trace/spanContext.ts create mode 100644 shared/AppInsightsCore/src/otel/api/trace/traceApi.ts create mode 100644 shared/AppInsightsCore/src/otel/api/trace/traceProvider.ts create mode 100644 shared/AppInsightsCore/src/otel/api/trace/traceState.ts create mode 100644 shared/AppInsightsCore/src/otel/api/trace/tracer.ts create mode 100644 shared/AppInsightsCore/src/otel/api/trace/tracerProvider.ts create mode 100644 shared/AppInsightsCore/src/otel/api/trace/utils.ts create mode 100644 shared/AppInsightsCore/src/otel/attribute/SemanticConventions.ts create mode 100644 shared/AppInsightsCore/src/otel/attribute/attributeContainer.ts rename shared/{AppInsightsCommon/src => AppInsightsCore/src/telemetry}/ConnectionStringParser.ts (87%) rename shared/{AppInsightsCommon/src => AppInsightsCore/src/telemetry}/RequestResponseHeaders.ts (91%) rename shared/{AppInsightsCommon/src => AppInsightsCore/src/telemetry}/TelemetryItemCreator.ts (81%) create mode 100644 shared/AppInsightsCore/src/telemetry/W3cTraceState.ts rename shared/{AppInsightsCommon/src/Telemetry => AppInsightsCore/src/telemetry/ai}/Common/Data.ts (68%) rename shared/{AppInsightsCommon/src/Telemetry => AppInsightsCore/src/telemetry/ai}/Common/DataPoint.ts (82%) rename shared/{AppInsightsCommon/src/Telemetry => AppInsightsCore/src/telemetry/ai}/Common/DataSanitizer.ts (88%) rename shared/{AppInsightsCommon/src/Telemetry => AppInsightsCore/src/telemetry/ai}/Common/Envelope.ts (87%) create mode 100644 shared/AppInsightsCore/src/telemetry/ai/DataTypes.ts create mode 100644 shared/AppInsightsCore/src/telemetry/ai/EnvelopeTypes.ts rename shared/{AppInsightsCommon/src/Telemetry => AppInsightsCore/src/telemetry/ai}/Event.ts (65%) rename shared/{AppInsightsCommon/src/Telemetry => AppInsightsCore/src/telemetry/ai}/Exception.ts (95%) rename shared/{AppInsightsCommon/src/Telemetry => AppInsightsCore/src/telemetry/ai}/Metric.ts (73%) rename shared/{AppInsightsCommon/src/Telemetry => AppInsightsCore/src/telemetry/ai}/PageView.ts (74%) rename shared/{AppInsightsCommon/src/Telemetry => AppInsightsCore/src/telemetry/ai}/PageViewPerformance.ts (79%) rename shared/{AppInsightsCommon/src/Telemetry => AppInsightsCore/src/telemetry/ai}/RemoteDependencyData.ts (78%) rename shared/{AppInsightsCommon/src/Telemetry => AppInsightsCore/src/telemetry/ai}/Trace.ts (66%) rename shared/AppInsightsCore/src/{JavaScriptSDK => utils}/CoreUtils.ts (97%) rename shared/AppInsightsCore/src/{JavaScriptSDK => utils}/DataCacheHelper.ts (97%) rename shared/{AppInsightsCommon/src => AppInsightsCore/src/utils}/DomHelperFuncs.ts (86%) rename shared/AppInsightsCore/src/{JavaScriptSDK => utils}/EnvUtils.ts (75%) rename shared/AppInsightsCore/src/{JavaScriptSDK => utils}/HelperFuncs.ts (71%) rename shared/{AppInsightsCommon/src => AppInsightsCore/src/utils}/Offline.ts (92%) rename shared/AppInsightsCore/src/{JavaScriptSDK => utils}/RandomHelper.ts (94%) rename shared/{AppInsightsCommon/src => AppInsightsCore/src/utils}/StorageHelperFuncs.ts (92%) rename shared/AppInsightsCore/src/{JavaScriptSDK/W3cTraceParent.ts => utils/TraceParent.ts} (89%) rename shared/{AppInsightsCommon/src => AppInsightsCore/src/utils}/UrlHelperFuncs.ts (85%) rename shared/{AppInsightsCommon/src => AppInsightsCore/src/utils}/Util.ts (79%) diff --git a/.aiAutoMinify.json b/.aiAutoMinify.json index 21ede5e68..8209d565e 100644 --- a/.aiAutoMinify.json +++ b/.aiAutoMinify.json @@ -5,6 +5,15 @@ "_eSetDynamicPropertyFlags", "ePendingOp", "CallbackType", + "eW3CTraceFlags", + "eRequestHeaders", + "eTraceStateKeyType", + "eOfflineValue", + "eDependencyTypes", + "eStorageType", + "FieldType", + "eDistributedTracingModes", + "EventPersistenceValue", "eEventsDiscardedReason", "eBatchDiscardedReason", "FeatureOptInMode", @@ -16,27 +25,28 @@ "TransportType", "eStatsType", "TelemetryUnloadReason", - "TelemetryUpdateReason" - ] - }, - "@microsoft/applicationinsights-perfmarkmeasure-js": { - "constEnums": [] - }, - "@microsoft/applicationinsights-common": { - "constEnums": [ - "eStorageType", - "FieldType", - "eDistributedTracingModes", - "EventPersistenceValue", - "eOfflineValue", - "eRequestHeaders", + "TelemetryUpdateReason", + "eTraceHeadersMode", + "eAttributeChangeOp", + "eOTelSamplingDecision", + "eOTelSpanKind", + "eOTelSpanStatusCode", + "eAttributeSource", + "AddAttributeResult", "DataPointType", "DependencyKind", "DependencySourceType", "eSeverityLevel", + "eAttributeFilter", "DataSanitizerValues" ] }, + "@microsoft/applicationinsights-perfmarkmeasure-js": { + "constEnums": [] + }, + "@microsoft/applicationinsights-common": { + "constEnums": [] + }, "@microsoft/applicationinsights-properties-js": { "constEnums": [] }, @@ -47,7 +57,9 @@ "constEnums": [] }, "@microsoft/applicationinsights-channel-js": { - "constEnums": [] + "constEnums": [ + "eSerializeType" + ] }, "@microsoft/applicationinsights-react-native": { "constEnums": [] @@ -61,7 +73,9 @@ "constEnums": [] }, "@microsoft/applicationinsights-analytics-js": { - "constEnums": [] + "constEnums": [ + "eRouteTraceStrategy" + ] }, "@microsoft/applicationinsights-web": { "constEnums": [] diff --git a/.github/copilot-instructions.md b/.github/copilot-instructions.md index 03c797bf0..1f3b0395a 100644 --- a/.github/copilot-instructions.md +++ b/.github/copilot-instructions.md @@ -188,17 +188,120 @@ export class MyPlugin extends BaseTelemetryPlugin { - Test both success and failure scenarios - Verify telemetry data structure and content +### Testing Framework Requirements +- **Extend AITestClass**: All test classes must extend `AITestClass` from `@microsoft/ai-test-framework` +- **Use Framework Tools**: Leverage existing framework helpers like `this.hookFetch()`, `this.useFakeTimers()`, and `this.onDone()` +- **Proper Registration**: Implement `registerTests()` method and use `this.testCase()` for test registration +- **Async Tests**: Return `IPromise` from test functions for asynchronous operations (do not use deprecated `testCaseAsync`) + +### Critical Cleanup & Resource Management +- **Mandatory Core Cleanup**: Always call `appInsightsCore.unload(false)` in test cleanup to prevent hook pollution between tests +- **Extension Teardown**: Only call `teardown()` on extension instances that were NOT added to a core instance; `core.unload()` handles teardown for initialized extensions +- **Hook Validation**: The framework validates that all hooks are properly removed; tests will fail if cleanup is incomplete +- **Resource Isolation**: Each test must be completely isolated - no shared state or leftover hooks + +### Configuration Testing Requirements +- **Static Configuration**: Test initial configuration setup and validation +- **Dynamic Configuration**: **REQUIRED** - All tests that touch configuration must include post-initialization configuration change tests +- **onConfigChange Testing**: Components using `onConfigChange` callbacks must be tested for runtime configuration updates +- **Configuration Validation**: Test both valid and invalid configuration scenarios with proper error handling + +```typescript +// Example dynamic configuration test pattern +public testDynamicConfig() { + // Initial setup with one config + let initialConfig = { enableFeature: false }; + core.initialize(initialConfig, channels); + + // Verify initial behavior + Assert.equal(false, component.isFeatureEnabled()); + + // Update configuration dynamically + core.config.enableFeature = true; + // Note: core.onConfigChange() only registers callbacks, it doesn't trigger changes + + // To trigger config change detection, use one of these patterns: + + // Option 1: Using fake timers (synchronous) + this.clock.tick(1); // Trigger 1ms timer for config change detection + + // Option 2: Async test without fake timers + // return createPromise((resolve) => { + // setTimeout(() => { + // Assert.equal(true, component.isFeatureEnabled()); + // resolve(); + // }, 10); + // }); + + // Verify behavior changed (when using fake timers) + Assert.equal(true, component.isFeatureEnabled()); +} +``` + +### Package Organization & Dependencies +- **Respect Package Boundaries**: Place tests in the package that owns the functionality being tested +- **Dependency Injection**: Extensions must include dependencies in `config.extensions` array for proper initialization +- **Cross-Package Coordination**: Understand which package owns which functionality when testing integrated features +- **Import Resolution**: Use proper module imports and avoid direct file path dependencies + +### HTTP API & Network Testing +- **Use Framework Helpers**: Use `this.hookFetch()` instead of custom fetch mocking implementations +- **XMLHttpRequest Testing**: Use framework's built-in mechanisms for XHR validation +- **Header Validation**: Test both presence and absence of headers based on different configuration modes +- **Network Scenarios**: Test success, failure, timeout, and edge cases consistently + +### Async Testing Patterns +- **IPromise Return**: Use `this.testCase()` and return `IPromise` for asynchronous operations instead of deprecated `testCaseAsync` +- **Promise Handling**: Handle both resolution and rejection paths in async tests +- **Timing Control**: Use `this.clock.tick()` when `useFakeTimers: true` for deterministic timing +- **Cleanup in Async**: Ensure cleanup happens in both success and failure paths of async tests + +```typescript +// Example async test pattern +this.testCase({ + name: "Async operation test", + test: () => { + return createPromise((resolve, reject) => { + // Setup async operation + someAsyncOperation().then(() => { + try { + // Assertions + Assert.ok(true, "Operation succeeded"); + resolve(); + } catch (e) { + reject(e); + } + }).catch(reject); + }); + } +}); +``` + +### Unit Testing Best Practices +- **Comprehensive Coverage**: Test all major code paths including edge cases and error conditions +- **Mock Browser APIs**: Mock browser APIs consistently using framework-provided mechanisms +- **Telemetry Validation**: Verify telemetry data structure, content, and proper formatting +- **State Testing**: Test both empty/null states and populated states for data structures + ### Browser Testing -- Cross-browser compatibility testing -- Performance regression testing -- Memory leak detection -- Network failure scenarios +- **Cross-browser Compatibility**: Test across different browser environments and API availability +- **Performance Regression**: Monitor test execution time and detect performance regressions +- **Memory Leak Detection**: Verify proper cleanup and resource management in long-running scenarios +- **API Graceful Degradation**: Test behavior when browser APIs are unavailable or disabled ### Test Organization -- Collocate tests with source code in `Tests/` directories -- Use descriptive test names -- Group related tests in test suites -- Mock external dependencies +- **Collocate Tests**: Place tests in `Tests/` directories within the same package as source code +- **Descriptive Naming**: Use clear, descriptive test names that explain the scenario being tested +- **Logical Grouping**: Group related tests in test suites within the same test class +- **Documentation**: Include comments explaining complex test scenarios and edge cases + +### Common Anti-Patterns to Avoid +- **Skipping Cleanup**: Not calling `unload()` or `teardown()` methods leads to test interference +- **Custom Implementations**: Implementing custom mocks/helpers instead of using framework-provided tools +- **Configuration Gaps**: Testing only static configuration without dynamic configuration change scenarios +- **Hook Pollution**: Leaving hooks active between tests causing false positives/negatives +- **Incomplete Coverage**: Missing edge cases, error conditions, or state transitions +- **Deprecated Async**: Using deprecated `testCaseAsync` instead of `testCase` with `IPromise` return ## Configuration & Initialization diff --git a/AISKU/README.md b/AISKU/README.md index 14205bb44..034ae081c 100644 --- a/AISKU/README.md +++ b/AISKU/README.md @@ -34,6 +34,7 @@ See [Breaking Changes](https://microsoft.github.io/ApplicationInsights-JS/upgrad | Version | Full Size | Raw Minified | GZip Size |---------|-----------|--------------|------------- | [<nightly3>](https://github.com/microsoft/ApplicationInsights-JS/tree/main/AISKU) | [![full size size](https://js.monitor.azure.com/nightly/ai.3-nightly3.js.svg)](https://js.monitor.azure.com/nightly/ai.3-nightly3.js.svg)| ![minified size size](https://js.monitor.azure.com/nightly/ai.3-nightly3.min.js.svg) | ![gzip size](https://js.monitor.azure.com/nightly/ai.3-nightly3.min.js.gzip.svg) +| 3.4.0-beta: | [![full size size](https://js.monitor.azure.com/scripts/b/ai.3.4.0-beta.js.svg)](https://js.monitor.azure.com/scripts/b/ai.3.4.0-beta.js.svg)| ![minified size size](https://js.monitor.azure.com/scripts/b/ai.3.4.0-beta.min.js.svg) | ![gzip size](https://js.monitor.azure.com/scripts/b/ai.3.3.11.min.js.gzip.svg) | 3.3.11: | [![full size size](https://js.monitor.azure.com/scripts/b/ai.3.3.11.js.svg)](https://js.monitor.azure.com/scripts/b/ai.3.3.11.js.svg)| ![minified size size](https://js.monitor.azure.com/scripts/b/ai.3.3.11.min.js.svg) | ![gzip size](https://js.monitor.azure.com/scripts/b/ai.3.3.11.min.js.gzip.svg) | 3.3.10: | [![full size size](https://js.monitor.azure.com/scripts/b/ai.3.3.10.js.svg)](https://js.monitor.azure.com/scripts/b/ai.3.3.10.js.svg)| ![minified size size](https://js.monitor.azure.com/scripts/b/ai.3.3.10.min.js.svg) | ![gzip size](https://js.monitor.azure.com/scripts/b/ai.3.3.10.min.js.gzip.svg) | 3.3.9: | [![full size size](https://js.monitor.azure.com/scripts/b/ai.3.3.9.js.svg)](https://js.monitor.azure.com/scripts/b/ai.3.3.9.js.svg)| ![minified size size](https://js.monitor.azure.com/scripts/b/ai.3.3.9.min.js.svg) | ![gzip size](https://js.monitor.azure.com/scripts/b/ai.3.3.9.min.js.gzip.svg) diff --git a/AISKU/Tests/Manual/README.md b/AISKU/Tests/Manual/README.md new file mode 100644 index 000000000..090aacdcf --- /dev/null +++ b/AISKU/Tests/Manual/README.md @@ -0,0 +1,262 @@ +# Span API End-to-End (E2E) Tests + +This directory contains end-to-end tests for the new Span APIs that send real telemetry to Azure Application Insights (Breeze endpoint) for manual validation in the Azure Portal. + +## 📁 Files + +- **`SpanE2E.Tests.ts`** - Automated E2E test suite that can be configured to send real telemetry +- **`span-e2e-manual-test.html`** - Interactive HTML page for manual testing with visual feedback + +## 🚀 Quick Start - Manual HTML Testing + +The easiest way to test is using the interactive HTML page: + +1. **Get your Application Insights credentials**: + - Go to [Azure Portal](https://portal.azure.com) + - Navigate to your Application Insights resource (or create a new one) + - Copy the **Instrumentation Key** or **Connection String** from the Overview page + +2. **Open the test page**: + ```bash + # Option 1: Open directly in browser + open AISKU/Tests/Manual/span-e2e-manual-test.html + + # Option 2: Serve via local web server + cd AISKU/Tests/Manual + python -m http.server 8080 + # Then open http://localhost:8080/span-e2e-manual-test.html + ``` + +3. **Run tests**: + - Paste your Instrumentation Key or Connection String + - Click "Initialize SDK" + - Run individual tests or click "Run All Tests" + - Watch the output log for confirmation + +4. **View results in Azure Portal**: + - Wait 1-2 minutes for telemetry to arrive + - Go to your Application Insights resource + - Navigate to **Performance** → **Dependencies** or **Requests** + - Use **Search** to find specific test scenarios + - Click **"View in End-to-End Transaction"** to see distributed traces + +## 🧪 Automated Test Suite + +### Configuration + +To run the automated test suite with real telemetry: + +1. Open [`SpanE2E.Tests.ts`](../Unit/src/SpanE2E.Tests.ts) + +2. Update the configuration: + ```typescript + // Set to true to send real telemetry + private static readonly MANUAL_E2E_TEST = true; + + // Replace with your instrumentation key + private static readonly _instrumentationKey = "YOUR-IKEY-HERE"; + ``` + +3. Run the tests: + ```bash + # From repository root + rush build + rush test + ``` + +### Test Scenarios Included + +The test suite covers: + +#### Basic Span Tests +- ✅ CLIENT span → RemoteDependency +- ✅ SERVER span → Request +- ✅ Failed span → success=false + +#### Distributed Trace Tests +- ✅ Parent-child relationships +- ✅ 3-level nested hierarchy +- ✅ Context propagation + +#### HTTP Dependency Tests +- ✅ Various HTTP methods (GET, POST, PUT, DELETE) +- ✅ Multiple status codes (2xx, 4xx, 5xx) +- ✅ Full HTTP details (headers, body size, response time) + +#### Database Dependency Tests +- ✅ MySQL, PostgreSQL, MongoDB, Redis, SQL Server +- ✅ SQL statements and operations +- ✅ Slow query scenarios + +#### Complex Scenarios +- ✅ E-commerce checkout flow (7 dependencies) +- ✅ Mixed success and failure operations +- ✅ Rich custom properties for filtering + +## 🔍 What to Look For in the Portal + +### Performance Blade + +**Dependencies Tab**: +- Look for CLIENT, PRODUCER, and INTERNAL spans +- Verify dependency types (Http, mysql, postgresql, redis, etc.) +- Check duration, target, and result codes +- Examine custom properties in the details pane + +**Requests Tab**: +- Look for SERVER and CONSUMER spans +- Verify URLs, methods, and status codes +- Check success/failure status +- View response codes and durations + +### Search Feature + +Filter by custom properties to find specific test runs: +``` +customDimensions.test.scenario == "ecommerce" +customDimensions.test.timestamp >= datetime(2025-12-01) +customDimensions.business.tenant == "manual-test-corp" +``` + +### End-to-End Transaction View + +1. Click any request or dependency +2. Click **"View in End-to-End Transaction"** +3. See the complete distributed trace: + - Timeline showing span durations + - Parent-child relationships + - All related dependencies + - Custom properties at each level + +### Transaction Timeline + +Look for: +- ✅ Correct parent-child relationships (indentation) +- ✅ Proper span nesting (visual hierarchy) +- ✅ Accurate duration calculations +- ✅ Operation IDs matching across spans +- ✅ Custom dimensions preserved throughout + +## 📊 Expected Results + +### Test: Basic CLIENT Span +- **Portal Location**: Performance → Dependencies +- **Dependency Type**: "Dependency" or "Http" +- **Custom Properties**: test.scenario, test.timestamp + +### Test: Parent-Child Trace +- **Portal Location**: End-to-End Transaction view +- **Expected**: 1 Request + 2 Dependencies +- **Relationship**: Both children reference same parent operation.id + +### Test: E-commerce Checkout +- **Portal Location**: End-to-End Transaction view +- **Expected**: 1 Request + 7 Dependencies +- **Types**: Http (inventory, payment, email), Database (create order), Redis (cache) +- **Duration**: Parent spans entire operation + +### Test: Rich Custom Properties +- **Portal Location**: Search → Custom dimensions filter +- **Expected Properties**: + - business.tenant + - user.subscription + - feature.* flags + - performance.* metrics + +## 🐛 Troubleshooting + +### Telemetry not appearing in portal + +1. **Wait longer**: Initial ingestion can take 1-3 minutes +2. **Check time filter**: Ensure portal is showing last 30 minutes +3. **Verify iKey**: Confirm instrumentation key is correct +4. **Check browser console**: Look for SDK errors +5. **Flush telemetry**: Call `appInsights.flush()` in tests + +### SDK initialization fails + +1. **Valid credentials**: Verify instrumentation key format +2. **CORS issues**: Ensure application is running on http/https (not file://) +3. **Browser compatibility**: Use modern browser (Chrome, Edge, Firefox) + +### Missing custom properties + +1. **Property name limits**: Check for truncation (8192 char limit) +2. **Reserved names**: Some property names are filtered (http.*, db.*, microsoft.*) +3. **Type preservation**: Ensure values are correct types (string, number, boolean) + +## 📝 Adding New Test Scenarios + +To add a new E2E test scenario: + +1. **In SpanE2E.Tests.ts**: + ```typescript + this.testCase({ + name: "E2E: Your new scenario", + test: () => { + const span = this._ai.startSpan("E2E-YourScenario", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "test.scenario": "your-scenario", + "custom.property": "value" + } + }); + + if (span) { + span.setStatus({ code: eOTelSpanStatusCode.OK }); + span.end(); + } + + this._ai.flush(); + Assert.ok(span, "Span created"); + } + }); + ``` + +2. **In span-e2e-manual-test.html**: + ```javascript + function testYourScenario() { + if (!appInsights) return; + + const span = appInsights.startSpan('E2E-Manual-YourScenario', { + kind: 1, // CLIENT + attributes: { + 'test.scenario': 'your-scenario', + 'custom.property': 'value' + } + }); + + if (span) { + span.setStatus({ code: 1 }); + span.end(); + log('✅ Your scenario sent', 'success'); + } + appInsights.flush(); + } + ``` + +## 🎯 Best Practices + +1. **Use descriptive names**: Prefix test spans with "E2E-" or "Manual-" +2. **Include timestamps**: Add test.timestamp for filtering +3. **Add scenario tags**: Use test.scenario for grouping +4. **Flush after tests**: Always call `flush()` to send immediately +5. **Wait before checking**: Give telemetry 1-2 minutes to arrive +6. **Use unique identifiers**: Help distinguish between test runs +7. **Clean up regularly**: Archive or delete old test data + +## 🔗 Resources + +- [Application Insights Overview](https://docs.microsoft.com/azure/azure-monitor/app/app-insights-overview) +- [OpenTelemetry Specification](https://opentelemetry.io/docs/specs/otel/trace/api/) +- [Azure Portal](https://portal.azure.com) +- [Application Insights SDK Documentation](../../../README.md) + +## 🤝 Contributing + +When adding new E2E tests: +1. Follow existing naming conventions (E2E-* prefix) +2. Include relevant custom properties +3. Document expected portal behavior +4. Update this README with new scenarios +5. Test manually before committing diff --git a/AISKU/Tests/Manual/span-e2e-manual-test.html b/AISKU/Tests/Manual/span-e2e-manual-test.html new file mode 100644 index 000000000..00172a525 --- /dev/null +++ b/AISKU/Tests/Manual/span-e2e-manual-test.html @@ -0,0 +1,818 @@ + + + + + + Application Insights - Span API Manual E2E Test + + + +
+

🔍 Application Insights - Span API Manual E2E Test

+

Send real telemetry to Azure Application Insights and verify in the portal

+
+ +
+

⚙️ Configuration

+
+ + +

+ Get this from: Azure Portal → Your Application Insights resource → Overview → Instrumentation Key +

+
+ +
+
+ +
+

🧪 Test Scenarios

+ +

Basic Tests

+ + + + +

Distributed Trace Tests

+ + + +

Dependency Tests

+ + + +

Complex Scenarios

+ + + + +

Batch Actions

+
+ + +
+ +

Utilities

+ + +
+ +
+

📊 Statistics

+
+
+
0
+
Spans Created
+
+
+
0
+
Dependencies
+
+
+
0
+
Requests
+
+
+
0
+
Trace IDs
+
+
+
+ +
+

📝 Output Log

+
+ Ready to start testing. Initialize the SDK first. +
+
+ +
+

🔗 View Results

+

After running tests, wait 1-2 minutes for telemetry to appear in the portal.

+ Open Azure Portal + +
+ 💡 Tips for viewing in portal: + +
+
+ + + + + + + diff --git a/AISKU/Tests/Perf/src/AISKUPerf.Tests.ts b/AISKU/Tests/Perf/src/AISKUPerf.Tests.ts index 19aa62857..023de9783 100644 --- a/AISKU/Tests/Perf/src/AISKUPerf.Tests.ts +++ b/AISKU/Tests/Perf/src/AISKUPerf.Tests.ts @@ -1,6 +1,6 @@ import { AITestClass, Assert } from "@microsoft/ai-test-framework"; import { AppInsightsInitPerfTestClass } from "./AISKUPerf"; -import { utlRemoveSessionStorage } from "@microsoft/applicationinsights-common"; +import { utlRemoveSessionStorage } from "@microsoft/applicationinsights-core-js"; import { createTimeoutPromise, doAwait } from "@nevware21/ts-async"; function isNullOrUndefined(value: any): boolean { @@ -153,8 +153,8 @@ export class AISKUPerf extends AITestClass { } public testCleanup() { - utlRemoveSessionStorage(null as any, "AI_sentBuffer", ); - utlRemoveSessionStorage(null as any, "AI_buffer", ); + utlRemoveSessionStorage(null as any, "AI_sentBuffer"); + utlRemoveSessionStorage(null as any, "AI_buffer"); } public registerTests() { diff --git a/AISKU/Tests/Unit/src/AISKUSize.Tests.ts b/AISKU/Tests/Unit/src/AISKUSize.Tests.ts index 2a7782040..e48d81793 100644 --- a/AISKU/Tests/Unit/src/AISKUSize.Tests.ts +++ b/AISKU/Tests/Unit/src/AISKUSize.Tests.ts @@ -2,7 +2,7 @@ import { AITestClass, Assert } from "@microsoft/ai-test-framework"; import { dumpObj } from '@nevware21/ts-utils'; import { createPromise, doAwait, IPromise } from '@nevware21/ts-async'; import { strUndefined } from "@microsoft/applicationinsights-core-js"; -import { utlRemoveSessionStorage } from "@microsoft/applicationinsights-common"; +import { utlRemoveSessionStorage } from "@microsoft/applicationinsights-core-js"; import * as pako from "pako"; import { Snippet } from "../../../src/Snippet"; @@ -46,18 +46,18 @@ function _loadPackageJson(cb:(isNightly: boolean, packageJson: any) => IPromise< } function _checkSize(checkType: string, maxSize: number, size: number, isNightly: boolean): void { - if (isNightly) { + if (isNightly) { maxSize += .5; } Assert.ok(size <= maxSize, `exceed ${maxSize} KB, current ${checkType} size is: ${size} KB`); -} +} export class AISKUSizeCheck extends AITestClass { - private readonly MAX_RAW_SIZE = 150; - private readonly MAX_BUNDLE_SIZE = 150; - private readonly MAX_RAW_DEFLATE_SIZE = 60; - private readonly MAX_BUNDLE_DEFLATE_SIZE = 60; + private readonly MAX_RAW_SIZE = 174; + private readonly MAX_BUNDLE_SIZE = 174; + private readonly MAX_RAW_DEFLATE_SIZE = 70; + private readonly MAX_BUNDLE_DEFLATE_SIZE = 70; private readonly rawFilePath = "../dist/es5/applicationinsights-web.min.js"; // Automatically updated by version scripts private readonly currentVer = "3.3.11"; @@ -85,8 +85,8 @@ export class AISKUSizeCheck extends AITestClass { } public testCleanup() { - utlRemoveSessionStorage(null as any, "AI_sentBuffer", ); - utlRemoveSessionStorage(null as any, "AI_buffer", ); + utlRemoveSessionStorage(null as any, "AI_sentBuffer"); + utlRemoveSessionStorage(null as any, "AI_buffer"); } public registerTests() { diff --git a/AISKU/Tests/Unit/src/CdnPackaging.tests.ts b/AISKU/Tests/Unit/src/CdnPackaging.tests.ts index 5d84b058b..2aa3fd8e5 100644 --- a/AISKU/Tests/Unit/src/CdnPackaging.tests.ts +++ b/AISKU/Tests/Unit/src/CdnPackaging.tests.ts @@ -2,7 +2,7 @@ import { AITestClass, Assert } from "@microsoft/ai-test-framework"; import { AnalyticsPluginIdentifier, BreezeChannelIdentifier, DEFAULT_BREEZE_ENDPOINT, DisabledPropertyName, DistributedTracingModes, PropertiesPluginIdentifier, RequestHeaders, SeverityLevel -} from "@microsoft/applicationinsights-common"; +} from "@microsoft/applicationinsights-core-js"; import { dumpObj, LoggingSeverity, objForEachKey, objKeys, strUndefined } from "@microsoft/applicationinsights-core-js"; import { Snippet } from "../../../src/Snippet"; diff --git a/AISKU/Tests/Unit/src/CdnThrottle.tests.ts b/AISKU/Tests/Unit/src/CdnThrottle.tests.ts index 2d3142132..f0e0f44c6 100644 --- a/AISKU/Tests/Unit/src/CdnThrottle.tests.ts +++ b/AISKU/Tests/Unit/src/CdnThrottle.tests.ts @@ -1,6 +1,6 @@ import { ApplicationInsights, ApplicationInsightsContainer, IApplicationInsights, IConfig, IConfiguration, LoggingSeverity, Snippet, _eInternalMessageId } from '../../../src/applicationinsights-web' import { AITestClass, Assert, IFetchArgs, PollingAssert} from '@microsoft/ai-test-framework'; -import { IThrottleInterval, IThrottleLimit, IThrottleMgrConfig } from '@microsoft/applicationinsights-common'; +import { IThrottleInterval, IThrottleLimit, IThrottleMgrConfig } from '@microsoft/applicationinsights-core-js'; import { SinonSpy } from 'sinon'; import { AppInsightsSku } from '../../../src/AISku'; import { createSnippetV5 } from './testSnippetV5'; diff --git a/AISKU/Tests/Unit/src/IAnalyticsConfig.Tests.ts b/AISKU/Tests/Unit/src/IAnalyticsConfig.Tests.ts index d9bd8c6f5..26d123f2a 100644 --- a/AISKU/Tests/Unit/src/IAnalyticsConfig.Tests.ts +++ b/AISKU/Tests/Unit/src/IAnalyticsConfig.Tests.ts @@ -1,6 +1,6 @@ import { ApplicationInsights, IAnalyticsConfig, IAppInsights, IConfig, ApplicationAnalytics } from "../../../src/applicationinsights-web"; import { AITestClass, Assert } from "@microsoft/ai-test-framework"; -import { AnalyticsPluginIdentifier, utlRemoveSessionStorage } from "@microsoft/applicationinsights-common"; +import { AnalyticsPluginIdentifier, utlRemoveSessionStorage } from "@microsoft/applicationinsights-core-js"; import { AppInsightsCore, IConfiguration, isFunction, onConfigChange } from "@microsoft/applicationinsights-core-js"; import { Sender } from "@microsoft/applicationinsights-channel-js"; diff --git a/AISKU/Tests/Unit/src/NonRecordingSpan.Tests.ts b/AISKU/Tests/Unit/src/NonRecordingSpan.Tests.ts new file mode 100644 index 000000000..12597ed63 --- /dev/null +++ b/AISKU/Tests/Unit/src/NonRecordingSpan.Tests.ts @@ -0,0 +1,773 @@ +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { ApplicationInsights } from "../../../src/applicationinsights-web"; +import { eOTelSpanKind, eOTelSpanStatusCode, ITelemetryItem } from "@microsoft/applicationinsights-core-js"; + +/** + * Comprehensive tests for non-recording span behavior + * + * Non-recording spans are used for: + * - Context propagation without telemetry overhead + * - Testing and debugging scenarios + * - Wrapping external span contexts + * - Performance-sensitive scenarios + */ +export class NonRecordingSpanTests extends AITestClass { + private static readonly _instrumentationKey = "b7170927-2d1c-44f1-acec-59f4e1751c11"; + private static readonly _connectionString = `InstrumentationKey=${NonRecordingSpanTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "NonRecordingSpanTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: NonRecordingSpanTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + } catch (e) { + console.error("Failed to initialize tests: " + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addBasicNonRecordingTests(); + this.addAttributeOperationTests(); + this.addStatusAndNameTests(); + this.addSpanKindTests(); + this.addHierarchyTests(); + this.addTelemetryGenerationTests(); + this.addPerformanceTests(); + this.addEdgeCaseTests(); + } + + private addBasicNonRecordingTests(): void { + this.testCase({ + name: "NonRecording: span created with recording:false is not recording", + test: () => { + // Act + const span = this._ai.startSpan("non-recording-basic", { recording: false }); + + // Assert + Assert.ok(span, "Span should be created"); + Assert.ok(!span.isRecording(), "Span should not be recording"); + Assert.equal(span.name, "non-recording-basic", "Span name should be set"); + } + }); + + this.testCase({ + name: "NonRecording: default recording:true creates recording span", + test: () => { + // Act + const span = this._ai.startSpan("recording-default"); + + // Assert + Assert.ok(span, "Span should be created"); + Assert.ok(span.isRecording(), "Span should be recording by default"); + } + }); + + this.testCase({ + name: "NonRecording: explicit recording:true creates recording span", + test: () => { + // Act + const span = this._ai.startSpan("recording-explicit", { recording: true }); + + // Assert + Assert.ok(span, "Span should be created"); + Assert.ok(span.isRecording(), "Span should be recording"); + } + }); + + this.testCase({ + name: "NonRecording: isRecording() returns false throughout lifecycle", + test: () => { + // Arrange + const span = this._ai.startSpan("non-recording-lifecycle", { recording: false }); + + // Act & Assert - Before operations + Assert.ok(!span.isRecording(), "Should not be recording initially"); + + // Perform operations + span!.setAttribute("key", "value"); + Assert.ok(!span.isRecording(), "Should not be recording after setAttribute"); + + span!.setStatus({ code: eOTelSpanStatusCode.OK }); + Assert.ok(!span.isRecording(), "Should not be recording after setStatus"); + + span!.updateName("new-name"); + Assert.ok(!span.isRecording(), "Should not be recording after updateName"); + + span!.end(); + Assert.ok(!span.isRecording(), "Should not be recording after end"); + } + }); + } + + private addAttributeOperationTests(): void { + this.testCase({ + name: "NonRecording: setAttribute does not store attributes", + test: () => { + // Arrange + const span = this._ai.startSpan("non-recording-attrs", { recording: false }); + + // Act + span!.setAttribute("key1", "value1"); + span!.setAttribute("key2", 123); + span!.setAttribute("key3", true); + + // Assert + const attrs = span!.attributes; + Assert.ok(attrs, "Attributes object should exist"); + // Non-recording spans don't store attributes + Assert.equal(Object.keys(attrs).length, 0, "No attributes should be stored"); + } + }); + + this.testCase({ + name: "NonRecording: setAttributes does not store attributes", + test: () => { + // Arrange + const span = this._ai.startSpan("non-recording-set-attrs", { recording: false }); + + // Act + span!.setAttributes({ + "attr1": "value1", + "attr2": 456, + "attr3": false, + "attr4": [1, 2, 3] + }); + + // Assert + const attrs = span!.attributes; + Assert.equal(Object.keys(attrs).length, 0, "No attributes should be stored"); + } + }); + + this.testCase({ + name: "NonRecording: setAttribute returns span for chaining", + test: () => { + // Arrange + const span = this._ai.startSpan("non-recording-chain", { recording: false }); + + // Act + const result = span!.setAttribute("key", "value"); + + // Assert + Assert.equal(result, span, "setAttribute should return the span for chaining"); + } + }); + + this.testCase({ + name: "NonRecording: setAttributes returns span for chaining", + test: () => { + // Arrange + const span = this._ai.startSpan("non-recording-chain-multi", { recording: false }); + + // Act + const result = span!.setAttributes({ "key1": "value1", "key2": "value2" }); + + // Assert + Assert.equal(result, span, "setAttributes should return the span for chaining"); + } + }); + + this.testCase({ + name: "NonRecording: multiple setAttribute calls increment dropped count", + test: () => { + // Arrange + const span = this._ai.startSpan("non-recording-dropped", { recording: false }); + + // Act + span!.setAttribute("key1", "value1"); + span!.setAttribute("key2", "value2"); + span!.setAttribute("key3", "value3"); + span!.setAttributes({ "key4": "value4", "key5": "value5" }); + + // Assert + const droppedCount = span!.droppedAttributesCount; + Assert.ok(droppedCount >= 5, `At least 5 attributes should be dropped, got ${droppedCount}`); + } + }); + + this.testCase({ + name: "NonRecording: setAttribute after end() increments dropped count", + test: () => { + // Arrange + const span = this._ai.startSpan("non-recording-after-end", { recording: false }); + span!.end(); + + // Act + span!.setAttribute("late-key", "late-value"); + + // Assert - Should not throw, just increment dropped count + Assert.ok(span.ended, "Span should be ended"); + Assert.ok(span.droppedAttributesCount > 0, "Dropped attribute count should be incremented"); + } + }); + } + + private addStatusAndNameTests(): void { + this.testCase({ + name: "NonRecording: setStatus changes status even when not recording", + test: () => { + // Arrange + const span = this._ai.startSpan("non-recording-status", { recording: false }); + + // Act + span!.setStatus({ code: eOTelSpanStatusCode.ERROR, message: "Test error" }); + + // Assert + Assert.equal(span.status.code, eOTelSpanStatusCode.ERROR, "Status code should be set"); + Assert.equal(span.status.message, "Test error", "Status message should be set"); + } + }); + + this.testCase({ + name: "NonRecording: setStatus returns span for chaining", + test: () => { + // Arrange + const span = this._ai.startSpan("non-recording-status-chain", { recording: false }); + + // Act + const result = span!.setStatus({ code: eOTelSpanStatusCode.OK }); + + // Assert + Assert.equal(result, span, "setStatus should return the span for chaining"); + } + }); + + this.testCase({ + name: "NonRecording: updateName changes name even when not recording", + test: () => { + // Arrange + const span = this._ai.startSpan("original-name", { recording: false }); + + // Act + span!.updateName("updated-name"); + + // Assert + Assert.equal(span.name, "updated-name", "Name should be updated"); + } + }); + + this.testCase({ + name: "NonRecording: updateName returns span for chaining", + test: () => { + // Arrange + const span = this._ai.startSpan("chain-name", { recording: false }); + + // Act + const result = span!.updateName("new-chain-name"); + + // Assert + Assert.equal(result, span, "updateName should return the span for chaining"); + } + }); + + this.testCase({ + name: "NonRecording: chained operations work correctly", + test: () => { + // Arrange + const span = this._ai.startSpan("chaining-test", { recording: false }); + + // Act + const result = span + .setAttribute("key1", "value1") + .setAttributes({ "key2": "value2" }) + .setStatus({ code: eOTelSpanStatusCode.OK }) + .updateName("chained-name"); + + // Assert + Assert.equal(result, span, "All operations should return the span"); + Assert.equal(span.name, "chained-name", "Name should be updated"); + Assert.equal(span.status.code, eOTelSpanStatusCode.OK, "Status should be set"); + } + }); + } + + private addSpanKindTests(): void { + this.testCase({ + name: "NonRecording: CLIENT kind non-recording span", + test: () => { + // Act + const span = this._ai.startSpan("client-non-recording", { + kind: eOTelSpanKind.CLIENT, + recording: false + }); + + // Assert + Assert.ok(span, "Span should be created"); + Assert.equal(span.kind, eOTelSpanKind.CLIENT, "Kind should be CLIENT"); + Assert.ok(!span.isRecording(), "Should not be recording"); + } + }); + + this.testCase({ + name: "NonRecording: SERVER kind non-recording span", + test: () => { + // Act + const span = this._ai.startSpan("server-non-recording", { + kind: eOTelSpanKind.SERVER, + recording: false + }); + + // Assert + Assert.equal(span.kind, eOTelSpanKind.SERVER, "Kind should be SERVER"); + Assert.ok(!span.isRecording(), "Should not be recording"); + } + }); + + this.testCase({ + name: "NonRecording: INTERNAL kind non-recording span", + test: () => { + // Act + const span = this._ai.startSpan("internal-non-recording", { + kind: eOTelSpanKind.INTERNAL, + recording: false + }); + + // Assert + Assert.equal(span.kind, eOTelSpanKind.INTERNAL, "Kind should be INTERNAL"); + Assert.ok(!span.isRecording(), "Should not be recording"); + } + }); + + this.testCase({ + name: "NonRecording: PRODUCER kind non-recording span", + test: () => { + // Act + const span = this._ai.startSpan("producer-non-recording", { + kind: eOTelSpanKind.PRODUCER, + recording: false + }); + + // Assert + Assert.equal(span.kind, eOTelSpanKind.PRODUCER, "Kind should be PRODUCER"); + Assert.ok(!span.isRecording(), "Should not be recording"); + } + }); + + this.testCase({ + name: "NonRecording: CONSUMER kind non-recording span", + test: () => { + // Act + const span = this._ai.startSpan("consumer-non-recording", { + kind: eOTelSpanKind.CONSUMER, + recording: false + }); + + // Assert + Assert.equal(span.kind, eOTelSpanKind.CONSUMER, "Kind should be CONSUMER"); + Assert.ok(!span.isRecording(), "Should not be recording"); + } + }); + } + + private addHierarchyTests(): void { + this.testCase({ + name: "NonRecording: parent recording, child non-recording", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("recording-parent", { + kind: eOTelSpanKind.SERVER, + recording: true + }); + const parentContext = parentSpan!.spanContext(); + + // Act + const childSpan = this._ai.startSpan("non-recording-child", { + kind: eOTelSpanKind.CLIENT, + recording: false + }, parentContext); + + // Assert + Assert.ok(parentSpan!.isRecording(), "Parent should be recording"); + Assert.ok(!childSpan.isRecording(), "Child should not be recording"); + Assert.equal(childSpan!.spanContext().traceId, parentContext.traceId, + "Child should share parent's trace ID"); + } + }); + + this.testCase({ + name: "NonRecording: parent non-recording, child recording", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("non-recording-parent", { + kind: eOTelSpanKind.SERVER, + recording: false + }); + const parentContext = parentSpan!.spanContext(); + + // Act + const childSpan = this._ai.startSpan("recording-child", { + kind: eOTelSpanKind.CLIENT, + recording: true + }, parentContext); + + // Assert + Assert.ok(!parentSpan.isRecording(), "Parent should not be recording"); + Assert.ok(childSpan!.isRecording(), "Child should be recording"); + Assert.equal(childSpan!.spanContext().traceId, parentContext.traceId, + "Child should share parent's trace ID"); + } + }); + + this.testCase({ + name: "NonRecording: both parent and child non-recording", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("non-recording-parent-2", { + recording: false + }); + const parentContext = parentSpan!.spanContext(); + + // Act + const childSpan = this._ai.startSpan("non-recording-child-2", { + recording: false + }, parentContext); + + // Assert + Assert.ok(!parentSpan.isRecording(), "Parent should not be recording"); + Assert.ok(!childSpan.isRecording(), "Child should not be recording"); + } + }); + + this.testCase({ + name: "NonRecording: multi-level hierarchy with mixed recording", + test: () => { + // Arrange + const level1 = this._ai.startSpan("level1-recording", { recording: true }); + const level1Context = level1!.spanContext(); + + const level2 = this._ai.startSpan("level2-non-recording", { + recording: false + }, level1Context); + const level2Context = level2!.spanContext(); + + const level3 = this._ai.startSpan("level3-recording", { + recording: true + }, level2Context); + + // Assert + Assert.ok(level1!.isRecording(), "Level 1 should be recording"); + Assert.ok(!level2.isRecording(), "Level 2 should not be recording"); + Assert.ok(level3!.isRecording(), "Level 3 should be recording"); + + // All should share the same trace ID + Assert.equal(level2!.spanContext().traceId, level1Context.traceId, + "Level 2 should share trace ID"); + Assert.equal(level3!.spanContext().traceId, level1Context.traceId, + "Level 3 should share trace ID"); + } + }); + } + + private addTelemetryGenerationTests(): void { + this.testCase({ + name: "NonRecording: no telemetry generated on end()", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("non-recording-no-telemetry", { + kind: eOTelSpanKind.CLIENT, + recording: false + }); + span!.setAttribute("should-not-appear", "in-telemetry"); + span!.setStatus({ code: eOTelSpanStatusCode.OK }); + span!.end(); + + // Assert + const telemetryItem = this._trackCalls.find( + item => item.baseData?.name === "non-recording-no-telemetry" + ); + Assert.ok(!telemetryItem, "Non-recording span should not generate telemetry"); + } + }); + + this.testCase({ + name: "NonRecording: recording span generates telemetry, non-recording does not", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const recordingSpan = this._ai.startSpan("recording-generates", { + kind: eOTelSpanKind.CLIENT, + recording: true + }); + recordingSpan.end(); + + const nonRecordingSpan = this._ai.startSpan("non-recording-silent", { + kind: eOTelSpanKind.CLIENT, + recording: false + }); + nonRecordingSpan.end(); + + // Assert + const recordingTelemetry = this._trackCalls.find( + item => item.baseData?.name === "recording-generates" + ); + const nonRecordingTelemetry = this._trackCalls.find( + item => item.baseData?.name === "non-recording-silent" + ); + + Assert.ok(recordingTelemetry, "Recording span should generate telemetry"); + Assert.ok(!nonRecordingTelemetry, "Non-recording span should not generate telemetry"); + } + }); + + this.testCase({ + name: "NonRecording: parent recording generates telemetry, child non-recording does not", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const parent = this._ai.startSpan("parent-with-telemetry", { + kind: eOTelSpanKind.SERVER, + recording: true + }); + const parentContext = parent.spanContext(); + + const child = this._ai.startSpan("child-without-telemetry", { + kind: eOTelSpanKind.CLIENT, + recording: false + }, parentContext); + + child.end(); + parent.end(); + + // Assert + const parentTelemetry = this._trackCalls.find( + item => item.baseData?.name === "parent-with-telemetry" + ); + const childTelemetry = this._trackCalls.find( + item => item.baseData?.name === "child-without-telemetry" + ); + + Assert.ok(parentTelemetry, "Parent recording span should generate telemetry"); + Assert.ok(!childTelemetry, "Child non-recording span should not generate telemetry"); + } + }); + } + + private addPerformanceTests(): void { + this.testCase({ + name: "NonRecording: multiple non-recording spans minimal overhead", + test: () => { + // Arrange + this._trackCalls = []; + const spanCount = 100; + + // Act + const startTime = Date.now(); + for (let i = 0; i < spanCount; i++) { + const span = this._ai.startSpan(`non-recording-perf-${i}`, { + recording: false + }); + span!.setAttribute("iteration", i); + span!.setStatus({ code: eOTelSpanStatusCode.OK }); + span!.end(); + } + const elapsed = Date.now() - startTime; + + // Assert + Assert.ok(elapsed < 1000, `Creating ${spanCount} non-recording spans should be fast, took ${elapsed}ms`); + Assert.equal(this._trackCalls.length, 0, "No telemetry should be generated for non-recording spans"); + } + }); + + this.testCase({ + name: "NonRecording: attribute operations are fast on non-recording spans", + test: () => { + // Arrange + const span = this._ai.startSpan("perf-attrs", { recording: false }); + const attrCount = 1000; + + // Act + const startTime = Date.now(); + for (let i = 0; i < attrCount; i++) { + span!.setAttribute(`key${i}`, `value${i}`); + } + const elapsed = Date.now() - startTime; + + // Assert + Assert.ok(elapsed < 500, `Setting ${attrCount} attributes should be fast, took ${elapsed}ms`); + Assert.equal(Object.keys(span.attributes).length, 0, "Attributes should not be stored"); + } + }); + } + + private addEdgeCaseTests(): void { + this.testCase({ + name: "NonRecording: end() can be called multiple times safely", + test: () => { + // Arrange + const span = this._ai.startSpan("multi-end", { recording: false }); + + // Act & Assert - Should not throw + span!.end(); + Assert.ok(span.ended, "Span should be ended"); + + span!.end(); + Assert.ok(span.ended, "Span should still be ended"); + + span!.end(); + Assert.ok(span.ended, "Span should still be ended"); + } + }); + + this.testCase({ + name: "NonRecording: operations after end() do not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("ops-after-end", { recording: false }); + span!.end(); + + // Act & Assert - Should not throw + span!.setAttribute("late-attr", "value"); + span!.setAttributes({ "late-attrs": "values" }); + span!.setStatus({ code: eOTelSpanStatusCode.ERROR }); + span!.updateName("late-name"); + + Assert.ok(span.ended, "Span should remain ended"); + } + }); + + this.testCase({ + name: "NonRecording: null and undefined attribute values handled", + test: () => { + // Arrange + const span = this._ai.startSpan("null-attrs", { recording: false }); + + // Act & Assert - Should not throw + span!.setAttribute("null-value", null as any); + span!.setAttribute("undefined-value", undefined as any); + span!.setAttributes({ + "null-in-set": null as any, + "undefined-in-set": undefined as any + }); + + Assert.ok(!span.isRecording(), "Span should still be non-recording"); + } + }); + + this.testCase({ + name: "NonRecording: empty string name allowed", + test: () => { + // Act + const span = this._ai.startSpan("", { recording: false }); + + // Assert + Assert.ok(span, "Span with empty name should be created"); + Assert.equal(span.name, "", "Name should be empty string"); + Assert.ok(!span.isRecording(), "Should not be recording"); + } + }); + + this.testCase({ + name: "NonRecording: special characters in span name", + test: () => { + // Arrange + const specialNames = [ + "span/with/slashes", + "span:with:colons", + "span-with-dashes", + "span.with.dots", + "span with spaces", + "span\twith\ttabs", + "span(with)parens", + "span[with]brackets", + "span{with}braces" + ]; + + // Act & Assert + specialNames.forEach(name => { + const span = this._ai.startSpan(name, { recording: false }); + Assert.ok(span, `Span with name '${name}' should be created`); + Assert.equal(span.name, name, "Name should be preserved"); + Assert.ok(!span.isRecording(), "Should not be recording"); + }); + } + }); + + this.testCase({ + name: "NonRecording: very long span name handled", + test: () => { + // Arrange + const longName = "a".repeat(10000); + + // Act + const span = this._ai.startSpan(longName, { recording: false }); + + // Assert + Assert.ok(span, "Span with very long name should be created"); + Assert.ok(!span.isRecording(), "Should not be recording"); + } + }); + + this.testCase({ + name: "NonRecording: spanContext() returns valid context", + test: () => { + // Act + const span = this._ai.startSpan("context-check", { recording: false }); + const context = span!.spanContext(); + + // Assert + Assert.ok(context, "Context should exist"); + Assert.ok(context.traceId, "Trace ID should exist"); + Assert.ok(context.spanId, "Span ID should exist"); + Assert.ok(context.traceId.length === 32, "Trace ID should be 32 characters"); + Assert.ok(context.spanId.length === 16, "Span ID should be 16 characters"); + } + }); + + this.testCase({ + name: "NonRecording: status object immutability", + test: () => { + // Arrange + const span = this._ai.startSpan("status-immutable", { recording: false }); + span!.setStatus({ code: eOTelSpanStatusCode.OK, message: "Initial" }); + + // Act + const status1 = span!.status; + span!.setStatus({ code: eOTelSpanStatusCode.ERROR, message: "Changed" }); + const status2 = span!.status; + + // Assert + Assert.equal(status1.code, eOTelSpanStatusCode.OK, "First status should be OK"); + Assert.equal(status2.code, eOTelSpanStatusCode.ERROR, "Second status should be ERROR"); + } + }); + } +} diff --git a/AISKU/Tests/Unit/src/OTelInit.Tests.ts b/AISKU/Tests/Unit/src/OTelInit.Tests.ts new file mode 100644 index 000000000..7b311d8e7 --- /dev/null +++ b/AISKU/Tests/Unit/src/OTelInit.Tests.ts @@ -0,0 +1,112 @@ +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { ApplicationInsights } from "../../../src/applicationinsights-web"; +import { eOTelSpanKind, eOTelSpanStatusCode, isTracingSuppressed, ITelemetryItem, unsuppressTracing } from "@microsoft/applicationinsights-core-js"; +import { objIs, setBypassLazyCache } from "@nevware21/ts-utils"; +import { AnalyticsPluginIdentifier, PropertiesPluginIdentifier } from "@microsoft/applicationinsights-core-js"; + +/** + * Integration Tests for Span APIs with Properties Plugin and Analytics Plugin + * + * Tests verify that span telemetry correctly integrates with: + * - PropertiesPlugin: session, user, device, application context + * - AnalyticsPlugin: telemetry creation, dependency tracking, page views + * - Telemetry Initializers: custom property injection + * - SDK configuration: sampling, disabled tracking, etc. + */ +export class OTelInitTests extends AITestClass { + private _ai!: ApplicationInsights; + + constructor(testName?: string) { + super(testName || "OTelInitTests"); + } + + public testInitialize() { + try { + setBypassLazyCache(true); + this.useFakeServer = true; + + this._ai = new ApplicationInsights({ + config: { + instrumentationKey: "test-ikey-123", + disableInstrumentationKeyValidation: true, + disableAjaxTracking: false, + disableXhr: false, + disableFetchTracking: false, + enableAutoRouteTracking: false, + disableExceptionTracking: false, + maxBatchInterval: 100, + enableDebug: false, + extensionConfig: { + ["AppInsightsPropertiesPlugin"]: { + accountId: "test-account-id" + } + }, + traceCfg: { + coreTrace: 1 + } as any + } + }); + + this._ai.loadAppInsights(); + } catch (e) { + Assert.ok(false, "Failed to initialize tests: " + e); + console.error("Failed to initialize tests: " + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + setBypassLazyCache(false); + } + + public registerTests() { + this.testCase({ + name: "OTelInitTests", + test: () => { + Assert.ok(this._ai, "ApplicationInsights instance should be initialized"); + Assert.ok(this._ai.getPlugin(PropertiesPluginIdentifier), "PropertiesPlugin should be loaded"); + Assert.ok(this._ai.getPlugin(AnalyticsPluginIdentifier), "AnalyticsPlugin should be loaded"); + Assert.ok(!isTracingSuppressed(this._ai.core), "Tracing should not be suppressed by default"); + } + }); + + this.testCase({ + name: "Validate OTelApi", + test: () => { + const otelApi = this._ai.otelApi; + Assert.ok(otelApi, "OTel API should be available"); + Assert.ok(otelApi.cfg, "OTel configuration should be available"); + Assert.ok(objIs(this._ai, otelApi.host), "OTel API host should be the same as the SKU instance"); + Assert.ok(objIs(otelApi.cfg.traceCfg, this._ai.core.config.traceCfg), "OTel trace configuration should be the same as the SDK config"); + Assert.ok(objIs(otelApi.host.config, this._ai.config), "OTel API config should be the same as the SDK config"); + Assert.ok(objIs(otelApi.host.config, this._ai.core.config), "OTel API config should be the same as the SDK core config"); + Assert.ok(objIs(this._ai.config, this._ai.core.config), "SDK config should be the same as the SDK core config"); + } + }); + + this.testCase({ + name: "Validate Trace suppression", + test: () => { + const otelApi = this._ai.otelApi; + Assert.ok(otelApi, "OTel API should be available"); + Assert.equal(false, isTracingSuppressed(this._ai.core), "Tracing should not be suppressed by default"); + Assert.equal(false, otelApi.cfg.traceCfg?.suppressTracing, "supressTracing should be false by default"); + Assert.equal(false, this._ai.core.config.traceCfg.suppressTracing, "suppressTracing should be false by default"); + + this._ai.core.config.traceCfg.suppressTracing = true; + Assert.equal(true, isTracingSuppressed(this._ai.core), "Tracing should be suppressed when suppressTracing is set to true"); + Assert.equal(true, otelApi.cfg.traceCfg?.suppressTracing, "supressTracing should be true when suppressTracing is set to true"); + Assert.equal(true, this._ai.core.config.traceCfg.suppressTracing, "suppressTracing should be true when suppressTracing is set to true"); + + unsuppressTracing(this._ai.core); + Assert.equal(false, isTracingSuppressed(this._ai.core), "Tracing should not be suppressed after unsuppressTracing"); + Assert.equal(false, this._ai.core.config.traceCfg.suppressTracing, "suppressTracing should be false by default"); + Assert.equal(false, otelApi.cfg.traceCfg?.suppressTracing, "supressTracing should be false after unsuppressTracing"); + } + }); + + } +} diff --git a/AISKU/Tests/Unit/src/SnippetInitialization.Tests.ts b/AISKU/Tests/Unit/src/SnippetInitialization.Tests.ts index 22c6c85f5..2812f78d9 100644 --- a/AISKU/Tests/Unit/src/SnippetInitialization.Tests.ts +++ b/AISKU/Tests/Unit/src/SnippetInitialization.Tests.ts @@ -10,10 +10,10 @@ import { BaseTelemetryPlugin, IProcessTelemetryContext, isNotNullOrUndefined, IT import { BreezeChannelIdentifier, ContextTagKeys, DistributedTracingModes, IConfig, IDependencyTelemetry, RequestHeaders, utlRemoveSessionStorage, utlSetSessionStorage -} from "@microsoft/applicationinsights-common"; +} from "@microsoft/applicationinsights-core-js"; import { getGlobal } from "@microsoft/applicationinsights-shims"; -import { TelemetryContext } from "@microsoft/applicationinsights-properties-js"; -import { dumpObj, isPromiseLike, objHasOwnProperty } from "@nevware21/ts-utils"; +import { IPropTelemetryContext } from "@microsoft/applicationinsights-properties-js"; +import { dumpObj, isPromiseLike, objHasOwnProperty, strSubstring } from "@nevware21/ts-utils"; import { AppInsightsSku } from "../../../src/AISku"; const TestInstrumentationKey = 'b7170927-2d1c-44f1-acec-59f4e1751c11'; @@ -678,7 +678,7 @@ export class SnippetInitializationTests extends AITestClass { () => { let theSnippet = this._initializeSnippet(snippetCreator(getSnippetConfig(this.sessionPrefix))); const xhr = new XMLHttpRequest(); - xhr.open('GET', 'https://httpbin.org/status/200'); + xhr.open('GET', 'http://localhost:9001/README.md'); xhr.send(); Assert.ok(true); } @@ -694,15 +694,15 @@ export class SnippetInitializationTests extends AITestClass { steps: [ () => { let theSnippet = this._initializeSnippet(snippetCreator(getSnippetConfig(this.sessionPrefix))); - fetch('https://httpbin.org/status/200', { method: 'GET', headers: { 'header': 'value'} }); + fetch('http://localhost:9001/README.md', { method: 'GET', headers: { 'header': 'value'} }); Assert.ok(true, "fetch monitoring is instrumented"); }, () => { - fetch('https://httpbin.org/status/200', { method: 'GET' }); + fetch('http://localhost:9001/README.md', { method: 'GET' }); Assert.ok(true, "fetch monitoring is instrumented"); }, () => { - fetch('https://httpbin.org/status/200'); + fetch('http://localhost:9001/README.md'); Assert.ok(true, "fetch monitoring is instrumented"); } ] @@ -733,6 +733,7 @@ export class SnippetInitializationTests extends AITestClass { Assert.ok(baseData.properties.requestHeaders[RequestHeaders.requestIdHeader], "Request-Id header"); Assert.ok(baseData.properties.requestHeaders[RequestHeaders.requestContextHeader], "Request-Context header"); Assert.ok(baseData.properties.requestHeaders[RequestHeaders.traceParentHeader], "traceparent"); + Assert.ok(!baseData.properties.requestHeaders[RequestHeaders.traceStateHeader], "traceState should not be present in outbound event"); const id: string = baseData.id; const regex = id.match(/\|.{32}\..{16}\./g); Assert.ok(id.length > 0); @@ -865,7 +866,7 @@ export class SnippetInitializationTests extends AITestClass { steps: [ () => { let theSnippet = this._initializeSnippet(snippetCreator(getSnippetConfig(this.sessionPrefix))); - const context = (theSnippet.context) as TelemetryContext; + const context = (theSnippet.context) as IPropTelemetryContext; context.user.setAuthenticatedUserContext('10001'); theSnippet.trackTrace({ message: 'authUserContext test' }); } @@ -897,7 +898,7 @@ export class SnippetInitializationTests extends AITestClass { steps: [ () => { let theSnippet = this._initializeSnippet(snippetCreator(getSnippetConfig(this.sessionPrefix))); - const context = (theSnippet.context) as TelemetryContext; + const context = (theSnippet.context) as IPropTelemetryContext; context.user.setAuthenticatedUserContext('10001', 'account123'); theSnippet.trackTrace({ message: 'authUserContext test' }); } @@ -928,7 +929,7 @@ export class SnippetInitializationTests extends AITestClass { steps: [ () => { let theSnippet = this._initializeSnippet(snippetCreator(getSnippetConfig(this.sessionPrefix))); - const context = (theSnippet.context) as TelemetryContext; + const context = (theSnippet.context) as IPropTelemetryContext; context.user.setAuthenticatedUserContext("\u0428", "\u0429"); theSnippet.trackTrace({ message: 'authUserContext test' }); } @@ -959,7 +960,7 @@ export class SnippetInitializationTests extends AITestClass { steps: [ () => { let theSnippet = this._initializeSnippet(snippetCreator(getSnippetConfig(this.sessionPrefix))); - const context = (theSnippet.context) as TelemetryContext; + const context = (theSnippet.context) as IPropTelemetryContext; context.user.setAuthenticatedUserContext('10002', 'account567'); context.user.clearAuthenticatedUserContext(); theSnippet.trackTrace({ message: 'authUserContext test' }); @@ -991,7 +992,7 @@ export class SnippetInitializationTests extends AITestClass { test: () => { // Setup let theSnippet = this._initializeSnippet(snippetCreator(getSnippetConfig(this.sessionPrefix))); - const context = (theSnippet.context) as TelemetryContext; + const context = (theSnippet.context) as IPropTelemetryContext; const authSpy: SinonSpy = this.sandbox.spy(context.user, 'setAuthenticatedUserContext'); let cookieMgr = theSnippet.getCookieMgr(); const cookieSpy: SinonSpy = this.sandbox.spy(cookieMgr, 'set'); diff --git a/AISKU/Tests/Unit/src/SpanContextPropagation.Tests.ts b/AISKU/Tests/Unit/src/SpanContextPropagation.Tests.ts new file mode 100644 index 000000000..8de152cfe --- /dev/null +++ b/AISKU/Tests/Unit/src/SpanContextPropagation.Tests.ts @@ -0,0 +1,733 @@ +import { AITestClass, Assert } from '@microsoft/ai-test-framework'; +import { ApplicationInsights } from '../../../src/applicationinsights-web'; +import { IReadableSpan, IDistributedTraceContext, ITelemetryItem, asString } from "@microsoft/applicationinsights-core-js"; +import { createPromise, IPromise } from '@nevware21/ts-async'; + +export class SpanContextPropagationTests extends AITestClass { + private static readonly _instrumentationKey = 'b7170927-2d1c-44f1-acec-59f4e1751c11'; + private static readonly _connectionString = `InstrumentationKey=${SpanContextPropagationTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "SpanContextPropagationTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: SpanContextPropagationTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + + } catch (e) { + console.error('Failed to initialize tests: ' + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addParentChildRelationshipTests(); + this.addMultiLevelHierarchyTests(); + this.addSiblingSpanTests(); + this.addAsyncBoundaryTests(); + this.addContextPropagationTests(); + } + + private addParentChildRelationshipTests(): void { + this.testCase({ + name: "ParentChild: child span should inherit parent's traceId", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("parent-span"); + Assert.ok(parentSpan, "Parent span should be created"); + + // Act + const parentContext = parentSpan!.spanContext(); + const childSpan = this._ai.startSpan("child-span", undefined, parentContext); + const childContext = childSpan!.spanContext(); + + // Assert + Assert.equal(childContext.traceId, parentContext.traceId, "Child span should inherit parent's traceId"); + Assert.notEqual(childContext.spanId, parentContext.spanId, "Child span should have different spanId from parent"); + + // Cleanup + childSpan?.end(); + parentSpan?.end(); + } + }); + + this.testCase({ + name: "ParentChild: child span should have unique spanId", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("parent-unique-id"); + const parentContext = parentSpan!.spanContext(); + + // Act + const child1 = this._ai.startSpan("child-1", undefined, parentContext); + const child2 = this._ai.startSpan("child-2", undefined, parentContext); + + const child1Context = child1!.spanContext(); + const child2Context = child2!.spanContext(); + + // Assert + Assert.notEqual(child1Context.spanId, child2Context.spanId, + "Sibling children should have unique spanIds"); + Assert.notEqual(child1Context.spanId, parentContext.spanId, + "Child 1 spanId should differ from parent"); + Assert.notEqual(child2Context.spanId, parentContext.spanId, + "Child 2 spanId should differ from parent"); + + // Cleanup + child1?.end(); + child2?.end(); + parentSpan?.end(); + } + }); + + this.testCase({ + name: "ParentChild: child spans created via getTraceCtx", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("parent-via-getTraceCtx"); + + this._ai.setActiveSpan(parentSpan!); + + // Act - Use getTraceCtx to get current context + const currentContext = this._ai.getTraceCtx(); + const childSpan = this._ai.startSpan("child-via-getTraceCtx", undefined, currentContext || undefined); + + // Assert + const parentContext = parentSpan!.spanContext(); + const childContext = childSpan!.spanContext(); + + Assert.equal(childContext.traceId, parentContext.traceId, + "Child should inherit traceId via getTraceCtx"); + Assert.notEqual(childContext.spanId, parentContext.spanId, + "Child should have unique spanId"); + + // Cleanup + childSpan?.end(); + parentSpan?.end(); + } + }); + + this.testCase({ + name: "ParentChild: parent context should preserve traceFlags", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("parent-traceflags"); + const parentContext = parentSpan!.spanContext(); + + // Act + const childSpan = this._ai.startSpan("child-traceflags", undefined, parentContext); + const childContext = childSpan!.spanContext(); + + // Assert + Assert.equal(childContext.traceFlags, parentContext.traceFlags, + "Child should preserve parent's traceFlags"); + + // Cleanup + childSpan?.end(); + parentSpan?.end(); + } + }); + + this.testCase({ + name: "ParentChild: parent context should preserve traceState if present", + test: () => { + // Arrange - Create parent span + const parentSpan = this._ai.startSpan("parent-tracestate"); + const parentContext = parentSpan!.spanContext(); + + // Manually set traceState (if the implementation supports it) + if (parentContext.traceState !== undefined) { + // Act + const childSpan = this._ai.startSpan("child-tracestate", undefined, parentContext); + const childContext = childSpan!.spanContext(); + + // Assert + Assert.equal(asString(childContext.traceState), asString(parentContext.traceState), + "Child should preserve parent's traceState"); + + // Cleanup + childSpan?.end(); + } + + parentSpan?.end(); + } + }); + + this.testCase({ + name: "ParentChild: multiple children from same parent share traceId", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("parent-multiple-children"); + const parentContext = parentSpan!.spanContext(); + + // Act - Create multiple children + const children: IReadableSpan[] = []; + for (let i = 0; i < 5; i++) { + const child = this._ai.startSpan(`child-${i}`, undefined, parentContext); + if (child) { + children.push(child); + } + } + + // Assert + children.forEach((child, index) => { + const childContext = child.spanContext(); + Assert.equal(childContext.traceId, parentContext.traceId, + `Child ${index} should have parent's traceId`); + }); + + // All children should have unique spanIds + for (let i = 0; i < children.length; i++) { + for (let j = i + 1; j < children.length; j++) { + const ctx1 = children[i].spanContext(); + const ctx2 = children[j].spanContext(); + Assert.notEqual(ctx1.spanId, ctx2.spanId, + `Child ${i} and child ${j} should have different spanIds`); + } + } + + // Cleanup + children.forEach(child => child.end()); + parentSpan?.end(); + } + }); + } + + private addMultiLevelHierarchyTests(): void { + this.testCase({ + name: "MultiLevel: grandchild inherits root traceId", + test: () => { + // Arrange & Act - Create 3-level hierarchy + const rootSpan = this._ai.startSpan("root-span"); + const rootContext = rootSpan!.spanContext(); + + const childSpan = this._ai.startSpan("child-span", undefined, rootContext); + const childContext = childSpan!.spanContext(); + + const grandchildSpan = this._ai.startSpan("grandchild-span", undefined, childContext); + const grandchildContext = grandchildSpan!.spanContext(); + + // Assert + Assert.equal(childContext.traceId, rootContext.traceId, + "Child should have root's traceId"); + Assert.equal(grandchildContext.traceId, rootContext.traceId, + "Grandchild should have root's traceId"); + + Assert.notEqual(childContext.spanId, rootContext.spanId, + "Child should have unique spanId"); + Assert.notEqual(grandchildContext.spanId, childContext.spanId, + "Grandchild should have unique spanId"); + Assert.notEqual(grandchildContext.spanId, rootContext.spanId, + "Grandchild spanId should differ from root"); + + // Cleanup + grandchildSpan?.end(); + childSpan?.end(); + rootSpan?.end(); + } + }); + + this.testCase({ + name: "MultiLevel: deep hierarchy maintains trace consistency", + test: () => { + // Arrange - Create deep hierarchy (5 levels) + const spans: IReadableSpan[] = []; + + // Act - Create root + const rootSpan = this._ai.startSpan("level-0-root"); + spans.push(rootSpan!); + + // Create nested spans + for (let i = 1; i <= 4; i++) { + const parentContext = spans[i - 1].spanContext(); + const childSpan = this._ai.startSpan(`level-${i}`, undefined, parentContext); + spans.push(childSpan!); + } + + // Assert - All spans share same traceId + const rootTraceId = spans[0].spanContext().traceId; + spans.forEach((span, index) => { + const context = span.spanContext(); + Assert.equal(context.traceId, rootTraceId, + `Level ${index} should have root traceId`); + }); + + // All spans should have unique spanIds + const spanIds = spans.map(span => span.spanContext().spanId); + const uniqueSpanIds = new Set(spanIds); + Assert.equal(uniqueSpanIds.size, spans.length, + "All spans should have unique spanIds"); + + // Cleanup + for (let i = spans.length - 1; i >= 0; i--) { + spans[i].end(); + } + } + }); + + this.testCase({ + name: "MultiLevel: intermediate span can be parent to multiple children", + test: () => { + // Arrange - Create hierarchy with branching + const rootSpan = this._ai.startSpan("root"); + const rootContext = rootSpan!.spanContext(); + + const intermediateSpan = this._ai.startSpan("intermediate", undefined, rootContext); + const intermediateContext = intermediateSpan!.spanContext(); + + // Act - Create multiple children from intermediate + const leaf1 = this._ai.startSpan("leaf-1", undefined, intermediateContext); + const leaf2 = this._ai.startSpan("leaf-2", undefined, intermediateContext); + const leaf3 = this._ai.startSpan("leaf-3", undefined, intermediateContext); + + // Assert + const leaf1Context = leaf1!.spanContext(); + const leaf2Context = leaf2!.spanContext(); + const leaf3Context = leaf3!.spanContext(); + + // All share same traceId + Assert.equal(leaf1Context.traceId, rootContext.traceId, + "Leaf 1 should have root traceId"); + Assert.equal(leaf2Context.traceId, rootContext.traceId, + "Leaf 2 should have root traceId"); + Assert.equal(leaf3Context.traceId, rootContext.traceId, + "Leaf 3 should have root traceId"); + + // All have unique spanIds + Assert.notEqual(leaf1Context.spanId, leaf2Context.spanId, + "Leaf 1 and 2 should have different spanIds"); + Assert.notEqual(leaf2Context.spanId, leaf3Context.spanId, + "Leaf 2 and 3 should have different spanIds"); + Assert.notEqual(leaf1Context.spanId, leaf3Context.spanId, + "Leaf 1 and 3 should have different spanIds"); + + // Cleanup + leaf3?.end(); + leaf2?.end(); + leaf1?.end(); + intermediateSpan?.end(); + rootSpan?.end(); + } + }); + } + + private addSiblingSpanTests(): void { + this.testCase({ + name: "Siblings: spans with same parent have same traceId", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("parent-for-siblings"); + const parentContext = parentSpan!.spanContext(); + + // Act - Create sibling spans + const sibling1 = this._ai.startSpan("sibling-1", undefined, parentContext); + const sibling2 = this._ai.startSpan("sibling-2", undefined, parentContext); + const sibling3 = this._ai.startSpan("sibling-3", undefined, parentContext); + + // Assert + const ctx1 = sibling1!.spanContext(); + const ctx2 = sibling2!.spanContext(); + const ctx3 = sibling3!.spanContext(); + + Assert.equal(ctx1.traceId, parentContext.traceId, + "Sibling 1 should have parent's traceId"); + Assert.equal(ctx2.traceId, parentContext.traceId, + "Sibling 2 should have parent's traceId"); + Assert.equal(ctx3.traceId, parentContext.traceId, + "Sibling 3 should have parent's traceId"); + + // Cleanup + sibling3?.end(); + sibling2?.end(); + sibling1?.end(); + parentSpan?.end(); + } + }); + + this.testCase({ + name: "Siblings: independent root spans have different traceIds", + test: () => { + // Act - Create independent root spans + const root1 = this._ai.startSpan("independent-root-1", { root: true }); + const root2 = this._ai.startSpan("independent-root-2", { root: true }); + const root3 = this._ai.startSpan("independent-root-3", { root: true }); + + // Assert + const ctx1 = root1!.spanContext(); + const ctx2 = root2!.spanContext(); + const ctx3 = root3!.spanContext(); + + Assert.notEqual(ctx1.traceId, ctx2.traceId, + "Independent root 1 and 2 should have different traceIds"); + Assert.notEqual(ctx2.traceId, ctx3.traceId, + "Independent root 2 and 3 should have different traceIds"); + Assert.notEqual(ctx1.traceId, ctx3.traceId, + "Independent root 1 and 3 should have different traceIds"); + + // Cleanup + root3?.end(); + root2?.end(); + root1?.end(); + } + }); + + this.testCase({ + name: "Siblings: sibling spans have unique spanIds", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("parent-unique-siblings"); + const parentContext = parentSpan!.spanContext(); + + // Act - Create many sibling spans + const siblings: IReadableSpan[] = []; + for (let i = 0; i < 10; i++) { + const sibling = this._ai.startSpan(`sibling-${i}`, undefined, parentContext); + if (sibling) { + siblings.push(sibling); + } + } + + // Assert - All spanIds should be unique + const spanIds = siblings.map(s => s.spanContext().spanId); + const uniqueSpanIds = new Set(spanIds); + Assert.equal(uniqueSpanIds.size, siblings.length, + "All sibling spans should have unique spanIds"); + + // Cleanup + siblings.forEach(s => s.end()); + parentSpan?.end(); + } + }); + } + + private addAsyncBoundaryTests(): void { + this.testCase({ + name: "AsyncBoundary: context can be captured and used across async operations", + test: () => { + // Arrange + const rootSpan = this._ai.startSpan("async-root"); + const capturedContext = rootSpan!.spanContext(); + + // Act - Simulate async boundary by creating child later + return createPromise((resolve) => { + setTimeout(() => { + // Create child span using captured context + const childSpan = this._ai.startSpan("async-child", undefined, capturedContext); + const childContext = childSpan!.spanContext(); + + // Assert + Assert.equal(childContext.traceId, capturedContext.traceId, + "Child created after async boundary should have parent's traceId"); + + // Cleanup + childSpan?.end(); + rootSpan?.end(); + resolve(); + }, 10); + }); + } + }); + + this.testCase({ + name: "AsyncBoundary: getTraceCtx can capture context for async operations", + test: () => { + // Arrange + const rootSpan = this._ai.startSpan("async-getTraceCtx-root"); + this._ai.setActiveSpan(rootSpan!); + + // Capture context using getTraceCtx + const capturedContext = this._ai.getTraceCtx(); + + // Act - Simulate async operation + return createPromise((resolve) => { + setTimeout(() => { + // Use captured context in async boundary + const asyncSpan = this._ai.startSpan("async-operation", undefined, capturedContext || undefined); + const asyncContext = asyncSpan!.spanContext(); + + // Assert + Assert.equal(asyncContext.traceId, capturedContext.traceId, "Async span should inherit captured traceId"); + + // Cleanup + asyncSpan?.end(); + rootSpan?.end(); + resolve(); + }, 10); + }); + } + }); + + this.testCase({ + name: "AsyncBoundary: nested async operations maintain trace", + test: () => { + // Arrange + const rootSpan = this._ai.startSpan("nested-async-root"); + const rootContext = rootSpan!.spanContext(); + + // Act - Chain async operations + return createPromise((resolve) => { + setTimeout(() => { + const child1 = this._ai.startSpan("async-child-1", undefined, rootContext); + const child1Context = child1!.spanContext(); + + setTimeout(() => { + const child2 = this._ai.startSpan("async-child-2", undefined, child1Context); + const child2Context = child2!.spanContext(); + + // Assert + Assert.equal(child1Context.traceId, rootContext.traceId, + "First async child should have root traceId"); + Assert.equal(child2Context.traceId, rootContext.traceId, + "Second async child should have root traceId"); + + // Cleanup + child2?.end(); + child1?.end(); + rootSpan?.end(); + resolve(); + }, 10); + }, 10); + }); + } + }); + + this.testCase({ + name: "AsyncBoundary: parallel async operations share traceId", + test: () => { + // Arrange + const rootSpan = this._ai.startSpan("parallel-async-root"); + const rootContext = rootSpan!.spanContext(); + + // Act - Create parallel async operations + const promises: IPromise[] = []; + const childContexts: IDistributedTraceContext[] = []; + + for (let i = 0; i < 3; i++) { + const promise = createPromise((resolve) => { + setTimeout(() => { + const childSpan = this._ai.startSpan(`parallel-child-${i}`, undefined, rootContext); + childContexts.push(childSpan!.spanContext()); + childSpan?.end(); + resolve(); + }, 10 + i * 5); + }); + promises.push(promise); + } + + return Promise.all(promises).then(() => { + // Assert - All parallel children should share root traceId + childContexts.forEach((ctx, index) => { + Assert.equal(ctx.traceId, rootContext.traceId, + `Parallel child ${index} should have root traceId`); + }); + + // All should have unique spanIds + const spanIds = childContexts.map(ctx => ctx.spanId); + const uniqueSpanIds = new Set(spanIds); + Assert.equal(uniqueSpanIds.size, childContexts.length, + "Parallel children should have unique spanIds"); + + // Cleanup + rootSpan?.end(); + }); + } + }); + } + + private addContextPropagationTests(): void { + this.testCase({ + name: "ContextPropagation: explicit parent context overrides active context", + test: () => { + // Arrange - Create two independent traces + const trace1Root = this._ai.startSpan("trace-1-root", { root: true }); + const trace2Root = this._ai.startSpan("trace-2-root", { root: true }); + + this._ai.setActiveSpan(trace1Root!); + + // Act - Create child with explicit trace2 parent + const trace2Context = trace2Root!.spanContext(); + const childSpan = this._ai.startSpan("explicit-parent-child", undefined, trace2Context); + const childContext = childSpan!.spanContext(); + + // Assert - Child should belong to trace2, not active trace1 + Assert.equal(childContext.traceId, trace2Context.traceId, + "Explicit parent context should override active context"); + Assert.notEqual(childContext.traceId, trace1Root!.spanContext().traceId, + "Child should not belong to active trace"); + + // Cleanup + childSpan?.end(); + trace2Root?.end(); + trace1Root?.end(); + } + }); + + this.testCase({ + name: "ContextPropagation: spans without parent create new trace", + test: () => { + // Act - Create spans without explicit parent + const span1 = this._ai.startSpan("no-parent-1"); + const span2 = this._ai.startSpan("no-parent-2"); + + const ctx1 = span1!.spanContext(); + const ctx2 = span2!.spanContext(); + + // Assert - Should create independent traces or share active context + // (depends on implementation - both are valid) + Assert.ok(ctx1.traceId, "Span 1 should have traceId"); + Assert.ok(ctx2.traceId, "Span 2 should have traceId"); + Assert.ok(ctx1.spanId !== ctx2.spanId, + "Spans should have unique spanIds"); + + // Cleanup + span2?.end(); + span1?.end(); + } + }); + + this.testCase({ + name: "ContextPropagation: root option creates new trace", + test: () => { + // Arrange - Create parent span + const parentSpan = this._ai.startSpan("existing-parent"); + this._ai.setActiveSpan(parentSpan!); + + // Act - Create root span (should ignore active parent) + const rootSpan = this._ai.startSpan("new-root", { root: true }); + + const parentContext = parentSpan!.spanContext(); + const rootContext = rootSpan!.spanContext(); + + // Assert - Root span should have different traceId + Assert.notEqual(rootContext.traceId, parentContext.traceId, + "Root option should create new independent trace"); + + // Cleanup + rootSpan?.end(); + parentSpan?.end(); + } + }); + + this.testCase({ + name: "ContextPropagation: context with all required fields propagates correctly", + test: () => { + // Arrange - Create context with all fields + const parentSpan = this._ai.startSpan("full-context-parent"); + const parentContext = parentSpan!.spanContext(); + + // Act - Create child + const childSpan = this._ai.startSpan("full-context-child", undefined, parentContext); + const childContext = childSpan!.spanContext(); + + // Assert - All fields should be present + Assert.ok(childContext.traceId, "Child should have traceId"); + Assert.ok(childContext.spanId, "Child should have spanId"); + Assert.equal(childContext.traceFlags, parentContext.traceFlags, + "Child should have traceFlags"); + + Assert.equal(childContext.traceId, parentContext.traceId, + "TraceId should propagate"); + Assert.equal(childContext.traceFlags, parentContext.traceFlags, + "TraceFlags should propagate"); + + // Cleanup + childSpan?.end(); + parentSpan?.end(); + } + }); + + this.testCase({ + name: "ContextPropagation: recording attribute propagates independently", + test: () => { + // Arrange - Create recording parent + const recordingParent = this._ai.startSpan("recording-parent", { recording: true }); + const recordingContext = recordingParent!.spanContext(); + + // Act - Create non-recording child from recording parent + const nonRecordingChild = this._ai.startSpan("non-recording-child", + { recording: false }, recordingContext); + + // Assert - Recording is per-span, not propagated + Assert.ok(recordingParent!.isRecording(), + "Parent should be recording"); + Assert.ok(!nonRecordingChild!.isRecording(), + "Child should not be recording despite recording parent"); + + // But traceId should still propagate + Assert.equal(nonRecordingChild!.spanContext().traceId, recordingContext.traceId, + "TraceId should propagate regardless of recording state"); + + // Cleanup + nonRecordingChild?.end(); + recordingParent?.end(); + } + }); + + this.testCase({ + name: "ContextPropagation: span attributes do not propagate to children", + test: () => { + // Arrange - Create parent with attributes + const parentAttrs = { + "parent.attr1": "value1", + "parent.attr2": "value2" + }; + const parentSpan = this._ai.startSpan("parent-with-attrs", + { attributes: parentAttrs }); + const parentContext = parentSpan!.spanContext(); + + // Act - Create child with different attributes + const childAttrs = { + "child.attr1": "childValue1" + }; + const childSpan = this._ai.startSpan("child-with-attrs", + { attributes: childAttrs }, parentContext); + + // Assert - Attributes are per-span, not inherited + Assert.ok(parentSpan!.attributes["parent.attr1"] === "value1", + "Parent should have its attributes"); + Assert.ok(childSpan!.attributes["child.attr1"] === "childValue1", + "Child should have its attributes"); + Assert.ok(!childSpan!.attributes["parent.attr1"], + "Child should not inherit parent's attributes"); + + // But context should propagate + Assert.equal(childSpan!.spanContext().traceId, parentContext.traceId, + "TraceId should propagate"); + + // Cleanup + childSpan?.end(); + parentSpan?.end(); + } + }); + } +} diff --git a/AISKU/Tests/Unit/src/SpanE2E.Tests.ts b/AISKU/Tests/Unit/src/SpanE2E.Tests.ts new file mode 100644 index 000000000..e9389d0a5 --- /dev/null +++ b/AISKU/Tests/Unit/src/SpanE2E.Tests.ts @@ -0,0 +1,751 @@ +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { ApplicationInsights } from "../../../src/applicationinsights-web"; +import { eOTelSpanKind, eOTelSpanStatusCode } from "@microsoft/applicationinsights-core-js"; + +/** + * E2E Tests for Span APIs that send real telemetry to Breeze endpoint + * + * These tests can be run manually to verify telemetry appears correctly in the Azure Portal: + * 1. Set MANUAL_E2E_TEST to true + * 2. Replace the instrumentationKey with a valid test iKey + * 3. Run the tests + * 4. Check the Azure Portal for the telemetry within 1-2 minutes + * + * Look for: + * - Dependencies in the "Performance" blade + * - Requests in the "Performance" blade + * - Custom properties and measurements + * - Distributed trace correlation + * - End-to-end transaction view + */ +export class SpanE2ETests extends AITestClass { + // Set to true to actually send telemetry to Breeze for manual validation + private static readonly MANUAL_E2E_TEST = false; + + // Replace with your test instrumentation key for manual E2E testing + private static readonly _instrumentationKey = "b7170927-2d1c-44f1-acec-59f4e1751c11"; + private static readonly _connectionString = `InstrumentationKey=${SpanE2ETests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + + constructor(testName?: string) { + super(testName || "SpanE2ETests"); + } + + public testInitialize() { + try { + this.useFakeServer = !SpanE2ETests.MANUAL_E2E_TEST; + + this._ai = new ApplicationInsights({ + config: { + connectionString: SpanE2ETests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + disableFetchTracking: false, + enableAutoRouteTracking: true, + disableExceptionTracking: false, + maxBatchInterval: 1000, // Send quickly for manual testing + enableDebug: true, + loggingLevelConsole: 2 // Show warnings and errors + } + }); + + this._ai.loadAppInsights(); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("=== MANUAL E2E TEST MODE ==="); + console.log("Telemetry will be sent to Breeze endpoint"); + console.log("Check Azure Portal in 1-2 minutes"); + console.log("Instrumentation Key:", SpanE2ETests._instrumentationKey); + console.log("============================"); + } + } catch (e) { + console.error("Failed to initialize tests: " + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + // Flush any pending telemetry before cleanup + this._ai.flush(); + this._ai.unload(false); + } + } + + public registerTests() { + this.addE2EBasicSpanTests(); + this.addE2EDistributedTraceTests(); + this.addE2EHttpDependencyTests(); + this.addE2EDatabaseDependencyTests(); + this.addE2EComplexScenarioTests(); + } + + private addE2EBasicSpanTests(): void { + this.testCase({ + name: "E2E: Basic CLIENT span creates RemoteDependency in portal", + test: () => { + // This will appear in the Azure Portal under Performance -> Dependencies + const span = this._ai.startSpan("E2E-BasicClientSpan", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "test.scenario": "basic-client", + "test.timestamp": new Date().toISOString(), + "test.type": "manual-validation", + "custom.property": "This should appear in custom properties" + } + }); + + // Simulate some work + if (span) { + span.setAttribute("work.completed", true); + span.setStatus({ code: eOTelSpanStatusCode.OK }); + span.end(); + } + + // Flush to ensure it's sent + this._ai.flush(); + + Assert.ok(span, "Span should be created"); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ Basic CLIENT span sent - Check Azure Portal Dependencies"); + } + } + }); + + this.testCase({ + name: "E2E: Basic SERVER span creates Request in portal", + test: () => { + // This will appear in the Azure Portal under Performance -> Requests + const span = this._ai.startSpan("E2E-BasicServerSpan", { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.method": "POST", + "http.url": "https://example.com/api/test", + "http.status_code": 200, + "test.scenario": "basic-server", + "test.timestamp": new Date().toISOString() + } + }); + + if (span) { + span.setStatus({ code: eOTelSpanStatusCode.OK }); + span.end(); + } + + this._ai.flush(); + + Assert.ok(span, "Span should be created"); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ Basic SERVER span sent - Check Azure Portal Requests"); + } + } + }); + + this.testCase({ + name: "E2E: Failed span shows as error in portal", + test: () => { + const span = this._ai.startSpan("E2E-FailedOperation", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "test.scenario": "failure-case", + "test.timestamp": new Date().toISOString() + } + }); + + if (span) { + // Simulate a failure + span.setAttribute("error.type", "TimeoutError"); + span.setAttribute("error.message", "Operation timed out after 5000ms"); + span.setStatus({ + code: eOTelSpanStatusCode.ERROR, + message: "Operation failed due to timeout" + }); + span.end(); + } + + this._ai.flush(); + + Assert.ok(span, "Span should be created"); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ Failed span sent - Should show success=false in portal"); + } + } + }); + } + + private addE2EDistributedTraceTests(): void { + this.testCase({ + name: "E2E: Parent-child span relationship visible in portal", + test: () => { + // Create parent span + const parentSpan = this._ai.startSpan("E2E-ParentOperation", { + kind: eOTelSpanKind.SERVER, + attributes: { + "test.scenario": "distributed-trace", + "test.timestamp": new Date().toISOString(), + "operation.level": "parent" + } + }); + + const parentContext = parentSpan?.spanContext(); + + // Create child span with explicit parent + const childSpan1 = this._ai.startSpan("E2E-ChildOperation1", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "operation.level": "child", + "child.index": 1 + } + }, parentContext); + + if (childSpan1) { + childSpan1.setAttribute("http.url", "https://api.example.com/users"); + childSpan1.setAttribute("http.method", "GET"); + childSpan1.setStatus({ code: eOTelSpanStatusCode.OK }); + childSpan1.end(); + } + + // Create another child + const childSpan2 = this._ai.startSpan("E2E-ChildOperation2", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "operation.level": "child", + "child.index": 2 + } + }, parentContext); + + if (childSpan2) { + childSpan2.setAttribute("http.url", "https://api.example.com/orders"); + childSpan2.setAttribute("http.method", "POST"); + childSpan2.setStatus({ code: eOTelSpanStatusCode.OK }); + childSpan2.end(); + } + + // End parent + if (parentSpan) { + parentSpan.setAttribute("children.count", 2); + parentSpan.setStatus({ code: eOTelSpanStatusCode.OK }); + parentSpan.end(); + } + + this._ai.flush(); + + Assert.ok(parentSpan && childSpan1 && childSpan2, "All spans should be created"); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ Distributed trace sent - Check End-to-End Transaction view"); + console.log(" Parent operation.id:", parentContext?.traceId); + } + } + }); + + this.testCase({ + name: "E2E: Nested span hierarchy (3 levels) visible in portal", + test: () => { + // Level 1: Root + const rootSpan = this._ai.startSpan("E2E-RootOperation", { + kind: eOTelSpanKind.SERVER, + attributes: { + "test.scenario": "nested-hierarchy", + "test.timestamp": new Date().toISOString(), + "span.level": 1 + } + }); + + const rootContext = rootSpan?.spanContext(); + + // Level 2: Child + const level2Span = this._ai.startSpan("E2E-Level2Operation", { + kind: eOTelSpanKind.INTERNAL, + attributes: { + "span.level": 2 + } + }, rootContext); + + const level2Context = level2Span?.spanContext(); + + // Level 3: Grandchild + const level3Span = this._ai.startSpan("E2E-Level3Operation", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "span.level": 3, + "http.url": "https://api.example.com/deep-call" + } + }, level2Context); + + // End in reverse order (child first, parent last) + if (level3Span) { + level3Span.setStatus({ code: eOTelSpanStatusCode.OK }); + level3Span.end(); + } + + if (level2Span) { + level2Span.setStatus({ code: eOTelSpanStatusCode.OK }); + level2Span.end(); + } + + if (rootSpan) { + rootSpan.setStatus({ code: eOTelSpanStatusCode.OK }); + rootSpan.end(); + } + + this._ai.flush(); + + Assert.ok(rootSpan && level2Span && level3Span, "All spans should be created"); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ 3-level nested trace sent - Check transaction timeline"); + } + } + }); + } + + private addE2EHttpDependencyTests(): void { + this.testCase({ + name: "E2E: HTTP dependency with full details in portal", + test: () => { + const span = this._ai.startSpan("E2E-HTTPDependency", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "POST", + "http.url": "https://api.example.com/v1/users/create", + "http.status_code": 201, + "http.request.header.content-type": "application/json", + "http.response.header.content-length": "1234", + "test.scenario": "http-dependency", + "test.timestamp": new Date().toISOString(), + "request.body.size": 512, + "response.time.ms": 145 + } + }); + + if (span) { + span.setStatus({ code: eOTelSpanStatusCode.OK }); + span.end(); + } + + this._ai.flush(); + + Assert.ok(span, "Span should be created"); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ HTTP dependency sent - Check Dependencies with full HTTP details"); + } + } + }); + + this.testCase({ + name: "E2E: HTTP dependency with various status codes in portal", + test: () => { + const statusCodes = [200, 201, 204, 400, 401, 403, 404, 500, 502, 503]; + + for (const statusCode of statusCodes) { + const isSuccess = statusCode >= 200 && statusCode < 400; + const span = this._ai.startSpan(`E2E-HTTP-${statusCode}`, { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "GET", + "http.url": `https://api.example.com/status/${statusCode}`, + "http.status_code": statusCode, + "test.scenario": "http-status-codes", + "test.timestamp": new Date().toISOString() + } + }); + + if (span) { + span.setStatus({ + code: isSuccess ? eOTelSpanStatusCode.OK : eOTelSpanStatusCode.ERROR + }); + span.end(); + } + } + + this._ai.flush(); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ Multiple HTTP status codes sent - Check success/failure in portal"); + } + + Assert.ok(true, "Multiple status codes tested"); + } + }); + } + + private addE2EDatabaseDependencyTests(): void { + this.testCase({ + name: "E2E: Database dependencies appear in portal", + test: () => { + const databases = [ + { system: "mysql", statement: "SELECT * FROM users WHERE id = ?", name: "production_db" }, + { system: "postgresql", statement: "INSERT INTO logs (message, level) VALUES ($1, $2)", name: "logs_db" }, + { system: "mongodb", statement: "db.products.find({category: 'electronics'})", name: "catalog_db" }, + { system: "redis", statement: "GET user:session:abc123", name: "cache_db" }, + { system: "mssql", statement: "EXEC sp_GetUserOrders @UserId=123", name: "orders_db" } + ]; + + for (const db of databases) { + const span = this._ai.startSpan(`E2E-DB-${db.system}`, { + kind: eOTelSpanKind.CLIENT, + attributes: { + "db.system": db.system, + "db.statement": db.statement, + "db.name": db.name, + "db.user": "app_user", + "net.peer.name": `${db.system}.example.com`, + "net.peer.port": 5432, + "test.scenario": "database-dependencies", + "test.timestamp": new Date().toISOString() + } + }); + + if (span) { + span.setAttribute("db.rows.affected", 42); + span.setAttribute("db.duration.ms", 23); + span.setStatus({ code: eOTelSpanStatusCode.OK }); + span.end(); + } + } + + this._ai.flush(); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ Database dependencies sent - Check Dependencies for SQL/NoSQL types"); + } + + Assert.ok(true, "Database dependencies tested"); + } + }); + + this.testCase({ + name: "E2E: Database slow query marked appropriately", + test: () => { + const span = this._ai.startSpan("E2E-SlowDatabaseQuery", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "db.system": "postgresql", + "db.statement": "SELECT * FROM orders JOIN users ON orders.user_id = users.id WHERE created_at > NOW() - INTERVAL '30 days'", + "db.name": "analytics_db", + "test.scenario": "slow-query", + "test.timestamp": new Date().toISOString(), + "db.query.execution.plan": "SeqScan on orders (cost=0.00..1000.00 rows=10000)", + "db.slow.query": true, + "db.duration.ms": 5432 + } + }); + + if (span) { + // Mark as warning (not error, but slow) + span.setStatus({ code: eOTelSpanStatusCode.OK }); + span.setAttribute("performance.warning", "Query exceeded 1000ms threshold"); + span.end(); + } + + this._ai.flush(); + + Assert.ok(span, "Slow query span created"); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ Slow database query sent - Check duration in portal"); + } + } + }); + } + + private addE2EComplexScenarioTests(): void { + this.testCase({ + name: "E2E: Complex e-commerce checkout scenario in portal", + test: () => { + // Simulate a complete e-commerce checkout flow with multiple dependencies + const timestamp = new Date().toISOString(); + + // 1. Initial checkout request + const checkoutSpan = this._ai.startSpan("E2E-CheckoutRequest", { + kind: eOTelSpanKind.SERVER, + attributes: { + "test.scenario": "complex-ecommerce", + "test.timestamp": timestamp, + "http.method": "POST", + "http.url": "https://shop.example.com/api/checkout", + "http.status_code": 200, + "user.id": "user_12345", + "cart.items.count": 3, + "cart.total.amount": 299.97 + } + }); + + const checkoutContext = checkoutSpan?.spanContext(); + + // 2. Validate inventory + const inventorySpan = this._ai.startSpan("E2E-ValidateInventory", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "POST", + "http.url": "https://inventory-api.example.com/validate", + "http.status_code": 200, + "items.validated": 3 + } + }, checkoutContext); + + if (inventorySpan) { + inventorySpan.setStatus({ code: eOTelSpanStatusCode.OK }); + inventorySpan.end(); + } + + // 3. Calculate shipping + const shippingSpan = this._ai.startSpan("E2E-CalculateShipping", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "POST", + "http.url": "https://shipping-api.example.com/calculate", + "http.status_code": 200, + "shipping.method": "express", + "shipping.cost": 15.99 + } + }, checkoutContext); + + if (shippingSpan) { + shippingSpan.setStatus({ code: eOTelSpanStatusCode.OK }); + shippingSpan.end(); + } + + // 4. Process payment + const paymentSpan = this._ai.startSpan("E2E-ProcessPayment", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "POST", + "http.url": "https://payments.example.com/charge", + "http.status_code": 200, + "payment.method": "credit_card", + "payment.amount": 315.96, + "payment.currency": "USD" + } + }, checkoutContext); + + if (paymentSpan) { + paymentSpan.setAttribute("payment.processor", "stripe"); + paymentSpan.setAttribute("payment.transaction.id", "txn_abc123xyz"); + paymentSpan.setStatus({ code: eOTelSpanStatusCode.OK }); + paymentSpan.end(); + } + + // 5. Create order in database + const createOrderSpan = this._ai.startSpan("E2E-CreateOrder", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "db.system": "postgresql", + "db.statement": "INSERT INTO orders (user_id, total, status) VALUES ($1, $2, $3) RETURNING id", + "db.name": "orders_db", + "db.operation": "INSERT" + } + }, checkoutContext); + + if (createOrderSpan) { + createOrderSpan.setAttribute("order.id", "ord_98765"); + createOrderSpan.setStatus({ code: eOTelSpanStatusCode.OK }); + createOrderSpan.end(); + } + + // 6. Send confirmation email + const emailSpan = this._ai.startSpan("E2E-SendConfirmationEmail", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "POST", + "http.url": "https://email-service.example.com/send", + "http.status_code": 202, + "email.recipient": "user@example.com", + "email.template": "order-confirmation" + } + }, checkoutContext); + + if (emailSpan) { + emailSpan.setStatus({ code: eOTelSpanStatusCode.OK }); + emailSpan.end(); + } + + // 7. Update cache + const cacheSpan = this._ai.startSpan("E2E-UpdateCache", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "db.system": "redis", + "db.statement": "SET user:12345:last_order ord_98765 EX 86400", + "cache.operation": "set", + "cache.key": "user:12345:last_order" + } + }, checkoutContext); + + if (cacheSpan) { + cacheSpan.setStatus({ code: eOTelSpanStatusCode.OK }); + cacheSpan.end(); + } + + // Complete checkout + if (checkoutSpan) { + checkoutSpan.setAttribute("checkout.status", "completed"); + checkoutSpan.setAttribute("order.id", "ord_98765"); + checkoutSpan.setAttribute("dependencies.count", 7); + checkoutSpan.setStatus({ code: eOTelSpanStatusCode.OK }); + checkoutSpan.end(); + } + + this._ai.flush(); + + Assert.ok(checkoutSpan, "Checkout span created"); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ Complex e-commerce scenario sent"); + console.log(" Trace ID:", checkoutContext?.traceId); + console.log(" Check End-to-End Transaction view for complete flow"); + console.log(" Expected: 1 Request + 7 Dependencies"); + } + } + }); + + this.testCase({ + name: "E2E: Mixed success and failure scenario in portal", + test: () => { + const timestamp = new Date().toISOString(); + + // Parent operation + const operationSpan = this._ai.startSpan("E2E-MixedResultsOperation", { + kind: eOTelSpanKind.SERVER, + attributes: { + "test.scenario": "mixed-success-failure", + "test.timestamp": timestamp + } + }); + + const operationContext = operationSpan?.spanContext(); + + // Successful child 1 + const successSpan1 = this._ai.startSpan("E2E-SuccessfulCall1", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.url": "https://api.example.com/service1", + "http.status_code": 200 + } + }, operationContext); + + if (successSpan1) { + successSpan1.setStatus({ code: eOTelSpanStatusCode.OK }); + successSpan1.end(); + } + + // Failed child + const failedSpan = this._ai.startSpan("E2E-FailedCall", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.url": "https://api.example.com/service2", + "http.status_code": 503 + } + }, operationContext); + + if (failedSpan) { + failedSpan.setAttribute("error.type", "ServiceUnavailable"); + failedSpan.setAttribute("retry.count", 3); + failedSpan.setStatus({ + code: eOTelSpanStatusCode.ERROR, + message: "Service temporarily unavailable" + }); + failedSpan.end(); + } + + // Successful child 2 (after retry) + const successSpan2 = this._ai.startSpan("E2E-SuccessfulCall2", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.url": "https://api.example.com/service3", + "http.status_code": 200 + } + }, operationContext); + + if (successSpan2) { + successSpan2.setStatus({ code: eOTelSpanStatusCode.OK }); + successSpan2.end(); + } + + // Parent partially successful + if (operationSpan) { + operationSpan.setAttribute("successful.calls", 2); + operationSpan.setAttribute("failed.calls", 1); + operationSpan.setAttribute("total.calls", 3); + operationSpan.setStatus({ code: eOTelSpanStatusCode.OK }); + operationSpan.end(); + } + + this._ai.flush(); + + Assert.ok(operationSpan, "Operation span created"); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ Mixed success/failure scenario sent"); + console.log(" Check for 2 successful + 1 failed dependency in transaction"); + } + } + }); + + this.testCase({ + name: "E2E: Span with rich custom properties for portal search", + test: () => { + const span = this._ai.startSpan("E2E-RichProperties", { + kind: eOTelSpanKind.SERVER, + attributes: { + "test.scenario": "rich-properties", + "test.timestamp": new Date().toISOString(), + + // Business context + "business.tenant": "acme-corp", + "business.region": "us-west-2", + "business.environment": "production", + + // User context + "user.id": "user_12345", + "user.email": "test@example.com", + "user.subscription": "premium", + "user.account.age.days": 456, + + // Request context + "request.id": "req_abc123", + "request.source": "web-app", + "request.version": "v2.3.1", + + // Performance metrics + "performance.db.queries": 5, + "performance.cache.hits": 3, + "performance.cache.misses": 2, + "performance.total.ms": 234, + + // Feature flags + "feature.new.checkout": true, + "feature.ab.test.group": "variant-b", + + // Custom measurements + "metrics.items.processed": 42, + "metrics.data.size.kb": 128 + } + }); + + if (span) { + span.setStatus({ code: eOTelSpanStatusCode.OK }); + span.end(); + } + + this._ai.flush(); + + Assert.ok(span, "Span with rich properties created"); + + if (SpanE2ETests.MANUAL_E2E_TEST) { + console.log("✓ Span with rich properties sent"); + console.log(" Use Application Insights search to filter by custom properties"); + console.log(" Example queries:"); + console.log(" - customDimensions.business.tenant == 'acme-corp'"); + console.log(" - customDimensions.user.subscription == 'premium'"); + console.log(" - customDimensions.feature.new.checkout == true"); + } + } + }); + } +} diff --git a/AISKU/Tests/Unit/src/SpanErrorHandling.Tests.ts b/AISKU/Tests/Unit/src/SpanErrorHandling.Tests.ts new file mode 100644 index 000000000..d6079a02f --- /dev/null +++ b/AISKU/Tests/Unit/src/SpanErrorHandling.Tests.ts @@ -0,0 +1,768 @@ +import { AITestClass, Assert } from '@microsoft/ai-test-framework'; +import { ApplicationInsights } from '../../../src/applicationinsights-web'; +import { IReadableSpan, eOTelSpanStatusCode, ITelemetryItem } from "@microsoft/applicationinsights-core-js"; + +export class SpanErrorHandlingTests extends AITestClass { + private static readonly _instrumentationKey = 'b7170927-2d1c-44f1-acec-59f4e1751c11'; + private static readonly _connectionString = `InstrumentationKey=${SpanErrorHandlingTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "SpanErrorHandlingTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: SpanErrorHandlingTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + + } catch (e) { + console.error('Failed to initialize tests: ' + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addInvalidSpanNameTests(); + this.addInvalidAttributeTests(); + this.addNullUndefinedInputTests(); + this.addInvalidParentContextTests(); + this.addInvalidOptionsTests(); + this.addEdgeCaseTests(); + } + + private addInvalidSpanNameTests(): void { + this.testCase({ + name: "SpanName: empty string name should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan(""); + span?.end(); + }, "Empty string name should not throw"); + } + }); + + this.testCase({ + name: "SpanName: null name should handle gracefully", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan(null as any); + span?.end(); + }, "Null name should not throw"); + } + }); + + this.testCase({ + name: "SpanName: undefined name should handle gracefully", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan(undefined as any); + span?.end(); + }, "Undefined name should not throw"); + } + }); + + this.testCase({ + name: "SpanName: very long name should be accepted", + test: () => { + // Arrange + const longName = "a".repeat(10000); + + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan(longName); + Assert.ok(span, "Should create span with long name"); + span?.end(); + }, "Very long name should not throw"); + } + }); + + this.testCase({ + name: "SpanName: special characters in name should be accepted", + test: () => { + // Arrange + const specialNames = [ + "span-with-dashes", + "span_with_underscores", + "span.with.dots", + "span/with/slashes", + "span:with:colons", + "span@with@at", + "span#with#hash", + "span$with$dollar" + ]; + + // Act & Assert + specialNames.forEach(name => { + Assert.doesNotThrow(() => { + const span = this._ai.startSpan(name); + span?.end(); + }, `Special character name '${name}' should not throw`); + }); + } + }); + } + + private addInvalidAttributeTests(): void { + this.testCase({ + name: "Attributes: null attribute value should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("null-attribute-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttribute("nullable.attr", null); + }, "Setting null attribute should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Attributes: undefined attribute value should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("undefined-attribute-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttribute("undefined.attr", undefined); + }, "Setting undefined attribute should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Attributes: empty string attribute key should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("empty-key-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttribute("", "value"); + }, "Empty string key should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Attributes: null attribute key should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("null-key-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttribute(null as any, "value"); + }, "Null key should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Attributes: undefined attribute key should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("undefined-key-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttribute(undefined as any, "value"); + }, "Undefined key should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Attributes: invalid attribute value types should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("invalid-type-test"); + + // Act & Assert - Test various invalid types + Assert.doesNotThrow(() => { + span?.setAttribute("object.attr", { nested: "object" } as any); + span?.setAttribute("array.attr", [1, 2, 3] as any); + span?.setAttribute("function.attr", (() => {}) as any); + span?.setAttribute("symbol.attr", Symbol("test") as any); + }, "Invalid attribute types should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Attributes: setAttributes with null should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("setAttributes-null-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttributes(null as any); + }, "setAttributes with null should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Attributes: setAttributes with undefined should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("setAttributes-undefined-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttributes(undefined as any); + }, "setAttributes with undefined should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Attributes: setAttributes with invalid object should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("setAttributes-invalid-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttributes({ + "valid": "value", + "null.value": null, + "undefined.value": undefined, + "object.value": { nested: "obj" } as any + }); + }, "setAttributes with mixed valid/invalid should not throw"); + + // Cleanup + span?.end(); + } + }); + } + + private addNullUndefinedInputTests(): void { + this.testCase({ + name: "NullUndefined: startSpan with null options should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("null-options-test", null as any); + span?.end(); + }, "Null options should not throw"); + } + }); + + this.testCase({ + name: "NullUndefined: startSpan with undefined options should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("undefined-options-test", undefined); + span?.end(); + }, "Undefined options should not throw"); + } + }); + + this.testCase({ + name: "NullUndefined: setStatus with null should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("null-status-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setStatus(null as any); + }, "setStatus with null should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "NullUndefined: setStatus with undefined should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("undefined-status-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setStatus(undefined as any); + }, "setStatus with undefined should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "NullUndefined: updateName with null should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("null-name-update-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.updateName(null as any); + }, "updateName with null should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "NullUndefined: updateName with undefined should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("undefined-name-update-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.updateName(undefined as any); + }, "updateName with undefined should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "NullUndefined: end with null time should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("null-end-time-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.end(null as any); + }, "end with null time should not throw"); + } + }); + + this.testCase({ + name: "NullUndefined: end with undefined time should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("undefined-end-time-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.end(undefined); + }, "end with undefined time should not throw"); + } + }); + + this.testCase({ + name: "NullUndefined: recordException with null should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("null-exception-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.recordException(null as any); + }, "recordException with null should not throw"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "NullUndefined: recordException with undefined should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("undefined-exception-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.recordException(undefined as any); + }, "recordException with undefined should not throw"); + + // Cleanup + span?.end(); + } + }); + } + + private addInvalidParentContextTests(): void { + this.testCase({ + name: "ParentContext: null parent context should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("null-parent-test", undefined, null as any); + span?.end(); + }, "Null parent context should not throw"); + } + }); + + this.testCase({ + name: "ParentContext: undefined parent context should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("undefined-parent-test", undefined, undefined); + span?.end(); + }, "Undefined parent context should not throw"); + } + }); + + this.testCase({ + name: "ParentContext: invalid parent context object should not throw", + test: () => { + // Arrange - Create invalid context objects + const invalidContexts = [ + {}, + { traceId: "invalid" }, + { spanId: "invalid" }, + { traceId: "", spanId: "" }, + { traceId: "123", spanId: "456" } // Too short + ]; + + // Act & Assert + invalidContexts.forEach((ctx, index) => { + Assert.doesNotThrow(() => { + const span = this._ai.startSpan(`invalid-context-${index}`, undefined, ctx as any); + span?.end(); + }, `Invalid context ${index} should not throw`); + }); + } + }); + + this.testCase({ + name: "ParentContext: parent context with missing fields should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("missing-fields-test", undefined, { + traceId: "12345678901234567890123456789012" + // Missing spanId and traceFlags + } as any); + span?.end(); + }, "Parent context with missing fields should not throw"); + } + }); + + this.testCase({ + name: "ParentContext: parent context with wrong types should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("wrong-types-test", undefined, { + traceId: 123456789, // Should be string + spanId: 987654321, // Should be string + traceFlags: "invalid" // Should be number + } as any); + span?.end(); + }, "Parent context with wrong types should not throw"); + } + }); + } + + private addInvalidOptionsTests(): void { + this.testCase({ + name: "Options: invalid kind value should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("invalid-kind-test", { + kind: 999 as any // Invalid kind value + }); + span?.end(); + }, "Invalid kind value should not throw"); + } + }); + + this.testCase({ + name: "Options: negative kind value should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("negative-kind-test", { + kind: -1 as any + }); + span?.end(); + }, "Negative kind value should not throw"); + } + }); + + this.testCase({ + name: "Options: null attributes in options should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("null-attrs-options-test", { + attributes: null as any + }); + span?.end(); + }, "Null attributes in options should not throw"); + } + }); + + this.testCase({ + name: "Options: undefined attributes in options should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("undefined-attrs-options-test", { + attributes: undefined + }); + span?.end(); + }, "Undefined attributes in options should not throw"); + } + }); + + this.testCase({ + name: "Options: invalid startTime should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("invalid-starttime-test", { + startTime: "invalid" as any + }); + span?.end(); + }, "Invalid startTime should not throw"); + } + }); + + this.testCase({ + name: "Options: negative startTime should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("negative-starttime-test", { + startTime: -1000 + }); + span?.end(); + }, "Negative startTime should not throw"); + } + }); + + this.testCase({ + name: "Options: future startTime should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("future-starttime-test", { + startTime: Date.now() + 1000000 + }); + span?.end(); + }, "Future startTime should not throw"); + } + }); + + this.testCase({ + name: "Options: multiple invalid options should not throw", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const span = this._ai.startSpan("multi-invalid-options-test", { + kind: -999 as any, + attributes: null as any, + startTime: "invalid" as any, + recording: "maybe" as any, + root: "yes" as any + } as any); + span?.end(); + }, "Multiple invalid options should not throw"); + } + }); + } + + private addEdgeCaseTests(): void { + this.testCase({ + name: "EdgeCase: operations on null span should not throw", + test: () => { + // Arrange - Force null span (though SDK shouldn't return null) + const span: IReadableSpan | null = null; + + // Act & Assert - All operations should be safe + Assert.doesNotThrow(() => { + span?.setAttribute("key", "value"); + span?.setAttributes({ "key": "value" }); + span?.setStatus({ code: eOTelSpanStatusCode.OK }); + span?.updateName("new-name"); + span?.end(); + span?.recordException(new Error("test")); + }, "Operations on null span should not throw"); + } + }); + + this.testCase({ + name: "EdgeCase: extremely large attribute count should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("large-attr-count-test"); + const largeAttrs: any = {}; + for (let i = 0; i < 1000; i++) { + largeAttrs[`attr_${i}`] = `value_${i}`; + } + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttributes(largeAttrs); + span?.end(); + }, "Large attribute count should not throw"); + } + }); + + this.testCase({ + name: "EdgeCase: very long attribute values should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("long-attr-value-test"); + const longValue = "x".repeat(100000); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttribute("long.attr", longValue); + span?.end(); + }, "Very long attribute values should not throw"); + } + }); + + this.testCase({ + name: "EdgeCase: rapid successive operations should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("rapid-ops-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + for (let i = 0; i < 100; i++) { + span?.setAttribute(`rapid_${i}`, i); + span?.setStatus({ code: eOTelSpanStatusCode.OK }); + span?.updateName(`name_${i}`); + } + span?.end(); + }, "Rapid successive operations should not throw"); + } + }); + + this.testCase({ + name: "EdgeCase: mixed valid and invalid operations should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("mixed-ops-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttribute("valid", "value"); + span?.setAttribute(null as any, "invalid-key"); + span?.setAttribute("another.valid", 123); + span?.setAttribute("", "empty-key"); + span?.setAttributes({ "good": "attr", "bad": null }); + span?.setStatus({ code: eOTelSpanStatusCode.OK }); + span?.updateName(null as any); + span?.updateName("valid-name"); + span?.end(); + }, "Mixed valid and invalid operations should not throw"); + } + }); + + this.testCase({ + name: "EdgeCase: special Unicode characters should not throw", + test: () => { + // Arrange + const unicodeStrings = [ + "Hello 世界", + "Emoji 😀🎉", + "RTL العربية", + "Combined ñ é ü", + "Zero-width\u200B\u200Ccharacters" + ]; + + // Act & Assert + unicodeStrings.forEach((str, index) => { + Assert.doesNotThrow(() => { + const span = this._ai.startSpan(str); + span?.setAttribute("unicode.attr", str); + span?.updateName(`unicode_${index}_${str}`); + span?.end(); + }, `Unicode string ${index} should not throw`); + }); + } + }); + + this.testCase({ + name: "EdgeCase: circular reference in error should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("circular-error-test"); + const circularError: any = new Error("Circular test"); + circularError.self = circularError; // Create circular reference + + // Act & Assert + Assert.doesNotThrow(() => { + span?.recordException(circularError); + span?.end(); + }, "Circular reference in error should not throw"); + } + }); + + this.testCase({ + name: "EdgeCase: NaN and Infinity values should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("special-numbers-test"); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttribute("nan.value", NaN as any); + span?.setAttribute("infinity.value", Infinity as any); + span?.setAttribute("neg.infinity.value", -Infinity as any); + span?.end(); + }, "NaN and Infinity values should not throw"); + } + }); + } +} diff --git a/AISKU/Tests/Unit/src/SpanHelperUtils.Tests.ts b/AISKU/Tests/Unit/src/SpanHelperUtils.Tests.ts new file mode 100644 index 000000000..32c89c4d7 --- /dev/null +++ b/AISKU/Tests/Unit/src/SpanHelperUtils.Tests.ts @@ -0,0 +1,992 @@ +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { ApplicationInsights } from "../../../src/applicationinsights-web"; +import { + createDistributedTraceContext, + eOTelSpanKind, + eOTelSpanStatusCode, + IDistributedTraceInit, + isReadableSpan, + isSpanContextValid, + ITelemetryItem, + wrapSpanContext +} from "@microsoft/applicationinsights-core-js"; + +/** + * Comprehensive tests for span helper utility functions + * + * Tests verify: + * - isSpanContextValid: validates span context + * - wrapSpanContext: wraps external span contexts + * - isReadableSpan: type guard for spans + * - createNonRecordingSpan: creates non-recording spans (tested via wrapSpanContext) + */ +export class SpanHelperUtilsTests extends AITestClass { + private static readonly _instrumentationKey = "b7170927-2d1c-44f1-acec-59f4e1751c11"; + private static readonly _connectionString = `InstrumentationKey=${SpanHelperUtilsTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "SpanHelperUtilsTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: SpanHelperUtilsTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + } catch (e) { + console.error("Failed to initialize tests: " + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addIsSpanContextValidTests(); + this.addWrapSpanContextTests(); + this.addIsReadableSpanTests(); + this.addHelperIntegrationTests(); + } + + private addIsSpanContextValidTests(): void { + this.testCase({ + name: "isSpanContextValid: valid span context returns true", + test: () => { + // Arrange + const span = this._ai.startSpan("test-span"); + const spanContext = span?.spanContext(); + + // Act + const isValid = isSpanContextValid(spanContext!); + + // Assert + Assert.ok(isValid, "Valid span context should return true"); + span?.end(); + } + }); + + this.testCase({ + name: "isSpanContextValid: valid traceId and spanId returns true", + test: () => { + // Arrange - create valid context + const validContext = createDistributedTraceContext({ + traceId: "0123456789abcdef0123456789abcdef", // 32 hex chars + spanId: "0123456789abcdef", // 16 hex chars + traceFlags: 1 + }); + + // Act + const isValid = isSpanContextValid(validContext); + + // Assert + Assert.ok(isValid, "Context with valid IDs should return true"); + } + }); + + this.testCase({ + name: "isSpanContextValid: invalid traceId returns false", + test: () => { + // Arrange - traceId too short (use IDistributedTraceInit directly, not createDistributedTraceContext which validates) + const invalidContext: IDistributedTraceInit = { + traceId: "0123456789abcdef", // Only 16 chars (should be 32) + spanId: "0123456789abcdef", + traceFlags: 1 + }; + + // Act + const isValid = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!isValid, "Context with invalid traceId should return false"); + } + }); + + this.testCase({ + name: "isSpanContextValid: invalid spanId returns false", + test: () => { + // Arrange - spanId too short (use IDistributedTraceInit directly, not createDistributedTraceContext which validates) + const invalidContext: IDistributedTraceInit = { + traceId: "0123456789abcdef0123456789abcdef", + spanId: "01234567", // Only 8 chars (should be 16) + traceFlags: 1 + }; + + // Act + const isValid = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!isValid, "Context with invalid spanId should return false"); + } + }); + + this.testCase({ + name: "isSpanContextValid: all zeros traceId returns false", + test: () => { + // Arrange - all zeros is invalid per spec (use IDistributedTraceInit directly, not createDistributedTraceContext which validates) + const invalidContext: IDistributedTraceInit = { + traceId: "00000000000000000000000000000000", + spanId: "0123456789abcdef", + traceFlags: 1 + }; + + // Act + const isValid = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!isValid, "Context with all-zero traceId should return false"); + } + }); + + this.testCase({ + name: "isSpanContextValid: all zeros spanId returns false", + test: () => { + // Arrange - all zeros is invalid per spec (use IDistributedTraceInit directly, not createDistributedTraceContext which validates) + const invalidContext: IDistributedTraceInit = { + traceId: "0123456789abcdef0123456789abcdef", + spanId: "0000000000000000", + traceFlags: 1 + }; + + // Act + const isValid = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!isValid, "Context with all-zero spanId should return false"); + } + }); + + this.testCase({ + name: "isSpanContextValid: null context returns false", + test: () => { + // Act + const isValid = isSpanContextValid(null as any); + + // Assert + Assert.ok(!isValid, "Null context should return false"); + } + }); + + this.testCase({ + name: "isSpanContextValid: undefined context returns false", + test: () => { + // Act + const isValid = isSpanContextValid(undefined as any); + + // Assert + Assert.ok(!isValid, "Undefined context should return false"); + } + }); + + this.testCase({ + name: "isSpanContextValid: empty traceId returns false", + test: () => { + // Arrange - use IDistributedTraceInit directly, not createDistributedTraceContext which validates + const invalidContext: IDistributedTraceInit = { + traceId: "", + spanId: "0123456789abcdef", + traceFlags: 1 + }; + + // Act + const isValid = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!isValid, "Empty traceId should return false"); + } + }); + + this.testCase({ + name: "isSpanContextValid: empty spanId returns false", + test: () => { + // Arrange - use IDistributedTraceInit directly, not createDistributedTraceContext which validates + const invalidContext: IDistributedTraceInit = { + traceId: "0123456789abcdef0123456789abcdef", + spanId: "", + traceFlags: 1 + }; + + // Act + const isValid = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!isValid, "Empty spanId should return false"); + } + }); + + this.testCase({ + name: "isSpanContextValid: non-hex characters in traceId returns false", + test: () => { + // Arrange - use IDistributedTraceInit directly, not createDistributedTraceContext which validates + const invalidContext: IDistributedTraceInit = { + traceId: "0123456789abcdefghij456789abcdef", // Contains g-j + spanId: "0123456789abcdef", + traceFlags: 1 + }; + + // Act + const isValid = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!isValid, "TraceId with non-hex chars should return false"); + } + }); + + this.testCase({ + name: "isSpanContextValid: uppercase hex characters are valid", + test: () => { + // Arrange + const validContext = createDistributedTraceContext({ + traceId: "0123456789ABCDEF0123456789ABCDEF", + spanId: "0123456789ABCDEF", + traceFlags: 1 + }); + + // Act + const isValid = isSpanContextValid(validContext); + + // Assert + Assert.ok(isValid, "Uppercase hex characters should be valid"); + } + }); + + this.testCase({ + name: "isSpanContextValid: mixed case hex characters are valid", + test: () => { + // Arrange + const validContext = createDistributedTraceContext({ + traceId: "0123456789AbCdEf0123456789AbCdEf", + spanId: "0123456789AbCdEf", + traceFlags: 1 + }); + + // Act + const isValid = isSpanContextValid(validContext); + + // Assert + Assert.ok(isValid, "Mixed case hex characters should be valid"); + } + }); + } + + private addWrapSpanContextTests(): void { + this.testCase({ + name: "wrapSpanContext: creates non-recording span from context", + test: () => { + // Arrange + const originalSpan = this._ai.startSpan("original-span"); + const spanContext = originalSpan?.spanContext(); + + // Act + const wrappedSpan = wrapSpanContext(this._ai.otelApi, spanContext!); + + // Assert + Assert.ok(wrappedSpan, "Wrapped span should be created"); + Assert.ok(!wrappedSpan.isRecording(), "Wrapped span should not be recording"); + Assert.equal(wrappedSpan.spanContext().traceId, spanContext?.traceId, "TraceId should match"); + Assert.equal(wrappedSpan.spanContext().spanId, spanContext?.spanId, "SpanId should match"); + + // Cleanup + originalSpan?.end(); + wrappedSpan.end(); + } + }); + + this.testCase({ + name: "wrapSpanContext: wrapped span name includes spanId", + test: () => { + // Arrange + const spanContext = createDistributedTraceContext({ + traceId: "0123456789abcdef0123456789abcdef", + spanId: "0123456789abcdef", + traceFlags: 1 + }); + + // Act + const wrappedSpan = wrapSpanContext(this._ai.otelApi, spanContext); + + // Assert + Assert.ok(wrappedSpan.name.includes(spanContext.spanId), + "Wrapped span name should include spanId"); + Assert.ok(wrappedSpan.name.includes("wrapped"), + "Wrapped span name should indicate it's wrapped"); + + wrappedSpan.end(); + } + }); + + this.testCase({ + name: "wrapSpanContext: wrapped span does not generate telemetry", + test: () => { + // Arrange + this._trackCalls = []; + const spanContext = createDistributedTraceContext({ + traceId: "abcdef0123456789abcdef0123456789", + spanId: "abcdef0123456789", + traceFlags: 1 + }); + + // Act + const wrappedSpan = wrapSpanContext(this._ai.otelApi, spanContext); + wrappedSpan.setAttribute("test", "value"); + wrappedSpan.setStatus({ code: eOTelSpanStatusCode.OK }); + wrappedSpan.end(); + + // Assert + Assert.equal(this._trackCalls.length, 0, + "Wrapped span should not generate telemetry"); + } + }); + + this.testCase({ + name: "wrapSpanContext: wrapped span kind is INTERNAL", + test: () => { + // Arrange + const spanContext = createDistributedTraceContext({ + traceId: "fedcba9876543210fedcba9876543210", + spanId: "fedcba9876543210", + traceFlags: 1 + }); + + // Act + const wrappedSpan = wrapSpanContext(this._ai.otelApi, spanContext); + + // Assert + Assert.equal(wrappedSpan.kind, eOTelSpanKind.INTERNAL, + "Wrapped span should have INTERNAL kind"); + + wrappedSpan.end(); + } + }); + + this.testCase({ + name: "wrapSpanContext: can use wrapped span as parent", + test: () => { + // Arrange + const parentContext = createDistributedTraceContext({ + traceId: "1234567890abcdef1234567890abcdef", + spanId: "1234567890abcdef", + traceFlags: 1 + }); + const wrappedParent = wrapSpanContext(this._ai.otelApi, parentContext); + + // Act - create child with wrapped parent + const childSpan = this._ai.startSpan("child-span", { + kind: eOTelSpanKind.CLIENT + }, wrappedParent.spanContext()); + + // Assert + Assert.ok(childSpan, "Child span should be created"); + Assert.equal(childSpan.spanContext().traceId, parentContext.traceId, + "Child should have same traceId as wrapped parent"); + + // Cleanup + childSpan?.end(); + wrappedParent.end(); + } + }); + + this.testCase({ + name: "wrapSpanContext: wrapped span supports all span operations", + test: () => { + // Arrange + const spanContext = createDistributedTraceContext({ + traceId: "aabbccddeeff00112233445566778899", + spanId: "aabbccddeeff0011", + traceFlags: 1 + }); + const wrappedSpan = wrapSpanContext(this._ai.otelApi, spanContext); + + // Act - perform various operations + wrappedSpan.setAttribute("key1", "value1"); + wrappedSpan.setAttributes({ + "key2": 123, + "key3": true + }); + wrappedSpan.updateName("new-name"); + wrappedSpan.setStatus({ code: eOTelSpanStatusCode.OK, message: "Success" }); + wrappedSpan.recordException(new Error("Test error")); + + // Assert - operations should not throw + Assert.ok(true, "All operations should complete without error"); + Assert.equal(wrappedSpan.name, "new-name", "Name should be updated"); + + wrappedSpan.end(); + } + }); + + this.testCase({ + name: "wrapSpanContext: preserves traceFlags if present", + test: () => { + // Arrange + const spanContext = createDistributedTraceContext({ + traceId: "11112222333344445555666677778888", + spanId: "1111222233334444", + traceFlags: 1 // Sampled + }); + + // Act + const wrappedSpan = wrapSpanContext(this._ai.otelApi, spanContext); + + // Assert + Assert.equal(wrappedSpan.spanContext().traceFlags, 1, + "TraceFlags should be preserved"); + + wrappedSpan.end(); + } + }); + + this.testCase({ + name: "wrapSpanContext: multiple wrapped spans from same context are independent", + test: () => { + // Arrange + const spanContext = createDistributedTraceContext({ + traceId: "99887766554433221100ffeeddccbbaa", + spanId: "9988776655443322", + traceFlags: 1 + }); + + // Act + const wrapped1 = wrapSpanContext(this._ai.otelApi, spanContext); + const wrapped2 = wrapSpanContext(this._ai.otelApi, spanContext); + + // Assert - both wrap same context but are different span objects + Assert.notEqual(wrapped1, wrapped2, "Should create different span objects"); + Assert.equal(wrapped1.spanContext().traceId, wrapped2.spanContext().traceId, + "Both should have same traceId"); + Assert.equal(wrapped1.spanContext().spanId, wrapped2.spanContext().spanId, + "Both should have same spanId"); + + // Operations on one should not affect the other + wrapped1.updateName("wrapped-1"); + wrapped2.updateName("wrapped-2"); + Assert.equal(wrapped1.name, "wrapped-1", "First span name"); + Assert.equal(wrapped2.name, "wrapped-2", "Second span name"); + + wrapped1.end(); + wrapped2.end(); + } + }); + + this.testCase({ + name: "wrapSpanContext: can wrap context from external system", + test: () => { + // Arrange - simulate receiving context from external system (e.g., HTTP header) + const externalContext = createDistributedTraceContext({ + traceId: "00112233445566778899aabbccddeeff", + spanId: "0011223344556677", + traceFlags: 1 + }); + + // Act + const wrappedSpan = wrapSpanContext(this._ai.otelApi, externalContext); + + // Create child span to continue the trace + const childSpan = this._ai.startSpan("continue-external-trace", { + kind: eOTelSpanKind.SERVER + }, wrappedSpan.spanContext()); + + // Assert + Assert.equal(childSpan?.spanContext().traceId, externalContext.traceId, + "Should continue external trace"); + Assert.notEqual(childSpan?.spanContext().spanId, externalContext.spanId, + "Should have new spanId"); + + // Cleanup + childSpan?.end(); + wrappedSpan.end(); + } + }); + } + + private addIsReadableSpanTests(): void { + this.testCase({ + name: "isReadableSpan: valid span returns true", + test: () => { + // Arrange + const span = this._ai.startSpan("test-span"); + + // Act + const isValid = isReadableSpan(span); + + // Assert + Assert.ok(isValid, "Valid span should return true"); + + span?.end(); + } + }); + + this.testCase({ + name: "isReadableSpan: null returns false", + test: () => { + // Act + const isValid = isReadableSpan(null); + + // Assert + Assert.ok(!isValid, "Null should return false"); + } + }); + + this.testCase({ + name: "isReadableSpan: undefined returns false", + test: () => { + // Act + const isValid = isReadableSpan(undefined); + + // Assert + Assert.ok(!isValid, "Undefined should return false"); + } + }); + + this.testCase({ + name: "isReadableSpan: plain object returns false", + test: () => { + // Arrange + const notASpan = { + name: "fake-span", + kind: eOTelSpanKind.INTERNAL + }; + + // Act + const isValid = isReadableSpan(notASpan); + + // Assert + Assert.ok(!isValid, "Plain object should return false"); + } + }); + + this.testCase({ + name: "isReadableSpan: object with partial span interface returns false", + test: () => { + // Arrange - object with some but not all required properties + const partialSpan = { + name: "partial", + kind: eOTelSpanKind.CLIENT, + spanContext: () => ({ traceId: "123", spanId: "456" }), + // Missing: duration, ended, startTime, endTime, etc. + }; + + // Act + const isValid = isReadableSpan(partialSpan); + + // Assert + Assert.ok(!isValid, "Partial span interface should return false"); + } + }); + + this.testCase({ + name: "isReadableSpan: recording span returns true", + test: () => { + // Arrange + const recordingSpan = this._ai.startSpan("recording", { recording: true }); + + // Act + const isValid = isReadableSpan(recordingSpan); + + // Assert + Assert.ok(isValid, "Recording span should return true"); + + recordingSpan?.end(); + } + }); + + this.testCase({ + name: "isReadableSpan: non-recording span returns true", + test: () => { + // Arrange + const nonRecordingSpan = this._ai.startSpan("non-recording", { recording: false }); + + // Act + const isValid = isReadableSpan(nonRecordingSpan); + + // Assert + Assert.ok(isValid, "Non-recording span should return true"); + + nonRecordingSpan?.end(); + } + }); + + this.testCase({ + name: "isReadableSpan: wrapped span context returns true", + test: () => { + // Arrange + const spanContext = createDistributedTraceContext({ + traceId: "aabbccdd00112233aabbccdd00112233", + spanId: "aabbccdd00112233", + traceFlags: 1 + }); + const wrappedSpan = wrapSpanContext(this._ai.otelApi, spanContext); + + // Act + const isValid = isReadableSpan(wrappedSpan); + + // Assert + Assert.ok(isValid, "Wrapped span should return true"); + + wrappedSpan.end(); + } + }); + + this.testCase({ + name: "isReadableSpan: ended span returns true", + test: () => { + // Arrange + const span = this._ai.startSpan("ended-span"); + span?.end(); + + // Act + const isValid = isReadableSpan(span); + + // Assert + Assert.ok(isValid, "Ended span should still return true"); + } + }); + + this.testCase({ + name: "isReadableSpan: span with all kinds returns true", + test: () => { + // Arrange & Act & Assert + const internalSpan = this._ai.startSpan("internal", { kind: eOTelSpanKind.INTERNAL }); + Assert.ok(isReadableSpan(internalSpan), "INTERNAL span should be valid"); + internalSpan?.end(); + + const clientSpan = this._ai.startSpan("client", { kind: eOTelSpanKind.CLIENT }); + Assert.ok(isReadableSpan(clientSpan), "CLIENT span should be valid"); + clientSpan?.end(); + + const serverSpan = this._ai.startSpan("server", { kind: eOTelSpanKind.SERVER }); + Assert.ok(isReadableSpan(serverSpan), "SERVER span should be valid"); + serverSpan?.end(); + + const producerSpan = this._ai.startSpan("producer", { kind: eOTelSpanKind.PRODUCER }); + Assert.ok(isReadableSpan(producerSpan), "PRODUCER span should be valid"); + producerSpan?.end(); + + const consumerSpan = this._ai.startSpan("consumer", { kind: eOTelSpanKind.CONSUMER }); + Assert.ok(isReadableSpan(consumerSpan), "CONSUMER span should be valid"); + consumerSpan?.end(); + } + }); + + this.testCase({ + name: "isReadableSpan: string returns false", + test: () => { + // Act + const isValid = isReadableSpan("not a span"); + + // Assert + Assert.ok(!isValid, "String should return false"); + } + }); + + this.testCase({ + name: "isReadableSpan: number returns false", + test: () => { + // Act + const isValid = isReadableSpan(12345); + + // Assert + Assert.ok(!isValid, "Number should return false"); + } + }); + + this.testCase({ + name: "isReadableSpan: array returns false", + test: () => { + // Act + const isValid = isReadableSpan([]); + + // Assert + Assert.ok(!isValid, "Array should return false"); + } + }); + + this.testCase({ + name: "isReadableSpan: function returns false", + test: () => { + // Act + const isValid = isReadableSpan(() => {}); + + // Assert + Assert.ok(!isValid, "Function should return false"); + } + }); + } + + private addHelperIntegrationTests(): void { + this.testCase({ + name: "Integration: validate context before wrapping", + test: () => { + // Arrange - good practice to validate before wrapping + const validContext: IDistributedTraceInit = { + traceId: "aaaabbbbccccddddeeeeffffaaaabbbb", + spanId: "aaaabbbbccccdddd", + traceFlags: 1 + }; + const invalidContext: IDistributedTraceInit = { + traceId: "invalid", + spanId: "also-bad", + traceFlags: 1 + }; + + // Act & Assert - validate before wrapping + Assert.ok(isSpanContextValid(validContext), "Valid context should pass validation"); + const wrappedSpan = wrapSpanContext(this._ai.otelApi, validContext); + Assert.ok(wrappedSpan, "Should wrap valid context"); + Assert.ok(isReadableSpan(wrappedSpan), "Wrapped span should be readable"); + wrappedSpan.end(); + + // Don't wrap invalid context + Assert.ok(!isSpanContextValid(invalidContext), + "Should detect invalid context before wrapping"); + } + }); + + this.testCase({ + name: "Integration: type-safe span handling with isReadableSpan", + test: () => { + // Arrange + const span = this._ai.startSpan("type-safe"); + const maybeSpan: any = span; + + // Act - type guard allows safe access + if (isReadableSpan(maybeSpan)) { + // TypeScript knows this is IReadableSpan now + const context = maybeSpan.spanContext(); + maybeSpan.setAttribute("safe", "access"); + maybeSpan.end(); + + // Assert + Assert.ok(context.traceId, "Can safely access span properties"); + } else { + Assert.ok(false, "Span should be readable"); + } + } + }); + + this.testCase({ + name: "Integration: wrap external trace and continue locally", + test: () => { + // Arrange - simulate receiving trace context from external service + const externalContext = createDistributedTraceContext({ + traceId: "1234567890abcdef1234567890abcdef", + spanId: "1234567890abcdef", + traceFlags: 1 + }); + + // Act - validate and wrap + Assert.ok(isSpanContextValid(externalContext), + "External context should be valid"); + + const wrappedExternal = wrapSpanContext( + this._ai.otelApi, + externalContext + ); + Assert.ok(isReadableSpan(wrappedExternal), + "Wrapped external should be readable"); + + // Continue trace with local spans + const localSpan1 = this._ai.startSpan("local-processing", { + kind: eOTelSpanKind.SERVER + }, wrappedExternal.spanContext()); + + const localSpan2 = this._ai.startSpan("database-call", { + kind: eOTelSpanKind.CLIENT + }, localSpan1?.spanContext()); + + // Assert - trace continuity + Assert.equal(localSpan1?.spanContext().traceId, externalContext.traceId, + "Local span should continue external trace"); + Assert.equal(localSpan2?.spanContext().traceId, externalContext.traceId, + "Nested span should continue external trace"); + Assert.notEqual(localSpan2?.spanContext().spanId, externalContext.spanId, + "Should have new span IDs"); + + // Cleanup + localSpan2?.end(); + localSpan1?.end(); + wrappedExternal.end(); + } + }); + + this.testCase({ + name: "Integration: helper functions work with all span kinds", + test: () => { + // Test each span kind + const kinds = [ + eOTelSpanKind.INTERNAL, + eOTelSpanKind.CLIENT, + eOTelSpanKind.SERVER, + eOTelSpanKind.PRODUCER, + eOTelSpanKind.CONSUMER + ]; + + for (const kind of kinds) { + // Create span + const span = this._ai.startSpan(`span-kind-${kind}`, { kind }); + Assert.ok(span, `Span with kind ${kind} should be created`); + + // Verify with isReadableSpan + Assert.ok(isReadableSpan(span), `Span kind ${kind} should be readable`); + + // Get and validate context + const context = span?.spanContext(); + Assert.ok(isSpanContextValid(context!), + `Span kind ${kind} should have valid context`); + + // Wrap the context + const wrapped = wrapSpanContext(this._ai.otelApi, context!); + Assert.ok(isReadableSpan(wrapped), + `Wrapped span from kind ${kind} should be readable`); + + // Cleanup + span?.end(); + wrapped.end(); + } + + Assert.ok(true, "All span kinds tested successfully"); + } + }); + + this.testCase({ + name: "Integration: helpers work after span lifecycle", + test: () => { + // Arrange - create and end span + const span = this._ai.startSpan("lifecycle-test"); + const context = span?.spanContext(); + span?.end(); + + // Act & Assert - helpers should still work with ended span + Assert.ok(isReadableSpan(span), + "isReadableSpan should work with ended span"); + Assert.ok(isSpanContextValid(context!), + "isSpanContextValid should work with context from ended span"); + + const wrapped = wrapSpanContext(this._ai.otelApi, context!); + Assert.ok(isReadableSpan(wrapped), + "Can wrap context from ended span"); + + wrapped.end(); + } + }); + + this.testCase({ + name: "Integration: defensive programming with helpers", + test: () => { + // Arrange - potentially problematic inputs + const nullValue: any = null; + const undefinedValue: any = undefined; + const emptyObject: any = {}; + const wrongType: any = "not a span"; + + // Act & Assert - helpers should handle gracefully + Assert.ok(!isReadableSpan(nullValue), "Handle null"); + Assert.ok(!isReadableSpan(undefinedValue), "Handle undefined"); + Assert.ok(!isReadableSpan(emptyObject), "Handle empty object"); + Assert.ok(!isReadableSpan(wrongType), "Handle wrong type"); + + Assert.ok(!isSpanContextValid(nullValue), "Validate null context"); + Assert.ok(!isSpanContextValid(undefinedValue), "Validate undefined context"); + Assert.ok(!isSpanContextValid(emptyObject), "Validate empty context"); + Assert.ok(!isSpanContextValid(wrongType), "Validate wrong type context"); + } + }); + + this.testCase({ + name: "Integration: wrap and use as active span", + test: () => { + // Arrange + const externalContext = createDistributedTraceContext({ + traceId: "activespan123456789012345678901234", + spanId: "activespan123456", + traceFlags: 1 + }); + + // Act - wrap and set as active + const wrappedSpan = wrapSpanContext(this._ai.otelApi, externalContext); + const scope = this._ai.core.setActiveSpan(wrappedSpan); + + // Create child that should automatically get wrapped span as parent + const childSpan = this._ai.startSpan("auto-child", { + kind: eOTelSpanKind.INTERNAL + }); + + // Assert + Assert.equal(childSpan?.spanContext().traceId, externalContext.traceId, + "Child should inherit traceId from active wrapped span"); + + // Cleanup + childSpan?.end(); + scope?.restore(); + wrappedSpan.end(); + } + }); + + this.testCase({ + name: "Integration: validation chain for incoming distributed trace", + test: () => { + // Simulate complete flow of receiving and processing distributed trace + + // Step 1: Receive trace context (e.g., from HTTP headers) + const receivedContext = createDistributedTraceContext({ + traceId: "abcdef0123456789abcdef0123456789", + spanId: "abcdef0123456789", + traceFlags: 1 + }); + + // Step 2: Validate received context + Assert.ok(isSpanContextValid(receivedContext), "Received trace context should be valid"); + + // Step 3: Wrap context to create local span representation + const remoteSpan = wrapSpanContext(this._ai.otelApi, receivedContext); + Assert.ok(isReadableSpan(remoteSpan), "Failed to create readable span"); + + // Step 4: Create local server span as child + const serverSpan = this._ai.startSpan("handle-request", { + kind: eOTelSpanKind.SERVER + }, remoteSpan.spanContext()); + + Assert.ok(isReadableSpan(serverSpan), "Server span should be readable"); + + // Step 5: Verify trace continuity + const serverContext = serverSpan?.spanContext(); + Assert.ok(isSpanContextValid(serverContext!), + "Server span should have valid context"); + Assert.equal(serverContext?.traceId, receivedContext.traceId, + "Trace ID should be preserved across process boundary"); + + // Step 6: Complete request handling + serverSpan?.setAttribute("http.status_code", 200); + serverSpan?.setStatus({ code: eOTelSpanStatusCode.OK }); + serverSpan?.end(); + remoteSpan.end(); + + Assert.ok(true, "Complete distributed trace flow validated"); + } + }); + } +} diff --git a/AISKU/Tests/Unit/src/SpanLifeCycle.Tests.ts b/AISKU/Tests/Unit/src/SpanLifeCycle.Tests.ts new file mode 100644 index 000000000..34cc169b5 --- /dev/null +++ b/AISKU/Tests/Unit/src/SpanLifeCycle.Tests.ts @@ -0,0 +1,655 @@ +import { AITestClass, Assert } from '@microsoft/ai-test-framework'; +import { ApplicationInsights } from '../../../src/applicationinsights-web'; +import { eOTelSpanStatusCode, ITelemetryItem } from "@microsoft/applicationinsights-core-js"; + +export class SpanLifeCycleTests extends AITestClass { + private static readonly _instrumentationKey = 'b7170927-2d1c-44f1-acec-59f4e1751c11'; + private static readonly _connectionString = `InstrumentationKey=${SpanLifeCycleTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "SpanLifeCycleTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: SpanLifeCycleTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + + } catch (e) { + console.error('Failed to initialize tests: ' + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addDoubleEndTests(); + this.addOperationsOnEndedSpansTests(); + this.addEndedPropertyTests(); + this.addIsRecordingAfterEndTests(); + this.addEndTimeTests(); + } + + private addDoubleEndTests(): void { + this.testCase({ + name: "DoubleEnd: calling end() twice should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("double-end-test"); + + // Act & Assert - First end should succeed + Assert.doesNotThrow(() => { + span?.end(); + }, "First end() should not throw"); + + // Second end should not throw but should be no-op + Assert.doesNotThrow(() => { + span?.end(); + }, "Second end() should not throw"); + } + }); + + this.testCase({ + name: "DoubleEnd: second end() should be no-op", + test: () => { + // Arrange + const span = this._ai.startSpan("double-end-noop"); + this._trackCalls = []; + + // Act - End twice + span?.end(); + const trackCountAfterFirst = this._trackCalls.length; + + span?.end(); + const trackCountAfterSecond = this._trackCalls.length; + + // Assert - Second end should not generate additional telemetry + Assert.equal(trackCountAfterSecond, trackCountAfterFirst, + "Second end() should not generate additional telemetry"); + } + }); + + this.testCase({ + name: "DoubleEnd: ended property remains true after second end", + test: () => { + // Arrange + const span = this._ai.startSpan("double-end-property"); + + // Act + span?.end(); + const endedAfterFirst = span?.ended; + + span?.end(); + const endedAfterSecond = span?.ended; + + // Assert + Assert.ok(endedAfterFirst, "Span should be ended after first end()"); + Assert.ok(endedAfterSecond, "Span should remain ended after second end()"); + } + }); + + this.testCase({ + name: "DoubleEnd: multiple end() calls are safe", + test: () => { + // Arrange + const span = this._ai.startSpan("multiple-end-test"); + + // Act & Assert - Multiple ends should all be safe + Assert.doesNotThrow(() => { + for (let i = 0; i < 10; i++) { + span?.end(); + } + }, "Multiple end() calls should not throw"); + + Assert.ok(span?.ended, "Span should be marked as ended"); + } + }); + + this.testCase({ + name: "DoubleEnd: end with different times only uses first", + test: () => { + // Arrange + const span = this._ai.startSpan("double-end-time"); + + // Act - End with specific time + const firstEndTime = Date.now(); + span?.end(firstEndTime); + const capturedEndTime1 = span?.endTime; + + // Try to end again with different time + const secondEndTime = Date.now() + 1000; + span?.end(secondEndTime); + const capturedEndTime2 = span?.endTime; + + // Assert - End time should not change + Assert.deepEqual(capturedEndTime1, capturedEndTime2, + "End time should not change on second end()"); + } + }); + } + + private addOperationsOnEndedSpansTests(): void { + this.testCase({ + name: "EndedSpan: setAttribute on ended span should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("ended-setAttribute"); + span?.end(); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttribute("after.end", "value"); + }, "setAttribute should not throw on ended span"); + } + }); + + this.testCase({ + name: "EndedSpan: setAttribute on ended span should be no-op", + test: () => { + // Arrange + const span = this._ai.startSpan("ended-setAttribute-noop"); + span?.setAttribute("before.end", "initialValue"); + + const attributesBeforeEnd = span?.attributes; + span?.end(); + + // Act + span?.setAttribute("after.end", "newValue"); + span?.setAttribute("before.end", "modifiedValue"); + + // Assert + const attributesAfterEnd = span?.attributes; + Assert.ok(!attributesAfterEnd["after.end"], + "New attribute should not be added after end"); + Assert.equal(attributesAfterEnd["before.end"], "initialValue", + "Existing attribute should not be modified after end"); + } + }); + + this.testCase({ + name: "EndedSpan: setAttributes on ended span should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("ended-setAttributes"); + span?.end(); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setAttributes({ + "attr1": "value1", + "attr2": "value2" + }); + }, "setAttributes should not throw on ended span"); + } + }); + + this.testCase({ + name: "EndedSpan: setAttributes on ended span should be no-op", + test: () => { + // Arrange + const span = this._ai.startSpan("ended-setAttributes-noop"); + span?.setAttributes({ "initial": "value" }); + span?.end(); + + // Act + span?.setAttributes({ + "after.end.1": "value1", + "after.end.2": "value2" + }); + + // Assert + const attributes = span?.attributes; + Assert.ok(!attributes["after.end.1"], + "Attributes should not be added after end"); + Assert.ok(!attributes["after.end.2"], + "Attributes should not be added after end"); + } + }); + + this.testCase({ + name: "EndedSpan: setStatus on ended span should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("ended-setStatus"); + span?.end(); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.setStatus({ + code: eOTelSpanStatusCode.ERROR, + message: "Error after end" + }); + }, "setStatus should not throw on ended span"); + } + }); + + this.testCase({ + name: "EndedSpan: setStatus on ended span should be no-op", + test: () => { + // Arrange + const span = this._ai.startSpan("ended-setStatus-noop"); + span?.setStatus({ + code: eOTelSpanStatusCode.OK, + message: "Initial status" + }); + + const statusBeforeEnd = span?.status; + span?.end(); + + // Act + span?.setStatus({ + code: eOTelSpanStatusCode.ERROR, + message: "Modified after end" + }); + + // Assert + const statusAfterEnd = span?.status; + Assert.equal(statusAfterEnd.code, statusBeforeEnd?.code, + "Status code should not change after end"); + } + }); + + this.testCase({ + name: "EndedSpan: updateName on ended span should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("original-name"); + span?.end(); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.updateName("new-name-after-end"); + }, "updateName should not throw on ended span"); + } + }); + + this.testCase({ + name: "EndedSpan: updateName on ended span should be no-op", + test: () => { + // Arrange + const originalName = "original-name-noop"; + const span = this._ai.startSpan(originalName); + span?.end(); + + // Act + span?.updateName("modified-name"); + + // Assert + Assert.equal(span?.name, originalName, + "Span name should not change after end"); + } + }); + + this.testCase({ + name: "EndedSpan: recordException on ended span should not throw", + test: () => { + // Arrange + const span = this._ai.startSpan("ended-recordException"); + span?.end(); + + // Act & Assert + Assert.doesNotThrow(() => { + span?.recordException(new Error("Exception after end")); + }, "recordException should not throw on ended span"); + } + }); + + this.testCase({ + name: "EndedSpan: multiple operations on ended span should all be safe", + test: () => { + // Arrange + const span = this._ai.startSpan("ended-multiple-ops"); + span?.end(); + + // Act & Assert - All operations should be safe + Assert.doesNotThrow(() => { + span?.setAttribute("key", "value"); + span?.setAttributes({ "key1": "val1", "key2": "val2" }); + span?.setStatus({ code: eOTelSpanStatusCode.ERROR }); + span?.updateName("new-name"); + span?.recordException(new Error("test")); + span?.end(); // Try to end again + }, "Multiple operations on ended span should not throw"); + } + }); + } + + private addEndedPropertyTests(): void { + this.testCase({ + name: "EndedProperty: span should not be ended initially", + test: () => { + // Arrange & Act + const span = this._ai.startSpan("initial-not-ended"); + + // Assert + Assert.ok(!span?.ended, "Span should not be ended initially"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "EndedProperty: span should be ended after end() call", + test: () => { + // Arrange + const span = this._ai.startSpan("ended-after-call"); + + // Act + span?.end(); + + // Assert + Assert.ok(span?.ended, "Span should be ended after end() call"); + } + }); + + this.testCase({ + name: "EndedProperty: ended property is read-only", + test: () => { + // Arrange + const span = this._ai.startSpan("readonly-ended") as any; + + // Act - Try to modify ended property + const canModify = () => { + try { + span.ended = true; + return true; + } catch (e) { + return false; + } + }; + + // Assert + Assert.ok(!span.ended, "Should start not ended"); + // Property should be read-only (or modification has no effect) + canModify(); + Assert.ok(!span.ended, "Manual modification should not affect ended state"); + + // Cleanup + span.end(); + } + }); + + this.testCase({ + name: "EndedProperty: ended state persists across property reads", + test: () => { + // Arrange + const span = this._ai.startSpan("persistent-ended"); + span?.end(); + + // Act - Read ended property multiple times + const ended1 = span?.ended; + const ended2 = span?.ended; + const ended3 = span?.ended; + + // Assert + Assert.ok(ended1, "First read should show ended"); + Assert.ok(ended2, "Second read should show ended"); + Assert.ok(ended3, "Third read should show ended"); + } + }); + + this.testCase({ + name: "EndedProperty: recording and non-recording spans both have ended property", + test: () => { + // Arrange + const recordingSpan = this._ai.startSpan("recording", { recording: true }); + const nonRecordingSpan = this._ai.startSpan("non-recording", { recording: false }); + + // Act + recordingSpan?.end(); + nonRecordingSpan?.end(); + + // Assert + Assert.ok(recordingSpan?.ended, "Recording span should be ended"); + Assert.ok(nonRecordingSpan?.ended, "Non-recording span should be ended"); + } + }); + } + + private addIsRecordingAfterEndTests(): void { + this.testCase({ + name: "IsRecording: isRecording() returns false after end()", + test: () => { + // Arrange + const span = this._ai.startSpan("recording-test"); + const isRecordingBefore = span?.isRecording(); + + // Act + span?.end(); + const isRecordingAfter = span?.isRecording(); + + // Assert + Assert.ok(isRecordingBefore, "Span should be recording before end"); + Assert.ok(!isRecordingAfter, "Span should not be recording after end"); + } + }); + + this.testCase({ + name: "IsRecording: non-recording span stays non-recording after end", + test: () => { + // Arrange + const span = this._ai.startSpan("non-recording-test", { recording: false }); + const isRecordingBefore = span?.isRecording(); + + // Act + span?.end(); + const isRecordingAfter = span?.isRecording(); + + // Assert + Assert.ok(!isRecordingBefore, "Non-recording span should not be recording before end"); + Assert.ok(!isRecordingAfter, "Non-recording span should not be recording after end"); + } + }); + + this.testCase({ + name: "IsRecording: isRecording() consistent with ended state", + test: () => { + // Arrange + const span = this._ai.startSpan("recording-consistency"); + + // Assert initial state + Assert.ok(span?.isRecording(), "Should be recording when not ended"); + Assert.ok(!span?.ended, "Should not be ended initially"); + + // Act + span?.end(); + + // Assert final state + Assert.ok(!span?.isRecording(), "Should not be recording when ended"); + Assert.ok(span?.ended, "Should be ended after end()"); + } + }); + + this.testCase({ + name: "IsRecording: multiple isRecording() calls after end return consistent value", + test: () => { + // Arrange + const span = this._ai.startSpan("recording-multiple-calls"); + span?.end(); + + // Act + const check1 = span?.isRecording(); + const check2 = span?.isRecording(); + const check3 = span?.isRecording(); + + // Assert + Assert.ok(!check1, "First check should return false"); + Assert.ok(!check2, "Second check should return false"); + Assert.ok(!check3, "Third check should return false"); + } + }); + } + + private addEndTimeTests(): void { + this.testCase({ + name: "EndTime: endTime is undefined before end()", + test: () => { + // Arrange & Act + const span = this._ai.startSpan("endtime-undefined"); + const endTime = span?.endTime; + + // Assert + Assert.ok(endTime === undefined || endTime === null, + "endTime should be undefined/null before end()"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "EndTime: endTime is set after end()", + test: () => { + // Arrange + const span = this._ai.startSpan("endtime-set"); + + // Act + span?.end(); + const endTime = span?.endTime; + + // Assert + Assert.ok(endTime !== undefined && endTime !== null, + "endTime should be set after end()"); + } + }); + + this.testCase({ + name: "EndTime: endTime is after startTime", + test: () => { + // Arrange + const span = this._ai.startSpan("endtime-after-start"); + + // Act + span?.end(); + + // Assert + const startTime = span?.startTime; + const endTime = span?.endTime; + + if (startTime && endTime) { + // Compare HrTime [seconds, nanoseconds] + const startMs = startTime[0] * 1000 + startTime[1] / 1000000; + const endMs = endTime[0] * 1000 + endTime[1] / 1000000; + + Assert.ok(endMs >= startMs, + "endTime should be after or equal to startTime"); + } + } + }); + + this.testCase({ + name: "EndTime: custom endTime is respected", + test: () => { + // Arrange + const span = this._ai.startSpan("endtime-custom"); + const customEndTime = Date.now(); + + // Act + span?.end(customEndTime); + const actualEndTime = span?.endTime; + + // Assert + if (actualEndTime) { + const actualMs = actualEndTime[0] * 1000 + actualEndTime[1] / 1000000; + const diff = Math.abs(actualMs - customEndTime); + + Assert.ok(diff < 10, // Allow 10ms difference for conversion + "Custom endTime should be approximately respected"); + } + } + }); + + this.testCase({ + name: "EndTime: duration is calculated from startTime to endTime", + test: () => { + // Arrange + const span = this._ai.startSpan("duration-calculation"); + + // Act - Add small delay + const startTime = Date.now(); + for (let i = 0; i < 1000; i++) { + // Small busy loop + } + span?.end(); + + // Assert + const duration = span?.duration; + if (duration) { + const durationMs = duration[0] * 1000 + duration[1] / 1000000; + Assert.ok(durationMs >= 0, "Duration should be non-negative"); + } + } + }); + + this.testCase({ + name: "EndTime: endTime does not change after span is ended", + test: () => { + // Arrange + const span = this._ai.startSpan("endtime-immutable"); + + // Act + span?.end(); + const endTime1 = span?.endTime; + + // Try to end again (should be no-op) + span?.end(); + const endTime2 = span?.endTime; + + // Assert + Assert.deepEqual(endTime1, endTime2, + "endTime should not change after first end()"); + } + }); + + this.testCase({ + name: "EndTime: negative duration is handled gracefully", + test: () => { + // Arrange + const span = this._ai.startSpan("negative-duration"); + + // Act - End with time before start + const futureTime = Date.now() + 10000; + span?.end(); + + // Try to set past end time after span started + // (Note: SDK should handle this internally and prevent negative duration) + + // Assert + const duration = span?.duration; + if (duration) { + const durationMs = duration[0] * 1000 + duration[1] / 1000000; + Assert.ok(durationMs >= 0, + "Duration should never be negative (SDK should handle this)"); + } + } + }); + } +} diff --git a/AISKU/Tests/Unit/src/SpanPluginIntegration.Tests.ts b/AISKU/Tests/Unit/src/SpanPluginIntegration.Tests.ts new file mode 100644 index 000000000..35b5fb077 --- /dev/null +++ b/AISKU/Tests/Unit/src/SpanPluginIntegration.Tests.ts @@ -0,0 +1,1020 @@ +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { ApplicationInsights } from "../../../src/applicationinsights-web"; +import { eOTelSpanKind, eOTelSpanStatusCode, isTracingSuppressed, ITelemetryItem } from "@microsoft/applicationinsights-core-js"; +import { setBypassLazyCache } from "@nevware21/ts-utils"; + +/** + * Integration Tests for Span APIs with Properties Plugin and Analytics Plugin + * + * Tests verify that span telemetry correctly integrates with: + * - PropertiesPlugin: session, user, device, application context + * - AnalyticsPlugin: telemetry creation, dependency tracking, page views + * - Telemetry Initializers: custom property injection + * - SDK configuration: sampling, disabled tracking, etc. + */ +export class SpanPluginIntegrationTests extends AITestClass { + private _ai!: ApplicationInsights; + + constructor(testName?: string) { + super(testName || "SpanPluginIntegrationTests"); + } + + public testInitialize() { + try { + setBypassLazyCache(true); + this.useFakeServer = true; + + this._ai = new ApplicationInsights({ + config: { + instrumentationKey: "test-ikey-123", + disableInstrumentationKeyValidation: true, + disableAjaxTracking: false, + disableXhr: false, + disableFetchTracking: false, + enableAutoRouteTracking: false, + disableExceptionTracking: false, + maxBatchInterval: 100, + enableDebug: false, + extensionConfig: { + ["AppInsightsPropertiesPlugin"]: { + accountId: "test-account-id" + } + } + } + }); + + this._ai.loadAppInsights(); + } catch (e) { + Assert.ok(false, "Failed to initialize tests: " + e); + console.error("Failed to initialize tests: " + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + setBypassLazyCache(false); + } + + public registerTests() { + this.addPropertiesPluginIntegrationTests(); + this.addAnalyticsPluginIntegrationTests(); + this.addTelemetryInitializerTests(); + this.addSessionContextTests(); + this.addUserContextTests(); + this.addDeviceContextTests(); + this.addDistributedTraceContextTests(); + this.addSamplingIntegrationTests(); + this.addConfigurationIntegrationTests(); + } + + private addPropertiesPluginIntegrationTests(): void { + this.testCase({ + name: "PropertiesPlugin: span telemetry includes session context", + useFakeTimers: true, + useFakeServer: true, + test: () => { + const span = this._ai.startSpan("test-operation", { + kind: eOTelSpanKind.CLIENT + }); + Assert.ok(span, "Span should be created"); + + Assert.equal(false, isTracingSuppressed(span), "Span should not be suppressed"); + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + span!.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.equal(1, sentItems.length, "Telemetry should be sent"); + + const payload = sentItems[0]; + Assert.ok(payload, "Payload should exist"); + Assert.ok(payload.tags, "Payload should have tags"); + + // Session ID is sent in tags with key "ai.session.id" + const sessionId = payload.tags["ai.session.id"]; + Assert.ok(sessionId, "Session ID should be in tags"); + Assert.ok(sessionId.length > 0, "Session ID should not be empty"); + } + }); + + this.testCase({ + name: "PropertiesPlugin: span telemetry includes user context", + useFakeTimers: true, + test: () => { + // Set user context before creating span + this._ai.context.user.authenticatedId = "test-auth-user-123"; + this._ai.context.user.accountId = "test-account-456"; + + const span = this._ai.startSpan("user-operation", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span, "Span should be created"); + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + span!.setAttribute("custom.prop", "value"); + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + span!.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.equal(1, sentItems.length, "Telemetry should be sent"); + + const payload = sentItems[0]; + Assert.ok(payload, "Payload should exist"); + Assert.ok(payload.tags, "Payload should have tags"); + + // User auth ID is sent in tags with key "ai.user.authUserId" + const authUserId = payload.tags["ai.user.authUserId"]; + Assert.equal(authUserId, "test-auth-user-123", "Authenticated ID should match"); + + // Account ID is sent in tags with key "ai.user.accountId" + const accountId = payload.tags["ai.user.accountId"]; + Assert.equal(accountId, "test-account-456", "Account ID should be in tags"); + } + }); + + this.testCase({ + name: "PropertiesPlugin: span telemetry includes device context", + useFakeTimers: true, + test: () => { + const span = this._ai.startSpan("device-operation", { + kind: eOTelSpanKind.CLIENT + }); + Assert.ok(span, "Span should be created"); + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + span!.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.equal(1, sentItems.length, "Telemetry should be sent"); + const payload = sentItems[0]; + Assert.ok(payload, "Payload should exist"); + Assert.ok(payload.tags, "Payload should have tags"); + + // Device info is sent in tags with keys "ai.device.type" and "ai.device.id" + const deviceType = payload.tags["ai.device.type"]; + Assert.equal(deviceType, "Browser", "Device type should be Browser"); + + const deviceId = payload.tags["ai.device.id"]; + Assert.equal(deviceId, "browser", "Device ID should be browser"); + } + }); + + this.testCase({ + name: "PropertiesPlugin: span telemetry includes SDK version from internal context", + useFakeTimers: true, + test: () => { + const span = this._ai.startSpan("sdk-version-check", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span, "Span should be created"); + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + span!.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.equal(1, sentItems.length, "Telemetry should be sent"); + const payload = sentItems[0]; + Assert.ok(payload, "Payload should exist"); + Assert.ok(payload.tags, "Payload should have tags"); + + // SDK version is sent in tags with key "ai.internal.sdkVersion" + const sdkVersion = payload.tags["ai.internal.sdkVersion"]; + Assert.ok(sdkVersion, "SDK version should exist"); + Assert.ok(sdkVersion.indexOf("javascript") >= 0 || sdkVersion.indexOf("ext1") >= 0, + "SDK version should contain javascript or extension prefix"); + } + }); + + this.testCase({ + name: "PropertiesPlugin: web context applied to span telemetry", + useFakeTimers: true, + test: () => { + const span = this._ai.startSpan("web-context-operation", { + kind: eOTelSpanKind.SERVER + }); + Assert.ok(span, "Span should be created"); + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + span!.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.equal(1, sentItems.length, "Telemetry should be sent"); + + const payload = sentItems[0]; + Assert.ok(payload, "Payload should exist"); + + // Web context info like browser is sent in data section or tags + // Just verify the payload was sent successfully with telemetry + Assert.ok(payload.data, "Payload should have data section"); + } + }); + } + + private addAnalyticsPluginIntegrationTests(): void { + this.testCase({ + name: "AnalyticsPlugin: CLIENT span creates RemoteDependencyData", + useFakeTimers: true, + test: () => { + const span = this._ai.startSpan("http-request", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "GET", + "http.url": "https://api.example.com/data", + "http.status_code": 200 + } + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + + span.setStatus({ code: eOTelSpanStatusCode.OK }); + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.ok(sentItems.length > 0, "Telemetry should be sent"); + + const item = sentItems[0] as ITelemetryItem; + Assert.ok(item.data, "Data should exist"); + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + Assert.equal(item.data!.baseType, "RemoteDependencyData", "BaseType should be RemoteDependencyData"); + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + Assert.ok(item.data!.baseData, "BaseData should exist"); + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + Assert.equal(item.data!.baseData.name, "GET /data", "Name should match span name"); + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + Assert.equal(item.data!.baseData.success, true, "Success should be true for OK status"); + } + }); + + this.testCase({ + name: "AnalyticsPlugin: PRODUCER span creates RemoteDependencyData with message type", + useFakeTimers: true, + test: () => { + const span = this._ai.startSpan("send-message", { + kind: eOTelSpanKind.PRODUCER, + attributes: { + "messaging.system": "rabbitmq", + "messaging.destination": "orders-queue", + "messaging.operation": "send" + } + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.ok(sentItems.length > 0, "Telemetry should be sent"); + + const item = sentItems[0] as ITelemetryItem; + Assert.ok(item.data, "Data should exist"); + if (!item.data) { + return; + } + Assert.equal(item.data.baseType, "RemoteDependencyData", "BaseType should be RemoteDependencyData"); + Assert.ok(item.data.baseData, "BaseData should exist"); + if (!item.data.baseData) { + return; + } + Assert.ok(item.data.baseData.type, "Type should be set for message dependency"); + } + }); + + this.testCase({ + name: "AnalyticsPlugin: custom properties merged into baseData", + useFakeTimers: true, + test: () => { + const span = this._ai.startSpan("operation-with-props", { + kind: eOTelSpanKind.INTERNAL, + attributes: { + "custom.string": "value", + "custom.number": 42, + "custom.boolean": true + } + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + + span.setAttribute("runtime.prop", "added-after-start"); + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.ok(sentItems.length > 0, "Telemetry should be sent"); + + const item = sentItems[0] as ITelemetryItem; + if (!item.data) { + return; + } + Assert.ok(item.data.baseData, "BaseData should exist"); + if (!item.data.baseData) { + return; + } + + // Custom properties should be in properties object + if (item.data.baseData.properties) { + Assert.equal(item.data.baseData.properties["custom.string"], "value", "String property should be preserved"); + Assert.equal(item.data.baseData.properties["custom.number"], 42, "Number property should be preserved"); + Assert.equal(item.data.baseData.properties["custom.boolean"], "true", "Boolean property should be preserved"); + Assert.equal(item.data.baseData.properties["runtime.prop"], "added-after-start", + "Runtime-added property should be present"); + } + } + }); + + this.testCase({ + name: "AnalyticsPlugin: span duration calculated correctly", + useFakeTimers: true, + test: () => { + const span = this._ai.startSpan("timed-operation", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + + // Simulate some time passing + this.clock.tick(250); + + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.ok(sentItems.length > 0, "Telemetry should be sent"); + + const item = sentItems[0] as ITelemetryItem; + if (!item.data) { + return; + } + Assert.ok(item.data.baseData, "BaseData should exist"); + if (!item.data.baseData) { + return; + } + Assert.equal(item.data.baseData.name, "timed-operation", "Name should match span name"); + Assert.ok(item.data.baseData.duration, "Duration should exist"); + + // Duration should be approximately 250ms (formatted as time span string) + const durationMs = this.parseDurationToMs(item.data.baseData.duration); + Assert.ok(durationMs >= 240 && durationMs <= 260, + "Duration should be ~250ms, got " + durationMs + "ms - " + JSON.stringify(item)); + } + }); + + this.testCase({ + name: "AnalyticsPlugin: failed span sets success=false", + useFakeTimers: true, + test: () => { + const span = this._ai.startSpan("failing-operation", { + kind: eOTelSpanKind.CLIENT + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + + span.setStatus({ + code: eOTelSpanStatusCode.ERROR, + message: "Operation failed" + }); + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.ok(sentItems.length > 0, "Telemetry should be sent"); + + const item = sentItems[0] as ITelemetryItem; + if (!item.data) { + return; + } + Assert.ok(item.data.baseData, "BaseData should exist"); + if (!item.data.baseData) { + return; + } + Assert.equal(item.data.baseData.success, false, "Success should be false for ERROR status"); + } + }); + } + + private addTelemetryInitializerTests(): void { + this.testCase({ + name: "TelemetryInitializer: can modify span telemetry", + useFakeTimers: true, + test: () => { + let initializerCalled = false; + + this._ai.addTelemetryInitializer((item: ITelemetryItem) => { + initializerCalled = true; + + if (item.baseType === "RemoteDependencyData") { + // Add custom property via initializer + item.baseData = item.baseData || {}; + item.baseData.properties = item.baseData.properties || {}; + item.baseData.properties["initializer.added"] = "custom-value"; + item.baseData.properties["initializer.timestamp"] = new Date().toISOString(); + } + + return true; + }); + + const span = this._ai.startSpan("initialized-span", { + kind: eOTelSpanKind.CLIENT + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + + span.end(); + this.clock.tick(500); + + Assert.ok(initializerCalled, "Telemetry initializer should be called"); + + const sentItems = this.getSentTelemetry(); + Assert.ok(sentItems.length > 0, "Telemetry should be sent"); + + const item = sentItems[0] as ITelemetryItem; + if (!item.data) { + return; + } + if (!item.data.baseData) { + return; + } + Assert.ok(item.data.baseData.properties, "Properties should exist"); + Assert.equal(item.data.baseData.properties["initializer.added"], "custom-value", + "Initializer-added property should be present"); + Assert.ok(item.data.baseData.properties["initializer.timestamp"], + "Timestamp should be added by initializer"); + } + }); + + this.testCase({ + name: "TelemetryInitializer: can filter span telemetry", + useFakeTimers: true, + test: () => { + this._ai.addTelemetryInitializer((item: ITelemetryItem) => { + // Filter out spans with specific attribute + if (item.baseType === "RemoteDependencyData" && + item.baseData && + item.baseData.properties && + item.baseData.properties["filter.me"] === "true") { + return false; // Reject this telemetry + } + return true; + }); + + // This span should be filtered out + const filteredSpan = this._ai.startSpan("filtered-span", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "filter.me": "true" + } + }); + Assert.ok(filteredSpan, "Filtered span should be created"); + if (filteredSpan) { + filteredSpan.end(); + } + + // This span should go through + const normalSpan = this._ai.startSpan("normal-span", { + kind: eOTelSpanKind.CLIENT + }); + Assert.ok(normalSpan, "Normal span should be created"); + if (normalSpan) { + normalSpan.end(); + } + + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.equal(sentItems.length, 1, "Only one span should be sent (filtered one rejected)"); + + const item = sentItems[0] as ITelemetryItem; + if (item.data && item.data.baseData) { + Assert.equal(item.data.baseData.name, "normal-span", "Only normal span should be sent"); + } + } + }); + + this.testCase({ + name: "TelemetryInitializer: can enrich with context data", + useFakeTimers: true, + test: () => { + this._ai.addTelemetryInitializer((item: ITelemetryItem) => { + // Add environment and build info to all span telemetry + if (item.baseType === "RemoteDependencyData") { + item.baseData = item.baseData || {}; + item.baseData.properties = item.baseData.properties || {}; + item.baseData.properties["environment"] = "test"; + item.baseData.properties["build.version"] = "1.2.3"; + item.baseData.properties["region"] = "us-west"; + } + return true; + }); + + const span = this._ai.startSpan("enriched-span", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + const item = sentItems[0] as ITelemetryItem; + if (!item.data) { + return; + } + if (!item.data.baseData) { + return; + } + + Assert.equal(item.data.baseData.properties["environment"], "test", "Environment should be added"); + Assert.equal(item.data.baseData.properties["build.version"], "1.2.3", "Build version should be added"); + Assert.equal(item.data.baseData.properties["region"], "us-west", "Region should be added"); + } + }); + } + + private addSessionContextTests(): void { + this.testCase({ + name: "SessionContext: consistent session ID across multiple spans", + useFakeTimers: true, + test: () => { + const span1 = this._ai.startSpan("operation-1", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span1, "First span should be created"); + if (span1) { + span1.end(); + } + + const span2 = this._ai.startSpan("operation-2", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span2, "Second span should be created"); + if (span2) { + span2.end(); + } + + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.equal(sentItems.length, 2, "Two telemetry items should be sent"); + + const payload1 = sentItems[0]; + const payload2 = sentItems[1]; + + const sessionId1 = payload1.tags ? payload1.tags["ai.session.id"] : undefined; + const sessionId2 = payload2.tags ? payload2.tags["ai.session.id"] : undefined; + + Assert.ok(sessionId1, "First item should have session ID"); + Assert.ok(sessionId2, "Second item should have session ID"); + Assert.equal(sessionId1, sessionId2, "Session IDs should be consistent"); + } + }); + + this.testCase({ + name: "SessionContext: session renewal doesn't affect active spans", + useFakeTimers: true, + test: () => { + const span1 = this._ai.startSpan("before-renewal", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span1, "Span before renewal should be created"); + if (span1) { + span1.end(); + } + this.clock.tick(500); + + // Simulate session renewal time passing (30+ minutes) + this.clock.tick(31 * 60 * 1000); + + const span2 = this._ai.startSpan("after-renewal", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span2, "Span after renewal should be created"); + if (span2) { + span2.end(); + } + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.equal(sentItems.length, 2, "Both spans should be sent"); + + // Session might have renewed, but both spans should have valid session IDs + const payload1 = sentItems[0]; + const payload2 = sentItems[1]; + + const sessionId1 = payload1.tags ? payload1.tags["ai.session.id"] : undefined; + const sessionId2 = payload2.tags ? payload2.tags["ai.session.id"] : undefined; + + Assert.ok(sessionId1, "First span should have session ID"); + Assert.ok(sessionId2, "Second span should have session ID"); + } + }); + } + + private addUserContextTests(): void { + this.testCase({ + name: "UserContext: setting user ID applies to subsequent spans", + useFakeTimers: true, + test: () => { + // Set user context + this._ai.context.user.id = "user-12345"; + this._ai.context.user.authenticatedId = "auth-user-67890"; + + const span = this._ai.startSpan("user-operation", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + const payload = sentItems[0]; + + Assert.ok(payload.tags, "Payload should have tags"); + + // User ID is sent in tags with key "ai.user.id" + const userId = payload.tags["ai.user.id"]; + Assert.equal(userId, "user-12345", "User ID should match"); + + // Auth user ID is sent in tags with key "ai.user.authUserId" + const authUserId = payload.tags["ai.user.authUserId"]; + Assert.equal(authUserId, "auth-user-67890", "Authenticated ID should match"); + } + }); + + this.testCase({ + name: "UserContext: clearing user context removes from spans", + useFakeTimers: true, + test: () => { + // Set then clear + this._ai.context.user.authenticatedId = "temp-user"; + this._ai.context.user.clearAuthenticatedUserContext(); + + const span = this._ai.startSpan("after-clear", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + const payload = sentItems[0]; + + // User context should still exist but authenticated ID should be undefined/missing + if (payload.tags) { + const authUserId = payload.tags["ai.user.authUserId"]; + Assert.ok(!authUserId || authUserId === undefined, + "Authenticated ID should be cleared"); + } + } + }); + } + + private addDeviceContextTests(): void { + this.testCase({ + name: "DeviceContext: device information included in all spans", + useFakeTimers: true, + test: () => { + const span = this._ai.startSpan("device-check", { + kind: eOTelSpanKind.CLIENT + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + const payload = sentItems[0]; + + Assert.ok(payload.tags, "Payload should have tags"); + + // Device info is sent in tags + const deviceType = payload.tags["ai.device.type"]; + const deviceId = payload.tags["ai.device.id"]; + + Assert.ok(deviceType, "Device type should be set"); + Assert.ok(deviceId, "Device ID should be set"); + } + }); + } + + private addDistributedTraceContextTests(): void { + this.testCase({ + name: "DistributedTrace: parent-child spans share trace ID", + useFakeTimers: true, + test: () => { + const parentSpan = this._ai.startSpan("parent-op", { + kind: eOTelSpanKind.SERVER + }); + Assert.ok(parentSpan, "Parent span should be created"); + if (!parentSpan) { + return; + } + + const childSpan = this._ai.startSpan("child-op", { + kind: eOTelSpanKind.CLIENT + }); + Assert.ok(childSpan, "Child span should be created"); + + parentSpan.end(); + if (childSpan) { + childSpan.end(); + } + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.equal(sentItems.length, 2, "Both spans should be sent"); + + const parentPayload = sentItems[0]; + const childPayload = sentItems[1]; + + // Both should have same operation ID (trace ID) in tags + const parentOpId = parentPayload.tags ? parentPayload.tags["ai.operation.id"] : undefined; + const childOpId = childPayload.tags ? childPayload.tags["ai.operation.id"] : undefined; + + Assert.ok(parentOpId, "Parent should have operation ID"); + Assert.ok(childOpId, "Child should have operation ID"); + Assert.equal(parentOpId, childOpId, "Trace IDs should match for parent and child"); + } + }); + + this.testCase({ + name: "DistributedTrace: span context propagates through telemetry", + useFakeTimers: true, + test: () => { + const span = this._ai.startSpan("traced-operation", { + kind: eOTelSpanKind.CLIENT + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + + const spanContext = span.spanContext(); + Assert.ok(spanContext.traceId, "Span should have trace ID"); + Assert.ok(spanContext.spanId, "Span should have span ID"); + + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + const payload = sentItems[0]; + + // Trace context should be in tags + if (payload.tags) { + const operationId = payload.tags["ai.operation.id"]; + const operationParentId = payload.tags["ai.operation.parentId"]; + + Assert.equal(operationId, spanContext.traceId, "Operation ID should match trace ID"); + Assert.ok(operationParentId, "Operation parent ID should be set"); + } + } + }); + } + + private addSamplingIntegrationTests(): void { + this.testCase({ + name: "Sampling: 1% sampling allows minimal span telemetry", + useFakeTimers: true, + test: () => { + // Recreate AI with 1% sampling (minimum valid value) + this._ai.unload(false); + + this._ai = new ApplicationInsights({ + config: { + instrumentationKey: "test-ikey-123", + samplingPercentage: 1 + } + }); + this._ai.loadAppInsights(); + + const span = this._ai.startSpan("low-sampled", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span, "Span should still be created"); + if (span) { + span.end(); + } + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + // With 1% sampling, telemetry may or may not be sent (depends on sample hash) + Assert.ok(sentItems.length >= 0, "Telemetry should respect 1% sampling rate"); + } + }); + + this.testCase({ + name: "Sampling: 100% sampling sends all span telemetry", + useFakeTimers: true, + test: () => { + // Default config has 100% sampling + const spans = []; + for (let i = 0; i < 10; i++) { + const span = this._ai.startSpan("operation-" + i, { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span, "Span " + i + " should be created"); + if (span) { + span.end(); + spans.push(span); + } + } + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.equal(sentItems.length, 10, "All 10 spans should be sent with 100% sampling"); + } + }); + } + + private addConfigurationIntegrationTests(): void { + this.testCase({ + name: "Config: disableAjaxTracking doesn't affect manual spans", + useFakeTimers: true, + test: () => { + // Config already has disableAjaxTracking: false, but manual spans should work regardless + const span = this._ai.startSpan("manual-span", { + kind: eOTelSpanKind.CLIENT + }); + Assert.ok(span, "Manual span should be created"); + if (!span) { + return; + } + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.ok(sentItems.length > 0, "Manual span should be sent regardless of ajax tracking config"); + } + }); + + this.testCase({ + name: "Config: maxBatchInterval affects when span telemetry is sent", + useFakeTimers: true, + test: () => { + // Current config has maxBatchInterval: 0 (send immediately) + const span = this._ai.startSpan("immediate-send", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + span.end(); + + // No tick needed with maxBatchInterval: 0 + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + Assert.ok(sentItems.length > 0, "Span should be sent immediately"); + } + }); + + this.testCase({ + name: "Config: extensionConfig reaches PropertiesPlugin", + useFakeTimers: true, + test: () => { + // We set accountId in extensionConfig during init + const span = this._ai.startSpan("config-test", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span, "Span should be created"); + if (!span) { + return; + } + span.end(); + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + const payload = sentItems[0]; + + // Check if account ID from config made it through tags + if (payload.tags) { + const accountId = payload.tags["ai.user.accountId"]; + if (accountId) { + Assert.equal(accountId, "test-account-id", + "Account ID from config should be present in tags"); + } + } + } + }); + + this.testCase({ + name: "Config: dynamic configuration changes affect new spans", + useFakeTimers: true, + test: () => { + // Create span with initial config + const span1 = this._ai.startSpan("before-config-change", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span1, "First span should be created"); + if (span1) { + span1.end(); + } + this.clock.tick(500); + + // Change configuration dynamically + this._ai.config.extensionConfig = this._ai.config.extensionConfig || {}; + this._ai.config.extensionConfig["AppInsightsChannelPlugin"] = + this._ai.config.extensionConfig["AppInsightsChannelPlugin"] || {}; + this._ai.config.extensionConfig["AppInsightsChannelPlugin"].samplingPercentage = 1; + this.clock.tick(500); // Allow config change to propagate + + // Create span after config change + const span2 = this._ai.startSpan("after-config-change", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(span2, "Second span should be created"); + if (span2) { + span2.end(); + } + this.clock.tick(500); + + const sentItems = this.getSentTelemetry(); + + // First span should be sent (100% default), second may be sampled (1%) + Assert.ok(sentItems.length >= 1, "At least first span should be sent"); + + const firstItem = sentItems[0] as ITelemetryItem; + if (firstItem.data && firstItem.data.baseData) { + Assert.equal(firstItem.data.baseData.name, "before-config-change", + "First span should be sent before config change"); + } + } + }); + } + + // Helper methods + private getSentTelemetry(): any[] { + const items: any[] = []; + const requests = this.activeXhrRequests; + if (requests) { + requests.forEach((request: any) => { + if (request.requestBody) { + try { + const payload = JSON.parse(request.requestBody); + if (payload && Array.isArray(payload)) { + items.push(...payload); + } else if (payload) { + items.push(payload); + } + } catch (e) { + // Ignore parse errors + } + } + }); + } + return items; + } + + private parseDurationToMs(duration: string): number { + // Duration format: "00:00:00.250" or similar + if (!duration) { + return 0; + } + + const parts = duration.split(":"); + if (parts.length !== 3) { + return 0; + } + + const hours = parseInt(parts[0], 10); + const minutes = parseInt(parts[1], 10); + const secondsParts = parts[2].split("."); + const seconds = parseInt(secondsParts[0], 10); + const milliseconds = secondsParts[1] ? parseInt(secondsParts[1].padEnd(3, "0").substring(0, 3), 10) : 0; + + return (hours * 3600000) + (minutes * 60000) + (seconds * 1000) + milliseconds; + } +} diff --git a/AISKU/Tests/Unit/src/SpanUtils.Tests.ts b/AISKU/Tests/Unit/src/SpanUtils.Tests.ts new file mode 100644 index 000000000..22ee416d7 --- /dev/null +++ b/AISKU/Tests/Unit/src/SpanUtils.Tests.ts @@ -0,0 +1,1971 @@ +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { ApplicationInsights, IDependencyTelemetry } from "../../../src/applicationinsights-web"; +import { + eOTelSpanKind, + eOTelSpanStatusCode, + ITelemetryItem, + SEMATTRS_HTTP_METHOD, + SEMATTRS_HTTP_URL, + SEMATTRS_HTTP_STATUS_CODE, + SEMATTRS_DB_SYSTEM, + SEMATTRS_DB_STATEMENT, + SEMATTRS_DB_NAME, + SEMATTRS_RPC_SYSTEM, + SEMATTRS_RPC_GRPC_STATUS_CODE, + ATTR_HTTP_REQUEST_METHOD, + ATTR_HTTP_RESPONSE_STATUS_CODE, + ATTR_URL_FULL, + ATTR_SERVER_ADDRESS, + ATTR_SERVER_PORT, + ATTR_ENDUSER_ID, + ATTR_ENDUSER_PSEUDO_ID, + ATTR_HTTP_ROUTE, + MicrosoftClientIp +} from "@microsoft/applicationinsights-core-js"; +import { IRequestTelemetry } from "@microsoft/applicationinsights-core-js"; + +export class SpanUtilsTests extends AITestClass { + private static readonly _instrumentationKey = "b7170927-2d1c-44f1-acec-59f4e1751c11"; + private static readonly _connectionString = `InstrumentationKey=${SpanUtilsTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "SpanUtilsTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: SpanUtilsTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + } catch (e) { + console.error("Failed to initialize tests: " + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addDependencyTelemetryTests(); + this.addRequestTelemetryTests(); + this.addHttpDependencyTests(); + this.addDbDependencyTests(); + this.addRpcDependencyTests(); + this.addAttributeMappingTests(); + this.addTagsCreationTests(); + this.addAzureSDKTests(); + this.addSemanticAttributeExclusionTests(); + this.addEdgeCaseTests(); + this.addCrossBrowserCompatibilityTests(); + } + + private addDependencyTelemetryTests(): void { + this.testCase({ + name: "createDependencyTelemetry: CLIENT span generates RemoteDependency telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("client-operation", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "custom.attr": "value" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.equal(item.name, "Microsoft.ApplicationInsights.RemoteDependency", "Should be RemoteDependency"); + Assert.equal(item.baseType, "RemoteDependencyData", "Should have correct baseType"); + Assert.ok(item.baseData, "Should have baseData"); + Assert.equal((item.baseData as any).name, "client-operation", "Should have span name"); + Assert.equal((item.baseData as any).type, "Dependency", "Should have default dependency type"); + } + }); + + this.testCase({ + name: "createDependencyTelemetry: PRODUCER span generates QueueMessage dependency", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("queue-producer", { + kind: eOTelSpanKind.PRODUCER, + attributes: { + "messaging.system": "kafka", + "messaging.destination": "orders" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).type, "Queue Message", "Should be QueueMessage type"); + } + }); + + this.testCase({ + name: "createDependencyTelemetry: INTERNAL span with parent generates InProc dependency", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const parentSpan = this._ai.startSpan("parent-operation"); + const parentContext = parentSpan?.spanContext(); + + const childSpan = this._ai.startSpan("internal-operation", { + kind: eOTelSpanKind.INTERNAL + }, parentContext); + childSpan?.end(); + parentSpan?.end(); + + // Assert + const childItem = this._trackCalls.find(t => t.baseData?.name === "internal-operation"); + Assert.ok(childItem, "Should have child telemetry"); + Assert.equal((childItem?.baseData as any).type, "InProc", "Should be InProc type"); + } + }); + + this.testCase({ + name: "createDependencyTelemetry: SUCCESS status based on span status code", + test: () => { + // Arrange + this._trackCalls = []; + + // Act - span with OK status + const okSpan = this._ai.startSpan("ok-span", { kind: eOTelSpanKind.CLIENT }); + okSpan?.setStatus({ code: eOTelSpanStatusCode.OK }); + okSpan?.end(); + + // Act - span with ERROR status + const errorSpan = this._ai.startSpan("error-span", { kind: eOTelSpanKind.CLIENT }); + errorSpan?.setStatus({ code: eOTelSpanStatusCode.ERROR, message: "Failed" }); + errorSpan?.end(); + + // Assert + const okItem = this._trackCalls.find(t => t.baseData?.name === "ok-span"); + const errorItem = this._trackCalls.find(t => t.baseData?.name === "error-span"); + + Assert.equal((okItem?.baseData as any).success, true, "OK span should have success=true"); + Assert.equal((errorItem?.baseData as any).success, false, "ERROR span should have success=false"); + } + }); + + this.testCase({ + name: "createDependencyTelemetry: includes span context IDs", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("test-span", { kind: eOTelSpanKind.CLIENT }); + const spanContext = span?.spanContext(); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).id, spanContext?.spanId, "Should have spanId"); + Assert.ok(item.tags, "Should have tags"); + Assert.equal((item.tags as any)["ai.operation.id"], spanContext?.traceId, "Should have traceId in tags"); + } + }); + } + + private addRequestTelemetryTests(): void { + this.testCase({ + name: "createRequestTelemetry: SERVER span generates Request telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("server-operation", { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.method": "GET", + "http.url": "https://example.com/api/users" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.equal(item.name, "Microsoft.ApplicationInsights.Request", "Should be Request"); + Assert.equal(item.baseType, "RequestData", "Should have correct baseType"); + Assert.ok(item.baseData, "Should have baseData"); + } + }); + + this.testCase({ + name: "createRequestTelemetry: CONSUMER span generates Request telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("queue-consumer", { + kind: eOTelSpanKind.CONSUMER, + attributes: { + "messaging.system": "rabbitmq" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal(item.name, "Microsoft.ApplicationInsights.Request", "Should be Request"); + } + }); + + this.testCase({ + name: "createRequestTelemetry: SUCCESS derived from status code", + test: () => { + // Arrange + this._trackCalls = []; + + // Act - UNSET status with 2xx HTTP code + const successSpan = this._ai.startSpan("success-request", { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.method": "POST", + "http.status_code": 201 + } + }); + successSpan?.end(); + + // Act - UNSET status with 5xx HTTP code + const failSpan = this._ai.startSpan("fail-request", { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.method": "GET", + "http.status_code": 500 + } + }); + failSpan?.end(); + + // Assert + const successItem = this._trackCalls.find(t => t.baseData?.name === "success-request"); + const failItem = this._trackCalls.find(t => t.baseData?.name === "fail-request"); + + Assert.equal((successItem?.baseData as any).success, true, "2xx should be success"); + Assert.equal((failItem?.baseData as any).success, false, "5xx should be failure"); + } + }); + + this.testCase({ + name: "createRequestTelemetry: OK status overrides HTTP status code", + test: () => { + // Arrange + this._trackCalls = []; + + // Act - OK status with 5xx code (shouldn't happen but testing precedence) + const span = this._ai.startSpan("explicit-ok", { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.method": "GET", + "http.status_code": 500 + } + }); + span?.setStatus({ code: eOTelSpanStatusCode.OK }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).success, true, "OK status should take precedence"); + } + }); + + this.testCase({ + name: "createRequestTelemetry: includes URL for HTTP requests", + test: () => { + // Arrange + this._trackCalls = []; + const testUrl = "https://api.example.com/v1/users?id=123"; + + // Act + const span = this._ai.startSpan("http-request", { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.method": "GET", + "http.url": testUrl, + "http.status_code": 200 + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const baseData = item.baseData as IRequestTelemetry; + Assert.equal(baseData.url, testUrl, "Should include URL"); + Assert.equal(baseData.responseCode, 200, "Should include status code"); + } + }); + + this.testCase({ + name: "createRequestTelemetry: gRPC status code mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("grpc-request", { + kind: eOTelSpanKind.SERVER, + attributes: { + "rpc.system": "grpc", + "rpc.grpc.status_code": 0 // OK + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const baseData = item.baseData as IRequestTelemetry; + Assert.equal(baseData.responseCode, 0, "Should map gRPC status code"); + } + }); + } + + private addHttpDependencyTests(): void { + this.testCase({ + name: "HTTP Dependency: legacy semantic conventions mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("http-call", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_HTTP_METHOD]: "POST", + [SEMATTRS_HTTP_URL]: "https://api.example.com/v1/users", + [SEMATTRS_HTTP_STATUS_CODE]: 201 + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const baseData = item.baseData as IDependencyTelemetry; + Assert.equal(baseData.type, "Http", "Should be HTTP type"); + Assert.ok(baseData.name?.startsWith("POST"), "Name should include method"); + Assert.equal(baseData.data, "https://api.example.com/v1/users", "Should include URL"); + Assert.equal(baseData.responseCode, 201, "Should include status code"); + } + }); + + this.testCase({ + name: "HTTP Dependency: new semantic conventions mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("http-call-new", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [ATTR_HTTP_REQUEST_METHOD]: "GET", + [ATTR_URL_FULL]: "https://api.example.com/v2/products", + [ATTR_HTTP_RESPONSE_STATUS_CODE]: 200, + [ATTR_SERVER_ADDRESS]: "api.example.com", + [ATTR_SERVER_PORT]: 443 + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const baseData = item.baseData as IDependencyTelemetry; + Assert.equal(baseData.type, "Http", "Should be HTTP type"); + Assert.ok(baseData.data, "Should have data field"); + } + }); + + this.testCase({ + name: "HTTP Dependency: target with default port removal", + test: () => { + // Arrange + this._trackCalls = []; + + // Act - HTTPS with default port 443 + const httpsSpan = this._ai.startSpan("https-call", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "GET", + "http.url": "https://example.com:443/api", + "net.peer.name": "example.com", + "net.peer.port": 443 + } + }); + httpsSpan?.end(); + + // Act - HTTP with default port 80 + const httpSpan = this._ai.startSpan("http-call", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "GET", + "http.url": "http://example.com:80/api", + "net.peer.name": "example.com", + "net.peer.port": 80 + } + }); + httpSpan?.end(); + + // Assert + const httpsItem = this._trackCalls.find(t => t.baseData?.name?.includes("https-call") || t.baseData?.data?.includes("https://example.com:443")); + const httpItem = this._trackCalls.find(t => t.baseData?.name?.includes("http-call") || t.baseData?.data?.includes("http://example.com:80")); + + // Default ports should be stripped from target + Assert.ok(httpsItem, "Should have HTTPS telemetry"); + Assert.ok(httpItem, "Should have HTTP telemetry"); + } + }); + + this.testCase({ + name: "HTTP Dependency: target with non-default port preserved", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("custom-port-call", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "GET", + "http.url": "https://example.com:8443/api", + "net.peer.name": "example.com", + "net.peer.port": 8443 + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok((item.baseData as any).target, "Should have target"); + // Non-default port should be preserved in target + } + }); + + this.testCase({ + name: "HTTP Dependency: name generated from URL pathname", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("generic-name", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "DELETE", + "http.url": "https://api.example.com/v1/users/123" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok((item.baseData as any).name.includes("DELETE"), "Name should include HTTP method"); + Assert.ok((item.baseData as any).name.includes("/v1/users/123"), "Name should include path"); + } + }); + } + + private addDbDependencyTests(): void { + this.testCase({ + name: "DB Dependency: MySQL mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("db-query", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_DB_SYSTEM]: "mysql", + [SEMATTRS_DB_STATEMENT]: "SELECT * FROM users WHERE id = ?", + [SEMATTRS_DB_NAME]: "myapp_db", + "net.peer.name": "db.example.com", + "net.peer.port": 3306 + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).type, "mysql", "Should be mysql type"); + Assert.equal((item.baseData as any).data, "SELECT * FROM users WHERE id = ?", "Should include statement"); + Assert.ok((item.baseData as any).target?.includes("myapp_db"), "Target should include DB name"); + } + }); + + this.testCase({ + name: "DB Dependency: PostgreSQL mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("postgres-query", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_DB_SYSTEM]: "postgresql", + [SEMATTRS_DB_STATEMENT]: "INSERT INTO logs (message) VALUES ($1)" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).type, "postgresql", "Should be postgresql type"); + } + }); + + this.testCase({ + name: "DB Dependency: MongoDB mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("mongo-query", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_DB_SYSTEM]: "mongodb", + [SEMATTRS_DB_STATEMENT]: "db.users.find({age: {$gt: 25}})" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).type, "mongodb", "Should be mongodb type"); + } + }); + + this.testCase({ + name: "DB Dependency: Redis mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("redis-cmd", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_DB_SYSTEM]: "redis", + [SEMATTRS_DB_STATEMENT]: "GET user:123" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).type, "redis", "Should be redis type"); + } + }); + + this.testCase({ + name: "DB Dependency: SQL Server mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("mssql-query", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_DB_SYSTEM]: "mssql", + [SEMATTRS_DB_STATEMENT]: "SELECT TOP 10 * FROM Orders" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).type, "SQL", "Should be SQL type for SQL Server"); + } + }); + + this.testCase({ + name: "DB Dependency: operation used when no statement", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("db-op", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_DB_SYSTEM]: "postgresql", + "db.operation": "SELECT", + [SEMATTRS_DB_NAME]: "products_db" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).data, "SELECT", "Should use operation when no statement"); + } + }); + + this.testCase({ + name: "DB Dependency: target formatting with host and dbname", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("db-call", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_DB_SYSTEM]: "mysql", + [SEMATTRS_DB_NAME]: "production_db", + "net.peer.name": "mysql-prod.example.com", + "net.peer.port": 3306 + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok((item.baseData as any).target?.includes("mysql-prod.example.com"), "Target should include host"); + Assert.ok((item.baseData as any).target?.includes("production_db"), "Target should include DB name"); + } + }); + } + + private addRpcDependencyTests(): void { + this.testCase({ + name: "RPC Dependency: gRPC mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("grpc-call", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_RPC_SYSTEM]: "grpc", + [SEMATTRS_RPC_GRPC_STATUS_CODE]: 0, + "rpc.service": "UserService", + "rpc.method": "GetUser" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + let baseData = item.baseData as IDependencyTelemetry; + Assert.equal(baseData.type, "GRPC", "Should be Dependency type"); + Assert.equal(baseData.responseCode, 0, "Should include gRPC status code"); + } + }); + + this.testCase({ + name: "RPC Dependency: WCF mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("wcf-call", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_RPC_SYSTEM]: "wcf", + "rpc.service": "CalculatorService" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).type, "WCF Service", "Should be Dependency type"); + } + }); + + this.testCase({ + name: "RPC Dependency: target from peer service", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("rpc-call", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_RPC_SYSTEM]: "grpc", + "net.peer.name": "grpc.example.com", + "net.peer.port": 50051 + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + let baseData = item.baseData as IDependencyTelemetry; + Assert.ok(baseData.target, "Should have target"); + } + }); + } + + private addAttributeMappingTests(): void { + this.testCase({ + name: "Attribute Mapping: custom attributes preserved in properties", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("custom-attrs", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "app.version": "1.2.3", + "user.tier": "premium", + "request.priority": 5, + "feature.enabled": true + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok((item.baseData as any).properties, "Should have properties"); + Assert.equal((item.baseData as any).properties["app.version"], "1.2.3", "String attribute preserved"); + Assert.equal((item.baseData as any).properties["user.tier"], "premium", "String attribute preserved"); + Assert.equal((item.baseData as any).properties["request.priority"], 5, "Number attribute preserved"); + Assert.equal((item.baseData as any).properties["feature.enabled"], true, "Boolean attribute preserved"); + } + }); + + this.testCase({ + name: "Attribute Mapping: dt.spanId and dt.traceId always added", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("test-span", { kind: eOTelSpanKind.CLIENT }); + const context = span?.spanContext(); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal(item.ext?.dt.spanId, context?.spanId, "Should have dt.spanId"); + Assert.equal(item.ext?.dt.traceId, context?.traceId, "Should have dt.traceId"); + } + }); + + this.testCase({ + name: "Attribute Mapping: sampling.probability mapped to sampleRate", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("sampled-span", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "microsoft.sample_rate": 25 + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0] as any; + Assert.equal(item.sampleRate, 25, "Should map sampling.probability to sampleRate"); + } + }); + } + + private addTagsCreationTests(): void { + this.testCase({ + name: "Tags: operation ID from trace ID", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("test-span", { kind: eOTelSpanKind.SERVER }); + const context = span?.spanContext(); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.tags as any)["ai.operation.id"], context?.traceId, "Should map traceId to operation.id"); + } + }); + + this.testCase({ + name: "Tags: operation parent ID from parent span", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const parentSpan = this._ai.startSpan("parent", { kind: eOTelSpanKind.SERVER }); + const parentContext = parentSpan?.spanContext(); + + const childSpan = this._ai.startSpan("child", { kind: eOTelSpanKind.INTERNAL }, parentContext); + childSpan?.end(); + parentSpan?.end(); + + // Assert + const childItem = this._trackCalls.find(t => t.baseData?.name === "child"); + Assert.equal((childItem?.tags as any)?.["ai.operation.parentId"], parentContext?.spanId, + "Should map parent spanId to operation.parentId"); + } + }); + + this.testCase({ + name: "Tags: enduser.id mapped to user auth ID", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("user-span", { + kind: eOTelSpanKind.SERVER, + attributes: { + [ATTR_ENDUSER_ID]: "user@example.com" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.tags as any)["ai.user.authUserId"], "user@example.com", + "Should map enduser.id to user.authUserId"); + } + }); + + this.testCase({ + name: "Tags: enduser.pseudo.id mapped to user ID", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("pseudo-user-span", { + kind: eOTelSpanKind.SERVER, + attributes: { + [ATTR_ENDUSER_PSEUDO_ID]: "anon-12345" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.tags as any)["ai.user.id"], "anon-12345", + "Should map enduser.pseudo.id to user.id"); + } + }); + + this.testCase({ + name: "Tags: microsoft.client.ip takes precedence", + test: () => { + // Arrange + this._trackCalls = []; + const clientIp = "203.0.113.42"; + + // Act + const span = this._ai.startSpan("ip-span", { + kind: eOTelSpanKind.SERVER, + attributes: { + [MicrosoftClientIp]: clientIp, + "client.address": "192.168.1.1", // Should be ignored + "http.client_ip": "10.0.0.1" // Should be ignored + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.tags as any)["ai.location.ip"], clientIp, + "microsoft.client.ip should take precedence"); + } + }); + + this.testCase({ + name: "Tags: operation name from http.route for SERVER spans", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("http-request", { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.method": "POST", + [ATTR_HTTP_ROUTE]: "/api/v1/users/:id", + "http.url": "https://example.com/api/v1/users/123" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok((item.tags as any)["ai.operation.name"]?.includes("POST"), "Should include method"); + Assert.ok((item.tags as any)["ai.operation.name"]?.includes("/api/v1/users/:id"), "Should include route"); + } + }); + + this.testCase({ + name: "Tags: operation name falls back to URL path when no route", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("http-request-no-route", { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.method": "GET", + "http.url": "https://example.com/products/search?q=laptop" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok((item.tags as any)["ai.operation.name"]?.includes("GET"), "Should include method"); + Assert.ok((item.tags as any)["ai.operation.name"]?.includes("/products/search"), "Should include path"); + } + }); + + this.testCase({ + name: "Tags: user agent mapped correctly", + test: () => { + // Arrange + this._trackCalls = []; + const userAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) Chrome/91.0"; + + // Act + const span = this._ai.startSpan("ua-span", { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.user_agent": userAgent + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.tags as any)["ai.user.userAgent"], userAgent, "Should map user agent"); + } + }); + + this.testCase({ + name: "Tags: synthetic source detection", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("bot-span", { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.user_agent": "Googlebot/2.1 (+http://www.google.com/bot.html)" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + // Synthetic source should be detected for bot user agents + if ((item.tags as any)["ai.operation.syntheticSource"]) { + Assert.equal((item.tags as any)["ai.operation.syntheticSource"], "True", + "Should detect synthetic source for bots"); + } + } + }); + } + + private addAzureSDKTests(): void { + this.testCase({ + name: "Azure SDK: EventHub PRODUCER span mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("EventHubs.send", { + kind: eOTelSpanKind.PRODUCER, + attributes: { + "az.namespace": "Microsoft.EventHub", + "message_bus.destination": "telemetry-events", + "net.peer.name": "eventhub.servicebus.windows.net" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok((item.baseData as any).type?.includes("Queue Message"), "Should be Queue Message type"); + Assert.ok((item.baseData as any).type?.includes("Microsoft.EventHub"), "Should include namespace"); + Assert.ok((item.baseData as any).target, "Should have target"); + } + }); + + this.testCase({ + name: "Azure SDK: EventHub CONSUMER span mapping", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("EventHubs.process", { + kind: eOTelSpanKind.CONSUMER, + attributes: { + "az.namespace": "Microsoft.EventHub", + "message_bus.destination": "telemetry-events", + "net.peer.name": "eventhub.servicebus.windows.net", + "enqueuedTime": "1638360000000" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok((item.baseData as any).source, "Consumer should have source"); + Assert.ok((item.baseData as any).measurements, "Should have measurements"); + Assert.ok("timeSinceEnqueued" in (item.baseData as any).measurements, "Should have timeSinceEnqueued measurement"); + } + }); + + this.testCase({ + name: "Azure SDK: INTERNAL span with Azure namespace", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("internal-azure-op", { + kind: eOTelSpanKind.INTERNAL, + attributes: { + "az.namespace": "Microsoft.Storage" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok((item.baseData as any).type?.includes("InProc"), "Should include InProc"); + Assert.ok((item.baseData as any).type?.includes("Microsoft.Storage"), "Should include namespace"); + } + }); + } + + private addSemanticAttributeExclusionTests(): void { + this.testCase({ + name: "Semantic Exclusion: HTTP attributes not in properties", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("http-span", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "POST", + "http.url": "https://example.com/api", + "http.status_code": 201, + "http.user_agent": "TestAgent/1.0", + "custom.attribute": "should-be-kept" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + + Assert.ok(!props["http.method"], "http.method should be excluded"); + Assert.ok(!props["http.url"], "http.url should be excluded"); + Assert.ok(!props["http.status_code"], "http.status_code should be excluded"); + Assert.ok(!props["http.user_agent"], "http.user_agent should be excluded"); + Assert.equal(props["custom.attribute"], "should-be-kept", "Custom attributes should be kept"); + } + }); + + this.testCase({ + name: "Semantic Exclusion: DB attributes not in properties", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("db-span", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "db.system": "postgresql", + "db.statement": "SELECT * FROM users", + "db.name": "mydb", + "db.operation": "SELECT", + "app.query.id": "query-123" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + + Assert.ok(!props["db.system"], "db.system should be excluded"); + Assert.ok(!props["db.statement"], "db.statement should be excluded"); + Assert.ok(!props["db.name"], "db.name should be excluded"); + Assert.ok(!props["db.operation"], "db.operation should be excluded"); + Assert.equal(props["app.query.id"], "query-123", "Custom attributes should be kept"); + } + }); + + this.testCase({ + name: "Semantic Exclusion: microsoft.* attributes excluded", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("microsoft-attrs", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "microsoft.internal.flag": true, + "microsoft.client.ip": "192.168.1.1", + "microsoft.custom": "value", + "app.microsoft": "not-excluded" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + + Assert.ok(!props["microsoft.internal.flag"], "microsoft.* should be excluded"); + Assert.ok(!props["microsoft.client.ip"], "microsoft.* should be excluded"); + Assert.ok(!props["microsoft.custom"], "microsoft.* should be excluded"); + Assert.equal(props["app.microsoft"], "not-excluded", + "Attributes containing 'microsoft' but not prefixed should be kept"); + } + }); + + this.testCase({ + name: "Semantic Exclusion: operation.name context tag excluded", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("op-name-span", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "ai.operation.name": "CustomOperation", + "custom.operation.name": "should-be-kept" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + + Assert.ok(!props["ai.operation.name"], "ai.operation.name should be excluded"); + Assert.equal(props["custom.operation.name"], "should-be-kept", + "Similar named custom attributes should be kept"); + } + }); + + this.testCase({ + name: "Semantic Exclusion: new semantic conventions excluded", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("new-semconv", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [ATTR_HTTP_REQUEST_METHOD]: "GET", + [ATTR_HTTP_RESPONSE_STATUS_CODE]: 200, + [ATTR_URL_FULL]: "https://example.com", + [ATTR_SERVER_ADDRESS]: "example.com", + [ATTR_SERVER_PORT]: 443, + "app.request.id": "req-123" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + + Assert.ok(!props["http.request.method"], "New http attributes should be excluded"); + Assert.ok(!props["http.response.status_code"], "New http attributes should be excluded"); + Assert.ok(!props["url.full"], "New url attributes should be excluded"); + Assert.ok(!props["server.address"], "New server attributes should be excluded"); + Assert.ok(!props["server.port"], "New server attributes should be excluded"); + Assert.equal(props["app.request.id"], "req-123", "Custom attributes should be kept"); + } + }); + } + + private addEdgeCaseTests(): void { + this.testCase({ + name: "Edge Case: Empty span name", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("", { + kind: eOTelSpanKind.CLIENT + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry for empty name"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + Assert.equal((item.baseData as any).name, "", "Should preserve empty name"); + } + }); + + this.testCase({ + name: "Edge Case: Span with null/undefined attributes", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("null-attrs", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "valid.attr": "value", + "null.attr": null as any, + "undefined.attr": undefined as any, + "zero.attr": 0, + "false.attr": false, + "empty.string": "" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + Assert.equal(props["valid.attr"], "value", "Valid attributes should be preserved"); + Assert.equal(props["zero.attr"], 0, "Zero values should be preserved"); + Assert.equal(props["false.attr"], false, "False values should be preserved"); + Assert.equal(props["empty.string"], "", "Empty strings should be preserved"); + } + }); + + this.testCase({ + name: "Edge Case: Span with extremely long attribute values", + test: () => { + // Arrange + this._trackCalls = []; + const veryLongValue = "a".repeat(20000); + + // Act + const span = this._ai.startSpan("long-attrs", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "long.value": veryLongValue + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + Assert.ok(props["long.value"], "Long value should be included"); + Assert.equal(props["long.value"], veryLongValue, "Long value should be preserved"); + } + }); + + this.testCase({ + name: "Edge Case: Span with special characters in name and attributes", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("span-with-特殊字符-émojis-🎉", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "unicode.key": "value with 中文 and émojis 🚀", + "special.chars": "tab\there\nnewline\r\ncarriage", + "quotes": "\"double\" and 'single' quotes" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should handle special characters"); + const item = this._trackCalls[0]; + Assert.ok((item.baseData as any).name.includes("特殊字符"), "Should preserve unicode in name"); + const props = (item.baseData as any).properties || {}; + Assert.ok(props["unicode.key"], "Should preserve unicode attributes"); + Assert.ok(props["special.chars"], "Should preserve special characters"); + Assert.ok(props["quotes"], "Should preserve quotes"); + } + }); + + this.testCase({ + name: "Edge Case: Span without explicit kind defaults appropriately", + test: () => { + // Arrange + this._trackCalls = []; + + // Act - startSpan with no kind specified + const span = this._ai.startSpan("no-kind-span"); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item, "Should have telemetry item"); + } + }); + + this.testCase({ + name: "Edge Case: Multiple rapid span creations and endings", + test: () => { + // Arrange + this._trackCalls = []; + const spanCount = 50; + + // Act - Create and end many spans rapidly + for (let i = 0; i < spanCount; i++) { + const span = this._ai.startSpan("rapid-span-" + i, { + kind: eOTelSpanKind.CLIENT, + attributes: { + "span.index": i + } + }); + span?.end(); + } + + // Assert + Assert.equal(this._trackCalls.length, spanCount, "Should track all spans"); + const firstItem = this._trackCalls[0]; + const lastItem = this._trackCalls[spanCount - 1]; + Assert.equal((firstItem.baseData as any).properties["span.index"], 0, "First span preserved"); + Assert.equal((lastItem.baseData as any).properties["span.index"], spanCount - 1, "Last span preserved"); + } + }); + + this.testCase({ + name: "Edge Case: Span with array attribute values", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("array-attrs", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "string.array": ["value1", "value2", "value3"], + "number.array": [1, 2, 3], + "mixed.array": ["string", 123, true] as any + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + Assert.ok(props["string.array"], "Array attributes should be included"); + } + }); + + this.testCase({ + name: "Edge Case: Span with nested object attributes", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("nested-attrs", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "nested.object": { key: "value", nested: { deep: "data" } } as any, + "simple.attr": "simple" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + Assert.ok(props["simple.attr"], "Simple attributes should work"); + } + }); + + this.testCase({ + name: "Edge Case: Span with malformed HTTP status codes", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("malformed-status", { + kind: eOTelSpanKind.SERVER, + attributes: { + [SEMATTRS_HTTP_STATUS_CODE]: "not-a-number" as any + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should handle malformed status codes"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + } + }); + + this.testCase({ + name: "Edge Case: Span with missing parent context", + test: () => { + // Arrange + this._trackCalls = []; + + // Act - Explicitly pass null/undefined parent context + const span = this._ai.startSpan("orphan-span", { + kind: eOTelSpanKind.CLIENT + }, undefined); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should handle missing parent"); + const item = this._trackCalls[0]; + Assert.ok((item.tags as any)["ai.operation.id"], "Should have operation ID"); + } + }); + + this.testCase({ + name: "Edge Case: Span ended multiple times", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("multi-end", { + kind: eOTelSpanKind.CLIENT + }); + span?.end(); + const firstCallCount = this._trackCalls.length; + span?.end(); // End again + const secondCallCount = this._trackCalls.length; + + // Assert + Assert.equal(firstCallCount, 1, "First end should generate telemetry"); + Assert.equal(secondCallCount, 1, "Second end should not generate duplicate telemetry"); + } + }); + + this.testCase({ + name: "Edge Case: Span with extremely large number of attributes", + test: () => { + // Arrange + this._trackCalls = []; + const attributes: any = {}; + for (let i = 0; i < 1000; i++) { + attributes["attr." + i] = "value" + i; + } + + // Act + const span = this._ai.startSpan("many-attrs", { + kind: eOTelSpanKind.CLIENT, + attributes: attributes + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should handle many attributes"); + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + Assert.ok(Object.keys(props).length > 0, "Should have some properties"); + } + }); + + this.testCase({ + name: "Edge Case: Zero duration span", + test: () => { + // Arrange + this._trackCalls = []; + + // Act - End span immediately + const span = this._ai.startSpan("instant-span", { + kind: eOTelSpanKind.CLIENT + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + const duration = (item.baseData as any).duration; + Assert.ok(duration !== undefined, "Should have duration field"); + } + }); + + this.testCase({ + name: "Edge Case: HTTP dependency with missing URL", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("http-no-url", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_HTTP_METHOD]: "GET", + [SEMATTRS_HTTP_STATUS_CODE]: 200 + // No URL attribute + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should handle missing URL"); + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).type, "Http", "Should still be HTTP type"); + } + }); + + this.testCase({ + name: "Edge Case: Database dependency with missing statement", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("db-no-statement", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_DB_SYSTEM]: "postgresql", + [SEMATTRS_DB_NAME]: "testdb" + // No statement + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.equal((item.baseData as any).type, "postgresql", "Should have DB type"); + } + }); + } + + private addCrossBrowserCompatibilityTests(): void { + this.testCase({ + name: "Cross-Browser: Handles performance.now() unavailable", + test: () => { + // Arrange + this._trackCalls = []; + + // Act - Create span when performance.now might not be available + const span = this._ai.startSpan("perf-test", { + kind: eOTelSpanKind.CLIENT + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should work without performance.now"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should generate valid telemetry"); + } + }); + + this.testCase({ + name: "Cross-Browser: Handles Date.now() for timing", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("date-timing", { + kind: eOTelSpanKind.CLIENT + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok((item.baseData as any).duration !== undefined, "Should have duration"); + Assert.ok((item.baseData as any).duration >= 0, "Duration should be non-negative"); + } + }); + + this.testCase({ + name: "Cross-Browser: String encoding compatibility", + test: () => { + // Arrange + this._trackCalls = []; + const testStrings = [ + "ASCII only", + "UTF-8: 你好世界", + "Emoji: 🎉🚀💻", + "Latin: café résumé", + "Mixed: Hello世界🌍" + ]; + + // Act + for (let i = 0; i < testStrings.length; i++) { + const span = this._ai.startSpan(testStrings[i], { + kind: eOTelSpanKind.CLIENT, + attributes: { + "test.string": testStrings[i] + } + }); + span?.end(); + } + + // Assert + Assert.equal(this._trackCalls.length, testStrings.length, "Should handle all encodings"); + for (let i = 0; i < testStrings.length; i++) { + const item = this._trackCalls[i]; + Assert.ok(item.baseData, "Should have baseData for encoding test " + i); + } + } + }); + + this.testCase({ + name: "Cross-Browser: JSON serialization of attributes", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("json-test", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "number": 123, + "string": "test", + "boolean": true, + "float": 123.456, + "negative": -999, + "zero": 0 + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + Assert.equal(typeof props["number"], "number", "Numbers should remain numbers"); + Assert.equal(typeof props["string"], "string", "Strings should remain strings"); + Assert.equal(typeof props["boolean"], "boolean", "Booleans should remain booleans"); + } + }); + + this.testCase({ + name: "Cross-Browser: Large payload handling", + test: () => { + // Arrange + this._trackCalls = []; + const largeAttributes: any = {}; + for (let i = 0; i < 100; i++) { + largeAttributes["large.attr." + i] = "x".repeat(100); + } + + // Act + const span = this._ai.startSpan("large-payload", { + kind: eOTelSpanKind.CLIENT, + attributes: largeAttributes + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should handle large payloads"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should generate telemetry"); + } + }); + + this.testCase({ + name: "Cross-Browser: Handles undefined vs null attributes", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("null-undefined", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "explicit.null": null as any, + "explicit.undefined": undefined as any, + "valid.value": "test" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + Assert.equal(props["valid.value"], "test", "Valid values should be preserved"); + } + }); + + this.testCase({ + name: "Cross-Browser: Whitespace handling in attribute keys", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("whitespace-keys", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "normal.key": "value1", + " leading.space": "value2", + "trailing.space ": "value3", + "has spaces": "value4" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should handle whitespace in keys"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + } + }); + + this.testCase({ + name: "Cross-Browser: Number precision and special values", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("number-precision", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "max.safe.integer": Number.MAX_SAFE_INTEGER, + "min.safe.integer": Number.MIN_SAFE_INTEGER, + "large.float": 1.7976931348623157e+308, + "small.float": 5e-324, + "infinity": Infinity as any, + "neg.infinity": -Infinity as any, + "not.a.number": NaN as any + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should handle special number values"); + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + Assert.ok(props["max.safe.integer"] !== undefined, "Should handle large integers"); + } + }); + + this.testCase({ + name: "Cross-Browser: URL parsing with various formats", + test: () => { + // Arrange + this._trackCalls = []; + const urls = [ + "http://example.com", + "https://example.com:8080/path", + "http://example.com/path?query=value", + "https://user:pass@example.com/path", + "http://192.168.1.1:3000", + "https://[::1]:8080/path" + ]; + + // Act + for (const url of urls) { + const span = this._ai.startSpan("url-test", { + kind: eOTelSpanKind.CLIENT, + attributes: { + [SEMATTRS_HTTP_URL]: url + } + }); + span?.end(); + } + + // Assert + Assert.equal(this._trackCalls.length, urls.length, "Should handle all URL formats"); + } + }); + + this.testCase({ + name: "Cross-Browser: Timestamp handling across timezones", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("timezone-test", { + kind: eOTelSpanKind.CLIENT + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + Assert.ok(item.time, "Should have timestamp"); + const timestamp = new Date(item.time || "").getTime(); + Assert.ok(timestamp > 0, "Timestamp should be valid"); + } + }); + + this.testCase({ + name: "Cross-Browser: Memory efficient attribute storage", + test: () => { + // Arrange + this._trackCalls = []; + + // Act - Create many spans to test memory handling + for (let i = 0; i < 10; i++) { + const span = this._ai.startSpan("memory-test-" + i, { + kind: eOTelSpanKind.CLIENT, + attributes: { + "iteration": i, + "data": "x".repeat(1000) + } + }); + span?.end(); + } + + // Assert + Assert.equal(this._trackCalls.length, 10, "Should handle multiple spans"); + Assert.ok(this._trackCalls[0].baseData, "First span should have data"); + Assert.ok(this._trackCalls[9].baseData, "Last span should have data"); + } + }); + + this.testCase({ + name: "Cross-Browser: Concurrent span operations", + test: () => { + // Arrange + this._trackCalls = []; + const spans: any[] = []; + + // Act - Create multiple spans before ending any + for (let i = 0; i < 5; i++) { + const span = this._ai.startSpan("concurrent-" + i, { + kind: eOTelSpanKind.CLIENT, + attributes: { + "index": i + } + }); + spans.push(span); + } + + // End all spans + for (const span of spans) { + span?.end(); + } + + // Assert + Assert.equal(this._trackCalls.length, 5, "Should handle concurrent spans"); + } + }); + + this.testCase({ + name: "Cross-Browser: RegExp in attribute values", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("regexp-test", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "pattern": "/test/gi" as any, + "normal": "value" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should handle RegExp-like values"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + } + }); + + this.testCase({ + name: "Cross-Browser: Function and Symbol values filtered", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("special-types", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "function": (() => "test") as any, + "symbol": Symbol("test") as any, + "normal": "value" + } + }); + span?.end(); + + // Assert + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + Assert.equal(props["normal"], "value", "Normal values should be preserved"); + } + }); + + this.testCase({ + name: "Cross-Browser: Circular reference handling", + test: () => { + // Arrange + this._trackCalls = []; + const circular: any = { a: "value" }; + circular.self = circular; + + // Act + const span = this._ai.startSpan("circular-test", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "circular": circular, + "normal": "value" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should handle circular references gracefully"); + const item = this._trackCalls[0]; + const props = (item.baseData as any).properties || {}; + Assert.equal(props["normal"], "value", "Normal attributes should still work"); + } + }); + } +} diff --git a/AISKU/Tests/Unit/src/StartSpan.Tests.ts b/AISKU/Tests/Unit/src/StartSpan.Tests.ts new file mode 100644 index 000000000..0ba05e737 --- /dev/null +++ b/AISKU/Tests/Unit/src/StartSpan.Tests.ts @@ -0,0 +1,301 @@ +import { AITestClass, Assert } from '@microsoft/ai-test-framework'; +import { ApplicationInsights } from '../../../src/applicationinsights-web'; +import { eOTelSpanKind, eOTelSpanStatusCode, ITelemetryItem } from "@microsoft/applicationinsights-core-js"; + +export class StartSpanTests extends AITestClass { + private static readonly _instrumentationKey = 'b7170927-2d1c-44f1-acec-59f4e1751c11'; + private static readonly _connectionString = `InstrumentationKey=${StartSpanTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + + // Track calls to track + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "StartSpanTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: StartSpanTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + // Initialize the SDK + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + + } catch (e) { + console.error('Failed to initialize tests: ' + e); + throw e; + } + } + + + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addTests(); + } + + private addTests(): void { + + this.testCase({ + name: "StartSpan: startSpan method should exist on ApplicationInsights instance", + test: () => { + // Verify that startSpan method exists + Assert.ok(this._ai, "ApplicationInsights should be initialized"); + Assert.ok(typeof this._ai.startSpan === 'function', "startSpan method should exist"); + + // Check core initialization + Assert.ok(this._ai.core, "Core should be available"); + const core = this._ai.core; + if (core) { + // Check if core has startSpan method + Assert.ok(typeof core.startSpan === 'function', "Core should have startSpan method"); + + // Test basic startSpan call on the core directly after initialization + const coreSpan = core.startSpan("debug-core-span"); + Assert.ok(coreSpan !== null, `Core startSpan returned ${coreSpan} instead of a span object`); + } + + // Test basic startSpan call after initialization + const span = this._ai.startSpan("debug-span"); + + // Should now return a valid span object + Assert.ok(span !== null, `startSpan returned ${span} instead of a span object`); + + Assert.ok(typeof span!.isRecording === 'function', "Span should have isRecording method"); + Assert.ok(typeof span!.end === 'function', "Span should have end method"); + const isRecording = span!.isRecording(); + Assert.ok(typeof isRecording === 'boolean', `isRecording should return boolean, got ${typeof isRecording}: ${isRecording}`); + } + }); + + this.testCase({ + name: "StartSpan: Recording span should trigger track when span ends", + test: () => { + // Clear previous calls + this._trackCalls = []; + + // Create a recording span using startSpan + const span = this._ai.startSpan("test-recording-span", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "test.attribute": "test-value", + "operation.type": "http" + } + }); + + Assert.ok(span, "Span should be created"); + + // Verify it's a recording span + Assert.ok(span!.isRecording(), "Span should be recording"); + + // End the span - this should trigger track via the onEnd callback + span!.end(); + + // Verify that track was called + Assert.equal(1, this._trackCalls.length, "track should have been called once for recording span"); + + // Add defensive check for the telemetry item + Assert.ok(this._trackCalls.length > 0, "Should have at least one track call"); + const item = this._trackCalls[0]; + Assert.ok(item, "Telemetry item should exist"); + Assert.ok(item.name, "Item name should be present"); + Assert.ok(item.baseData, "Base data should be present"); + + Assert.ok(item.baseData.properties, "Custom properties should be present"); + Assert.equal("test-value", item.baseData.properties["test.attribute"], "Should include span attributes"); + Assert.equal("http", item.baseData.properties["operation.type"], "Should include all span attributes"); + } + }); + + this.testCase({ + name: "StartSpan: Non-recording span should NOT trigger track when span ends", + test: () => { + // Clear previous calls + this._trackCalls = []; + + // NOTE: Currently all spans are recording by default + // When the recording: false option is implemented, this test will need to be updated + // For now, we'll create a regular span and document the expected behavior + const span = this._ai.startSpan("test-would-be-non-recording-span", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "test.attribute": "non-recording-value" + } + }); + + Assert.ok(span, "Span should be created"); + + // Currently, all spans are recording by default + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + Assert.ok(span!.isRecording(), "Span should be recording (default behavior)"); + + // End the span - this WILL trigger track since it's recording + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + span!.end(); + + // Currently expecting 1 call since all spans are recording + // When non-recording spans are implemented, this should be 0 + Assert.equal(1, this._trackCalls.length, "track should be called for recording span (current default behavior)"); + + // TODO: Update this test when recording: false option is implemented + // The test should then use recording: false and expect 0 track calls + } + }); + + this.testCase({ + name: "StartSpan: Multiple recording spans should each trigger track", + test: () => { + // Clear previous calls + this._trackCalls = []; + + // Create multiple recording spans + const span1 = this._ai.startSpan("span-1", { + attributes: { "span.number": 1 } + }); + const span2 = this._ai.startSpan("span-2", { + attributes: { "span.number": 2 } + }); + + Assert.ok(span1 && span2, "Both spans should be created"); + + // End both spans + span1!.end(); + span2!.end(); + + // Should have two track calls + Assert.equal(2, this._trackCalls.length, "track should have been called twice"); + + // Verify both calls have the correct data + const item1 = this._trackCalls.find(item => + item.baseData && item.baseData.properties && item.baseData.name === "span-1"); + const item2 = this._trackCalls.find(item => + item.baseData && item.baseData.properties && item.baseData.name === "span-2"); + + Assert.ok(item1, "Should have item for span-1"); + Assert.ok(item2, "Should have item for span-2"); + + if (item1 && item2) { + Assert.equal(1, item1.baseData.properties["span.number"], "First span should have correct attribute"); + Assert.equal(2, item2.baseData.properties["span.number"], "Second span should have correct attribute"); + } + } + }); + + this.testCase({ + name: "StartSpan: Error recording spans should generate telemetry with error status", + test: () => { + // Clear previous calls + this._trackCalls = []; + + // Create an error span + const span = this._ai.startSpan("error-span", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "error": true, + "error.message": "Something went wrong" + } + }); + + Assert.ok(span, "Span should be created"); + + // Set error status on the span + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + span!.setStatus({ + code: eOTelSpanStatusCode.ERROR, + message: "Test error occurred" + }); + + // End the span + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + span!.end(); + + // Verify track was called + Assert.equal(1, this._trackCalls.length, "track should have been called once"); + + const item = this._trackCalls[0]; + Assert.ok(item, "Telemetry item should be present"); + Assert.ok(item.baseData, "Base data should be present"); + Assert.ok(item.baseData.properties, "Properties should be present"); + Assert.equal("error-span", item.baseData.name, "Should include span name"); + + + Assert.ok(item.baseData.properties, "Custom properties should be present"); + Assert.equal(true, item.baseData.properties["error"], "Should include error attribute"); + Assert.equal("Something went wrong", item.baseData.properties["error.message"], "Should include error message"); + } + }); + + this.testCase({ + name: "StartSpan: startSpan with parent context should work", + test: () => { + // Clear previous calls + this._trackCalls = []; + + // Create span with optional parent context parameter + // (We'll pass null for now since we're not testing context propagation yet) + const parentContext = null; + + // Create span with parent context + const span = this._ai.startSpan("child-span", { + attributes: { "has.parent": false } + }); + + Assert.ok(span, "Span should be created with parent context"); + + // End the span + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + span!.end(); + + // Verify track was called + Assert.equal(1, this._trackCalls.length, "track should have been called once"); + + const item = this._trackCalls[0]; + Assert.ok(item, "Telemetry item should be present"); + Assert.ok(item.baseData && item.baseData.properties, "Properties should be present"); + Assert.equal("child-span", item.baseData.name, "Should include span name"); + Assert.equal(false, item.baseData.properties["has.parent"], "Should include span attributes"); + } + }); + + this.testCase({ + name: "StartSpan: startSpan should return valid span when trace provider is available", + test: () => { + // After initialization, the trace provider should be available + // and startSpan should return a valid span object + const span = this._ai.startSpan("test-span"); + + // Now that initialization is complete, we should get a valid span + Assert.ok(span !== null, "startSpan should return a valid span after initialization"); + Assert.ok(typeof span === 'object', "Span should be an object"); + + Assert.ok(typeof span!.end === 'function', "Span should have end method"); + Assert.ok(typeof span!.isRecording === 'function', "Span should have isRecording method"); + span!.end(); + } + }); + } +} \ No newline at end of file diff --git a/AISKU/Tests/Unit/src/TelemetryItemGeneration.Tests.ts b/AISKU/Tests/Unit/src/TelemetryItemGeneration.Tests.ts new file mode 100644 index 000000000..9d827180c --- /dev/null +++ b/AISKU/Tests/Unit/src/TelemetryItemGeneration.Tests.ts @@ -0,0 +1,831 @@ +import { AITestClass, Assert } from '@microsoft/ai-test-framework'; +import { ApplicationInsights } from '../../../src/applicationinsights-web'; +import { eOTelSpanKind, eOTelSpanStatusCode, ITelemetryItem } from "@microsoft/applicationinsights-core-js"; + +export class TelemetryItemGenerationTests extends AITestClass { + private static readonly _instrumentationKey = 'b7170927-2d1c-44f1-acec-59f4e1751c11'; + private static readonly _connectionString = `InstrumentationKey=${TelemetryItemGenerationTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "TelemetryItemGenerationTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: TelemetryItemGenerationTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + + } catch (e) { + console.error('Failed to initialize tests: ' + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addSpanKindTests(); + this.addStatusCodeTests(); + this.addAttributeTests(); + this.addTelemetryItemStructureTests(); + this.addComplexScenarioTests(); + } + + private addSpanKindTests(): void { + this.testCase({ + name: "SpanKind: INTERNAL span generates telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("internal-operation", { + kind: eOTelSpanKind.INTERNAL, + attributes: { "operation.name": "internal-task" } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate one telemetry item"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + Assert.ok(item.baseData.properties, "Should have properties"); + } + }); + + this.testCase({ + name: "SpanKind: CLIENT span generates telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("client-request", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "GET", + "http.url": "https://example.com/api", + "custom.attribute": "custom-value" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate one telemetry item"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + // Semantic attributes like http.method are excluded from properties + Assert.ok(!item.baseData.properties || !item.baseData.properties["http.method"], + "http.method should not be in properties (mapped to baseData)"); + // Custom attributes should be in properties + Assert.equal(item.baseData.properties?.["custom.attribute"], "custom-value", + "Custom attributes should be in properties"); + } + }); + + this.testCase({ + name: "SpanKind: SERVER span generates telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("server-handler", { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.method": "POST", + "http.status_code": 200, + "custom.server.id": "server-123" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate one telemetry item"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + // Semantic attributes are excluded from properties + Assert.ok(!item.baseData.properties || !item.baseData.properties["http.method"], + "http.method should not be in properties (mapped to baseData)"); + // Custom attributes should be in properties + Assert.equal(item.baseData.properties?.["custom.server.id"], "server-123", + "Custom attributes should be in properties"); + } + }); + + this.testCase({ + name: "SpanKind: PRODUCER span generates telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("message-producer", { + kind: eOTelSpanKind.PRODUCER, + attributes: { + "messaging.system": "kafka", + "messaging.destination": "orders-topic", + "producer.id": "producer-456" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate one telemetry item"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + // messaging.* attributes may or may not be excluded depending on semantic conventions + // Custom attributes should be in properties + Assert.equal(item.baseData.properties?.["producer.id"], "producer-456", + "Custom attributes should be in properties"); + } + }); + + this.testCase({ + name: "SpanKind: CONSUMER span generates telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("message-consumer", { + kind: eOTelSpanKind.CONSUMER, + attributes: { + "messaging.system": "rabbitmq", + "messaging.operation": "receive", + "consumer.group": "group-789" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate one telemetry item"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + // messaging.* attributes may or may not be excluded depending on semantic conventions + // Custom attributes should be in properties + Assert.equal(item.baseData.properties?.["consumer.group"], "group-789", + "Custom attributes should be in properties"); + } + }); + + this.testCase({ + name: "SpanKind: all span kinds generate independent telemetry", + test: () => { + // Arrange + this._trackCalls = []; + const spanKinds = [ + eOTelSpanKind.INTERNAL, + eOTelSpanKind.CLIENT, + eOTelSpanKind.SERVER, + eOTelSpanKind.PRODUCER, + eOTelSpanKind.CONSUMER + ]; + + // Act + spanKinds.forEach((kind, index) => { + const span = this._ai.startSpan(`span-kind-${index}`, { kind }); + span?.end(); + }); + + // Assert + Assert.equal(this._trackCalls.length, spanKinds.length, + "Each span kind should generate telemetry"); + } + }); + } + + private addStatusCodeTests(): void { + this.testCase({ + name: "StatusCode: UNSET status generates telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("unset-status-span"); + // Don't set status - defaults to UNSET + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + } + }); + + this.testCase({ + name: "StatusCode: OK status generates telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("ok-status-span"); + span?.setStatus({ + code: eOTelSpanStatusCode.OK, + message: "Operation successful" + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + } + }); + + this.testCase({ + name: "StatusCode: ERROR status generates telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("error-status-span"); + span?.setStatus({ + code: eOTelSpanStatusCode.ERROR, + message: "Operation failed" + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + } + }); + + this.testCase({ + name: "StatusCode: status with message includes message in telemetry", + test: () => { + // Arrange + this._trackCalls = []; + const errorMessage = "Database connection timeout"; + + // Act + const span = this._ai.startSpan("status-with-message"); + span?.setStatus({ + code: eOTelSpanStatusCode.ERROR, + message: errorMessage + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + // Note: Implementation may include status message in properties or elsewhere + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData with status information"); + } + }); + + this.testCase({ + name: "StatusCode: changing status before end affects telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("changing-status-span"); + span?.setStatus({ code: eOTelSpanStatusCode.OK }); + span?.setStatus({ code: eOTelSpanStatusCode.ERROR, message: "Changed to error" }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + // The final status (ERROR) should be reflected in telemetry + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData with final status"); + } + }); + + this.testCase({ + name: "StatusCode: multiple spans with different statuses", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span1 = this._ai.startSpan("span-ok"); + span1?.setStatus({ code: eOTelSpanStatusCode.OK }); + span1?.end(); + + const span2 = this._ai.startSpan("span-error"); + span2?.setStatus({ code: eOTelSpanStatusCode.ERROR }); + span2?.end(); + + const span3 = this._ai.startSpan("span-unset"); + // No status set + span3?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 3, "Should generate telemetry for all spans"); + } + }); + } + + private addAttributeTests(): void { + this.testCase({ + name: "Attributes: span with no attributes generates telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("no-attributes-span"); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + } + }); + + this.testCase({ + name: "Attributes: span with string attributes", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("string-attrs-span", { + attributes: { + "user.id": "user123", + "session.id": "session456", + "operation.name": "checkout" + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData?.properties, "Should have properties"); + // These custom attributes should be in properties + Assert.equal(item.baseData.properties["user.id"], "user123", + "Should include custom string attributes"); + Assert.equal(item.baseData.properties["session.id"], "session456", + "Should include custom string attributes"); + // operation.name is a context tag key and gets excluded from properties + } + }); + + this.testCase({ + name: "Attributes: span with number attributes", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("number-attrs-span", { + attributes: { + "request.size": 1024, + "response.time": 156.78, + "retry.count": 3 + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData?.properties, "Should have properties"); + // Custom number attributes should be in properties + Assert.equal(item.baseData.properties["request.size"], 1024, + "Should include custom number attributes"); + Assert.equal(item.baseData.properties["response.time"], 156.78, + "Should include custom number attributes"); + } + }); + + this.testCase({ + name: "Attributes: span with boolean attributes", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("boolean-attrs-span", { + attributes: { + "cache.hit": true, + "auth.required": false, + "retry.enabled": true + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData?.properties, "Should have properties"); + Assert.equal(item.baseData.properties["cache.hit"], true, + "Should include boolean attributes"); + } + }); + + this.testCase({ + name: "Attributes: span with mixed type attributes", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("mixed-attrs-span", { + attributes: { + "string.attr": "value", + "number.attr": 42, + "boolean.attr": true, + "float.attr": 3.14 + } + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData?.properties, "Should have properties"); + Assert.equal(item.baseData.properties["string.attr"], "value"); + Assert.equal(item.baseData.properties["number.attr"], 42); + Assert.equal(item.baseData.properties["boolean.attr"], true); + } + }); + + this.testCase({ + name: "Attributes: setAttribute after creation adds to telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("dynamic-attrs-span"); + span?.setAttribute("initial.attr", "initial"); + span?.setAttribute("added.later", "later-value"); + span?.setAttribute("number.added", 999); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData?.properties, "Should have properties"); + Assert.equal(item.baseData.properties["initial.attr"], "initial"); + Assert.equal(item.baseData.properties["added.later"], "later-value"); + Assert.equal(item.baseData.properties["number.added"], 999); + } + }); + + this.testCase({ + name: "Attributes: setAttributes adds multiple attributes to telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("batch-attrs-span"); + span?.setAttributes({ + "batch.attr1": "value1", + "batch.attr2": "value2", + "batch.attr3": 123 + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData?.properties, "Should have properties"); + Assert.equal(item.baseData.properties["batch.attr1"], "value1"); + Assert.equal(item.baseData.properties["batch.attr2"], "value2"); + Assert.equal(item.baseData.properties["batch.attr3"], 123); + } + }); + + this.testCase({ + name: "Attributes: updating attribute value reflects in telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("update-attr-span"); + span?.setAttribute("status", "pending"); + span?.setAttribute("status", "in-progress"); + span?.setAttribute("status", "completed"); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData?.properties, "Should have properties"); + Assert.equal(item.baseData.properties["status"], "completed", + "Should reflect final attribute value"); + } + }); + } + + private addTelemetryItemStructureTests(): void { + this.testCase({ + name: "Structure: telemetry item has required fields", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("structure-test-span"); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + + Assert.ok(item.name, "Should have name"); + Assert.ok(item.baseData, "Should have baseData"); + Assert.ok(item.baseData.properties, "Should have properties"); + } + }); + + this.testCase({ + name: "Structure: span name is in telemetry", + test: () => { + // Arrange + this._trackCalls = []; + const spanName = "my-custom-operation"; + + // Act + const span = this._ai.startSpan(spanName); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + + // Span name is in baseData.name, not properties.name + Assert.ok(item.baseData, "Should have baseData"); + Assert.equal(item.baseData.name, spanName, + "Span name should be in baseData.name"); + } + }); + + this.testCase({ + name: "Structure: updated span name reflects in telemetry", + test: () => { + // Arrange + this._trackCalls = []; + const originalName = "original-name"; + const updatedName = "updated-name"; + + // Act + const span = this._ai.startSpan(originalName); + span?.updateName(updatedName); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + + // Updated span name should be in baseData.name + Assert.ok(item.baseData, "Should have baseData"); + Assert.equal(item.baseData.name, updatedName, + "Updated span name should be in baseData.name"); + } + }); + + this.testCase({ + name: "Structure: trace context is in telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("trace-context-span"); + const spanContext = span?.spanContext(); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + + // Telemetry should include trace context information + Assert.ok(spanContext, "Span should have context"); + Assert.ok(spanContext?.traceId, "Should have traceId"); + Assert.ok(spanContext?.spanId, "Should have spanId"); + } + }); + + this.testCase({ + name: "Structure: multiple spans generate separate telemetry items", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span1 = this._ai.startSpan("span-1"); + span1?.end(); + + const span2 = this._ai.startSpan("span-2"); + span2?.end(); + + const span3 = this._ai.startSpan("span-3"); + span3?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 3, "Should generate 3 telemetry items"); + + // Span names are in baseData.name, not properties.name + const names = this._trackCalls.map(item => item.baseData?.name); + Assert.ok(names.includes("span-1"), "Should include span-1"); + Assert.ok(names.includes("span-2"), "Should include span-2"); + Assert.ok(names.includes("span-3"), "Should include span-3"); + } + }); + } + + private addComplexScenarioTests(): void { + this.testCase({ + name: "Complex: span with kind, status, and attributes", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("complex-span", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "http.method": "POST", + "http.url": "https://api.example.com/users", + "http.status_code": 201, + "request.id": "req-12345", + "user.action": "create" + } + }); + span?.setStatus({ + code: eOTelSpanStatusCode.OK, + message: "User created successfully" + }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData, "Should have baseData"); + // Semantic attributes are excluded from properties + Assert.ok(!item.baseData.properties || !item.baseData.properties["http.method"], + "http.method should not be in properties"); + // Custom attributes should be in properties + Assert.equal(item.baseData.properties?.["request.id"], "req-12345", + "Custom attributes should be in properties"); + Assert.equal(item.baseData.properties?.["user.action"], "create", + "Custom attributes should be in properties"); + } + }); + + this.testCase({ + name: "Complex: parent-child spans generate separate telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const parentSpan = this._ai.startSpan("parent-operation"); + const parentContext = parentSpan?.spanContext(); + + const childSpan = this._ai.startSpan("child-operation", undefined, parentContext); + childSpan?.end(); + + parentSpan?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 2, "Should generate telemetry for both spans"); + + // Span names are in baseData.name, not properties.name + const names = this._trackCalls.map(item => item.baseData?.name); + Assert.ok(names.includes("child-operation"), "Should include child telemetry"); + Assert.ok(names.includes("parent-operation"), "Should include parent telemetry"); + } + }); + + this.testCase({ + name: "Complex: span with dynamic attributes during execution", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const span = this._ai.startSpan("dynamic-execution-span", { + attributes: { "phase": "start" } + }); + + span?.setAttribute("phase", "processing"); + span?.setAttribute("items.processed", 50); + + span?.setAttribute("phase", "finalizing"); + span?.setAttribute("items.processed", 100); + + span?.setStatus({ code: eOTelSpanStatusCode.OK }); + span?.end(); + + // Assert + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + const item = this._trackCalls[0]; + Assert.ok(item.baseData?.properties, "Should have properties"); + Assert.equal(item.baseData.properties["phase"], "finalizing", + "Should have final phase value"); + Assert.equal(item.baseData.properties["items.processed"], 100, + "Should have final processed count"); + } + }); + + this.testCase({ + name: "Complex: all span kinds with attributes and status", + test: () => { + // Arrange + this._trackCalls = []; + const testData = [ + { kind: eOTelSpanKind.INTERNAL, name: "internal-op", attr: "internal-value" }, + { kind: eOTelSpanKind.CLIENT, name: "client-op", attr: "client-value" }, + { kind: eOTelSpanKind.SERVER, name: "server-op", attr: "server-value" }, + { kind: eOTelSpanKind.PRODUCER, name: "producer-op", attr: "producer-value" }, + { kind: eOTelSpanKind.CONSUMER, name: "consumer-op", attr: "consumer-value" } + ]; + + // Act + testData.forEach(data => { + const span = this._ai.startSpan(data.name, { + kind: data.kind, + attributes: { "operation.type": data.attr } + }); + span?.setStatus({ code: eOTelSpanStatusCode.OK }); + span?.end(); + }); + + // Assert + Assert.equal(this._trackCalls.length, testData.length, + "Should generate telemetry for all span types"); + + testData.forEach(data => { + // Span names are in baseData.name, not properties.name + const telemetry = this._trackCalls.find( + item => item.baseData?.name === data.name + ); + Assert.ok(telemetry, `Should have telemetry for ${data.name}`); + // Custom attributes should be in properties + Assert.equal(telemetry?.baseData?.properties?.["operation.type"], data.attr, + `Should have correct attributes for ${data.name}`); + }); + } + }); + + this.testCase({ + name: "Complex: non-recording spans do not generate telemetry", + test: () => { + // Arrange + this._trackCalls = []; + + // Act + const recordingSpan = this._ai.startSpan("recording-span", { recording: true }); + recordingSpan?.end(); + + const nonRecordingSpan = this._ai.startSpan("non-recording-span", { recording: false }); + nonRecordingSpan?.end(); + + // Assert + // Recording span should generate telemetry, non-recording should not + // Span names are in baseData.name, not properties.name + const recordingTelemetry = this._trackCalls.find( + item => item.baseData?.name === "recording-span" + ); + const nonRecordingTelemetry = this._trackCalls.find( + item => item.baseData?.name === "non-recording-span" + ); + + Assert.ok(recordingTelemetry, "Recording span should generate telemetry"); + Assert.ok(!nonRecordingTelemetry, "Non-recording span should not generate telemetry"); + } + }); + } +} diff --git a/AISKU/Tests/Unit/src/ThrottleSentMessage.tests.ts b/AISKU/Tests/Unit/src/ThrottleSentMessage.tests.ts index 1e4fc388a..50caf5325 100644 --- a/AISKU/Tests/Unit/src/ThrottleSentMessage.tests.ts +++ b/AISKU/Tests/Unit/src/ThrottleSentMessage.tests.ts @@ -1,6 +1,6 @@ import { ApplicationInsights, ApplicationInsightsContainer, IApplicationInsights, IConfig, IConfiguration, LoggingSeverity, Snippet, _eInternalMessageId } from '../../../src/applicationinsights-web' import { AITestClass, Assert} from '@microsoft/ai-test-framework'; -import { IThrottleInterval, IThrottleLimit, IThrottleMgrConfig } from '@microsoft/applicationinsights-common'; +import { IThrottleInterval, IThrottleLimit, IThrottleMgrConfig } from '@microsoft/applicationinsights-core-js'; import { SinonSpy } from 'sinon'; import { AppInsightsSku } from '../../../src/AISku'; import { createSnippetV5 } from './testSnippetV5'; diff --git a/AISKU/Tests/Unit/src/TraceContext.Tests.ts b/AISKU/Tests/Unit/src/TraceContext.Tests.ts new file mode 100644 index 000000000..49b75be42 --- /dev/null +++ b/AISKU/Tests/Unit/src/TraceContext.Tests.ts @@ -0,0 +1,735 @@ +import { AITestClass, Assert } from '@microsoft/ai-test-framework'; +import { ApplicationInsights } from '../../../src/applicationinsights-web'; +import { IOTelSpanOptions, eOTelSpanKind, ITelemetryItem, isUndefined, useSpan, isNumber } from "@microsoft/applicationinsights-core-js"; +import { isFunction, objIs } from '@nevware21/ts-utils'; + +export class TraceContextTests extends AITestClass { + private static readonly _instrumentationKey = 'b7170927-2d1c-44f1-acec-59f4e1751c11'; + private static readonly _connectionString = `InstrumentationKey=${TraceContextTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "TraceContextTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: TraceContextTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + + } catch (e) { + console.error('Failed to initialize tests: ' + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addGetTraceCtxTests(); + this.addActiveSpanTests(); + this.addsetActiveSpanTests(); + this.addIntegrationTests(); + } + + private addGetTraceCtxTests(): void { + this.testCase({ + name: "getTraceCtx: should return valid trace context after starting a span", + test: () => { + // Arrange + const spanName = "test-span-with-context"; + + // Act + const span = this._ai.startSpan(spanName); + const traceCtx = this._ai.getTraceCtx(); + + // Assert + Assert.ok(span !== null, "Span should be created"); + Assert.ok(traceCtx !== null && traceCtx !== undefined, "Should return trace context after starting span"); + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + Assert.ok(traceCtx!.traceId, "Trace context should have traceId"); + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + Assert.equal("", traceCtx!.spanId, "Trace context should not have a spanId (the default SDK initialization)"); + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + Assert.ok(isUndefined(traceCtx!.traceFlags), "Trace context should NOT have have traceFlags"); + + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + useSpan(this._ai.core, span!, () => { + const nestedTraceCtx = this._ai.getTraceCtx(); + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + Assert.equal(traceCtx!.traceId, nestedTraceCtx!.traceId, "TraceId should be the same within the span context"); + Assert.equal(span?.spanContext().traceId, nestedTraceCtx?.traceId, "TraceId should match the active span's traceId"); + Assert.equal(span?.spanContext().spanId, nestedTraceCtx?.spanId, "SpanId should match the active span's spanId"); + Assert.equal(span?.spanContext().traceFlags, nestedTraceCtx?.traceFlags, "TraceFlags should match the active span's traceFlags"); + }); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "getTraceCtx: should return trace context matching active span", + test: () => { + // Arrange + const spanName = "context-matching-span"; + + // Act + const span = this._ai.startSpan(spanName); + const traceCtx = this._ai.getTraceCtx(); + const spanContext = span?.spanContext(); + + // Assert + Assert.ok(span !== null, "Span should be created"); + Assert.ok(traceCtx !== null && traceCtx !== undefined, "Trace context should exist"); + Assert.ok(spanContext !== null && spanContext !== undefined, "Span context should exist"); + + Assert.equal(traceCtx.traceId, spanContext.traceId, "Trace context traceId should match span context"); + Assert.notEqual(traceCtx.spanId, spanContext.spanId, "Trace context spanId should match span context"); + + useSpan(this._ai.core, span!, () => { + const activeTraceCtx = this._ai.getTraceCtx(); + Assert.equal(activeTraceCtx?.traceId, spanContext.traceId, "The active traceId should match span context"); + Assert.equal(activeTraceCtx?.spanId, spanContext.spanId, "The active spanId should match span context"); + }); + + Assert.equal(traceCtx.traceId, spanContext.traceId, "Trace context traceId should match span context"); + Assert.notEqual(traceCtx.spanId, spanContext.spanId, "Trace context spanId should match span context"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "getTraceCtx: should have valid traceId format (32 hex chars)", + test: () => { + // Arrange + const span = this._ai.startSpan("format-test-span"); + + // Act + const traceCtx = this._ai.getTraceCtx(); + + // Assert + Assert.ok(traceCtx !== null && traceCtx !== undefined, "Trace context should exist"); + + if (traceCtx && traceCtx.traceId) { + Assert.equal(traceCtx.traceId.length, 32, "TraceId should be 32 characters"); + Assert.ok(/^[0-9a-f]{32}$/i.test(traceCtx.traceId), + "TraceId should be 32 hex characters"); + } + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "getTraceCtx: should have valid spanId format (16 hex chars)", + test: () => { + // Arrange + const span = this._ai.startSpan("spanid-test-span"); + + // Act + const traceCtx = this._ai.getTraceCtx(); + + // Assert + Assert.ok(traceCtx !== null && traceCtx !== undefined, "Trace context should exist"); + + if (traceCtx && traceCtx.spanId) { + Assert.equal(traceCtx.spanId.length, 16, "SpanId should be 16 characters"); + Assert.ok(/^[0-9a-f]{16}$/i.test(traceCtx.spanId), + "SpanId should be 16 hex characters"); + } + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "getTraceCtx: should persist context across multiple calls", + test: () => { + // Arrange + const span = this._ai.startSpan("persistence-span"); + + // Act + const traceCtx1 = this._ai.getTraceCtx(); + const traceCtx2 = this._ai.getTraceCtx(); + const traceCtx3 = this._ai.getTraceCtx(); + + // Assert + Assert.ok(traceCtx1 !== null && traceCtx1 !== undefined, "First call should return context"); + Assert.ok(traceCtx2 !== null && traceCtx2 !== undefined, "Second call should return context"); + Assert.ok(traceCtx3 !== null && traceCtx3 !== undefined, "Third call should return context"); + + if (traceCtx1 && traceCtx2 && traceCtx3) { + Assert.equal(traceCtx1.traceId, traceCtx2.traceId, "TraceId should be consistent"); + Assert.equal(traceCtx2.traceId, traceCtx3.traceId, "TraceId should be consistent"); + Assert.equal(traceCtx1.spanId, traceCtx2.spanId, "SpanId should be consistent"); + Assert.equal(traceCtx2.spanId, traceCtx3.spanId, "SpanId should be consistent"); + } + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "getTraceCtx: should return context for child spans with same traceId", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("parent-span"); + const parentCtx = this._ai.getTraceCtx(); + + // Act - create child span + const childSpan = this._ai.startSpan("child-span", undefined, parentCtx || undefined); + + let childCtx; + useSpan(this._ai.core, childSpan!, () => { + childCtx = this._ai.getTraceCtx(); + }); + + // Assert + Assert.ok(parentCtx !== null && parentCtx !== undefined, "Parent context should exist"); + Assert.ok(childCtx !== null && childCtx !== undefined, "Child context should exist"); + + Assert.equal(childCtx.traceId, parentCtx.traceId, "Child span should have same traceId as parent"); + Assert.notEqual(childCtx.spanId, parentCtx.spanId, "Child span should have different spanId from parent"); + + Assert.equal(childSpan?.spanContext().traceId, parentCtx.traceId, "Child span should have same traceId as parent"); + Assert.notEqual(childSpan?.spanContext().spanId, parentCtx.spanId, "Child span should have different spanId from parent"); + Assert.equal(childSpan?.spanContext().spanId, childCtx.spanId, "Child spanId should match its context"); + Assert.equal(childSpan?.spanContext().traceId, childCtx.traceId, "Child traceId should match its context"); + + // Cleanup + childSpan?.end(); + parentSpan?.end(); + } + }); + } + + private addActiveSpanTests(): void { + this.testCase({ + name: "activeSpan: should return null when no span is active (via trace provider)", + test: () => { + // Assert + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + const activeSpan = this._ai.getActiveSpan(); + Assert.ok(activeSpan, "Should Always return a non-null span when no span is active"); + Assert.equal(false, activeSpan.isRecording(), "Returned span should be a non-recording span"); + } + }); + + this.testCase({ + name: "activeSpan: should return null when createNew is false and no span is active", + test: () => { + // Act + const activeSpan = this._ai.getActiveSpan(false); + + // Assert + Assert.equal(activeSpan, null, "Should return null when createNew is false and no active span exists"); + } + }); + + this.testCase({ + name: "activeSpan: should return existing span when createNew is false and span is active", + test: () => { + // Arrange + const span = this._ai.startSpan("test-span"); + this._ai.setActiveSpan(span); + + // Act + const activeSpan = this._ai.getActiveSpan(false); + + // Assert + Assert.ok(activeSpan, "Should return the active span"); + Assert.equal(activeSpan, span, "Should return the same span object"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "activeSpan: should return active span after setActiveSpan (via trace provider)", + test: () => { + // Arrange + const span = this._ai.startSpan("active-span-test"); + + // Act + this._ai.setActiveSpan(span); + const activeSpan = this._ai.getActiveSpan(); + + // Assert + Assert.ok(activeSpan !== null, "Should return the active span"); + Assert.equal(activeSpan.name, span.name, "Active span should match the set span"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "activeSpan: should return the most recently set active span", + test: () => { + // Arrange + const span1 = this._ai.startSpan("span-1"); + const span2 = this._ai.startSpan("span-2"); + const span3 = this._ai.startSpan("span-3"); + + // Act & Assert + this._ai.setActiveSpan(span1); + let activeSpan = this._ai.getActiveSpan(); + Assert.equal(activeSpan.name, span1.name, "Should return span1 as active"); + + this._ai.setActiveSpan(span2); + activeSpan = this._ai.getActiveSpan(); + Assert.equal(activeSpan.name, span2.name, "Should return span2 as active"); + + this._ai.setActiveSpan(span3); + activeSpan = this._ai.getActiveSpan(); + Assert.equal(activeSpan.name, span3.name, "Should return span3 as active"); + + // Cleanup + span1?.end(); + span2?.end(); + span3?.end(); + } + }); + + this.testCase({ + name: "activeSpan: active span should have valid span context", + test: () => { + // Arrange + const span = this._ai.startSpan("context-check-span"); + + // Act + this._ai.setActiveSpan(span); + const activeSpan = this._ai.getActiveSpan(); + const spanContext = activeSpan.spanContext(); + + // Assert + Assert.ok(activeSpan !== null, "Active span should exist"); + Assert.ok(objIs(activeSpan, span), "Active span should match the set span"); + Assert.ok(spanContext !== null && spanContext !== undefined, "Active span should have valid context"); + + Assert.ok(spanContext.traceId, "Should have traceId"); + Assert.ok(spanContext.spanId, "Should have spanId"); + Assert.ok(isUndefined(spanContext.traceFlags), "Should have default traceFlags (undefined)"); + Assert.equal(undefined, spanContext?.traceFlags, "TraceFlags should not have sampled flag set by default"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "activeSpan: should work with recording spans", + test: () => { + // Arrange + const options: IOTelSpanOptions = { + recording: true, + kind: eOTelSpanKind.CLIENT + }; + const span = this._ai.startSpan("recording-span", options); + + // Act + this._ai.setActiveSpan(span); + const activeSpan = this._ai.getActiveSpan(); + + // Assert + Assert.ok(activeSpan !== null, "Active span should exist"); + Assert.ok(activeSpan.isRecording(), "Active span should be recording"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "activeSpan: should work with non-recording spans", + test: () => { + // Arrange + const options: IOTelSpanOptions = { + recording: false + }; + const span = this._ai.startSpan("non-recording-span", options); + + // Act + this._ai.setActiveSpan(span); + const activeSpan = this._ai.getActiveSpan(); + + // Assert + Assert.ok(activeSpan !== null, "Active span should exist"); + Assert.ok(!activeSpan.isRecording(), "Active span should not be recording"); + + // Cleanup + span?.end(); + } + }); + } + + private addsetActiveSpanTests(): void { + this.testCase({ + name: "setActiveSpan: should set a span as active", + test: () => { + // Arrange + const span = this._ai.startSpan("set-active-test"); + + // Act + const scope = this._ai.setActiveSpan(span); + const activeSpan = this._ai.getActiveSpan(); + + // Assert + Assert.ok(scope !== null, "Scope should be returned"); + Assert.equal(scope.span, span, "Scope.span should equal the passed span"); + Assert.ok(activeSpan !== null, "Active span should be set"); + Assert.equal(activeSpan, span, "ActiveSpan() should return the same span object"); + Assert.equal(activeSpan.name, span.name, "Set span should be the active span"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "setActiveSpan: should update getTraceCtx to reflect active span", + test: () => { + // Arrange + const span = this._ai.startSpan("trace-ctx-update-test"); + + // Act + const scope = this._ai.setActiveSpan(span); + const activeSpan = this._ai.getActiveSpan(); + const traceCtx = this._ai.getTraceCtx(); + const spanContext = span.spanContext(); + + // Assert + Assert.ok(scope !== null, "Scope should be returned"); + Assert.equal(scope.span, span, "Scope.span should equal the passed span"); + Assert.equal(activeSpan, span, "ActiveSpan() should return the same span object"); + Assert.ok(traceCtx !== null && traceCtx !== undefined, + "Trace context should be updated"); + + if (traceCtx && spanContext) { + Assert.equal(traceCtx.traceId, spanContext.traceId, + "Trace context should match active span"); + Assert.equal(traceCtx.spanId, spanContext.spanId, + "Trace context spanId should match active span"); + } + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "setActiveSpan: should allow switching between multiple spans", + test: () => { + // Arrange + const span1 = this._ai.startSpan("switch-span-1"); + const span2 = this._ai.startSpan("switch-span-2"); + + // Act & Assert + // Set first span as active + let scope = this._ai.setActiveSpan(span1); + let activeSpan = this._ai.getActiveSpan(); + Assert.equal(scope.span, span1, "Scope.span should equal span1"); + Assert.equal(activeSpan, span1, "ActiveSpan() should return span1"); + Assert.equal(activeSpan.name, span1.name, "First span should be active"); + + // Switch to second span + scope = this._ai.setActiveSpan(span2); + activeSpan = this._ai.getActiveSpan(); + Assert.equal(scope.span, span2, "Scope.span should equal span2"); + Assert.equal(activeSpan, span2, "ActiveSpan() should return span2"); + Assert.equal(activeSpan.name, span2.name, "Second span should be active"); + + // Switch back to first span + scope = this._ai.setActiveSpan(span1); + activeSpan = this._ai.getActiveSpan(); + Assert.equal(scope.span, span1, "Scope.span should equal span1 again"); + Assert.equal(activeSpan, span1, "ActiveSpan() should return span1 again"); + Assert.equal(activeSpan.name, span1.name, "First span should be active again"); + + // Cleanup + span1?.end(); + span2?.end(); + } + }); + + this.testCase({ + name: "setActiveSpan: should work with spans of different kinds", + test: () => { + // Arrange + const clientSpan = this._ai.startSpan("client-span", { kind: eOTelSpanKind.CLIENT }); + const serverSpan = this._ai.startSpan("server-span", { kind: eOTelSpanKind.SERVER }); + + // Act & Assert + let scope = this._ai.setActiveSpan(clientSpan); + let activeSpan = this._ai.getActiveSpan(); + Assert.equal(scope.span, clientSpan, "Scope.span should equal clientSpan"); + Assert.equal(activeSpan, clientSpan, "ActiveSpan() should return clientSpan"); + Assert.equal(activeSpan.kind, eOTelSpanKind.CLIENT, + "Client span should be active with correct kind"); + + scope = this._ai.setActiveSpan(serverSpan); + activeSpan = this._ai.getActiveSpan(); + Assert.equal(scope.span, serverSpan, "Scope.span should equal serverSpan"); + Assert.equal(activeSpan, serverSpan, "ActiveSpan() should return serverSpan"); + Assert.equal(activeSpan.kind, eOTelSpanKind.SERVER, + "Server span should be active with correct kind"); + + // Cleanup + clientSpan?.end(); + serverSpan?.end(); + } + }); + + this.testCase({ + name: "setActiveSpan: should work with spans that have attributes", + test: () => { + // Arrange + const attributes = { + "http.method": "GET", + "http.url": "https://example.com", + "custom.attribute": "test-value" + }; + const span = this._ai.startSpan("attributed-span", { attributes }); + + // Act + const scope = this._ai.setActiveSpan(span); + const activeSpan = this._ai.getActiveSpan(); + + // Assert + Assert.ok(scope !== null, "Scope should be returned"); + Assert.equal(scope.span, span, "Scope.span should equal the passed span"); + Assert.ok(activeSpan !== null, "Active span should exist"); + Assert.equal(activeSpan, span, "ActiveSpan() should return the same span object"); + Assert.equal(activeSpan.name, span.name, "Span name should match"); + + const spanAttributes = activeSpan.attributes; + Assert.equal(spanAttributes["http.method"], "GET", + "Attributes should be preserved"); + Assert.equal(spanAttributes["http.url"], "https://example.com", + "Attributes should be preserved"); + Assert.equal(spanAttributes["custom.attribute"], "test-value", + "Custom attributes should be preserved"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "setActiveSpan: should handle ended spans", + test: () => { + // Arrange + const span = this._ai.startSpan("ended-span"); + + // Act + span.end(); + + const scope = this._ai.setActiveSpan(span); + const activeSpan = this._ai.getActiveSpan(); + + // Assert + Assert.ok(scope !== null, "Scope should be returned"); + Assert.equal(scope.span, span, "Scope.span should equal the passed ended span"); + Assert.ok(activeSpan !== null, "Should be able to set ended span as active"); + Assert.equal(activeSpan, span, "ActiveSpan() should return the same ended span object"); + Assert.ok(activeSpan.ended, "Active span should be marked as ended"); + + // Cleanup is already done (span.end() called) + } + }); + } + + private addIntegrationTests(): void { + this.testCase({ + name: "Integration: getTraceCtx, activeSpan, and setActiveSpan should work together", + test: () => { + // Arrange + const span1 = this._ai.startSpan("integration-span-1"); + const span2 = this._ai.startSpan("integration-span-2"); + + // Act & Assert - Set first span active + this._ai.setActiveSpan(span1); + + let activeSpan = this._ai.getActiveSpan(); + let traceCtx = this._ai.getTraceCtx(); + let span1Context = span1.spanContext(); + + Assert.equal(activeSpan.name, span1.name, "Active span should be span1"); + Assert.equal(traceCtx?.spanId, span1Context.spanId, + "Trace context should match span1"); + + // Switch to second span + this._ai.setActiveSpan(span2); + + activeSpan = this._ai.getActiveSpan(); + traceCtx = this._ai.getTraceCtx(); + let span2Context = span2.spanContext(); + + Assert.equal(activeSpan.name, span2.name, "Active span should be span2"); + Assert.equal(traceCtx?.spanId, span2Context.spanId, + "Trace context should match span2"); + + // Cleanup + span1?.end(); + span2?.end(); + } + }); + + this.testCase({ + name: "Integration: parent-child span relationship via getTraceCtx", + test: () => { + // Arrange & Act + const parentSpan = this._ai.startSpan("integration-parent"); + + this._ai.setActiveSpan(parentSpan); + const parentCtx = this._ai.getTraceCtx(); + + // Create child span using parent context + const childSpan = this._ai.startSpan("integration-child", undefined, parentCtx || undefined); + this._ai.setActiveSpan(childSpan); + + const childCtx = this._ai.getTraceCtx(); + const activeSpan = this._ai.getActiveSpan(); + + // Assert + Assert.ok(parentCtx !== null && parentCtx !== undefined, "Parent context should exist"); + Assert.ok(childCtx !== null && childCtx !== undefined, "Child context should exist"); + + Assert.equal(childCtx.traceId, parentCtx.traceId, "Child should inherit parent's traceId"); + Assert.notEqual(childCtx.spanId, parentCtx.spanId, "Child should have different spanId"); + + Assert.equal(activeSpan.name, "integration-child", "Active span should be the child span"); + + // Cleanup + childSpan?.end(); + + parentSpan?.end(); + } + }); + + this.testCase({ + name: "Integration: multiple spans with trace context propagation", + test: () => { + // Arrange + const rootSpan = this._ai.startSpan("root-span"); + + this._ai.setActiveSpan(rootSpan); + const rootCtx = this._ai.getTraceCtx(); + + // Create first child + const child1Span = this._ai.startSpan("child-1", undefined, rootCtx || undefined); + this._ai.setActiveSpan(child1Span); + const child1Ctx = this._ai.getTraceCtx(); + + // Create second child (sibling to first child) + const child2Span = this._ai.startSpan("child-2", undefined, rootCtx || undefined); + this._ai.setActiveSpan(child2Span); + const child2Ctx = this._ai.getTraceCtx(); + + // Assert - all should share the same traceId + Assert.equal(child1Ctx.traceId, rootCtx.traceId, + "Child 1 should share root traceId"); + Assert.equal(child2Ctx.traceId, rootCtx.traceId, + "Child 2 should share root traceId"); + + // But have different spanIds + Assert.notEqual(child1Ctx.spanId, rootCtx.spanId, + "Child 1 should have different spanId"); + Assert.notEqual(child2Ctx.spanId, rootCtx.spanId, + "Child 2 should have different spanId"); + Assert.notEqual(child1Ctx.spanId, child2Ctx.spanId, + "Siblings should have different spanIds"); + + // Cleanup + child2Span?.end(); + child1Span?.end(); + + rootSpan?.end(); + } + }); + + this.testCase({ + name: "Integration: trace provider availability check", + test: () => { + // Act + const provider = this._ai.core.getTraceProvider(); + Assert.equal(this._ai.getTraceProvider(), provider, "Core and AI instance should return same trace provider"); + + // Assert + Assert.ok(provider !== null && provider !== undefined, "Trace provider should be available"); + + Assert.ok(isFunction(provider.createSpan), "Provider should have createSpan method"); + Assert.ok(isFunction(this._ai.getActiveSpan), "Provider should have activeSpan method"); + Assert.ok(isFunction(this._ai.setActiveSpan), "Provider should have setActiveSpan method"); + Assert.ok(isFunction(provider.getProviderId), "Provider should have getProviderId method"); + Assert.ok(isFunction(provider.isAvailable), "Provider should have isAvailable method"); + } + }); + + this.testCase({ + name: "Integration: trace provider isAvailable should reflect initialization state", + test: () => { + // Act + const provider = this._ai.core.getTraceProvider(); + + // Assert + const isAvailable = provider.isAvailable(); + Assert.ok(typeof isAvailable === 'boolean', "isAvailable should return boolean"); + Assert.ok(isAvailable, "Provider should be available after SDK initialization"); + } + }); + + this.testCase({ + name: "Integration: trace provider should have identifiable providerId", + test: () => { + // Act + const provider = this._ai.core.getTraceProvider(); + + // Assert + if (provider) { + const providerId = provider.getProviderId(); + Assert.ok(providerId, "Provider should have an ID"); + Assert.ok(typeof providerId === 'string', + "Provider ID should be a string"); + Assert.ok(providerId.length > 0, + "Provider ID should not be empty"); + } + } + }); + } +} diff --git a/AISKU/Tests/Unit/src/TraceProvider.Tests.ts b/AISKU/Tests/Unit/src/TraceProvider.Tests.ts new file mode 100644 index 000000000..3e11d09aa --- /dev/null +++ b/AISKU/Tests/Unit/src/TraceProvider.Tests.ts @@ -0,0 +1,716 @@ +import { AITestClass, Assert } from '@microsoft/ai-test-framework'; +import { ApplicationInsights } from '../../../src/applicationinsights-web'; +import { + IReadableSpan, IOTelSpanOptions, eOTelSpanKind, ITraceProvider, ITelemetryItem, + isFunction +} from "@microsoft/applicationinsights-core-js"; + +export class TraceProviderTests extends AITestClass { + private static readonly _instrumentationKey = 'b7170927-2d1c-44f1-acec-59f4e1751c11'; + private static readonly _connectionString = `InstrumentationKey=${TraceProviderTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "TraceProviderTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: TraceProviderTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + + } catch (e) { + console.error('Failed to initialize tests: ' + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addProviderAvailabilityTests(); + this.addGetProviderIdTests(); + this.addIsAvailableTests(); + this.addCreateSpanTests(); + this.addProviderIntegrationTests(); + } + + private addProviderAvailabilityTests(): void { + this.testCase({ + name: "TraceProvider: getTraceProvider should return provider instance", + test: () => { + // Act + const provider = this._ai.core.getTraceProvider(); + + // Assert + Assert.ok(provider !== null && provider !== undefined, + "Should return a trace provider instance"); + Assert.ok(typeof provider === 'object', + "Provider should be an object"); + } + }); + + this.testCase({ + name: "TraceProvider: provider should have all required methods", + test: () => { + // Act + const provider = this._ai.core.getTraceProvider(); + Assert.equal(this._ai.getTraceProvider(), provider, "Core and AI instance should return same trace provider"); + + // Assert + Assert.ok(provider, "Provider should exist"); + if (provider) { + Assert.ok(isFunction(provider.createSpan), "Should have createSpan method"); + Assert.ok(isFunction(this._ai.getActiveSpan), "Should have activeSpan method"); + Assert.ok(isFunction(this._ai.setActiveSpan), "Should have setActiveSpan method"); + Assert.ok(isFunction(provider.getProviderId), "Should have getProviderId method"); + Assert.ok(isFunction(provider.isAvailable), "Should have isAvailable method"); + } + } + }); + + this.testCase({ + name: "TraceProvider: provider should be available after SDK initialization", + test: () => { + // Act + const provider = this._ai.core.getTraceProvider(); + + // Assert + Assert.ok(provider !== null, "Provider should not be null"); + Assert.ok(provider !== undefined, "Provider should not be undefined"); + } + }); + + this.testCase({ + name: "TraceProvider: multiple calls to getTraceProvider should return same provider", + test: () => { + // Act + const provider1 = this._ai.core.getTraceProvider(); + const provider2 = this._ai.core.getTraceProvider(); + const provider3 = this._ai.core.getTraceProvider(); + + // Assert + Assert.ok(provider1 === provider2, + "First and second calls should return same provider"); + Assert.ok(provider2 === provider3, + "Second and third calls should return same provider"); + } + }); + } + + private addGetProviderIdTests(): void { + this.testCase({ + name: "getProviderId: should return a string identifier", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + const providerId = provider?.getProviderId(); + + // Assert + Assert.ok(providerId !== null && providerId !== undefined, + "Provider ID should not be null or undefined"); + Assert.ok(typeof providerId === 'string', + "Provider ID should be a string"); + } + }); + + this.testCase({ + name: "getProviderId: should return non-empty string", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + const providerId = provider?.getProviderId(); + + // Assert + if (providerId) { + Assert.ok(providerId.length > 0, + "Provider ID should not be empty"); + } + } + }); + + this.testCase({ + name: "getProviderId: should return consistent ID across multiple calls", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + const providerId1 = provider?.getProviderId(); + const providerId2 = provider?.getProviderId(); + const providerId3 = provider?.getProviderId(); + + // Assert + Assert.equal(providerId1, providerId2, + "Provider ID should be consistent across calls"); + Assert.equal(providerId2, providerId3, + "Provider ID should be consistent across calls"); + } + }); + + this.testCase({ + name: "getProviderId: should return identifiable name", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + const providerId = provider?.getProviderId(); + + // Assert + Assert.ok(providerId, "Provider ID should exist"); + if (providerId) { + // Provider ID should be a meaningful identifier, not just random characters + Assert.ok(providerId.length > 2, + "Provider ID should be more than 2 characters"); + } + } + }); + } + + private addIsAvailableTests(): void { + this.testCase({ + name: "isAvailable: should return boolean value", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + const isAvailable = provider?.isAvailable(); + + // Assert + Assert.ok(typeof isAvailable === 'boolean', + "isAvailable should return a boolean"); + } + }); + + this.testCase({ + name: "isAvailable: should return true after SDK initialization", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + const isAvailable = provider?.isAvailable(); + + // Assert + Assert.ok(isAvailable === true, + "Provider should be available after SDK initialization"); + } + }); + + this.testCase({ + name: "isAvailable: should be consistent across multiple calls", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + const isAvailable1 = provider?.isAvailable(); + const isAvailable2 = provider?.isAvailable(); + const isAvailable3 = provider?.isAvailable(); + + // Assert + Assert.equal(isAvailable1, isAvailable2, + "Availability should be consistent"); + Assert.equal(isAvailable2, isAvailable3, + "Availability should be consistent"); + } + }); + + this.testCase({ + name: "isAvailable: available provider should allow span creation", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + const isAvailable = provider?.isAvailable(); + + // Act + let canCreateSpan = false; + if (provider && isAvailable) { + const span = provider.createSpan("availability-test-span"); + canCreateSpan = span !== null && span !== undefined; + span?.end(); + } + + // Assert + Assert.ok(isAvailable, "Provider should be available"); + Assert.ok(canCreateSpan, + "Available provider should allow span creation"); + } + }); + + this.testCase({ + name: "isAvailable: should reflect provider initialization state", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // After full SDK initialization, provider should be available + Assert.ok(provider !== null, "Provider should not be null"); + + // Act + const isAvailable = provider?.isAvailable(); + + Assert.ok(isAvailable !== undefined, "isAvailable should not be undefined"); + + // Assert + + Assert.ok(isFunction(provider.createSpan), "Available provider should have createSpan"); + Assert.ok(isFunction(provider.getProviderId), "Available provider should have getProviderId"); + Assert.ok(isFunction(provider.isAvailable), "Available provider should have isAvailable"); + Assert.ok(isFunction(this._ai.getActiveSpan), "Available provider should have activeSpan"); + Assert.ok(isFunction(this._ai.setActiveSpan), "Available provider should have setActiveSpan"); + Assert.ok(isFunction(this._ai.core.getActiveSpan), "Available core should have activeSpan"); + Assert.ok(isFunction(this._ai.core.setActiveSpan), "Available core should have setActiveSpan"); + } + }); + } + + private addCreateSpanTests(): void { + this.testCase({ + name: "Provider createSpan: should create valid span", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + const spanName = "provider-create-span-test"; + + // Act + let span: IReadableSpan | null = null; + if (provider) { + span = provider.createSpan(spanName); + } + + // Assert + Assert.ok(span !== null && span !== undefined, + "Provider should create a span"); + if (span) { + Assert.equal(span.name, spanName, "Span name should match"); + Assert.ok(typeof span.isRecording === 'function', + "Span should have isRecording method"); + } + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Provider createSpan: should create span with options", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + const spanName = "provider-span-with-options"; + const options: IOTelSpanOptions = { + kind: eOTelSpanKind.CLIENT, + attributes: { + "test.attribute": "value" + } + }; + + // Act + let span: IReadableSpan | null = null; + if (provider) { + span = provider.createSpan(spanName, options); + } + + // Assert + Assert.ok(span !== null, "Provider should create span with options"); + if (span) { + Assert.equal(span.kind, eOTelSpanKind.CLIENT, + "Span kind should match options"); + Assert.ok(span.attributes["test.attribute"] === "value", + "Span attributes should be set"); + } + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Provider createSpan: should create span with parent context", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + let parentSpan: IReadableSpan | null = null; + let childSpan: IReadableSpan | null = null; + + if (provider) { + parentSpan = provider.createSpan("parent-span"); + const parentCtx = parentSpan.spanContext(); + + // Act + childSpan = provider.createSpan("child-span", undefined, parentCtx); + + // Assert + Assert.ok(childSpan !== null, "Child span should be created"); + if (childSpan && parentSpan) { + const childCtx = childSpan.spanContext(); + Assert.equal(childCtx.traceId, parentCtx.traceId, + "Child should inherit parent traceId"); + Assert.notEqual(childCtx.spanId, parentCtx.spanId, + "Child should have different spanId"); + } + } + + // Cleanup + childSpan?.end(); + parentSpan?.end(); + } + }); + + this.testCase({ + name: "Provider createSpan: should create multiple independent spans", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + let span1: IReadableSpan | null = null; + let span2: IReadableSpan | null = null; + let span3: IReadableSpan | null = null; + + if (provider) { + span1 = provider.createSpan("span-1"); + span2 = provider.createSpan("span-2"); + span3 = provider.createSpan("span-3"); + } + + // Assert + Assert.ok(span1 !== null, "First span should be created"); + Assert.ok(span2 !== null, "Second span should be created"); + Assert.ok(span3 !== null, "Third span should be created"); + + if (span1 && span2 && span3) { + const ctx1 = span1.spanContext(); + const ctx2 = span2.spanContext(); + const ctx3 = span3.spanContext(); + + Assert.notEqual(ctx1.spanId, ctx2.spanId, + "Spans should have different spanIds"); + Assert.notEqual(ctx2.spanId, ctx3.spanId, + "Spans should have different spanIds"); + Assert.notEqual(ctx1.spanId, ctx3.spanId, + "Spans should have different spanIds"); + } + + // Cleanup + span1?.end(); + span2?.end(); + span3?.end(); + } + }); + + this.testCase({ + name: "Provider createSpan: should create recording spans by default", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + let span: IReadableSpan | null = null; + if (provider) { + span = provider.createSpan("recording-test"); + } + + // Assert + Assert.ok(span !== null, "Span should be created"); + if (span) { + Assert.ok(span.isRecording(), + "Span should be recording by default"); + } + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Provider createSpan: should respect recording option when false", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + const options: IOTelSpanOptions = { + recording: false + }; + + // Act + let span: IReadableSpan | null = null; + if (provider) { + span = provider.createSpan("non-recording-test", options); + } + + // Assert + Assert.ok(span !== null, "Span should be created"); + if (span) { + Assert.ok(!span.isRecording(), + "Span should not be recording when options.recording is false"); + } + + // Cleanup + span?.end(); + } + }); + } + + private addProviderIntegrationTests(): void { + this.testCase({ + name: "Integration: provider operations should work with SDK instance", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act - Create span via provider + let providerSpan: IReadableSpan | null = null; + if (provider) { + providerSpan = provider.createSpan("provider-integration-span"); + } + + // Create span via SDK + const sdkSpan = this._ai.startSpan("sdk-integration-span"); + + // Assert + Assert.ok(providerSpan !== null, + "Provider should create span successfully"); + Assert.ok(sdkSpan !== null, + "SDK should create span successfully"); + + if (providerSpan && sdkSpan) { + // Both spans should have valid contexts + const providerCtx = providerSpan.spanContext(); + const sdkCtx = sdkSpan.spanContext(); + + Assert.ok(providerCtx.traceId, "Provider span should have traceId"); + Assert.ok(providerCtx.spanId, "Provider span should have spanId"); + Assert.ok(sdkCtx.traceId, "SDK span should have traceId"); + Assert.ok(sdkCtx.spanId, "SDK span should have spanId"); + } + + // Cleanup + providerSpan?.end(); + sdkSpan?.end(); + } + }); + + this.testCase({ + name: "Integration: provider activeSpan and setActiveSpan work together", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + let span: IReadableSpan | null = null; + if (provider) { + span = provider.createSpan("active-integration-span"); + this._ai.setActiveSpan(span); + const activeSpan = this._ai.getActiveSpan(); + + // Assert + Assert.ok(activeSpan !== null, "Active span should be retrievable"); + Assert.equal(activeSpan.name, span.name, + "Active span should match the set span"); + Assert.equal(activeSpan, this._ai.core.getActiveSpan(), "Active span from core should match active span from SDK"); + } + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Integration: provider availability affects span creation", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + const isAvailable = provider?.isAvailable(); + let canCreateSpan = false; + + if (provider) { + try { + const span = provider.createSpan("availability-integration-test"); + canCreateSpan = span !== null; + span?.end(); + } catch (e) { + canCreateSpan = false; + } + } + + // Assert + if (isAvailable) { + Assert.ok(canCreateSpan, + "Available provider should successfully create spans"); + } else { + // If provider is not available, we should handle it gracefully + Assert.ok(!canCreateSpan || canCreateSpan, + "Provider availability state should be consistent with span creation"); + } + } + }); + + this.testCase({ + name: "Integration: provider ID is consistent with trace operations", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + const providerId = provider?.getProviderId(); + let span: IReadableSpan | null = null; + + if (provider) { + span = provider.createSpan("provider-id-integration"); + } + + // Assert + Assert.ok(providerId, "Provider should have an ID"); + Assert.ok(span !== null, + "Provider with ID should be able to create spans"); + + // Cleanup + span?.end(); + } + }); + + this.testCase({ + name: "Integration: provider methods are callable without errors", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act & Assert - All methods should be callable + Assert.ok(provider !== null, "Provider should exist"); + + if (provider) { + // Test getProviderId + Assert.doesNotThrow(() => { + const id = provider.getProviderId(); + Assert.ok(typeof id === 'string', "getProviderId should return string"); + }, "getProviderId should not throw"); + + // Test isAvailable + Assert.doesNotThrow(() => { + const available = provider.isAvailable(); + Assert.ok(typeof available === 'boolean', + "isAvailable should return boolean"); + }, "isAvailable should not throw"); + + // Test createSpan + Assert.doesNotThrow(() => { + const span = provider.createSpan("error-test-span"); + Assert.ok(span !== null, "createSpan should return span"); + span?.end(); + }, "createSpan should not throw"); + + // Test activeSpan + Assert.doesNotThrow(() => { + const active = this._ai.getActiveSpan(); + // Can be null, that's ok + Assert.ok(active === null || typeof active === 'object', + "activeSpan should return null or span object"); + }, "activeSpan should not throw"); + + const span = provider.createSpan("set-active-error-test"); + + // Test setActiveSpan + Assert.doesNotThrow(() => { + this._ai.setActiveSpan(span); + span?.end(); + }, "setActiveSpan should not throw"); + + // Test setActiveSpan + Assert.doesNotThrow(() => { + this._ai.setActiveSpan(span); + }, "setActiveSpan should not throw when the span has already ended"); + + // Test setActiveSpan + Assert.doesNotThrow(() => { + span?.end(); + }, "ending an already ended span should not throw"); + } + } + }); + + this.testCase({ + name: "Integration: provider supports root span creation", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + + // Act + let rootSpan: IReadableSpan | null = null; + if (provider) { + rootSpan = provider.createSpan("root-span", { root: true }); + } + + // Assert + Assert.ok(rootSpan !== null, "Root span should be created"); + if (rootSpan) { + const ctx = rootSpan.spanContext(); + Assert.ok(ctx.traceId, "Root span should have traceId"); + Assert.ok(ctx.spanId, "Root span should have spanId"); + } + + // Cleanup + rootSpan?.end(); + } + }); + + this.testCase({ + name: "Integration: provider supports different span kinds", + test: () => { + // Arrange + const provider = this._ai.core.getTraceProvider(); + const spanKinds = [ + eOTelSpanKind.INTERNAL, + eOTelSpanKind.SERVER, + eOTelSpanKind.CLIENT, + eOTelSpanKind.PRODUCER, + eOTelSpanKind.CONSUMER + ]; + + // Act & Assert + if (provider) { + spanKinds.forEach(kind => { + const span = provider.createSpan(`span-kind-${kind}`, { kind }); + Assert.ok(span !== null, `Span with kind ${kind} should be created`); + Assert.equal(span?.kind, kind, + `Span should have kind ${kind}`); + span?.end(); + }); + } + } + }); + } +} diff --git a/AISKU/Tests/Unit/src/TraceSuppression.Tests.ts b/AISKU/Tests/Unit/src/TraceSuppression.Tests.ts new file mode 100644 index 000000000..820f5a08a --- /dev/null +++ b/AISKU/Tests/Unit/src/TraceSuppression.Tests.ts @@ -0,0 +1,706 @@ +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { ApplicationInsights } from "../../../src/applicationinsights-web"; +import { eOTelSpanKind, ITelemetryItem, suppressTracing, unsuppressTracing, isTracingSuppressed } from "@microsoft/applicationinsights-core-js"; + + +function _createAndInitializeSDK(connectionString: string): ApplicationInsights { + let newInst = new ApplicationInsights({ + config: { + connectionString: connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + // Initialize the SDK + newInst.loadAppInsights(); + + return newInst +} + +export class TraceSuppressionTests extends AITestClass { + private static readonly _instrumentationKey = "b7170927-2d1c-44f1-acec-59f4e1751c11"; + private static readonly _connectionString = `InstrumentationKey=${TraceSuppressionTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + + // Track calls to track for validation + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "TraceSuppressionTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + this._ai = _createAndInitializeSDK(TraceSuppressionTests._connectionString); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + + } catch (e) { + console.error("Failed to initialize TraceSuppressionTests: " + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addTests(); + } + + private addTests(): void { + + this.testCase({ + name: "TraceSuppression: new SDK instance should have tracing enabled by default and state should not leak between instances", + test: () => { + // Step 1: Verify first instance has tracing enabled by default + Assert.ok(!isTracingSuppressed(this._ai.core), "Instance 1: Tracing should NOT be suppressed in new instance"); + Assert.ok(!isTracingSuppressed(this._ai.core.config), "Instance 1: Tracing should NOT be suppressed on config"); + Assert.ok(!isTracingSuppressed(this._ai.otelApi), "Instance 1: Tracing should NOT be suppressed on otelApi"); + + // Verify that spans can record by default + const span1 = this._ai.startSpan("default-span"); + Assert.ok(span1, "Instance 1: Span should be created"); + Assert.ok(span1!.isRecording(), "Instance 1: Span should be recording by default"); + span1!.end(); + Assert.equal(this._trackCalls.length, 1, "Instance 1: Telemetry should be tracked by default"); + + // Step 2: Suppress tracing on this instance + suppressTracing(this._ai.core); + Assert.ok(isTracingSuppressed(this._ai.core), "Instance 1: Tracing should be suppressed after suppressTracing()"); + + // Verify suppression works - span still reports isRecording()=true but doesn't send telemetry + const span2 = this._ai.startSpan("suppressed-span"); + Assert.ok(!span2!.isRecording(), "Instance 2: Span reports isRecording()=false when suppressed"); + span2!.end(); + Assert.equal(this._trackCalls.length, 1, "Instance 1: No additional telemetry when suppressed"); + + // Step 3: Clean up first instance and create a new instance + this._ai.unload(false); + + this._ai = _createAndInitializeSDK(TraceSuppressionTests._connectionString); + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + + // Step 4: Verify new instance has tracing enabled by default (not inheriting suppressed state) + Assert.ok(!isTracingSuppressed(this._ai.core), "Instance 2: Tracing should NOT be suppressed in new instance"); + Assert.ok(!isTracingSuppressed(this._ai.core.config), "Instance 2: Tracing should NOT be suppressed on config"); + Assert.ok(!isTracingSuppressed(this._ai.otelApi), "Instance 2: Tracing should NOT be suppressed on otelApi"); + + // Verify that spans can record in the new instance + const span3 = this._ai.startSpan("new-instance-span"); + Assert.ok(span3, "Instance 2: Span should be created"); + Assert.ok(span3!.isRecording(), "Instance 2: Span should be recording by default (state should not leak)"); + span3!.end(); + Assert.equal(this._trackCalls.length, 2, "Instance 2: Telemetry should be tracked in new instance"); + } + }); + + this.testCase({ + name: "TraceSuppression: suppressTracing should be available as exported function", + test: () => { + // Verify that suppressTracing functions are available as imports + Assert.ok(typeof suppressTracing === "function", "suppressTracing should be available as exported function"); + Assert.ok(typeof unsuppressTracing === "function", "unsuppressTracing should be available as exported function"); + Assert.ok(typeof isTracingSuppressed === "function", "isTracingSuppressed should be available as exported function"); + } + }); + + this.testCase({ + name: "TraceSuppression: suppressTracing on core should prevent span recording", + test: () => { + // Arrange + this._trackCalls = []; + Assert.ok(!isTracingSuppressed(this._ai.core), "Tracing should not be suppressed initially"); + + // Act - suppress tracing + suppressTracing(this._ai.core); + Assert.ok(isTracingSuppressed(this._ai.core), "Tracing should be suppressed after calling suppressTracing"); + + // Create span while tracing is suppressed + const span = this._ai.startSpan("suppressed-span", { + kind: eOTelSpanKind.INTERNAL, + attributes: { + "test.suppressed": true + } + }); + + // Assert + Assert.ok(span, "Span should still be created"); + Assert.ok(!span!.isRecording(), "Span reports isRecording()=false"); + + // End the span - should not generate telemetry + span!.end(); + + Assert.equal(this._trackCalls.length, 0, "No telemetry should be tracked when tracing is suppressed"); + } + }); + + this.testCase({ + name: "TraceSuppression: unsuppressTracing should restore span recording", + test: () => { + // Arrange + this._trackCalls = []; + suppressTracing(this._ai.core); + Assert.ok(isTracingSuppressed(this._ai.core), "Tracing should be suppressed"); + + // Create span while suppressed - still reports isRecording()=true but won't send telemetry + const suppressedSpan = this._ai.startSpan("suppressed-span"); + Assert.ok(!suppressedSpan!.isRecording(), "Span reports isRecording()=false even when suppressed"); + suppressedSpan!.end(); + + // Act - unsuppress tracing + unsuppressTracing(this._ai.core); + Assert.ok(!isTracingSuppressed(this._ai.core), "Tracing should not be suppressed after unsuppressTracing"); + + // Create new span after unsuppressing + const recordingSpan = this._ai.startSpan("recording-span", { + attributes: { + "test.recording": true + } + }); + + // Assert + Assert.ok(recordingSpan, "Span should be created"); + Assert.ok(recordingSpan!.isRecording(), "Span should be recording after unsuppressing"); + + // End the span - should generate telemetry + recordingSpan!.end(); + + Assert.equal(this._trackCalls.length, 1, "Telemetry should be tracked after unsuppressing"); + Assert.equal(this._trackCalls[0].baseData?.name, "recording-span", "Tracked span should have correct name"); + } + }); + + this.testCase({ + name: "TraceSuppression: suppressTracing on config should prevent span recording", + test: () => { + // Arrange + this._trackCalls = []; + + // Suppress via config object + suppressTracing(this._ai.core.config); + Assert.ok(isTracingSuppressed(this._ai.core.config), "Tracing should be suppressed on config"); + Assert.ok(isTracingSuppressed(this._ai.core), "Tracing should be suppressed on core"); + + // Act - create span + const span = this._ai.startSpan("config-suppressed-span"); + + // Assert + Assert.ok(span, "Span should be created"); + Assert.ok(!span!.isRecording(), "Span reports isRecording()=false"); + span!.end(); + + Assert.equal(this._trackCalls.length, 0, "No telemetry when suppressed via config"); + } + }); + + this.testCase({ + name: "TraceSuppression: multiple startSpan calls while suppressed should all create non-recording spans", + test: () => { + // Arrange + this._trackCalls = []; + suppressTracing(this._ai.core); + + // Act - create multiple spans + const span1 = this._ai.startSpan("span-1"); + const span2 = this._ai.startSpan("span-2", { kind: eOTelSpanKind.CLIENT }); + const span3 = this._ai.startSpan("span-3", { kind: eOTelSpanKind.SERVER }); + + // Assert - spans still report isRecording()=true, suppression only affects telemetry output + Assert.ok(!span1!.isRecording(), "Span 1 reports isRecording()=false"); + Assert.ok(!span2!.isRecording(), "Span 2 reports isRecording()=false"); + Assert.ok(!span3!.isRecording(), "Span 3 reports isRecording()=false"); + + // All spans should still be valid and support operations + span1!.setAttribute("test", "value1"); + span2!.setStatus({ code: 0 }); + span3!.updateName("updated-span-3"); + + span1!.end(); + span2!.end(); + span3!.end(); + + Assert.equal(this._trackCalls.length, 0, "No telemetry should be generated for any suppressed span"); + } + }); + + this.testCase({ + name: "TraceSuppression: parent-child span hierarchy with suppression", + test: () => { + // Arrange + this._trackCalls = []; + suppressTracing(this._ai.core); + + // Act - create parent and child spans while suppressed + const parentSpan = this._ai.startSpan("parent-span", { + kind: eOTelSpanKind.SERVER + }); + Assert.ok(!parentSpan!.isRecording(), "Parent span reports isRecording()=false"); + + const childSpan = this._ai.startSpan("child-span", { + kind: eOTelSpanKind.INTERNAL + }); + Assert.ok(!childSpan!.isRecording(), "Child span reports isRecording()=false"); + + // Verify parent-child relationship still established + const childContext = childSpan!.spanContext(); + const parentContext = parentSpan!.spanContext(); + Assert.equal(childContext.traceId, parentContext.traceId, "Child should share traceId with parent"); + Assert.notEqual(childContext.spanId, parentContext.spanId, "Child should have different spanId"); + + childSpan!.end(); + parentSpan!.end(); + + Assert.equal(this._trackCalls.length, 0, "No telemetry for suppressed hierarchy"); + } + }); + + this.testCase({ + name: "TraceSuppression: toggle suppression during span lifecycle", + test: () => { + // Arrange + this._trackCalls = []; + + // Create recording span + const span1 = this._ai.startSpan("recording-span"); + Assert.ok(span1!.isRecording(), "Span should be recording initially"); + + // Suppress tracing mid-lifecycle + suppressTracing(this._ai.core); + + // Create new span while suppressed + const span2 = this._ai.startSpan("suppressed-span"); + Assert.ok(!span2!.isRecording(), "Span reports isRecording()=false when suppressed"); + + // End both spans + span1!.end(); // Was recording but tracing has been suppressed before it ends + span2!.end(); // Was not recording + + // Verify telemetry + Assert.equal(this._trackCalls.length, 0, "Only the recording span should generate telemetry"); + + // Unsuppress and create another span + unsuppressTracing(this._ai.core); + const span3 = this._ai.startSpan("restored-span"); + Assert.ok(span3!.isRecording(), "New span should be recording after unsuppressing"); + span3!.end(); + + Assert.equal(this._trackCalls.length, 1, "Restored span should generate telemetry"); + } + }); + + this.testCase({ + name: "TraceSuppression: toggle suppression during span lifecycle", + test: () => { + // Arrange + this._trackCalls = []; + + // Create recording span + const span1 = this._ai.startSpan("recording-span"); + Assert.ok(span1!.isRecording(), "Span should be recording initially"); + + // Suppress tracing mid-lifecycle + suppressTracing(this._ai.core); + + // Create new span while suppressed + const span2 = this._ai.startSpan("suppressed-span"); + Assert.ok(!span2!.isRecording(), "Span reports isRecording()=false when suppressed"); + + // Unsuppress and create another span + unsuppressTracing(this._ai.core); + + // End both spans + span1!.end(); // Was recording + span2!.end(); // Was not recording as tracing was suppressed when created + + // Verify telemetry + Assert.equal(this._trackCalls.length, 1, "Only the recording span should generate telemetry"); + Assert.equal(this._trackCalls[0].baseData?.name, "recording-span", "Recording span telemetry"); + + const span3 = this._ai.startSpan("restored-span"); + Assert.ok(span3!.isRecording(), "New span should be recording after unsuppressing"); + span3!.end(); + + Assert.equal(this._trackCalls.length, 2, "Restored span should generate telemetry"); + } + }); + + this.testCase({ + name: "TraceSuppression: suppressTracing should affect all span kinds", + test: () => { + // Arrange + this._trackCalls = []; + suppressTracing(this._ai.core); + + // Act - create spans of all kinds + const internalSpan = this._ai.startSpan("internal", { kind: eOTelSpanKind.INTERNAL }); + const clientSpan = this._ai.startSpan("client", { kind: eOTelSpanKind.CLIENT }); + const serverSpan = this._ai.startSpan("server", { kind: eOTelSpanKind.SERVER }); + const producerSpan = this._ai.startSpan("producer", { kind: eOTelSpanKind.PRODUCER }); + const consumerSpan = this._ai.startSpan("consumer", { kind: eOTelSpanKind.CONSUMER }); + + // Assert - all spans still report isRecording()=true, suppression only prevents telemetry output + Assert.ok(!internalSpan!.isRecording(), "INTERNAL span reports isRecording()=false"); + Assert.ok(!clientSpan!.isRecording(), "CLIENT span reports isRecording()=false"); + Assert.ok(!serverSpan!.isRecording(), "SERVER span reports isRecording()=false"); + Assert.ok(!producerSpan!.isRecording(), "PRODUCER span reports isRecording()=false"); + Assert.ok(!consumerSpan!.isRecording(), "CONSUMER span reports isRecording()=false"); + + // End all spans + internalSpan!.end(); + clientSpan!.end(); + serverSpan!.end(); + producerSpan!.end(); + consumerSpan!.end(); + + Assert.equal(this._trackCalls.length, 0, "No telemetry for any span kind when suppressed"); + } + }); + + this.testCase({ + name: "TraceSuppression: span operations should still work when tracing is suppressed", + test: () => { + // Arrange + suppressTracing(this._ai.core); + const span = this._ai.startSpan("suppressed-span"); + Assert.ok(!span!.isRecording(), "Span reports isRecording()=false"); + + // Act - perform various span operations + span!.setAttribute("string-attr", "value"); + span!.setAttribute("number-attr", 42); + span!.setAttribute("boolean-attr", true); + + span!.setAttributes({ + "batch-1": "test1", + "batch-2": 123 + }); + + span!.setStatus({ + code: 0, + message: "Test status" + }); + + span!.updateName("updated-name"); + + span!.recordException(new Error("Test exception")); + + // Assert - operations should not throw + Assert.ok(true, "All operations completed without throwing"); + + // Verify span properties + Assert.equal(span!.name, "updated-name", "Name should be updated"); + Assert.ok(!span!.ended, "Span should not be ended yet"); + + span!.end(); + Assert.ok(span!.ended, "Span should be ended"); + } + }); + + this.testCase({ + name: "TraceSuppression: isTracingSuppressed should return false when not suppressed", + test: () => { + // Ensure no suppression + unsuppressTracing(this._ai.core); + + // Assert + Assert.ok(!isTracingSuppressed(this._ai.core), "Should return false when not suppressed"); + Assert.ok(!isTracingSuppressed(this._ai.core.config), "Config should also not be suppressed"); + } + }); + + this.testCase({ + name: "TraceSuppression: suppressTracing should return the same context", + test: () => { + // Act + const returnedCore = suppressTracing(this._ai.core); + const returnedConfig = suppressTracing(this._ai.core.config); + + // Assert + Assert.equal(returnedCore, this._ai.core, "suppressTracing should return the same core instance"); + Assert.equal(returnedConfig, this._ai.core.config, "suppressTracing should return the same config instance"); + Assert.ok(isTracingSuppressed(returnedCore), "Returned core should have suppression enabled"); + Assert.ok(isTracingSuppressed(returnedConfig), "Returned config should have suppression enabled"); + } + }); + + this.testCase({ + name: "TraceSuppression: unsuppressTracing should return the same context", + test: () => { + // Arrange + suppressTracing(this._ai.core); + + // Act + const returnedCore = unsuppressTracing(this._ai.core); + const returnedConfig = unsuppressTracing(this._ai.core.config); + + // Assert + Assert.equal(returnedCore, this._ai.core, "unsuppressTracing should return the same core instance"); + Assert.equal(returnedConfig, this._ai.core.config, "unsuppressTracing should return the same config instance"); + Assert.ok(!isTracingSuppressed(returnedCore), "Returned core should have suppression disabled"); + Assert.ok(!isTracingSuppressed(returnedConfig), "Returned config should have suppression disabled"); + } + }); + + this.testCase({ + name: "TraceSuppression: suppression state should persist across multiple checks", + test: () => { + // Initial state + Assert.ok(!isTracingSuppressed(this._ai.core), "Initially not suppressed"); + + // Suppress + suppressTracing(this._ai.core); + Assert.ok(isTracingSuppressed(this._ai.core), "Should be suppressed - check 1"); + Assert.ok(isTracingSuppressed(this._ai.core), "Should be suppressed - check 2"); + Assert.ok(isTracingSuppressed(this._ai.core), "Should be suppressed - check 3"); + + // Unsuppress + unsuppressTracing(this._ai.core); + Assert.ok(!isTracingSuppressed(this._ai.core), "Should not be suppressed - check 1"); + Assert.ok(!isTracingSuppressed(this._ai.core), "Should not be suppressed - check 2"); + Assert.ok(!isTracingSuppressed(this._ai.core), "Should not be suppressed - check 3"); + } + }); + + this.testCase({ + name: "TraceSuppression: span attributes should be preserved when tracing is suppressed", + test: () => { + // Arrange + this._trackCalls = []; + suppressTracing(this._ai.core); + + // Act - create span with attributes + const span = this._ai.startSpan("suppressed-with-attrs", { + attributes: { + "initial.attr1": "value1", + "initial.attr2": 100 + } + }); + + Assert.ok(!span!.isRecording(), "Span reports isRecording()=false"); + + // Add more attributes + span!.setAttribute("runtime.attr", "added-later"); + + // Assert - attributes should still be accessible + const attributes = (span as any).attributes || {}; + Assert.ok(attributes["initial.attr1"] === undefined || attributes["runtime.attr"] === undefined, + "Attributes should not be stored as span was not recording"); + + span!.end(); + Assert.equal(this._trackCalls.length, 0, "No telemetry should be generated"); + } + }); + + this.testCase({ + name: "TraceSuppression: span context should be valid when tracing is suppressed", + test: () => { + // Arrange + suppressTracing(this._ai.core); + + // Act + const span = this._ai.startSpan("suppressed-context-test"); + Assert.ok(!span!.isRecording(), "Span reports isRecording()=false"); + + // Assert - span context should be valid + const spanContext = span!.spanContext(); + Assert.ok(spanContext, "Span context should exist"); + Assert.ok(spanContext.traceId, "Trace ID should exist"); + Assert.ok(spanContext.spanId, "Span ID should exist"); + Assert.equal(spanContext.traceId.length, 32, "Trace ID should be 32 hex characters"); + Assert.equal(spanContext.spanId.length, 16, "Span ID should be 16 hex characters"); + + span!.end(); + } + }); + + this.testCase({ + name: "TraceSuppression: rapid suppression toggling should work correctly", + test: () => { + // Arrange + this._trackCalls = []; + + // Act - rapidly toggle suppression + for (let i = 0; i < 5; i++) { + suppressTracing(this._ai.core); + Assert.ok(isTracingSuppressed(this._ai.core), `Should be suppressed on iteration ${i}`); + + unsuppressTracing(this._ai.core); + Assert.ok(!isTracingSuppressed(this._ai.core), `Should not be suppressed on iteration ${i}`); + } + + // Final state check + Assert.ok(!isTracingSuppressed(this._ai.core), "Should end in unsuppressed state"); + + // Create a recording span + const span = this._ai.startSpan("final-span"); + Assert.ok(span!.isRecording(), "Span should be recording after toggles"); + span!.end(); + + Assert.equal(this._trackCalls.length, 1, "Telemetry should be tracked"); + } + }); + + this.testCase({ + name: "TraceSuppression: suppression should work with explicit parent context", + test: () => { + // Arrange + this._trackCalls = []; + + // Create a recording parent span first + const parentSpan = this._ai.startSpan("parent-recording"); + Assert.ok(parentSpan!.isRecording(), "Parent should be recording"); + const parentContext = this._ai.getTraceCtx(); + + // Suppress tracing + suppressTracing(this._ai.core); + + // Act - create child with explicit parent while suppressed + const childSpan = this._ai.startSpan("child-suppressed", { + kind: eOTelSpanKind.INTERNAL + }, parentContext); + + // Assert + Assert.ok(!childSpan!.isRecording(), "Child span reports isRecording()=false when suppressed"); + + const childContext = childSpan!.spanContext(); + Assert.equal(childContext.traceId, parentContext!.traceId, "Child should have same traceId as parent"); + + unsuppressTracing(this._ai.core); + + childSpan!.end(); + parentSpan!.end(); + + // Only parent should generate telemetry + Assert.equal(this._trackCalls.length, 1, "Only parent span should generate telemetry"); + Assert.equal(this._trackCalls[0].baseData?.name, "parent-recording", "Parent span telemetry"); + } + }); + + this.testCase({ + name: "TraceSuppression: suppression should work even with parent context", + test: () => { + // Arrange + this._trackCalls = []; + + // Create a recording parent span first + const parentSpan = this._ai.startSpan("parent-recording"); + Assert.ok(parentSpan!.isRecording(), "Parent should be recording"); + const parentContext = this._ai.getTraceCtx(); + + // Suppress tracing + suppressTracing(this._ai.core); + + // Act - create child with explicit parent while suppressed + const childSpan = this._ai.startSpan("child-suppressed", { + kind: eOTelSpanKind.INTERNAL + }, parentContext); + + // Assert + Assert.ok(!childSpan!.isRecording(), "Child span reports isRecording()=false when suppressed"); + + const childContext = childSpan!.spanContext(); + Assert.equal(childContext.traceId, parentContext!.traceId, "Child should have same traceId as parent"); + + childSpan!.end(); + parentSpan!.end(); + + // Only parent should generate telemetry + Assert.equal(this._trackCalls.length, 0, "Parent span should not generate telemetry either as suppression is active"); + } + }); + + this.testCase({ + name: "TraceSuppression: isTracingSuppressed should handle null/undefined gracefully", + test: () => { + // Act & Assert - should not throw + let result1: boolean; + let result2: boolean; + + try { + result1 = isTracingSuppressed(null as any); + result2 = isTracingSuppressed(undefined as any); + Assert.ok(true, "isTracingSuppressed should handle null/undefined without throwing"); + Assert.ok(!result1, "Should return false for null"); + Assert.ok(!result2, "Should return false for undefined"); + } catch (e) { + Assert.ok(false, "isTracingSuppressed should not throw for null/undefined"); + } + } + }); + + this.testCase({ + name: "TraceSuppression: suppressTracing with startSpan integration test", + test: () => { + // Arrange + this._trackCalls = []; + + // Test 1: Normal recording + const span1 = this._ai.startSpan("normal-1"); + Assert.ok(span1!.isRecording(), "Span 1 should be recording"); + span1!.end(); + Assert.equal(this._trackCalls.length, 1, "Should have 1 telemetry item"); + + // Test 2: Suppress and verify spans still report isRecording()=true but don't send telemetry + suppressTracing(this._ai.core); + const span2 = this._ai.startSpan("suppressed-1"); + const span3 = this._ai.startSpan("suppressed-2"); + Assert.ok(!span2!.isRecording(), "Span 2 reports isRecording()=false when suppressed"); + Assert.ok(!span3!.isRecording(), "Span 3 reports isRecording()=false when suppressed"); + span2!.end(); + span3!.end(); + Assert.equal(this._trackCalls.length, 1, "Should still have only 1 telemetry item"); + + // Test 3: Unsuppress and verify startSpan creates recording spans again + unsuppressTracing(this._ai.core); + const span4 = this._ai.startSpan("normal-2"); + Assert.ok(span4!.isRecording(), "Span 4 should be recording"); + span4!.end(); + Assert.equal(this._trackCalls.length, 2, "Should have 2 telemetry items"); + + // Verify telemetry content + Assert.equal(this._trackCalls[0].baseData?.name, "normal-1", "First telemetry is from span1"); + Assert.equal(this._trackCalls[1].baseData?.name, "normal-2", "Second telemetry is from span4"); + } + }); + + this.testCase({ + name: "TraceSuppression: suppression with nested spans", + test: () => { + // Arrange + this._trackCalls = []; + suppressTracing(this._ai.core); + + // Create and set active span manually + const span1 = this._ai.startSpan("outer-span"); + Assert.ok(!span1!.isRecording(), "Outer span reports isRecording()=false"); + + // Simulate nested operation + const span2 = this._ai.startSpan("inner-span"); + Assert.ok(!span2!.isRecording(), "Inner span reports isRecording()=false"); + span2!.end(); + span1!.end(); + + Assert.equal(this._trackCalls.length, 0, "No telemetry for suppressed nested spans"); + } + }); + } +} diff --git a/AISKU/Tests/Unit/src/UseSpan.Tests.ts b/AISKU/Tests/Unit/src/UseSpan.Tests.ts new file mode 100644 index 000000000..207c35a61 --- /dev/null +++ b/AISKU/Tests/Unit/src/UseSpan.Tests.ts @@ -0,0 +1,1164 @@ +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { ApplicationInsights } from "../../../src/applicationinsights-web"; +import { + IAppInsightsCore, IReadableSpan, eOTelSpanKind, eOTelSpanStatusCode, useSpan, ITelemetryItem, ISpanScope, ITraceHost +} from "@microsoft/applicationinsights-core-js"; + +export class UseSpanTests extends AITestClass { + private static readonly _instrumentationKey = "b7170927-2d1c-44f1-acec-59f4e1751c11"; + private static readonly _connectionString = `InstrumentationKey=${UseSpanTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + + // Track calls to track for validation + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "UseSpanTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: UseSpanTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + // Initialize the SDK + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + + } catch (e) { + console.error("Failed to initialize UseSpan tests: " + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addTests(); + } + + private addTests(): void { + + this.testCase({ + name: "UseSpan: useSpan should be available as exported function", + test: () => { + // Verify that useSpan is available as an import + Assert.ok(typeof useSpan === "function", "useSpan should be available as exported function"); + } + }); + + this.testCase({ + name: "UseSpan: should execute function within span context", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("useSpan-context-test", { + kind: eOTelSpanKind.SERVER, + attributes: { + "test.type": "context-execution" + } + }); + + Assert.ok(testSpan, "Test span should be created"); + Assert.ok(this._ai.core, "Core should be available"); + + let capturedActiveSpan: IReadableSpan | null = null; + let capturedHost: ITraceHost | null = null; + const testFunction = function(this: ISpanScope) { + capturedActiveSpan = this.host.getActiveSpan(); + capturedHost = this.host; + return "context-success"; + }; + + // Act + const result = useSpan(this._ai.core!, testSpan!, testFunction); + + // Assert + Assert.equal(result, "context-success", "Function should execute and return result"); + Assert.ok(capturedActiveSpan, "Function should have access to active span"); + Assert.equal(capturedActiveSpan, testSpan, "Active span should be the provided test span"); + Assert.equal(capturedHost, this._ai.core, "Active host should be the core instance (passed to useSpan)"); + } + }); + + this.testCase({ + name: "UseSpan: should work with telemetry tracking inside span context", + test: () => { + // Arrange + this._trackCalls = []; + const testSpan = this._ai.startSpan("useSpan-telemetry-test", { + attributes: { + "operation.name": "telemetry-tracking" + } + }); + + Assert.ok(testSpan, "Test span should be created"); + + const telemetryFunction = () => { + // Track some telemetry within the span context + this._ai.trackEvent({ + name: "operation-event", + properties: { + "event.source": "useSpan-context" + } + }); + + this._ai.trackMetric({ + name: "operation.duration", + average: 123.45 + }); + + return "telemetry-tracked"; + }; + + // Act + const result = useSpan(this._ai.core!, testSpan!, telemetryFunction); + + // Assert + Assert.equal(result, "telemetry-tracked", "Function should complete successfully"); + + // End the span to trigger trace generation + testSpan!.end(); + + // Verify track was called for the span + Assert.equal(this._trackCalls.length, 3, "Should have one track call from span ending"); + const item = this._trackCalls[2]; + Assert.ok(item.baseData && item.baseData.properties, "Item should have properties"); + Assert.equal("useSpan-telemetry-test", item.baseData.name, "Should include span name in properties"); + } + }); + + this.testCase({ + name: "UseSpan: should handle complex function arguments and return values", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("useSpan-arguments-test"); + Assert.ok(testSpan, "Test span should be created"); + + const complexFunction = ( + _scope: ISpanScope, + stringArg: string, + numberArg: number, + objectArg: { key: string; value: number }, + arrayArg: string[] + ) => { + return { + processedString: stringArg!.toUpperCase(), + doubledNumber: numberArg! * 2, + extractedValue: objectArg!.value, + joinedArray: arrayArg!.join("-"), + timestamp: Date.now() + }; + }; + + const inputObject = { key: "test-key", value: 42 }; + const inputArray = ["item1", "item2", "item3"]; + + // Act + const result = useSpan( + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + this._ai.core!, + // eslint-disable-next-line @typescript-eslint/no-non-null-assertion + testSpan!, + complexFunction, + undefined, + "hello world", + 10, + inputObject, + inputArray + ); + + // Assert + Assert.equal(result.processedString, "HELLO WORLD", "String should be processed correctly"); + Assert.equal(result.doubledNumber, 20, "Number should be doubled correctly"); + Assert.equal(result.extractedValue, 42, "Object value should be extracted correctly"); + Assert.equal(result.joinedArray, "item1-item2-item3", "Array should be joined correctly"); + Assert.ok(result.timestamp > 0, "Timestamp should be generated"); + } + }); + + this.testCase({ + name: "UseSpan: should handle function with this context binding", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("useSpan-this-binding-test"); + Assert.ok(testSpan, "Test span should be created"); + + class TestService { + private _serviceId: string; + private _multiplier: number; + + constructor(id: string, multiplier: number) { + this._serviceId = id; + this._multiplier = multiplier; + } + + public processValue(_scope: ISpanScope, input: number): { serviceId: string; result: number; multiplied: number } { + return { + serviceId: this._serviceId, + result: input + 100, + multiplied: input * this._multiplier + }; + } + } + + const service = new TestService("test-service-123", 3); + + // Act + const result = useSpan( + this._ai.core!, + testSpan!, + service.processValue, + service, + 25 + ); + + // Assert + Assert.equal(result.serviceId, "test-service-123", "Service ID should be preserved via this binding"); + Assert.equal(result.result, 125, "Input should be processed correctly"); + Assert.equal(result.multiplied, 75, "Multiplication should use instance property"); + } + }); + + this.testCase({ + name: "UseSpan: should maintain span context across async-like operations", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("useSpan-async-like-test", { + attributes: { + "operation.type": "async-simulation" + } + }); + Assert.ok(testSpan, "Test span should be created"); + + let spanDuringCallback: IReadableSpan | null = null; + let callbackExecuted = false; + + const asyncLikeFunction = (scope: ISpanScope, callback: (data: string) => void) => { + // Simulate async work that completes synchronously in test + let currentSpan = scope.span; + + // Simulate callback execution (would normally be async) + setTimeout(() => { + spanDuringCallback = this._ai.core!.getActiveSpan(); + callback("async-data"); + callbackExecuted = true; + }, 0); + + return currentSpan ? (currentSpan as IReadableSpan).name : "no-span"; + }; + + // Act + let callbackData = ""; + const callback = (data: string) => { + callbackData = data; + }; + + const result = useSpan(this._ai.core!, testSpan!, asyncLikeFunction, undefined, callback); + + // Assert + Assert.equal(result, "useSpan-async-like-test", "Function should have access to span name"); + + // Note: In a real async scenario, the span context wouldn't automatically + // propagate to the setTimeout callback without additional context management + // This test validates the synchronous behavior of useSpan + } + }); + + this.testCase({ + name: "UseSpan: should handle exceptions and preserve span operations", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("useSpan-exception-test", { + attributes: { + "test.expects": "exception" + } + }); + Assert.ok(testSpan, "Test span should be created"); + + const exceptionFunction = () => { + // Perform some span operations before throwing + const activeSpan = this._ai.core!.getActiveSpan(); + Assert.ok(activeSpan, "Should have active span before exception"); + + activeSpan!.setAttribute("operation.status", "error"); + activeSpan!.setStatus({ + code: eOTelSpanStatusCode.ERROR, + message: "Operation failed with test exception" + }); + + throw new Error("Test exception for useSpan handling"); + }; + + // Act & Assert + let caughtException: Error | null = null; + try { + useSpan(this._ai.core!, testSpan!, exceptionFunction); + } catch (error) { + caughtException = error as Error; + } + + Assert.ok(caughtException, "Exception should be thrown and caught"); + Assert.equal(caughtException!.message, "Test exception for useSpan handling", "Exception message should be preserved"); + + // Verify span is still valid and operations were applied + Assert.ok(testSpan!.isRecording(), "Span should still be recording after exception"); + const readableSpan = testSpan! as IReadableSpan; + Assert.ok(!readableSpan.ended, "Span should not be ended by useSpan after exception"); + } + }); + + this.testCase({ + name: "UseSpan: should work with nested span operations and child spans", + test: () => { + // Arrange + this._trackCalls = []; + const parentSpan = this._ai.startSpan("parent-operation", { + kind: eOTelSpanKind.SERVER, + attributes: { + "operation.name": "parent-process" + } + }); + Assert.ok(parentSpan, "Parent span should be created"); + + const nestedOperations = () => { + // Verify we have the parent span as active + const currentActive = this._ai.core!.getActiveSpan(); + Assert.equal(currentActive, parentSpan, "Parent span should be active"); + + // Create child operations within the parent span context + const childSpan1 = this._ai.startSpan("child-operation-1", { + attributes: { "child.order": 1 } + }); + childSpan1!.setAttribute("child.status", "completed"); + childSpan1!.end(); + + const childSpan2 = this._ai.startSpan("child-operation-2", { + attributes: { "child.order": 2 } + }); + childSpan2!.setAttribute("child.status", "completed"); + childSpan2!.end(); + + return "nested-operations-completed"; + }; + + // Act + const result = useSpan(this._ai.core!, parentSpan!, nestedOperations); + + // Assert + Assert.equal(result, "nested-operations-completed", "Nested operations should complete successfully"); + + // End parent span to generate telemetry + parentSpan!.end(); + + // Should have 3 telemetry items: parent + 2 children + Assert.equal(this._trackCalls.length, 3, "Should have telemetry for parent and child spans"); + + // Verify span names in properties + const spanNames = this._trackCalls.map(item => item.baseData?.name).filter(n => n); + Assert.ok(spanNames.some(name => name === "parent-operation"), "Should have parent span telemetry"); + Assert.ok(spanNames.some(name => name === "child-operation-1"), "Should have child-1 span telemetry"); + Assert.ok(spanNames.some(name => name === "child-operation-2"), "Should have child-2 span telemetry"); + } + }); + + this.testCase({ + name: "UseSpan: should support different return value types", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("useSpan-return-types-test"); + Assert.ok(testSpan, "Test span should be created"); + + // Test various return types + const stringResult = useSpan(this._ai.core!, testSpan!, () => "string-result"); + const numberResult = useSpan(this._ai.core!, testSpan!, () => 42.5); + const booleanResult = useSpan(this._ai.core!, testSpan!, () => true); + const arrayResult = useSpan(this._ai.core!, testSpan!, () => [1, 2, 3]); + const objectResult = useSpan(this._ai.core!, testSpan!, () => ({ key: "value", nested: { prop: 123 } })); + const nullResult = useSpan(this._ai.core!, testSpan!, () => null); + const undefinedResult = useSpan(this._ai.core!, testSpan!, () => undefined); + + // Assert + Assert.equal(stringResult, "string-result", "String return should work"); + Assert.equal(numberResult, 42.5, "Number return should work"); + Assert.equal(booleanResult, true, "Boolean return should work"); + Assert.equal(arrayResult.length, 3, "Array return should work"); + Assert.equal(arrayResult[1], 2, "Array elements should be preserved"); + Assert.equal(objectResult.key, "value", "Object properties should be preserved"); + Assert.equal(objectResult.nested.prop, 123, "Nested object properties should be preserved"); + Assert.equal(nullResult, null, "Null return should work"); + Assert.equal(undefinedResult, undefined, "Undefined return should work"); + } + }); + + this.testCase({ + name: "UseSpan: should handle rapid successive calls efficiently", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("useSpan-performance-test"); + Assert.ok(testSpan, "Test span should be created"); + + const iterations = 100; + let totalResult = 0; + + // Simple computation function + const computeFunction = (_scope: ISpanScope, input: number) => { + return input * 2 + 1; + }; + + const startTime = Date.now(); + + // Act - Multiple rapid useSpan calls + for (let i = 0; i < iterations; i++) { + const result = useSpan(this._ai.core!, testSpan!, computeFunction, undefined, i); + totalResult += result; + } + + const endTime = Date.now(); + const duration = endTime - startTime; + + // Assert + const expectedTotal = Array.from({length: iterations}, (_, i) => i * 2 + 1).reduce((sum, val) => sum + val, 0); + Assert.equal(totalResult, expectedTotal, "All computations should be correct"); + + // Performance assertion - should complete reasonably quickly + Assert.ok(duration < 1000, `Performance test should complete quickly: ${duration}ms for ${iterations} iterations`); + + // Verify span is still valid after many operations + Assert.ok(testSpan!.isRecording(), "Span should still be recording after multiple useSpan calls"); + } + }); + + this.testCase({ + name: "UseSpan: should integrate with AI telemetry correlation", + test: () => { + // Arrange + this._trackCalls = []; + const operationSpan = this._ai.startSpan("user-operation", { + kind: eOTelSpanKind.SERVER, + attributes: { + "user.id": "user-123", + "operation.type": "data-processing" + } + }); + Assert.ok(operationSpan, "Operation span should be created"); + + const businessLogicFunction = (_scope: ISpanScope, userId: string, dataType: string) => { + // Track multiple telemetry items within span context + this._ai.trackEvent({ + name: "data-processing-started", + properties: { + "user.id": userId, + "data.type": dataType, + "processing.stage": "initialization" + } + }); + + // Simulate some processing steps + for (let step = 1; step <= 3; step++) { + this._ai.trackMetric({ + name: "processing.step.duration", + average: step * 10.5, + properties: { + "step.number": step.toString() + } + }); + } + + this._ai.trackEvent({ + name: "data-processing-completed", + properties: { + "user.id": userId, + "data.type": dataType, + "processing.stage": "completion", + "steps.completed": "3" + } + }); + + return { + userId: userId, + dataType: dataType, + stepsCompleted: 3, + status: "success" + }; + }; + + // Act + const result = useSpan( + this._ai.core!, + operationSpan!, + businessLogicFunction, + undefined, + "user-123", + "customer-data" + ); + + // End the span to generate trace + operationSpan!.end(); + + // Assert + Assert.equal(result.userId, "user-123", "User ID should be processed correctly"); + Assert.equal(result.dataType, "customer-data", "Data type should be processed correctly"); + Assert.equal(result.stepsCompleted, 3, "All processing steps should be completed"); + Assert.equal(result.status, "success", "Operation should complete successfully"); + + // Verify span telemetry was generated + Assert.equal(this._trackCalls.length, 6, "Should have one track call from span ending"); + const spanItem = this._trackCalls[5]; + Assert.ok(spanItem.baseData && spanItem.baseData.properties, "Item should have properties"); + Assert.equal("user-operation", spanItem.baseData.name, "Should include span name"); + + // Verify span attributes are included in properties + Assert.equal(spanItem.baseData.properties["user.id"], "user-123", "Span attributes should be included in telemetry"); + Assert.equal(spanItem.baseData.properties["operation.type"], "data-processing", "All span attributes should be preserved"); + } + }); + + this.testCase({ + name: "UseSpan: should handle empty or no-op functions gracefully", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("useSpan-noop-test"); + Assert.ok(testSpan, "Test span should be created"); + + // Test empty function + const emptyFunction = () => {}; + + // Test function that just returns without doing anything + const noOpFunction = () => { + return; + }; + + // Test function that returns undefined explicitly + const undefinedFunction = () => { + return undefined; + }; + + // Act + const emptyResult = useSpan(this._ai.core!, testSpan!, emptyFunction); + const noOpResult = useSpan(this._ai.core!, testSpan!, noOpFunction); + const undefinedResult = useSpan(this._ai.core!, testSpan!, undefinedFunction); + + // Assert + Assert.equal(emptyResult, undefined, "Empty function should return undefined"); + Assert.equal(noOpResult, undefined, "No-op function should return undefined"); + Assert.equal(undefinedResult, undefined, "Undefined function should return undefined"); + + // Verify span is still valid + Assert.ok(testSpan!.isRecording(), "Span should still be recording after no-op functions"); + } + }); + + this.testCase({ + name: "UseSpan: should use ISpanScope as 'this' when no thisArg provided", + test: () => { + // Arrange + const span = this._ai.startSpan("usespan-test", { + attributes: { "test.id": "useSpan-this-test" } + }); + + let capturedThis: any = null; + let capturedScopeParam: any = null; + + // Act - call useSpan without thisArg (function receives scope as parameter) + const result = useSpan(this._ai.core, span!, function(this: ISpanScope, scope: ISpanScope, arg1: string) { + capturedThis = this; + capturedScopeParam = scope; + + // Verify 'this' is ISpanScope + Assert.ok(this.host, "'this.core' should exist"); + Assert.ok(this.span, "'this.span' should exist"); + + // Verify scope parameter is also ISpanScope + Assert.ok(scope.host, "scope.host should exist"); + Assert.ok(scope.span, "scope.span should exist"); + + return `${arg1}-${scope.span.name}`; + }, undefined, "result"); + + // Assert + Assert.equal(result, "result-usespan-test", "Function should execute and return result"); + Assert.ok(capturedThis, "'this' should be defined"); + Assert.ok(capturedThis.host, "'this.host' should exist"); + Assert.ok(capturedThis.span, "'this.span' should exist"); + Assert.equal(capturedThis.host, this._ai.core, "'this.host' should be the AI core"); + Assert.equal(capturedThis.span, span, "'this.span' should be the passed span"); + + Assert.ok(capturedScopeParam, "scope parameter should be defined"); + Assert.equal(capturedScopeParam.host, this._ai.core, "scope.host should be the AI core"); + Assert.equal(capturedScopeParam.span, span, "scope.span should be the passed span"); + + // Both 'this' and scope param should be the same ISpanScope instance + Assert.equal(capturedThis, capturedScopeParam, "'this' and scope param should be the same ISpanScope instance"); + + span!.end(); + } + }); + + this.testCase({ + name: "UseSpan: should use provided thisArg when specified", + test: () => { + // Arrange + const span = this._ai.startSpan("usespan-thisarg-test"); + + class ServiceClass { + public serviceId: string = "service-456"; + public getData(prefix: string): string { + return `${prefix}-${this.serviceId}`; + } + } + + const service = new ServiceClass(); + let capturedThis: any = null; + let capturedScopeParam: any = null; + + // Act - call useSpan with explicit thisArg + const result = useSpan(this._ai.core, span!, function(this: ServiceClass, scope: ISpanScope) { + capturedThis = this; + capturedScopeParam = scope; + + // 'this' should be the service instance + Assert.equal(this.serviceId, "service-456", "'this.serviceId' should match"); + Assert.ok(typeof this.getData === "function", "'this.getData' should be a function"); + + // scope parameter should still be ISpanScope + Assert.ok(scope.host, "scope.host should exist"); + Assert.ok(scope.span, "scope.span should exist"); + + return this.getData("custom"); + }, service); + + // Assert + Assert.equal(result, "custom-service-456", "Function should execute with custom this context"); + Assert.ok(capturedThis, "'this' should be defined"); + Assert.equal(capturedThis, service, "'this' should be the service instance"); + Assert.equal(capturedThis.serviceId, "service-456", "'this.serviceId' should match"); + Assert.ok(!capturedThis.host, "Custom this should not have host property"); + Assert.ok(!capturedThis.span, "Custom this should not have span property"); + + Assert.ok(capturedScopeParam, "scope parameter should be defined"); + Assert.ok(capturedScopeParam.host, "scope.host should exist even with custom this"); + Assert.ok(capturedScopeParam.span, "scope.span should exist even with custom this"); + + span!.end(); + } + }); + + this.testCase({ + name: "UseSpan: scope parameter should provide access to core and span operations", + test: () => { + // Arrange + this._trackCalls = []; + const span = this._ai.startSpan("scope-operations-test"); + + // Act - use scope parameter to perform operations + useSpan(this._ai.core, span!, (scope: ISpanScope) => { + // Use scope.span to set attributes + scope.span.setAttribute("operation.name", "data-processing"); + scope.span.setAttribute("operation.step", 1); + + // Use scope.span to set status + scope.span.setStatus({ + code: 0, + message: "Operation successful" + }); + + // Use scope.span to get context + const spanContext = scope.span.spanContext(); + Assert.ok(spanContext.traceId, "Should access span context via scope"); + Assert.ok(spanContext.spanId, "Should access span ID via scope"); + + // Verify span name + Assert.equal(scope.span.name, "scope-operations-test", "Span name should be accessible"); + }); + + // Assert + Assert.ok(span, "Span should exist"); + Assert.equal(span!.name, "scope-operations-test", "Span name should match"); + + span!.end(); + + Assert.equal(this._trackCalls.length, 1, "Should generate telemetry"); + Assert.ok(this._trackCalls[0].baseData?.properties, "Should have properties"); + Assert.equal(this._trackCalls[0].baseData.properties["operation.name"], "data-processing", "Attributes should be preserved"); + } + }); + + this.testCase({ + name: "UseSpan: 'this' binding with nested useSpan calls", + test: () => { + // Arrange + const span = this._ai.startSpan("nested-calls-test"); + + const outerContext = { + contextName: "outer", + value: 100 + }; + + let outerThisCapture: any = null; + let innerThisCapture: any = null; + let ai = this._ai; + + // Act - nested useSpan calls with different thisArg + useSpan(this._ai.core, span!, function(this: typeof outerContext, outerScope: ISpanScope) { + outerThisCapture = this; + Assert.equal(this.contextName, "outer", "Outer 'this' should be outer context"); + Assert.equal(this.value, 100, "Outer 'this.value' should match"); + + const innerSpan = ai.startSpan("inner-nested-span"); + + useSpan(ai.core, innerSpan!, function(this: ISpanScope, innerScope: ISpanScope) { + innerThisCapture = this; + // Inner call without explicit thisArg - should be ISpanScope + Assert.ok(this.host, "Inner 'this' should be ISpanScope"); + Assert.ok(this.span, "Inner 'this.span' should exist"); + Assert.equal(this.span.name, "inner-nested-span", "Inner span name should match"); + }); + + innerSpan!.end(); + }, outerContext); + + // Assert + Assert.ok(outerThisCapture, "Outer 'this' should be captured"); + Assert.equal(outerThisCapture.contextName, "outer", "Outer context should be preserved"); + + Assert.ok(innerThisCapture, "Inner 'this' should be captured"); + Assert.ok(innerThisCapture.host, "Inner 'this' should have host"); + Assert.ok(innerThisCapture.span, "Inner 'this' should have span"); + + span!.end(); + } + }); + + this.testCase({ + name: "UseSpan: verify scope.restore() is called to restore previous active span", + test: () => { + // Arrange + const outerSpan = this._ai.startSpan("outer-span"); + const innerSpan = this._ai.startSpan("inner-span"); + + let activeSpanBeforeUseSpan: any = null; + let activeSpanInsideUseSpan: any = null; + let activeSpanAfterUseSpan: any = null; + + // Act + activeSpanBeforeUseSpan = this._ai.core.getActiveSpan ? this._ai.core.getActiveSpan() : null; + + useSpan(this._ai.core, innerSpan!, (scope: ISpanScope) => { + activeSpanInsideUseSpan = this._ai.core.getActiveSpan ? this._ai.core.getActiveSpan() : null; + Assert.equal(activeSpanInsideUseSpan, innerSpan, "Active span inside useSpan should be inner span"); + }); + + activeSpanAfterUseSpan = this._ai.core.getActiveSpan ? this._ai.core.getActiveSpan() : null; + + // Assert + // Active span should be restored after useSpan completes + Assert.equal(activeSpanAfterUseSpan, activeSpanBeforeUseSpan, + "Active span should be restored after useSpan completes"); + + innerSpan!.end(); + outerSpan!.end(); + } + }); + + this.testCase({ + name: "UseSpan: trace context should match active span context inside useSpan", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("trace-context-match-test", { + attributes: { + "test.type": "trace-context-validation" + } + }); + Assert.ok(testSpan, "Test span should be created"); + + let traceCtxInsideUseSpan: any = null; + let spanContextInsideUseSpan: any = null; + let activeSpanInsideUseSpan: any = null; + + // Act + useSpan(this._ai.core, testSpan!, (scope: ISpanScope) => { + // Get trace context from core + traceCtxInsideUseSpan = this._ai.core.getTraceCtx(false); + + // Get span context from the span + spanContextInsideUseSpan = scope.span.spanContext(); + + // Get active span + activeSpanInsideUseSpan = this._ai.core.getActiveSpan ? this._ai.core.getActiveSpan() : null; + }); + + // Assert + Assert.ok(traceCtxInsideUseSpan, "Trace context should exist inside useSpan"); + Assert.ok(spanContextInsideUseSpan, "Span context should exist"); + Assert.ok(activeSpanInsideUseSpan, "Active span should be set"); + + // Verify active span matches the useSpan span + Assert.equal(activeSpanInsideUseSpan, testSpan, "Active span should be the useSpan span"); + + // Verify trace context matches span context + Assert.equal(traceCtxInsideUseSpan.traceId, spanContextInsideUseSpan.traceId, + "Trace context traceId should match span context traceId"); + Assert.equal(traceCtxInsideUseSpan.spanId, spanContextInsideUseSpan.spanId, + "Trace context spanId should match span context spanId"); + Assert.equal(traceCtxInsideUseSpan.traceFlags, spanContextInsideUseSpan.traceFlags, + "Trace context traceFlags should match span context traceFlags"); + + testSpan!.end(); + } + }); + + this.testCase({ + name: "UseSpan: trace context updates when switching between nested useSpan calls", + test: () => { + // Arrange + const outerSpan = this._ai.startSpan("outer-trace-span"); + const innerSpan = this._ai.startSpan("inner-trace-span"); + + let outerTraceCtx: any = null; + let outerSpanCtx: any = null; + let innerTraceCtx: any = null; + let innerSpanCtx: any = null; + + // Act + useSpan(this._ai.core, outerSpan!, (outerScope: ISpanScope) => { + outerTraceCtx = this._ai.core.getTraceCtx(false); + outerSpanCtx = outerScope.span.spanContext(); + + // Verify outer trace context matches outer span + Assert.equal(outerTraceCtx.spanId, outerSpanCtx.spanId, + "Outer trace context should match outer span"); + + // Nested useSpan with different span + useSpan(this._ai.core, innerSpan!, (innerScope: ISpanScope) => { + innerTraceCtx = this._ai.core.getTraceCtx(false); + innerSpanCtx = innerScope.span.spanContext(); + + // Verify inner trace context matches inner span + Assert.equal(innerTraceCtx.spanId, innerSpanCtx.spanId, + "Inner trace context should match inner span"); + + // Verify inner context is different from outer + Assert.notEqual(innerTraceCtx.spanId, outerTraceCtx.spanId, + "Inner and outer trace contexts should have different spanIds"); + }); + + // After inner useSpan, verify we're back to outer context + const restoredTraceCtx = this._ai.core.getTraceCtx(false); + Assert.equal(restoredTraceCtx.spanId, outerSpanCtx.spanId, + "Trace context should be restored to outer span after inner useSpan completes"); + }); + + outerSpan!.end(); + innerSpan!.end(); + } + }); + + this.testCase({ + name: "UseSpan: child spans created inside useSpan inherit correct parent context", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("parent-for-child-test"); + + let childSpanContext: any = null; + let parentSpanContext: any = null; + + // Act + useSpan(this._ai.core, parentSpan!, (scope: ISpanScope) => { + parentSpanContext = scope.span.spanContext(); + + // Create a child span while parent is active + const childSpan = this._ai.startSpan("child-span-in-useSpan"); + childSpanContext = childSpan!.spanContext(); + + // Verify trace context matches parent + const traceCtx = this._ai.core.getTraceCtx(false); + Assert.equal(traceCtx.spanId, parentSpanContext.spanId, + "Trace context should match parent span inside useSpan"); + + childSpan!.end(); + }); + + // Assert + Assert.ok(childSpanContext, "Child span context should exist"); + Assert.ok(parentSpanContext, "Parent span context should exist"); + + // Child should have same traceId as parent but different spanId + Assert.equal(childSpanContext.traceId, parentSpanContext.traceId, + "Child span should have same traceId as parent"); + Assert.notEqual(childSpanContext.spanId, parentSpanContext.spanId, + "Child span should have different spanId from parent"); + + parentSpan!.end(); + } + }); + + this.testCase({ + name: "UseSpan: trace context is restored after useSpan completes", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("temporary-trace-span"); + + let traceCtxBefore: any = null; + let traceCtxInside: any = null; + let traceCtxAfter: any = null; + + // Act + traceCtxBefore = this._ai.core.getTraceCtx(false); + + useSpan(this._ai.core, testSpan!, () => { + traceCtxInside = this._ai.core.getTraceCtx(false); + }); + + traceCtxAfter = this._ai.core.getTraceCtx(false); + + // Assert + Assert.ok(traceCtxBefore, "Trace context should exist before useSpan (created by startSpan)"); + Assert.ok(traceCtxInside, "Trace context should exist inside useSpan"); + Assert.equal(traceCtxInside.spanId, testSpan!.spanContext().spanId, + "Trace context inside useSpan should match the test span"); + Assert.ok(traceCtxAfter, "Trace context should exist after useSpan"); + Assert.equal(traceCtxAfter.spanId, traceCtxBefore.spanId, + "Trace context should be restored to previous state after useSpan"); + + testSpan!.end(); + } + }); + + this.testCase({ + name: "UseSpan: trace context reflects parent span when useSpan is nested in another active span", + test: () => { + // Arrange + const outerSpan = this._ai.startSpan("outer-active-span"); + const provider = this._ai.core.getTraceProvider(); + + this._ai.setActiveSpan(outerSpan!); + + const innerSpan = this._ai.startSpan("inner-usespan-span"); + + let outerSpanCtx: any = null; + let traceCtxBeforeUseSpan: any = null; + let traceCtxInsideUseSpan: any = null; + let traceCtxAfterUseSpan: any = null; + + // Act + outerSpanCtx = outerSpan!.spanContext(); + traceCtxBeforeUseSpan = this._ai.core.getTraceCtx(false); + + // Verify initial trace context matches outer span + Assert.equal(traceCtxBeforeUseSpan.spanId, outerSpanCtx.spanId, + "Trace context should initially match outer span"); + + useSpan(this._ai.core, innerSpan!, (scope: ISpanScope) => { + traceCtxInsideUseSpan = this._ai.core.getTraceCtx(false); + const innerSpanCtx = scope.span.spanContext(); + + // Inside useSpan, trace context should match inner span + Assert.equal(traceCtxInsideUseSpan.spanId, innerSpanCtx.spanId, + "Trace context inside useSpan should match inner span"); + }); + + traceCtxAfterUseSpan = this._ai.core.getTraceCtx(false); + + // After useSpan, trace context should be restored to outer span + Assert.equal(traceCtxAfterUseSpan.spanId, outerSpanCtx.spanId, + "Trace context should be restored to outer span after useSpan"); + + innerSpan!.end(); + outerSpan!.end(); + } + }); + + this.testCase({ + name: "UseSpan: trace context traceState is accessible inside useSpan", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("tracestate-test-span"); + + let traceStateInside: any = null; + + // Act + useSpan(this._ai.core, testSpan!, () => { + const traceCtx = this._ai.core.getTraceCtx(false); + traceStateInside = traceCtx ? traceCtx.traceState : null; + }); + + // Assert + Assert.ok(traceStateInside !== undefined, + "Trace state should be accessible inside useSpan"); + + testSpan!.end(); + } + }); + + this.testCase({ + name: "UseSpan: span created inside useSpan has parent context matching outer trace context", + test: () => { + // Arrange + const outerSpan = this._ai.startSpan("outer-parent-span"); + + this._ai.setActiveSpan(outerSpan!); + + let outerTraceCtx: any = null; + let innerSpanParentCtx: any = null; + let innerSpanCreated: any = null; + + // Act + outerTraceCtx = this._ai.core.getTraceCtx(false); + + useSpan(this._ai.core, outerSpan!, (scope: ISpanScope) => { + // Create a new span inside useSpan + innerSpanCreated = this._ai.startSpan("inner-child-span"); + + // Get the parent context of the newly created span + if (innerSpanCreated) { + innerSpanParentCtx = innerSpanCreated.parentSpanContext; + } + + innerSpanCreated!.end(); + }); + + // Assert + Assert.ok(outerTraceCtx, "Outer trace context should exist"); + Assert.ok(innerSpanParentCtx, "Inner span should have parent context"); + + // Verify parent context matches outer trace context + Assert.equal(innerSpanParentCtx.traceId, outerTraceCtx.traceId, + "Inner span parent traceId should match outer trace context traceId"); + Assert.equal(innerSpanParentCtx.spanId, outerTraceCtx.spanId, + "Inner span parent spanId should match outer trace context spanId"); + + outerSpan!.end(); + } + }); + + this.testCase({ + name: "UseSpan: span parent context matches trace context when useSpan wraps different span", + test: () => { + // Arrange - Create initial trace context + const contextSpan = this._ai.startSpan("context-span"); + + this._ai.setActiveSpan(contextSpan!); + + const contextTraceCtx = this._ai.core.getTraceCtx(false); + + // Create a different span to use in useSpan + const wrapperSpan = this._ai.startSpan("wrapper-span"); + + let spanCreatedInCallback: any = null; + let spanParentCtx: any = null; + + // Act - useSpan with a different span than what's in trace context + useSpan(this._ai.core, wrapperSpan!, (scope: ISpanScope) => { + // The active span is now wrapperSpan + // Create a child span - it should have wrapperSpan as parent + spanCreatedInCallback = this._ai.startSpan("child-of-wrapper"); + + if (spanCreatedInCallback) { + spanParentCtx = spanCreatedInCallback.parentSpanContext; + } + + spanCreatedInCallback!.end(); + }); + + // Assert + Assert.ok(spanParentCtx, "Child span should have parent context"); + + // Parent should be wrapperSpan (the useSpan span), not contextSpan + const wrapperSpanCtx = wrapperSpan!.spanContext(); + Assert.equal(spanParentCtx.spanId, wrapperSpanCtx.spanId, + "Child span parent should be the wrapper span from useSpan"); + Assert.notEqual(spanParentCtx.spanId, contextTraceCtx.spanId, + "Child span parent should NOT be the original context span"); + + wrapperSpan!.end(); + contextSpan!.end(); + } + }); + + this.testCase({ + name: "UseSpan: multiple nested spans maintain correct parent-child relationships with trace context", + test: () => { + // Arrange + const rootSpan = this._ai.startSpan("root-span"); + + this._ai.setActiveSpan(rootSpan!); + + const rootTraceCtx = this._ai.core.getTraceCtx(false); + const level1Span = this._ai.startSpan("level1-span"); + + let level2SpanParent: any = null; + let level2SpanCtx: any = null; + let level3SpanParent: any = null; + + // Act - Nested useSpan calls + useSpan(this._ai.core, level1Span!, (scope1: ISpanScope) => { + const level1TraceCtx = this._ai.core.getTraceCtx(false); + + // Create level2 span - should have level1 as parent + const level2Span = this._ai.startSpan("level2-span"); + if (level2Span) { + level2SpanParent = level2Span.parentSpanContext; + level2SpanCtx = level2Span.spanContext(); + } + + useSpan(this._ai.core, level2Span!, (scope2: ISpanScope) => { + const level2TraceCtx = this._ai.core.getTraceCtx(false); + + // Create level3 span - should have level2 as parent + const level3Span = this._ai.startSpan("level3-span"); + if (level3Span) { + level3SpanParent = level3Span.parentSpanContext; + } + + // Verify level3 parent matches level2 trace context + Assert.equal(level3SpanParent.spanId, level2TraceCtx.spanId, + "Level3 span parent should match level2 trace context"); + + level3Span!.end(); + }); + + // Verify level2 parent matches level1 trace context + Assert.equal(level2SpanParent.spanId, level1TraceCtx.spanId, + "Level2 span parent should match level1 trace context"); + + level2Span!.end(); + }); + + // Assert + Assert.ok(level2SpanParent, "Level2 span should have parent context"); + Assert.ok(level2SpanCtx, "Level2 span context should exist"); + Assert.ok(level3SpanParent, "Level3 span should have parent context"); + + // Verify the chain: root -> level1 -> level2 -> level3 + Assert.equal(level2SpanParent.spanId, level1Span!.spanContext().spanId, + "Level2 parent should be level1"); + Assert.equal(level3SpanParent.spanId, level2SpanCtx.spanId, + "Level3 parent should be level2"); + + level1Span!.end(); + rootSpan!.end(); + } + }); + } +} \ No newline at end of file diff --git a/AISKU/Tests/Unit/src/WithSpan.Tests.ts b/AISKU/Tests/Unit/src/WithSpan.Tests.ts new file mode 100644 index 000000000..534520037 --- /dev/null +++ b/AISKU/Tests/Unit/src/WithSpan.Tests.ts @@ -0,0 +1,1114 @@ +import { AITestClass, Assert } from '@microsoft/ai-test-framework'; +import { ApplicationInsights } from '../../../src/applicationinsights-web'; +import { + IReadableSpan, eOTelSpanKind, eOTelSpanStatusCode, withSpan, ITelemetryItem, ISpanScope, ITraceHost +} from "@microsoft/applicationinsights-core-js"; +export class WithSpanTests extends AITestClass { + private static readonly _instrumentationKey = 'b7170927-2d1c-44f1-acec-59f4e1751c11'; + private static readonly _connectionString = `InstrumentationKey=${WithSpanTests._instrumentationKey}`; + + private _ai!: ApplicationInsights; + + // Track calls to track for validation + private _trackCalls: ITelemetryItem[] = []; + + constructor(testName?: string) { + super(testName || "WithSpanTests"); + } + + public testInitialize() { + try { + this.useFakeServer = false; + this._trackCalls = []; + + this._ai = new ApplicationInsights({ + config: { + connectionString: WithSpanTests._connectionString, + disableAjaxTracking: false, + disableXhr: false, + maxBatchInterval: 0, + disableExceptionTracking: false + } + }); + + // Initialize the SDK + this._ai.loadAppInsights(); + + // Hook core.track to capture calls + const originalTrack = this._ai.core.track; + this._ai.core.track = (item: ITelemetryItem) => { + this._trackCalls.push(item); + return originalTrack.call(this._ai.core, item); + }; + + } catch (e) { + console.error('Failed to initialize WithSpan tests: ' + e); + throw e; + } + } + + public testFinishedCleanup() { + if (this._ai && this._ai.unload) { + this._ai.unload(false); + } + } + + public registerTests() { + this.addTests(); + } + + private addTests(): void { + + this.testCase({ + name: "WithSpan: withSpan should be available as exported function", + test: () => { + // Verify that withSpan is available as an import + Assert.ok(typeof withSpan === 'function', "withSpan should be available as exported function"); + } + }); + + this.testCase({ + name: "WithSpan: should execute function within span context", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("withSpan-context-test", { + kind: eOTelSpanKind.SERVER, + attributes: { + "test.type": "context-execution" + } + }); + + Assert.ok(testSpan, "Test span should be created"); + Assert.ok(this._ai.core, "Core should be available"); + + let capturedActiveSpan: IReadableSpan | null = null; + const testFunction = () => { + capturedActiveSpan = this._ai.core!.getActiveSpan(); + return "context-success"; + }; + + // Act + const result = withSpan(this._ai.core!, testSpan!, testFunction); + + // Assert + Assert.equal(result, "context-success", "Function should execute and return result"); + Assert.ok(capturedActiveSpan, "Function should have access to active span"); + Assert.equal(capturedActiveSpan, testSpan, "Active span should be the provided test span"); + } + }); + + this.testCase({ + name: "WithSpan: should work with telemetry tracking inside span context", + test: () => { + // Arrange + this._trackCalls = []; + const testSpan = this._ai.startSpan("withSpan-telemetry-test", { + attributes: { + "operation.name": "telemetry-tracking" + } + }); + + Assert.ok(testSpan, "Test span should be created"); + + const telemetryFunction = () => { + // Track some telemetry within the span context + this._ai.trackEvent({ + name: "operation-event", + properties: { + "event.source": "withSpan-context" + } + }); + + this._ai.trackMetric({ + name: "operation.duration", + average: 123.45 + }); + + return "telemetry-tracked"; + }; + + // Act + const result = withSpan(this._ai.core!, testSpan!, telemetryFunction); + + // Assert + Assert.equal(result, "telemetry-tracked", "Function should complete successfully"); + + // End the span to trigger trace generation + testSpan!.end(); + + // Verify track was called for the span + Assert.equal(this._trackCalls.length, 3, "Should have one track call from span ending"); + const item = this._trackCalls[2]; + Assert.ok(item.baseData && item.baseData.properties, "Item should have properties"); + Assert.equal("withSpan-telemetry-test", item.baseData.name, "Should include span name in properties"); + } + }); + + this.testCase({ + name: "WithSpan: should handle complex function arguments and return values", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("withSpan-arguments-test"); + Assert.ok(testSpan, "Test span should be created"); + + const complexFunction = ( + stringArg: string, + numberArg: number, + objectArg: { key: string; value: number }, + arrayArg: string[] + ) => { + return { + processedString: stringArg.toUpperCase(), + doubledNumber: numberArg * 2, + extractedValue: objectArg.value, + joinedArray: arrayArg.join('-'), + timestamp: Date.now() + }; + }; + + const inputObject = { key: "test-key", value: 42 }; + const inputArray = ["item1", "item2", "item3"]; + + // Act + const result = withSpan( + this._ai.core!, + testSpan!, + complexFunction, + undefined, + "hello world", + 10, + inputObject, + inputArray + ); + + // Assert + Assert.equal(result.processedString, "HELLO WORLD", "String should be processed correctly"); + Assert.equal(result.doubledNumber, 20, "Number should be doubled correctly"); + Assert.equal(result.extractedValue, 42, "Object value should be extracted correctly"); + Assert.equal(result.joinedArray, "item1-item2-item3", "Array should be joined correctly"); + Assert.ok(result.timestamp > 0, "Timestamp should be generated"); + } + }); + + this.testCase({ + name: "WithSpan: should handle function with this context binding", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("withSpan-this-binding-test"); + Assert.ok(testSpan, "Test span should be created"); + + class TestService { + private _serviceId: string; + private _multiplier: number; + + constructor(id: string, multiplier: number) { + this._serviceId = id; + this._multiplier = multiplier; + } + + public processValue(input: number): { serviceId: string; result: number; multiplied: number } { + return { + serviceId: this._serviceId, + result: input + 100, + multiplied: input * this._multiplier + }; + } + } + + const service = new TestService("test-service-123", 3); + + // Act + const result = withSpan( + this._ai.core!, + testSpan!, + service.processValue, + service, + 25 + ); + + // Assert + Assert.equal(result.serviceId, "test-service-123", "Service ID should be preserved via this binding"); + Assert.equal(result.result, 125, "Input should be processed correctly"); + Assert.equal(result.multiplied, 75, "Multiplication should use instance property"); + } + }); + + this.testCase({ + name: "WithSpan: should maintain span context across async-like operations", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("withSpan-async-like-test", { + attributes: { + "operation.type": "async-simulation" + } + }); + Assert.ok(testSpan, "Test span should be created"); + + let spanDuringCallback: IReadableSpan | null = null; + let callbackExecuted = false; + + const asyncLikeFunction = (callback: (data: string) => void) => { + // Simulate async work that completes synchronously in test + const currentSpan = this._ai.core!.getActiveSpan(); + + // Simulate callback execution (would normally be async) + setTimeout(() => { + spanDuringCallback = this._ai.core!.getActiveSpan(); + callback("async-data"); + callbackExecuted = true; + }, 0); + + return currentSpan ? (currentSpan as IReadableSpan).name : "no-span"; + }; + + // Act + let callbackData = ""; + const callback = (data: string) => { + callbackData = data; + }; + + const result = withSpan(this._ai.core!, testSpan!, asyncLikeFunction, undefined, callback); + + // Assert + Assert.equal(result, "withSpan-async-like-test", "Function should have access to span name"); + + // Note: In a real async scenario, the span context wouldn't automatically + // propagate to the setTimeout callback without additional context management + // This test validates the synchronous behavior of withSpan + } + }); + + this.testCase({ + name: "WithSpan: should handle exceptions and preserve span operations", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("withSpan-exception-test", { + attributes: { + "test.expects": "exception" + } + }); + Assert.ok(testSpan, "Test span should be created"); + + const exceptionFunction = () => { + // Perform some span operations before throwing + const activeSpan = this._ai.core!.getActiveSpan(); + Assert.ok(activeSpan, "Should have active span before exception"); + + activeSpan!.setAttribute("operation.status", "error"); + activeSpan!.setStatus({ + code: eOTelSpanStatusCode.ERROR, + message: "Operation failed with test exception" + }); + + throw new Error("Test exception for withSpan handling"); + }; + + // Act & Assert + let caughtException: Error | null = null; + try { + withSpan(this._ai.core!, testSpan!, exceptionFunction); + } catch (error) { + caughtException = error as Error; + } + + Assert.ok(caughtException, "Exception should be thrown and caught"); + Assert.equal(caughtException!.message, "Test exception for withSpan handling", "Exception message should be preserved"); + + // Verify span is still valid and operations were applied + Assert.ok(testSpan!.isRecording(), "Span should still be recording after exception"); + const readableSpan = testSpan! as IReadableSpan; + Assert.ok(!readableSpan.ended, "Span should not be ended by withSpan after exception"); + } + }); + + this.testCase({ + name: "WithSpan: should work with nested span operations and child spans", + test: () => { + // Arrange + this._trackCalls = []; + const parentSpan = this._ai.startSpan("parent-operation", { + kind: eOTelSpanKind.SERVER, + attributes: { + "operation.name": "parent-process" + } + }); + Assert.ok(parentSpan, "Parent span should be created"); + + const nestedOperations = () => { + // Verify we have the parent span as active + const currentActive = this._ai.core!.getActiveSpan(); + Assert.equal(currentActive, parentSpan, "Parent span should be active"); + + // Create child operations within the parent span context + const childSpan1 = this._ai.startSpan("child-operation-1", { + attributes: { "child.order": 1 } + }); + childSpan1!.setAttribute("child.status", "completed"); + childSpan1!.end(); + + const childSpan2 = this._ai.startSpan("child-operation-2", { + attributes: { "child.order": 2 } + }); + childSpan2!.setAttribute("child.status", "completed"); + childSpan2!.end(); + + return "nested-operations-completed"; + }; + + // Act + const result = withSpan(this._ai.core!, parentSpan!, nestedOperations); + + // Assert + Assert.equal(result, "nested-operations-completed", "Nested operations should complete successfully"); + + // End parent span to generate telemetry + parentSpan!.end(); + + // Should have 3 telemetry items: parent + 2 children + Assert.equal(this._trackCalls.length, 3, "Should have telemetry for parent and child spans"); + + // Verify span names in properties + const spanNames = this._trackCalls.map(item => item.baseData?.name).filter(n => n); + Assert.ok(spanNames.some(name => name === "parent-operation"), "Should have parent span telemetry"); + Assert.ok(spanNames.some(name => name === "child-operation-1"), "Should have child-1 span telemetry"); + Assert.ok(spanNames.some(name => name === "child-operation-2"), "Should have child-2 span telemetry"); + } + }); + + this.testCase({ + name: "WithSpan: should support different return value types", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("withSpan-return-types-test"); + Assert.ok(testSpan, "Test span should be created"); + + // Test various return types + const stringResult = withSpan(this._ai.core!, testSpan!, () => "string-result"); + const numberResult = withSpan(this._ai.core!, testSpan!, () => 42.5); + const booleanResult = withSpan(this._ai.core!, testSpan!, () => true); + const arrayResult = withSpan(this._ai.core!, testSpan!, () => [1, 2, 3]); + const objectResult = withSpan(this._ai.core!, testSpan!, () => ({ key: "value", nested: { prop: 123 } })); + const nullResult = withSpan(this._ai.core!, testSpan!, () => null); + const undefinedResult = withSpan(this._ai.core!, testSpan!, () => undefined); + + // Assert + Assert.equal(stringResult, "string-result", "String return should work"); + Assert.equal(numberResult, 42.5, "Number return should work"); + Assert.equal(booleanResult, true, "Boolean return should work"); + Assert.equal(arrayResult.length, 3, "Array return should work"); + Assert.equal(arrayResult[1], 2, "Array elements should be preserved"); + Assert.equal(objectResult.key, "value", "Object properties should be preserved"); + Assert.equal(objectResult.nested.prop, 123, "Nested object properties should be preserved"); + Assert.equal(nullResult, null, "Null return should work"); + Assert.equal(undefinedResult, undefined, "Undefined return should work"); + } + }); + + this.testCase({ + name: "WithSpan: should handle rapid successive calls efficiently", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("withSpan-performance-test"); + Assert.ok(testSpan, "Test span should be created"); + + const iterations = 100; + let totalResult = 0; + + // Simple computation function + const computeFunction = (input: number) => { + return input * 2 + 1; + }; + + const startTime = Date.now(); + + // Act - Multiple rapid withSpan calls + for (let i = 0; i < iterations; i++) { + const result = withSpan(this._ai.core!, testSpan!, computeFunction, undefined, i); + totalResult += result; + } + + const endTime = Date.now(); + const duration = endTime - startTime; + + // Assert + const expectedTotal = Array.from({length: iterations}, (_, i) => i * 2 + 1).reduce((sum, val) => sum + val, 0); + Assert.equal(totalResult, expectedTotal, "All computations should be correct"); + + // Performance assertion - should complete reasonably quickly + Assert.ok(duration < 1000, `Performance test should complete quickly: ${duration}ms for ${iterations} iterations`); + + // Verify span is still valid after many operations + Assert.ok(testSpan!.isRecording(), "Span should still be recording after multiple withSpan calls"); + } + }); + + this.testCase({ + name: "WithSpan: should integrate with AI telemetry correlation", + test: () => { + // Arrange + this._trackCalls = []; + const operationSpan = this._ai.startSpan("user-operation", { + kind: eOTelSpanKind.SERVER, + attributes: { + "user.id": "user-123", + "operation.type": "data-processing" + } + }); + Assert.ok(operationSpan, "Operation span should be created"); + + const businessLogicFunction = (userId: string, dataType: string) => { + // Track multiple telemetry items within span context + this._ai.trackEvent({ + name: "data-processing-started", + properties: { + "user.id": userId, + "data.type": dataType, + "processing.stage": "initialization" + } + }); + + // Simulate some processing steps + for (let step = 1; step <= 3; step++) { + this._ai.trackMetric({ + name: "processing.step.duration", + average: step * 10.5, + properties: { + "step.number": step.toString() + } + }); + } + + this._ai.trackEvent({ + name: "data-processing-completed", + properties: { + "user.id": userId, + "data.type": dataType, + "processing.stage": "completion", + "steps.completed": "3" + } + }); + + return { + userId: userId, + dataType: dataType, + stepsCompleted: 3, + status: "success" + }; + }; + + // Act + const result = withSpan( + this._ai.core!, + operationSpan!, + businessLogicFunction, + undefined, + "user-123", + "customer-data" + ); + + // End the span to generate trace + operationSpan!.end(); + + // Assert + Assert.equal(result.userId, "user-123", "User ID should be processed correctly"); + Assert.equal(result.dataType, "customer-data", "Data type should be processed correctly"); + Assert.equal(result.stepsCompleted, 3, "All processing steps should be completed"); + Assert.equal(result.status, "success", "Operation should complete successfully"); + + // Verify span telemetry was generated + Assert.equal(this._trackCalls.length, 6, "Should have one track call from span ending"); + const spanItem = this._trackCalls[5]; + Assert.ok(spanItem.baseData && spanItem.baseData.properties, "Item should have properties"); + Assert.equal("user-operation", spanItem.baseData.name, "Should include span name"); + + // Verify span attributes are included in properties + Assert.equal(spanItem.baseData.properties["user.id"], "user-123", "Span attributes should be included in telemetry"); + Assert.equal(spanItem.baseData.properties["operation.type"], "data-processing", "All span attributes should be preserved"); + } + }); + + this.testCase({ + name: "WithSpan: should handle empty or no-op functions gracefully", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("withSpan-noop-test"); + Assert.ok(testSpan, "Test span should be created"); + + // Test empty function + const emptyFunction = () => {}; + + // Test function that just returns without doing anything + const noOpFunction = () => { + return; + }; + + // Test function that returns undefined explicitly + const undefinedFunction = () => { + return undefined; + }; + + // Act + const emptyResult = withSpan(this._ai.core!, testSpan!, emptyFunction); + const noOpResult = withSpan(this._ai.core!, testSpan!, noOpFunction); + const undefinedResult = withSpan(this._ai.core!, testSpan!, undefinedFunction); + + // Assert + Assert.equal(emptyResult, undefined, "Empty function should return undefined"); + Assert.equal(noOpResult, undefined, "No-op function should return undefined"); + Assert.equal(undefinedResult, undefined, "Undefined function should return undefined"); + + // Verify span is still valid + Assert.ok(testSpan!.isRecording(), "Span should still be recording after no-op functions"); + } + }); + + this.testCase({ + name: "WithSpan: should use ISpanScope as 'this' when no thisArg provided", + test: () => { + // Arrange + const span = this._ai.startSpan("test-span", { + attributes: { "test.id": "withSpan-this-test" } + }); + + let capturedThis: any = null; + let capturedHost: ITraceHost | null = null; + let capturedSpan: any = null; + + // Act - call withSpan without thisArg + const result = withSpan(this._ai.core, span!, function(this: ISpanScope, arg1: string, arg2: number) { + capturedThis = this; + capturedHost = this.host; + capturedSpan = this.span; + return `${arg1}-${arg2}`; + }, undefined, "test", 42); + + // Assert + Assert.equal(result, "test-42", "Function should execute and return result"); + Assert.ok(capturedThis, "'this' should be defined"); + Assert.ok(capturedThis.host, "'this.host' should exist"); + Assert.ok(capturedThis.span, "'this.span' should exist"); + Assert.equal(capturedHost, this._ai.core, "'this.host' should be the AI core"); + Assert.equal(capturedSpan, span, "'this.span' should be the passed span"); + Assert.equal(capturedThis.span.name, "test-span", "'this.span.name' should match"); + + span!.end(); + } + }); + + this.testCase({ + name: "WithSpan: should use provided thisArg when specified", + test: () => { + // Arrange + const span = this._ai.startSpan("test-span-thisarg"); + + const customContext = { + contextId: "custom-123", + multiplier: 10 + }; + + let capturedThis: any = null; + + // Act - call withSpan with explicit thisArg + const result = withSpan(this._ai.core, span!, function(this: typeof customContext, arg1: number) { + capturedThis = this; + return arg1 * this.multiplier; + }, customContext, 5); + + // Assert + Assert.equal(result, 50, "Function should execute with custom this context"); + Assert.ok(capturedThis, "'this' should be defined"); + Assert.equal(capturedThis, customContext, "'this' should be the custom context"); + Assert.equal(capturedThis.contextId, "custom-123", "'this.contextId' should match"); + Assert.equal(capturedThis.multiplier, 10, "'this.multiplier' should match"); + Assert.ok(!capturedThis.core, "Custom this should not have core property"); + Assert.ok(!capturedThis.span, "Custom this should not have span property"); + + span!.end(); + } + }); + + this.testCase({ + name: "WithSpan: arrow functions should not override 'this' binding", + test: () => { + // Arrange + const span = this._ai.startSpan("arrow-function-test"); + + // Act - arrow functions capture their lexical 'this' + const result = withSpan(this._ai.core, span!, (arg: string) => { + // Arrow function - 'this' is lexically bound to the test class instance + Assert.ok(this._ai, "Arrow function should have access to test class 'this'"); + return `arrow-${arg}`; + }, undefined, "result"); + + // Assert + Assert.equal(result, "arrow-result", "Arrow function should execute correctly"); + Assert.ok(this._ai, "Test class instance should still be accessible"); + + span!.end(); + } + }); + + this.testCase({ + name: "WithSpan: verify ISpanScope.restore() is called to restore previous active span", + test: () => { + // Arrange + const outerSpan = this._ai.startSpan("outer-span"); + const innerSpan = this._ai.startSpan("inner-span"); + + let activeSpanBeforeWithSpan: any = null; + let activeSpanInsideWithSpan: any = null; + let activeSpanAfterWithSpan: any = null; + + // Act + activeSpanBeforeWithSpan = this._ai.core.getActiveSpan ? this._ai.core.getActiveSpan() : null; + + withSpan(this._ai.core, innerSpan!, () => { + activeSpanInsideWithSpan = this._ai.core.getActiveSpan ? this._ai.core.getActiveSpan() : null; + Assert.equal(activeSpanInsideWithSpan, innerSpan, "Active span inside withSpan should be inner span"); + }); + + activeSpanAfterWithSpan = this._ai.core.getActiveSpan ? this._ai.core.getActiveSpan() : null; + + // Assert + // Active span should be restored after withSpan completes + Assert.equal(activeSpanAfterWithSpan, activeSpanBeforeWithSpan, + "Active span should be restored after withSpan completes"); + + innerSpan!.end(); + outerSpan!.end(); + } + }); + + this.testCase({ + name: "WithSpan: 'this' binding with nested withSpan calls", + test: () => { + // Arrange + const outerSpan = this._ai.startSpan("outer-withspan"); + const innerSpan = this._ai.startSpan("inner-withspan"); + + const outerContext = { + contextName: "outer", + value: 100 + }; + + let outerThisCapture: any = null; + let innerThisCapture: any = null; + let ai = this._ai; + + // Act - nested withSpan calls with different thisArg + withSpan(ai.core, outerSpan!, function(this: typeof outerContext, arg: number) { + outerThisCapture = this; + Assert.equal(this.contextName, "outer", "Outer 'this' should be outer context"); + Assert.equal(this.value, 100, "Outer 'this.value' should match"); + + withSpan(ai.core, innerSpan!, function(this: ISpanScope) { + innerThisCapture = this; + // Inner call without explicit thisArg - should be ISpanScope + Assert.ok(this.host, "Inner 'this' should be ISpanScope"); + Assert.ok(this.span, "Inner 'this.span' should exist"); + Assert.equal(this.span.name, "inner-withspan", "Inner span name should match"); + }); + + return arg * this.value; + }, outerContext, 2); + + // Assert + Assert.ok(outerThisCapture, "Outer 'this' should be captured"); + Assert.equal(outerThisCapture.contextName, "outer", "Outer context should be preserved"); + + Assert.ok(innerThisCapture, "Inner 'this' should be captured"); + Assert.ok(innerThisCapture.host, "Inner 'this' should have host"); + Assert.ok(innerThisCapture.span, "Inner 'this' should have span"); + + innerSpan!.end(); + outerSpan!.end(); + } + }); + + this.testCase({ + name: "WithSpan: trace context should match active span context inside withSpan", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("trace-context-match-test", { + attributes: { + "test.type": "trace-context-validation" + } + }); + Assert.ok(testSpan, "Test span should be created"); + + let traceCtxInsideWithSpan: any = null; + let spanContextInsideWithSpan: any = null; + let activeSpanInsideWithSpan: any = null; + + // Act + withSpan(this._ai.core, testSpan!, function(this: ISpanScope) { + // Get trace context from core + traceCtxInsideWithSpan = this.host.getTraceCtx(false); + + // Get span context from the span + spanContextInsideWithSpan = this.span.spanContext(); + + // Get active span + activeSpanInsideWithSpan = this.host.getActiveSpan ? this.host.getActiveSpan() : null; + }); + + // Assert + Assert.ok(traceCtxInsideWithSpan, "Trace context should exist inside withSpan"); + Assert.ok(spanContextInsideWithSpan, "Span context should exist"); + Assert.ok(activeSpanInsideWithSpan, "Active span should be set"); + + // Verify active span matches the withSpan span + Assert.equal(activeSpanInsideWithSpan, testSpan, "Active span should be the withSpan span"); + + // Verify trace context matches span context + Assert.equal(traceCtxInsideWithSpan.traceId, spanContextInsideWithSpan.traceId, + "Trace context traceId should match span context traceId"); + Assert.equal(traceCtxInsideWithSpan.spanId, spanContextInsideWithSpan.spanId, + "Trace context spanId should match span context spanId"); + Assert.equal(traceCtxInsideWithSpan.traceFlags, spanContextInsideWithSpan.traceFlags, + "Trace context traceFlags should match span context traceFlags"); + + testSpan!.end(); + } + }); + + this.testCase({ + name: "WithSpan: trace context updates when switching between nested withSpan calls", + test: () => { + // Arrange + const outerSpan = this._ai.startSpan("outer-trace-span"); + const innerSpan = this._ai.startSpan("inner-trace-span"); + + let outerTraceCtx: any = null; + let outerSpanCtx: any = null; + let innerTraceCtx: any = null; + let innerSpanCtx: any = null; + let ai = this._ai; + + // Act + withSpan(ai.core, outerSpan!, function(this: ISpanScope) { + outerTraceCtx = this.host.getTraceCtx(false); + outerSpanCtx = this.span.spanContext(); + + // Verify outer trace context matches outer span + Assert.equal(outerTraceCtx.spanId, outerSpanCtx.spanId, + "Outer trace context should match outer span"); + + // Nested withSpan with different span + withSpan(ai.core, innerSpan!, function(this: ISpanScope) { + innerTraceCtx = this.host.getTraceCtx(false); + innerSpanCtx = this.span.spanContext(); + + // Verify inner trace context matches inner span + Assert.equal(innerTraceCtx.spanId, innerSpanCtx.spanId, + "Inner trace context should match inner span"); + + // Verify inner context is different from outer + Assert.notEqual(innerTraceCtx.spanId, outerTraceCtx.spanId, + "Inner and outer trace contexts should have different spanIds"); + }); + + // After inner withSpan, verify we're back to outer context + const restoredTraceCtx = this.host.getTraceCtx(false); + Assert.equal(restoredTraceCtx.spanId, outerSpanCtx.spanId, + "Trace context should be restored to outer span after inner withSpan completes"); + }); + + outerSpan!.end(); + innerSpan!.end(); + } + }); + + this.testCase({ + name: "WithSpan: child spans created inside withSpan inherit correct parent context", + test: () => { + // Arrange + const parentSpan = this._ai.startSpan("parent-for-child-test"); + let ai = this._ai; + + let childSpanContext: any = null; + let parentSpanContext: any = null; + + // Act + withSpan(ai.core, parentSpan!, function(this: ISpanScope) { + parentSpanContext = this.span.spanContext(); + + // Create a child span while parent is active + const childSpan = ai.startSpan("child-span-in-withSpan"); + childSpanContext = childSpan!.spanContext(); + + // Verify trace context matches parent + const traceCtx = this.host.getTraceCtx(false); + Assert.equal(traceCtx.spanId, parentSpanContext.spanId, + "Trace context should match parent span inside withSpan"); + + childSpan!.end(); + }); + + // Assert + Assert.ok(childSpanContext, "Child span context should exist"); + Assert.ok(parentSpanContext, "Parent span context should exist"); + + // Child should have same traceId as parent but different spanId + Assert.equal(childSpanContext.traceId, parentSpanContext.traceId, + "Child span should have same traceId as parent"); + Assert.notEqual(childSpanContext.spanId, parentSpanContext.spanId, + "Child span should have different spanId from parent"); + + parentSpan!.end(); + } + }); + + this.testCase({ + name: "WithSpan: trace context is restored after withSpan completes", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("temporary-trace-span"); + + let traceCtxBefore: any = null; + let traceCtxInside: any = null; + let traceCtxAfter: any = null; + let ai = this._ai; + + // Act + traceCtxBefore = this._ai.core.getTraceCtx(false); + + withSpan(ai.core, testSpan!, function(this: ISpanScope) { + traceCtxInside = this.host.getTraceCtx(false); + }); + + traceCtxAfter = this._ai.core.getTraceCtx(false); + + // Assert + Assert.ok(traceCtxBefore, "Trace context should exist before withSpan (created by startSpan)"); + Assert.ok(traceCtxInside, "Trace context should exist inside withSpan"); + Assert.equal(traceCtxInside.spanId, testSpan!.spanContext().spanId, + "Trace context inside withSpan should match the test span"); + Assert.ok(traceCtxAfter, "Trace context should exist after withSpan"); + Assert.equal(traceCtxAfter.spanId, traceCtxBefore.spanId, + "Trace context should be restored to previous state after withSpan"); + + testSpan!.end(); + } + }); + + this.testCase({ + name: "WithSpan: trace context reflects parent span when withSpan is nested in another active span", + test: () => { + // Arrange + const outerSpan = this._ai.startSpan("outer-active-span"); + + this._ai.setActiveSpan(outerSpan!); + + const innerSpan = this._ai.startSpan("inner-withspan-span"); + let ai = this._ai; + + let outerSpanCtx: any = null; + let traceCtxBeforeWithSpan: any = null; + let traceCtxInsideWithSpan: any = null; + let traceCtxAfterWithSpan: any = null; + + // Act + outerSpanCtx = outerSpan!.spanContext(); + traceCtxBeforeWithSpan = this._ai.core.getTraceCtx(false); + + // Verify initial trace context matches outer span + Assert.equal(traceCtxBeforeWithSpan.spanId, outerSpanCtx.spanId, + "Trace context should initially match outer span"); + + withSpan(ai.core, innerSpan!, function(this: ISpanScope) { + traceCtxInsideWithSpan = this.host.getTraceCtx(false); + const innerSpanCtx = this.span.spanContext(); + + // Inside withSpan, trace context should match inner span + Assert.equal(traceCtxInsideWithSpan.spanId, innerSpanCtx.spanId, + "Trace context inside withSpan should match inner span"); + }); + + traceCtxAfterWithSpan = this._ai.core.getTraceCtx(false); + + // After withSpan, trace context should be restored to outer span + Assert.equal(traceCtxAfterWithSpan.spanId, outerSpanCtx.spanId, + "Trace context should be restored to outer span after withSpan"); + + innerSpan!.end(); + outerSpan!.end(); + } + }); + + this.testCase({ + name: "WithSpan: trace context traceState is accessible inside withSpan", + test: () => { + // Arrange + const testSpan = this._ai.startSpan("tracestate-test-span"); + let ai = this._ai; + + let traceStateInside: any = null; + + // Act + withSpan(ai.core, testSpan!, function(this: ISpanScope) { + const traceCtx = this.host.getTraceCtx(false); + traceStateInside = traceCtx ? traceCtx.traceState : null; + }); + + // Assert + Assert.ok(traceStateInside !== undefined, + "Trace state should be accessible inside withSpan"); + + testSpan!.end(); + } + }); + + this.testCase({ + name: "WithSpan: span created inside withSpan has parent context matching outer trace context", + test: () => { + // Arrange + const outerSpan = this._ai.startSpan("outer-parent-span"); + + this._ai.setActiveSpan(outerSpan!); + + let outerTraceCtx: any = null; + let innerSpanParentCtx: any = null; + let innerSpanCreated: any = null; + let ai = this._ai; + + // Act + outerTraceCtx = this._ai.core.getTraceCtx(false); + + withSpan(ai.core, outerSpan!, function(this: ISpanScope) { + // Create a new span inside withSpan + innerSpanCreated = ai.startSpan("inner-child-span"); + + // Get the parent context of the newly created span + if (innerSpanCreated) { + innerSpanParentCtx = innerSpanCreated.parentSpanContext; + } + + innerSpanCreated!.end(); + }); + + // Assert + Assert.ok(outerTraceCtx, "Outer trace context should exist"); + Assert.ok(innerSpanParentCtx, "Inner span should have parent context"); + + // Verify parent context matches outer trace context + Assert.equal(innerSpanParentCtx.traceId, outerTraceCtx.traceId, + "Inner span parent traceId should match outer trace context traceId"); + Assert.equal(innerSpanParentCtx.spanId, outerTraceCtx.spanId, + "Inner span parent spanId should match outer trace context spanId"); + + outerSpan!.end(); + } + }); + + this.testCase({ + name: "WithSpan: span parent context matches trace context when withSpan wraps different span", + test: () => { + // Arrange - Create initial trace context + const contextSpan = this._ai.startSpan("context-span"); + + this._ai.setActiveSpan(contextSpan!); + + const contextTraceCtx = this._ai.core.getTraceCtx(false); + + // Create a different span to use in withSpan + const wrapperSpan = this._ai.startSpan("wrapper-span"); + + let spanCreatedInCallback: any = null; + let spanParentCtx: any = null; + let ai = this._ai; + + // Act - withSpan with a different span than what's in trace context + withSpan(ai.core, wrapperSpan!, function(this: ISpanScope) { + // The active span is now wrapperSpan + // Create a child span - it should have wrapperSpan as parent + spanCreatedInCallback = ai.startSpan("child-of-wrapper"); + + if (spanCreatedInCallback) { + spanParentCtx = spanCreatedInCallback.parentSpanContext; + } + + spanCreatedInCallback!.end(); + }); + + // Assert + Assert.ok(spanParentCtx, "Child span should have parent context"); + + // Parent should be wrapperSpan (the withSpan span), not contextSpan + const wrapperSpanCtx = wrapperSpan!.spanContext(); + Assert.equal(spanParentCtx.spanId, wrapperSpanCtx.spanId, + "Child span parent should be the wrapper span from withSpan"); + Assert.notEqual(spanParentCtx.spanId, contextTraceCtx.spanId, + "Child span parent should NOT be the original context span"); + + wrapperSpan!.end(); + contextSpan!.end(); + } + }); + + this.testCase({ + name: "WithSpan: multiple nested spans maintain correct parent-child relationships with trace context", + test: () => { + // Arrange + const rootSpan = this._ai.startSpan("root-span"); + + this._ai.setActiveSpan(rootSpan!); + + const level1Span = this._ai.startSpan("level1-span"); + + let level2SpanParent: any = null; + let level2SpanCtx: any = null; + let level3SpanParent: any = null; + let ai = this._ai; + + // Act - Nested withSpan calls + withSpan(ai.core, level1Span!, function(this: ISpanScope) { + const level1TraceCtx = this.host.getTraceCtx(false); + + // Create level2 span - should have level1 as parent + const level2Span = ai.startSpan("level2-span"); + if (level2Span) { + level2SpanParent = level2Span.parentSpanContext; + level2SpanCtx = level2Span.spanContext(); + } + + withSpan(ai.core, level2Span!, function(this: ISpanScope) { + const level2TraceCtx = this.host.getTraceCtx(false); + + // Create level3 span - should have level2 as parent + const level3Span = ai.startSpan("level3-span"); + if (level3Span) { + level3SpanParent = level3Span.parentSpanContext; + } + + // Verify level3 parent matches level2 trace context + Assert.equal(level3SpanParent.spanId, level2TraceCtx.spanId, + "Level3 span parent should match level2 trace context"); + + level3Span!.end(); + }); + + // Verify level2 parent matches level1 trace context + Assert.equal(level2SpanParent.spanId, level1TraceCtx.spanId, + "Level2 span parent should match level1 trace context"); + + level2Span!.end(); + }); + + // Assert + Assert.ok(level2SpanParent, "Level2 span should have parent context"); + Assert.ok(level2SpanCtx, "Level2 span context should exist"); + Assert.ok(level3SpanParent, "Level3 span should have parent context"); + + // Verify the chain: root -> level1 -> level2 -> level3 + Assert.equal(level2SpanParent.spanId, level1Span!.spanContext().spanId, + "Level2 parent should be level1"); + Assert.equal(level3SpanParent.spanId, level2SpanCtx.spanId, + "Level3 parent should be level2"); + + level1Span!.end(); + rootSpan!.end(); + } + }); + } +} \ No newline at end of file diff --git a/AISKU/Tests/Unit/src/aiskuunittests.ts b/AISKU/Tests/Unit/src/aiskuunittests.ts index 8a53729b0..3088c034d 100644 --- a/AISKU/Tests/Unit/src/aiskuunittests.ts +++ b/AISKU/Tests/Unit/src/aiskuunittests.ts @@ -1,17 +1,44 @@ import { AISKUSizeCheck } from "./AISKUSize.Tests"; -import { ApplicationInsightsTests } from './applicationinsights.e2e.tests'; -import { ApplicationInsightsFetchTests } from './applicationinsights.e2e.fetch.tests'; -import { CdnPackagingChecks } from './CdnPackaging.tests'; -import { GlobalTestHooks } from './GlobalTestHooks.Test'; -import { SanitizerE2ETests } from './sanitizer.e2e.tests'; -import { ValidateE2ETests } from './validate.e2e.tests'; -import { SenderE2ETests } from './sender.e2e.tests'; -import { SnippetInitializationTests } from './SnippetInitialization.Tests'; +import { ApplicationInsightsTests } from "./applicationinsights.e2e.tests"; +import { ApplicationInsightsFetchTests } from "./applicationinsights.e2e.fetch.tests"; +import { CdnPackagingChecks } from "./CdnPackaging.tests"; +import { GlobalTestHooks } from "./GlobalTestHooks.Test"; +import { SanitizerE2ETests } from "./sanitizer.e2e.tests"; +import { ValidateE2ETests } from "./validate.e2e.tests"; +import { SenderE2ETests } from "./sender.e2e.tests"; +import { SnippetInitializationTests } from "./SnippetInitialization.Tests"; import { CdnThrottle} from "./CdnThrottle.tests"; import { ThrottleSentMessage } from "./ThrottleSentMessage.tests"; -import { IAnalyticsConfigTests } from './IAnalyticsConfig.Tests'; +import { IAnalyticsConfigTests } from "./IAnalyticsConfig.Tests"; +import { StartSpanTests } from "./StartSpan.Tests"; +import { UseSpanTests } from "./UseSpan.Tests"; +import { WithSpanTests } from "./WithSpan.Tests"; +import { SpanContextPropagationTests } from "./SpanContextPropagation.Tests"; +import { SpanLifeCycleTests } from "./SpanLifeCycle.Tests" +import { TelemetryItemGenerationTests } from "./TelemetryItemGeneration.Tests"; +import { SpanErrorHandlingTests } from "./SpanErrorHandling.Tests"; +import { SpanUtilsTests } from "./SpanUtils.Tests"; +import { SpanE2ETests } from "./SpanE2E.Tests"; +import { NonRecordingSpanTests } from "./NonRecordingSpan.Tests"; +import { SpanPluginIntegrationTests } from "./SpanPluginIntegration.Tests"; +import { SpanHelperUtilsTests } from "./SpanHelperUtils.Tests"; +import { TraceSuppressionTests } from "./TraceSuppression.Tests"; +import { TraceProviderTests } from "./TraceProvider.Tests"; +import { TraceContextTests } from "./TraceContext.Tests"; +import { OTelInitTests } from "./OTelInit.Tests"; export function runTests() { + new OTelInitTests().registerTests(); + new TraceSuppressionTests().registerTests(); + new SpanErrorHandlingTests().registerTests(); + new SpanUtilsTests().registerTests(); + new SpanE2ETests().registerTests(); + new NonRecordingSpanTests().registerTests(); + new SpanPluginIntegrationTests().registerTests(); + new SpanHelperUtilsTests().registerTests(); + new TraceProviderTests().registerTests(); + new TraceContextTests().registerTests(); + new GlobalTestHooks().registerTests(); new AISKUSizeCheck().registerTests(); new ApplicationInsightsTests().registerTests(); @@ -25,4 +52,10 @@ export function runTests() { new ThrottleSentMessage().registerTests(); new CdnThrottle().registerTests(); new IAnalyticsConfigTests().registerTests(); + new StartSpanTests().registerTests(); + new WithSpanTests().registerTests(); + new UseSpanTests().registerTests(); + new SpanContextPropagationTests().registerTests(); + new SpanLifeCycleTests().registerTests(); + new TelemetryItemGenerationTests().registerTests(); } \ No newline at end of file diff --git a/AISKU/Tests/Unit/src/applicationinsights.e2e.fetch.tests.ts b/AISKU/Tests/Unit/src/applicationinsights.e2e.fetch.tests.ts index 3d5a044a5..a2c601760 100644 --- a/AISKU/Tests/Unit/src/applicationinsights.e2e.fetch.tests.ts +++ b/AISKU/Tests/Unit/src/applicationinsights.e2e.fetch.tests.ts @@ -1,4 +1,4 @@ -import { DistributedTracingModes, IConfig } from '@microsoft/applicationinsights-common'; +import { DistributedTracingModes, IConfig } from '@microsoft/applicationinsights-core-js'; import { ApplicationInsightsTests } from './applicationinsights.e2e.tests'; import { IConfiguration } from '@microsoft/applicationinsights-core-js'; diff --git a/AISKU/Tests/Unit/src/applicationinsights.e2e.tests.ts b/AISKU/Tests/Unit/src/applicationinsights.e2e.tests.ts index a607c3b42..490a72de2 100644 --- a/AISKU/Tests/Unit/src/applicationinsights.e2e.tests.ts +++ b/AISKU/Tests/Unit/src/applicationinsights.e2e.tests.ts @@ -2,21 +2,21 @@ import { AITestClass, Assert, PollingAssert, EventValidator, TraceValidator, Exc import { SinonSpy } from 'sinon'; import { ApplicationInsights } from '../../../src/applicationinsights-web' import { Sender } from '@microsoft/applicationinsights-channel-js'; -import { IDependencyTelemetry, ContextTagKeys, Event, Trace, Exception, Metric, PageView, PageViewPerformance, RemoteDependencyData, DistributedTracingModes, RequestHeaders, IAutoExceptionTelemetry, BreezeChannelIdentifier, IConfig, EventPersistence } from '@microsoft/applicationinsights-common'; +import { IDependencyTelemetry, ContextTagKeys, Exception, DistributedTracingModes, RequestHeaders, IAutoExceptionTelemetry, BreezeChannelIdentifier, IConfig, EventPersistence, EventDataType, PageViewDataType, TraceDataType, ExceptionDataType, MetricDataType, PageViewPerformanceDataType, RemoteDependencyDataType } from '@microsoft/applicationinsights-core-js'; import { ITelemetryItem, getGlobal, newId, dumpObj, BaseTelemetryPlugin, IProcessTelemetryContext, __getRegisteredEvents, arrForEach, IConfiguration, ActiveStatus, FeatureOptInMode } from "@microsoft/applicationinsights-core-js"; -import { TelemetryContext } from '@microsoft/applicationinsights-properties-js'; +import { IPropTelemetryContext } from '@microsoft/applicationinsights-properties-js'; import { createAsyncResolvedPromise } from '@nevware21/ts-async'; import { CONFIG_ENDPOINT_URL } from '../../../src/InternalConstants'; import { OfflineChannel } from '@microsoft/applicationinsights-offlinechannel-js'; -import { IStackFrame } from '@microsoft/applicationinsights-common/src/Interfaces/Contracts/IStackFrame'; +import { IStackFrame } from '@microsoft/applicationinsights-core-js'; import { utcNow } from '@nevware21/ts-utils'; -function _checkExpectedFrame(expectedFrame: IStackFrame, actualFrame: IStackFrame, index: number) { +function _checkExpectedFrame(expectedFrame: IStackFrame, actualFrame: IStackFrame, index: number) { Assert.equal(expectedFrame.assembly, actualFrame.assembly, index + ") Assembly is not as expected"); Assert.equal(expectedFrame.fileName, actualFrame.fileName, index + ") FileName is not as expected"); Assert.equal(expectedFrame.line, actualFrame.line, index + ") Line is not as expected"); Assert.equal(expectedFrame.method, actualFrame.method, index + ") Method is not as expected"); - Assert.equal(expectedFrame.level, actualFrame.level, index + ") Level is not as expected"); + Assert.equal(expectedFrame.level, actualFrame.level, index + ") Level is not as expected"); } export class ApplicationInsightsTests extends AITestClass { @@ -40,7 +40,7 @@ export class ApplicationInsightsTests extends AITestClass { private _ai: ApplicationInsights; private _aiName: string = 'AppInsightsSDK'; - private isFetchPolyfill:boolean = false; + private isFetchPolyfill: boolean = false; // Sinon private errorSpy: SinonSpy; @@ -61,7 +61,7 @@ export class ApplicationInsightsTests extends AITestClass { constructor(testName?: string) { super(testName || "ApplicationInsightsTests"); } - + protected _getTestConfig(sessionPrefix: string) { let config: IConfiguration | IConfig = { connectionString: ApplicationInsightsTests._connectionString, @@ -76,7 +76,7 @@ export class ApplicationInsightsTests extends AITestClass { distributedTracingMode: DistributedTracingModes.AI_AND_W3C, samplingPercentage: 50, convertUndefined: "test-value", - disablePageUnloadEvents: [ "beforeunload" ], + disablePageUnloadEvents: ["beforeunload"], extensionConfig: { ["AppInsightsCfgSyncPlugin"]: { //cfgUrl: "" @@ -229,7 +229,7 @@ export class ApplicationInsightsTests extends AITestClass { let onChangeCalled = 0; let handler = this._ai.onCfgChange((details) => { - onChangeCalled ++; + onChangeCalled++; Assert.equal(expectedIkey, details.cfg.instrumentationKey, "Expect the iKey to be set"); Assert.equal(expectedEndpointUrl, details.cfg.endpointUrl, "Expect the endpoint to be set"); Assert.equal(expectedLoggingLevel, details.cfg.diagnosticLogInterval, "Expect the diagnosticLogInterval to be set"); @@ -256,7 +256,7 @@ export class ApplicationInsightsTests extends AITestClass { Assert.equal(3, onChangeCalled, "Expected the onChanged was called"); this.clock.tick(1); Assert.equal(4, onChangeCalled, "Expected the onChanged was called again"); - + // Remove the handler handler.rm(); } @@ -274,15 +274,15 @@ export class ApplicationInsightsTests extends AITestClass { // force unload oriInst.unload(false); } - + if (oriInst && oriInst["dependencies"]) { oriInst["dependencies"].teardown(); } - + this._config = this._getTestConfig(this._sessionPrefix); let csPromise = createAsyncResolvedPromise("InstrumentationKey=testIkey;ingestionendpoint=testUrl"); this._config.connectionString = csPromise; - this._config.initTimeOut= 80000; + this._config.initTimeOut = 80000; this._ctx.csPromise = csPromise; @@ -295,17 +295,17 @@ export class ApplicationInsightsTests extends AITestClass { let core = this._ai.core; let status = core.activeStatus && core.activeStatus(); Assert.equal(status, ActiveStatus.PENDING, "status should be set to pending"); - - + + }].concat(PollingAssert.createPollingAssert(() => { let core = this._ai.core let activeStatus = core.activeStatus && core.activeStatus(); let csPromise = this._ctx.csPromise; let config = this._ai.config; - + if (csPromise.state === "resolved" && activeStatus === ActiveStatus.ACTIVE) { Assert.equal("testIkey", core.config.instrumentationKey, "ikey should be set"); - Assert.equal("testUrl/v2/track", core.config.endpointUrl ,"endpoint shoule be set"); + Assert.equal("testUrl/v2/track", core.config.endpointUrl, "endpoint shoule be set"); config.connectionString = "InstrumentationKey=testIkey1;ingestionendpoint=testUrl1"; this.clock.tick(1); @@ -318,10 +318,10 @@ export class ApplicationInsightsTests extends AITestClass { }, "Wait for promise response" + new Date().toISOString(), 60) as any).concat(PollingAssert.createPollingAssert(() => { let core = this._ai.core let activeStatus = core.activeStatus && core.activeStatus(); - + if (activeStatus === ActiveStatus.ACTIVE) { Assert.equal("testIkey1", core.config.instrumentationKey, "ikey should be set test1"); - Assert.equal("testUrl1/v2/track", core.config.endpointUrl ,"endpoint shoule be set test1"); + Assert.equal("testUrl1/v2/track", core.config.endpointUrl, "endpoint shoule be set test1"); return true; } return false; @@ -340,15 +340,15 @@ export class ApplicationInsightsTests extends AITestClass { // force unload oriInst.unload(false); } - + if (oriInst && oriInst["dependencies"]) { oriInst["dependencies"].teardown(); } - + this._config = this._getTestConfig(this._sessionPrefix); let csPromise = createAsyncResolvedPromise("InstrumentationKey=testIkey;ingestionendpoint=testUrl"); this._config.connectionString = csPromise; - this._config.initTimeOut= 80000; + this._config.initTimeOut = 80000; this._ctx.csPromise = csPromise; @@ -361,22 +361,22 @@ export class ApplicationInsightsTests extends AITestClass { let core = this._ai.core; let status = core.activeStatus && core.activeStatus(); Assert.equal(status, ActiveStatus.PENDING, "status should be set to pending"); - + config.connectionString = "InstrumentationKey=testIkey1;ingestionendpoint=testUrl1"; this.clock.tick(1); status = core.activeStatus && core.activeStatus(); Assert.equal(status, ActiveStatus.ACTIVE, "active status should be set to active in next executing cycle"); // Assert.equal(status, ActiveStatus.PENDING, "status should be set to pending test1"); - - + + }].concat(PollingAssert.createPollingAssert(() => { let core = this._ai.core let activeStatus = core.activeStatus && core.activeStatus(); - + if (activeStatus === ActiveStatus.ACTIVE) { Assert.equal("testIkey", core.config.instrumentationKey, "ikey should be set"); - Assert.equal("testUrl/v2/track", core.config.endpointUrl ,"endpoint shoule be set"); + Assert.equal("testUrl/v2/track", core.config.endpointUrl, "endpoint shoule be set"); return true; } return false; @@ -396,17 +396,17 @@ export class ApplicationInsightsTests extends AITestClass { // force unload oriInst.unload(false); } - + if (oriInst && oriInst["dependencies"]) { oriInst["dependencies"].teardown(); } - + this._config = this._getTestConfig(this._sessionPrefix); let csPromise = createAsyncResolvedPromise("InstrumentationKey=testIkey;ingestionendpoint=testUrl"); this._config.connectionString = csPromise; let offlineChannel = new OfflineChannel(); this._config.channels = [[offlineChannel]]; - this._config.initTimeOut= 80000; + this._config.initTimeOut = 80000; let init = new ApplicationInsights({ @@ -419,21 +419,21 @@ export class ApplicationInsightsTests extends AITestClass { let status = core.activeStatus && core.activeStatus(); Assert.equal(status, ActiveStatus.PENDING, "status should be set to pending"); - + config.connectionString = "InstrumentationKey=testIkey1;ingestionendpoint=testUrl1" this.clock.tick(1); status = core.activeStatus && core.activeStatus(); Assert.equal(status, ActiveStatus.ACTIVE, "active status should be set to active in next executing cycle"); // Assert.equal(status, ActiveStatus.PENDING, "status should be set to pending test1"); - - + + }].concat(PollingAssert.createPollingAssert(() => { let core = this._ai.core let activeStatus = core.activeStatus && core.activeStatus(); - + if (activeStatus === ActiveStatus.ACTIVE) { Assert.equal("testIkey", core.config.instrumentationKey, "ikey should be set"); - Assert.equal("testUrl/v2/track", core.config.endpointUrl ,"endpoint shoule be set"); + Assert.equal("testUrl/v2/track", core.config.endpointUrl, "endpoint shoule be set"); let sendChannel = this._ai.getPlugin(BreezeChannelIdentifier); let offlineChannelPlugin = this._ai.getPlugin("OfflineChannel").plugin; Assert.equal(sendChannel.plugin.isInitialized(), true, "sender is initialized"); @@ -447,7 +447,7 @@ export class ApplicationInsightsTests extends AITestClass { }); - + this.testCaseAsync({ name: "Init: init with cs string, change with cs promise", stepDelay: 100, @@ -472,15 +472,15 @@ export class ApplicationInsightsTests extends AITestClass { status = core.activeStatus && core.activeStatus(); Assert.equal(status, ActiveStatus.ACTIVE, "active status should be set to active in next executing cycle"); //Assert.equal(status, ActiveStatus.PENDING, "status should be set to pending"); - - + + }].concat(PollingAssert.createPollingAssert(() => { let core = this._ai.core let activeStatus = core.activeStatus && core.activeStatus(); - + if (activeStatus === ActiveStatus.ACTIVE) { Assert.equal("testIkey", core.config.instrumentationKey, "ikey should be set"); - Assert.equal("testUrl/v2/track", core.config.endpointUrl ,"endpoint shoule be set"); + Assert.equal("testUrl/v2/track", core.config.endpointUrl, "endpoint shoule be set"); return true; } return false; @@ -499,11 +499,11 @@ export class ApplicationInsightsTests extends AITestClass { // force unload oriInst.unload(false); } - + if (oriInst && oriInst["dependencies"]) { oriInst["dependencies"].teardown(); } - + this._config = this._getTestConfig(this._sessionPrefix); let ikeyPromise = createAsyncResolvedPromise("testIkey"); let endpointPromise = createAsyncResolvedPromise("testUrl"); @@ -512,7 +512,7 @@ export class ApplicationInsightsTests extends AITestClass { this._config.connectionString = null; this._config.instrumentationKey = ikeyPromise; this._config.endpointUrl = endpointPromise; - this._config.initTimeOut= 80000; + this._config.initTimeOut = 80000; @@ -525,16 +525,16 @@ export class ApplicationInsightsTests extends AITestClass { let core = this._ai.core; let status = core.activeStatus && core.activeStatus(); Assert.equal(status, ActiveStatus.PENDING, "status should be set to pending"); - Assert.equal(config.connectionString,null, "connection string shoule be null"); - - + Assert.equal(config.connectionString, null, "connection string shoule be null"); + + }].concat(PollingAssert.createPollingAssert(() => { let core = this._ai.core let activeStatus = core.activeStatus && core.activeStatus(); - + if (activeStatus === ActiveStatus.ACTIVE) { Assert.equal("testIkey", core.config.instrumentationKey, "ikey should be set"); - Assert.equal("testUrl", core.config.endpointUrl ,"endpoint shoule be set"); + Assert.equal("testUrl", core.config.endpointUrl, "endpoint shoule be set"); return true; } return false; @@ -548,21 +548,21 @@ export class ApplicationInsightsTests extends AITestClass { test: () => { let fetchcalled = 0; let overrideFetchFn = (url: string, oncomplete: any, isAutoSync?: boolean) => { - fetchcalled ++; + fetchcalled++; Assert.equal(url, CONFIG_ENDPOINT_URL, "fetch should be called with prod cdn"); }; let config = { instrumentationKey: "testIKey", - extensionConfig:{ + extensionConfig: { ["AppInsightsCfgSyncPlugin"]: { overrideFetchFn: overrideFetchFn } } } as IConfiguration & IConfig; - let ai = new ApplicationInsights({config: config}); + let ai = new ApplicationInsights({ config: config }); ai.loadAppInsights(); - + ai.config.extensionConfig = ai.config.extensionConfig || {}; let extConfig = ai.config.extensionConfig["AppInsightsCfgSyncPlugin"]; Assert.equal(extConfig.cfgUrl, CONFIG_ENDPOINT_URL, "default cdn endpoint should be set"); @@ -570,7 +570,7 @@ export class ApplicationInsightsTests extends AITestClass { let featureOptIn = config.featureOptIn || {}; Assert.equal(featureOptIn["iKeyUsage"].mode, FeatureOptInMode.enable, "ikey message should be turned on"); - + Assert.equal(fetchcalled, 1, "fetch should be called once"); config.extensionConfig = config.extensionConfig || {}; let expectedTimeout = 2000000000; @@ -596,15 +596,15 @@ export class ApplicationInsightsTests extends AITestClass { let config = { instrumentationKey: "testIKey", endpointUrl: "testUrl", - extensionConfig:{ + extensionConfig: { ["AppInsightsCfgSyncPlugin"]: { cfgUrl: "" } }, - extensions:[offlineChannel] + extensions: [offlineChannel] } as IConfiguration & IConfig; - let ai = new ApplicationInsights({config: config}); + let ai = new ApplicationInsights({ config: config }); ai.loadAppInsights(); this.clock.tick(1); @@ -620,7 +620,7 @@ export class ApplicationInsightsTests extends AITestClass { ai["dependencies"].teardown(); } //offlineChannel.teardown(); - + } }); @@ -633,15 +633,15 @@ export class ApplicationInsightsTests extends AITestClass { let config = { instrumentationKey: "testIKey", endpointUrl: "testUrl", - extensionConfig:{ + extensionConfig: { ["AppInsightsCfgSyncPlugin"]: { cfgUrl: "" } }, - channels:[[offlineChannel]] + channels: [[offlineChannel]] } as IConfiguration & IConfig; - let ai = new ApplicationInsights({config: config}); + let ai = new ApplicationInsights({ config: config }); ai.loadAppInsights(); this.clock.tick(1); @@ -651,13 +651,13 @@ export class ApplicationInsightsTests extends AITestClass { Assert.equal(offlineChannelPlugin.isInitialized(), true, "offline channel is initialized"); let urlConfig = offlineChannelPlugin["_getDbgPlgTargets"]()[0]; Assert.ok(urlConfig, "offline url config is initialized"); - + ai.unload(false); if (ai && ai["dependencies"]) { ai["dependencies"].teardown(); } - + } }); @@ -669,15 +669,15 @@ export class ApplicationInsightsTests extends AITestClass { let offlineChannel = new OfflineChannel(); let config = { connectionString: "InstrumentationKey=testIKey", - extensionConfig:{ + extensionConfig: { ["AppInsightsCfgSyncPlugin"]: { cfgUrl: "" } }, - channels:[[offlineChannel]] + channels: [[offlineChannel]] } as IConfiguration & IConfig; - let ai = new ApplicationInsights({config: config}); + let ai = new ApplicationInsights({ config: config }); ai.loadAppInsights(); this.clock.tick(1); @@ -696,7 +696,7 @@ export class ApplicationInsightsTests extends AITestClass { } } }); - + } public addCDNOverrideTests(): void { @@ -766,15 +766,15 @@ export class ApplicationInsightsTests extends AITestClass { headers.forEach((val, key) => { if (key === "content-type") { Assert.deepEqual(val, "text/javascript; charset=utf-8", "should have correct content-type response header"); - headerCnt ++; + headerCnt++; } if (key === "x-ms-meta-aijssdksrc") { Assert.ok(val, "should have sdk src response header"); - headerCnt ++; + headerCnt++; } if (key === "x-ms-meta-aijssdkver") { Assert.ok(val, "should have version number for response header"); - headerCnt ++; + headerCnt++; } }); Assert.equal(headerCnt, 3, "all expected headers should be present"); @@ -819,15 +819,15 @@ export class ApplicationInsightsTests extends AITestClass { headers.forEach((val, key) => { if (key === "content-type") { Assert.deepEqual(val, "text/javascript; charset=utf-8", "should have correct content-type response header"); - headerCnt ++; + headerCnt++; } if (key === "x-ms-meta-aijssdksrc") { Assert.ok(val, "should have sdk src response header"); - headerCnt ++; + headerCnt++; } if (key === "x-ms-meta-aijssdkver") { Assert.ok(val, "should have version number for response header"); - headerCnt ++; + headerCnt++; } }); Assert.equal(headerCnt, 3, "all expected headers should be present"); @@ -855,7 +855,60 @@ export class ApplicationInsightsTests extends AITestClass { if (res.ok) { let val = await res.text(); - Assert.ok(val, "Response text should be returned" ); + Assert.ok(val, "Response text should be returned"); + } else { + Assert.fail("Fetch failed with status: " + dumpObj(res)); + } + } catch (e) { + Assert.fail("Fetch Error: " + dumpObj(e)); + } + } + }); + + this.testCase({ + name: "E2E.GenericTests: Fetch Static Web js1 - CDN V3", + useFakeServer: false, + useFakeFetch: false, + fakeFetchAutoRespond: false, + test: async () => { + // Use beta endpoint to pre-test any changes before public V3 cdn + let random = utcNow(); + // Under Cors Mode, Options request will be auto-triggered + try { + let res = await fetch(`https://js1.tst.applicationinsights.io/scripts/b/ai.3.gbl.min.js?${random}`, { + method: "GET" + }); + + if (res.ok) { + let val = await res.text(); + Assert.ok(val, "Response text should be returned"); + } else { + Assert.fail("Fetch failed with status: " + dumpObj(res)); + } + } catch (e) { + this._ctx.err = e; + Assert.fail("Fetch Error: " + dumpObj(e)); + } + } + }); + + this.testCase({ + name: "E2E.GenericTests: Fetch Static Web js2 - CDN V3", + useFakeServer: false, + useFakeFetch: false, + fakeFetchAutoRespond: false, + test: async () => { + // Use beta endpoint to pre-test any changes before public V3 cdn + let random = utcNow(); + // Under Cors Mode, Options request will be auto-triggered + try { + let res = await fetch(`https://js2.tst.applicationinsights.io/scripts/b/ai.3.gbl.min.js?${random}`, { + method: "GET" + }); + + if (res.ok) { + let val = await res.text(); + Assert.ok(val, "Response text should be returned"); } else { Assert.fail("Fetch failed with status: " + dumpObj(res)); } @@ -906,8 +959,8 @@ export class ApplicationInsightsTests extends AITestClass { if (payloadStr.length > 0) { const payload = JSON.parse(payloadStr[0]); const data = payload.data; - Assert.ok( payload && payload.iKey); - Assert.equal( ApplicationInsightsTests._instrumentationKey,payload.iKey,"payload ikey is not set correctly" ); + Assert.ok(payload && payload.iKey); + Assert.equal(ApplicationInsightsTests._instrumentationKey, payload.iKey, "payload ikey is not set correctly"); Assert.ok(data && data.baseData && data.baseData.properties["prop1"]); Assert.ok(data && data.baseData && data.baseData.measurements["measurement1"]); } @@ -923,8 +976,8 @@ export class ApplicationInsightsTests extends AITestClass { if (payloadStr.length > 0) { const payload = JSON.parse(payloadStr[0]); const data = payload.data; - Assert.ok( payload && payload.iKey); - Assert.equal( ApplicationInsightsTests._instrumentationKey,payload.iKey,"payload ikey is not set correctly" ); + Assert.ok(payload && payload.iKey); + Assert.equal(ApplicationInsightsTests._instrumentationKey, payload.iKey, "payload ikey is not set correctly"); Assert.ok(data && data.baseData && data.baseData.properties["prop1"]); Assert.ok(data && data.baseData && data.baseData.measurements["measurement1"]); } @@ -981,7 +1034,7 @@ export class ApplicationInsightsTests extends AITestClass { error: e, evt: null } as IAutoExceptionTelemetry; - + exception = e; this._ai.trackException({ exception: autoTelemetry }); } @@ -1007,7 +1060,7 @@ export class ApplicationInsightsTests extends AITestClass { error: e, evt: null } as IAutoExceptionTelemetry; - + exception = e; this._ai.trackException({ exception: autoTelemetry }, { custom: "custom value" }); } @@ -1033,7 +1086,7 @@ export class ApplicationInsightsTests extends AITestClass { error: e.toString(), evt: null } as IAutoExceptionTelemetry; - + exception = e; this._ai.trackException({ exception: autoTelemetry }); } @@ -1059,7 +1112,7 @@ export class ApplicationInsightsTests extends AITestClass { error: undefined, evt: null } as IAutoExceptionTelemetry; - + try { exception = e; this._ai.trackException({ exception: autoTelemetry }); @@ -1091,7 +1144,7 @@ export class ApplicationInsightsTests extends AITestClass { error: undefined, evt: null } as IAutoExceptionTelemetry; - + try { exception = e; this._ai.trackException({ exception: autoTelemetry }, { custom: "custom value" }); @@ -1155,7 +1208,7 @@ export class ApplicationInsightsTests extends AITestClass { name: 'E2E.GenericTests: trackException will keep id from the original exception', stepDelay: 1, steps: [() => { - this._ai.trackException({id:"testId", error: new Error("test local exception"), severityLevel: 3}); + this._ai.trackException({ id: "testId", error: new Error("test local exception"), severityLevel: 3 }); }].concat(this.asserts(1)).concat(() => { const payloadStr: string[] = this.getPayloadMessages(this.successSpy); if (payloadStr.length > 0) { @@ -1213,7 +1266,7 @@ export class ApplicationInsightsTests extends AITestClass { steps: [() => { let errObj = { name: "E2E.GenericTests", - reason:{ + reason: { message: "Test_Error_Throwing_Inside_UseCallback", stack: "Error: Test_Error_Throwing_Inside_UseCallback\n" + "at http://localhost:3000/static/js/main.206f4846.js:2:296748\n" + // Anonymous function with no function name attribution (firefox/ios) @@ -1233,7 +1286,7 @@ export class ApplicationInsightsTests extends AITestClass { " Line 11 of inline#1 script in http://localhost:3000/static/js/main.206f4846.js:2:296748\n" + // With Line 11 of inline#1 script attribution " Line 68 of inline#2 script in file://localhost/teststack.html\n" + // With Line 68 of inline#2 script attribution "at Function.Module._load (module.js:407:3)\n" + - " at Function.Module.runMain (module.js:575:10)\n"+ + " at Function.Module.runMain (module.js:575:10)\n" + " at startup (node.js:159:18)\n" + "at Global code (http://example.com/stacktrace.js:11:1)\n" + "at Object.Module._extensions..js (module.js:550:10)\n" + @@ -1306,7 +1359,7 @@ export class ApplicationInsightsTests extends AITestClass { Assert.equal(ex.parsedStack.length, 29); for (let lp = 0; lp < ex.parsedStack.length; lp++) { _checkExpectedFrame(expectedParsedStack[lp], ex.parsedStack[lp], lp); - } + } Assert.ok(baseData.properties, "Has BaseData properties"); Assert.equal(baseData.properties.custom, "custom value"); @@ -1322,13 +1375,13 @@ export class ApplicationInsightsTests extends AITestClass { stepDelay: 1, steps: [() => { let message = "Invalid hook call. Hooks can only be called inside of the body of a function component. This could happen for one of the following reasons:\n" + - "1. You might have mismatching versions of React and the renderer (such as React DOM)\n" + - "2. You might be breaking the Rules of Hooks\n" + - "3. You might have more than one copy of React in the same app\n" + - "See https://reactjs.org/link/invalid-hook-call for tips about how to debug and fix this problem."; + "1. You might have mismatching versions of React and the renderer (such as React DOM)\n" + + "2. You might be breaking the Rules of Hooks\n" + + "3. You might have more than one copy of React in the same app\n" + + "See https://reactjs.org/link/invalid-hook-call for tips about how to debug and fix this problem."; let errObj = { typeName: "Error", - reason:{ + reason: { message: "Error: " + message, stack: "Error: " + message + "\n" + " at Object.throwInvalidHookError (https://localhost:44365/static/js/bundle.js:201419:13)\n" + @@ -1383,7 +1436,7 @@ export class ApplicationInsightsTests extends AITestClass { Assert.equal(ex.parsedStack.length, 10); for (let lp = 0; lp < ex.parsedStack.length; lp++) { _checkExpectedFrame(expectedParsedStack[lp], ex.parsedStack[lp], lp); - } + } Assert.ok(baseData.properties, "Has BaseData properties"); Assert.equal(baseData.properties.custom, "custom value"); @@ -1401,7 +1454,7 @@ export class ApplicationInsightsTests extends AITestClass { () => { console.log("* calling trackMetric " + new Date().toISOString()); for (let i = 0; i < 100; i++) { - this._ai.trackMetric({ name: "test" + i, average: Math.round(100 * Math.random()), min: 1, max: i+1, stdDev: 10.0 * Math.random() }); + this._ai.trackMetric({ name: "test" + i, average: Math.round(100 * Math.random()), min: 1, max: i + 1, stdDev: 10.0 * Math.random() }); } console.log("* done calling trackMetric " + new Date().toISOString()); } @@ -1431,21 +1484,21 @@ export class ApplicationInsightsTests extends AITestClass { this._ai.trackPageView(); // sends 2 } ] - .concat(this.asserts(2)) - .concat(() => { + .concat(this.asserts(2)) + .concat(() => { - const payloadStr: string[] = this.getPayloadMessages(this.successSpy); - if (payloadStr.length > 0) { - const payload = JSON.parse(payloadStr[0]); - const data = payload.data; - Assert.ok(data.baseData.id, "pageView id is defined"); - Assert.ok(data.baseData.id.length > 0); - Assert.ok(payload.tags["ai.operation.id"]); - Assert.equal(data.baseData.id, payload.tags["ai.operation.id"], "pageView id matches current operation id"); - } else { - Assert.ok(false, "successSpy not called"); - } - }) + const payloadStr: string[] = this.getPayloadMessages(this.successSpy); + if (payloadStr.length > 0) { + const payload = JSON.parse(payloadStr[0]); + const data = payload.data; + Assert.ok(data.baseData.id, "pageView id is defined"); + Assert.ok(data.baseData.id.length > 0); + Assert.ok(payload.tags["ai.operation.id"]); + Assert.equal(data.baseData.id, payload.tags["ai.operation.id"], "pageView id matches current operation id"); + } else { + Assert.ok(false, "successSpy not called"); + } + }) }); this.testCaseAsync({ @@ -1545,20 +1598,20 @@ export class ApplicationInsightsTests extends AITestClass { name: 'E2E.GenericTests: undefined properties are replaced by customer defined value with config convertUndefined.', stepDelay: 1, steps: [() => { - this._ai.trackPageView({ name: 'pageview', properties: { 'prop1': 'val1' }}); + this._ai.trackPageView({ name: 'pageview', properties: { 'prop1': 'val1' } }); this._ai.trackEvent({ name: 'event', properties: { 'prop2': undefined } }); }].concat(this.asserts(3)).concat(() => { const payloadStr: string[] = this.getPayloadMessages(this.successSpy); for (let i = 0; i < payloadStr.length; i++) { - const payload = JSON.parse(payloadStr[i]);const baseType = payload.data.baseType; + const payload = JSON.parse(payloadStr[i]); const baseType = payload.data.baseType; // Make the appropriate assersion depending on the baseType switch (baseType) { - case Event.dataType: + case EventDataType: const eventData = payload.data; Assert.ok(eventData && eventData.baseData && eventData.baseData.properties['prop2']); Assert.equal(eventData.baseData.properties['prop2'], 'test-value'); break; - case PageView.dataType: + case PageViewDataType: const pageViewData = payload.data; Assert.ok(pageViewData && pageViewData.baseData && pageViewData.baseData.properties['prop1']); Assert.equal(pageViewData.baseData.properties['prop1'], 'val1'); @@ -1597,7 +1650,7 @@ export class ApplicationInsightsTests extends AITestClass { steps: [ () => { const xhr = new XMLHttpRequest(); - xhr.open('GET', 'https://httpbin.org/status/200'); + xhr.open('GET', 'https://localhost:9001/AISKU'); xhr.send(); Assert.ok(true); } @@ -1612,22 +1665,22 @@ export class ApplicationInsightsTests extends AITestClass { fakeFetchAutoRespond: true, steps: [ () => { - fetch('https://httpbin.org/status/200', { method: 'GET', headers: { 'header': 'value'} }); + fetch('http://localhost:9001/README.md', { method: 'GET', headers: { 'header': 'value' } }); Assert.ok(true, "fetch monitoring is instrumented"); }, () => { - fetch('https://httpbin.org/status/200', { method: 'GET' }); + fetch('http://localhost:9001/README.md', { method: 'GET' }); Assert.ok(true, "fetch monitoring is instrumented"); }, () => { - fetch('https://httpbin.org/status/200'); + fetch('http://localhost:9001/README.md'); Assert.ok(true, "fetch monitoring is instrumented"); } ].concat(this.asserts(3, false, false)) .concat(() => { let args = []; this.trackSpy.args.forEach(call => { - let message = call[0].baseData.message||""; + let message = call[0].baseData.message || ""; // Ignore the internal SendBrowserInfoOnUserInit message (Only occurs when running tests in a browser) if (message.indexOf("AI (Internal): 72 ") == -1) { args.push(call[0]); @@ -1651,6 +1704,7 @@ export class ApplicationInsightsTests extends AITestClass { Assert.ok(baseData.properties.requestHeaders[RequestHeaders.requestIdHeader], "Request-Id header"); Assert.ok(baseData.properties.requestHeaders[RequestHeaders.requestContextHeader], "Request-Context header"); Assert.ok(baseData.properties.requestHeaders[RequestHeaders.traceParentHeader], "traceparent"); + Assert.ok(!baseData.properties.requestHeaders[RequestHeaders.traceStateHeader], "traceState should not be present in outbound event"); const id: string = baseData.id; const regex = id.match(/\|.{32}\..{16}\./g); Assert.ok(id.length > 0); @@ -1680,23 +1734,23 @@ export class ApplicationInsightsTests extends AITestClass { this._ai.trackEvent({ name: "Custom event via addTelemetryInitializer" }); } ] - .concat(this.asserts(1, false, false)) - .concat(PollingAssert.createPollingAssert(() => { - const payloadStr: string[] = this.getPayloadMessages(this.successSpy); - if (payloadStr.length) { - const payload = JSON.parse(payloadStr[0]); + .concat(this.asserts(1, false, false)) + .concat(PollingAssert.createPollingAssert(() => { + const payloadStr: string[] = this.getPayloadMessages(this.successSpy); + if (payloadStr.length) { + const payload = JSON.parse(payloadStr[0]); Assert.equal(1, payloadStr.length, 'Only 1 track item is sent - ' + payload.name); Assert.ok(payload); - if (payload && payload.tags) { - const tagResult: string = payload.tags && payload.tags[this.tagKeys.cloudName]; - const tagExpect: string = 'my.custom.cloud.name'; - Assert.equal(tagResult, tagExpect, 'telemetryinitializer tag override successful'); - return true; + if (payload && payload.tags) { + const tagResult: string = payload.tags && payload.tags[this.tagKeys.cloudName]; + const tagExpect: string = 'my.custom.cloud.name'; + Assert.equal(tagResult, tagExpect, 'telemetryinitializer tag override successful'); + return true; + } + return false; } - return false; - } - }, 'Set custom tags') as any) + }, 'Set custom tags') as any) }); this.testCaseAsync({ @@ -1705,28 +1759,28 @@ export class ApplicationInsightsTests extends AITestClass { steps: [ () => { this._ai.addTelemetryInitializer((item: ITelemetryItem) => { - item.tags.push({[this.tagKeys.cloudName]: "my.shim.cloud.name"}); + item.tags.push({ [this.tagKeys.cloudName]: "my.shim.cloud.name" }); }); this._ai.trackEvent({ name: "Custom event" }); } ] - .concat(this.asserts(1)) - .concat(PollingAssert.createPollingAssert(() => { - const payloadStr: string[] = this.getPayloadMessages(this.successSpy); - if (payloadStr.length > 0) { - Assert.equal(1, payloadStr.length, 'Only 1 track item is sent'); - const payload = JSON.parse(payloadStr[0]); - Assert.ok(payload); + .concat(this.asserts(1)) + .concat(PollingAssert.createPollingAssert(() => { + const payloadStr: string[] = this.getPayloadMessages(this.successSpy); + if (payloadStr.length > 0) { + Assert.equal(1, payloadStr.length, 'Only 1 track item is sent'); + const payload = JSON.parse(payloadStr[0]); + Assert.ok(payload); - if (payload && payload.tags) { - const tagResult: string = payload.tags && payload.tags[this.tagKeys.cloudName]; - const tagExpect: string = 'my.shim.cloud.name'; - Assert.equal(tagResult, tagExpect, 'telemetryinitializer tag override successful'); - return true; + if (payload && payload.tags) { + const tagResult: string = payload.tags && payload.tags[this.tagKeys.cloudName]; + const tagExpect: string = 'my.shim.cloud.name'; + Assert.equal(tagResult, tagExpect, 'telemetryinitializer tag override successful'); + return true; + } + return false; } - return false; - } - }, 'Set custom tags') as any) + }, 'Set custom tags') as any) }); this.testCaseAsync({ @@ -1737,41 +1791,41 @@ export class ApplicationInsightsTests extends AITestClass { this._ai.addTelemetryInitializer((item: ITelemetryItem) => { item.tags[this.tagKeys.cloudName] = "my.custom.cloud.name"; item.tags[this.tagKeys.locationCity] = "my.custom.location.city"; - item.tags.push({[this.tagKeys.locationCountry]: "my.custom.location.country"}); - item.tags.push({[this.tagKeys.operationId]: "my.custom.operation.id"}); + item.tags.push({ [this.tagKeys.locationCountry]: "my.custom.location.country" }); + item.tags.push({ [this.tagKeys.operationId]: "my.custom.operation.id" }); }); this._ai.trackEvent({ name: "Custom event via shimmed addTelemetryInitializer" }); } ] - .concat(this.asserts(1)) - .concat(PollingAssert.createPollingAssert(() => { - const payloadStr: string[] = this.getPayloadMessages(this.successSpy); - if (payloadStr.length > 0) { - const payload = JSON.parse(payloadStr[0]); - Assert.equal(1, payloadStr.length, 'Only 1 track item is sent - ' + payload.name); - if (payloadStr.length > 1) { - this.dumpPayloadMessages(this.successSpy); - } - Assert.ok(payload); - - if (payload && payload.tags) { - const tagResult1: string = payload.tags && payload.tags[this.tagKeys.cloudName]; - const tagExpect1: string = 'my.custom.cloud.name'; - Assert.equal(tagResult1, tagExpect1, 'telemetryinitializer tag override successful'); - const tagResult2: string = payload.tags && payload.tags[this.tagKeys.locationCity]; - const tagExpect2: string = 'my.custom.location.city'; - Assert.equal(tagResult2, tagExpect2, 'telemetryinitializer tag override successful'); - const tagResult3: string = payload.tags && payload.tags[this.tagKeys.locationCountry]; - const tagExpect3: string = 'my.custom.location.country'; - Assert.equal(tagResult3, tagExpect3, 'telemetryinitializer tag override successful'); - const tagResult4: string = payload.tags && payload.tags[this.tagKeys.operationId]; - const tagExpect4: string = 'my.custom.operation.id'; - Assert.equal(tagResult4, tagExpect4, 'telemetryinitializer tag override successful'); - return true; + .concat(this.asserts(1)) + .concat(PollingAssert.createPollingAssert(() => { + const payloadStr: string[] = this.getPayloadMessages(this.successSpy); + if (payloadStr.length > 0) { + const payload = JSON.parse(payloadStr[0]); + Assert.equal(1, payloadStr.length, 'Only 1 track item is sent - ' + payload.name); + if (payloadStr.length > 1) { + this.dumpPayloadMessages(this.successSpy); + } + Assert.ok(payload); + + if (payload && payload.tags) { + const tagResult1: string = payload.tags && payload.tags[this.tagKeys.cloudName]; + const tagExpect1: string = 'my.custom.cloud.name'; + Assert.equal(tagResult1, tagExpect1, 'telemetryinitializer tag override successful'); + const tagResult2: string = payload.tags && payload.tags[this.tagKeys.locationCity]; + const tagExpect2: string = 'my.custom.location.city'; + Assert.equal(tagResult2, tagExpect2, 'telemetryinitializer tag override successful'); + const tagResult3: string = payload.tags && payload.tags[this.tagKeys.locationCountry]; + const tagExpect3: string = 'my.custom.location.country'; + Assert.equal(tagResult3, tagExpect3, 'telemetryinitializer tag override successful'); + const tagResult4: string = payload.tags && payload.tags[this.tagKeys.operationId]; + const tagExpect4: string = 'my.custom.operation.id'; + Assert.equal(tagResult4, tagExpect4, 'telemetryinitializer tag override successful'); + return true; + } + return false; } - return false; - } - }, 'Set custom tags') as any) + }, 'Set custom tags') as any) }); this.testCaseAsync({ @@ -1779,7 +1833,7 @@ export class ApplicationInsightsTests extends AITestClass { stepDelay: 1, steps: [ () => { - const context = (this._ai.context) as TelemetryContext; + const context = (this._ai.context) as IPropTelemetryContext; context.user.setAuthenticatedUserContext('10001'); this._ai.trackTrace({ message: 'authUserContext test' }); } @@ -1789,7 +1843,7 @@ export class ApplicationInsightsTests extends AITestClass { let payloadStr = this.getPayloadMessages(this.successSpy); if (payloadStr.length > 0) { let payloadEvents = payloadStr.length; - let thePayload:string = payloadStr[0]; + let thePayload: string = payloadStr[0]; if (payloadEvents !== 1) { // Only 1 track should be sent @@ -1810,7 +1864,7 @@ export class ApplicationInsightsTests extends AITestClass { stepDelay: 1, steps: [ () => { - const context = (this._ai.context) as TelemetryContext; + const context = (this._ai.context) as IPropTelemetryContext; context.user.setAuthenticatedUserContext('10001', 'account123'); this._ai.trackTrace({ message: 'authUserContext test' }); } @@ -1840,7 +1894,7 @@ export class ApplicationInsightsTests extends AITestClass { stepDelay: 1, steps: [ () => { - const context = (this._ai.context) as TelemetryContext; + const context = (this._ai.context) as IPropTelemetryContext; context.user.setAuthenticatedUserContext("\u0428", "\u0429"); this._ai.trackTrace({ message: 'authUserContext test' }); } @@ -1870,7 +1924,7 @@ export class ApplicationInsightsTests extends AITestClass { stepDelay: 1, steps: [ () => { - const context = (this._ai.context) as TelemetryContext; + const context = (this._ai.context) as IPropTelemetryContext; context.user.setAuthenticatedUserContext('10002', 'account567'); context.user.clearAuthenticatedUserContext(); this._ai.trackTrace({ message: 'authUserContext test' }); @@ -1901,7 +1955,7 @@ export class ApplicationInsightsTests extends AITestClass { name: 'AuthenticatedUserContext: setAuthenticatedUserContext does not set the cookie by default', test: () => { // Setup - const context = (this._ai.context) as TelemetryContext; + const context = (this._ai.context) as IPropTelemetryContext; const authSpy: SinonSpy = this.sandbox.spy(context.user, 'setAuthenticatedUserContext'); let cookieMgr = this._ai.getCookieMgr(); const cookieSpy: SinonSpy = this.sandbox.spy(cookieMgr, 'set'); @@ -1930,7 +1984,7 @@ export class ApplicationInsightsTests extends AITestClass { this.testCase({ name: 'iKey replacement: envelope will use the non-empty iKey defined in track method', test: () => { - this._ai.trackEvent({ name: 'event1', properties: { "prop1": "value1" }, measurements: { "measurement1": 200 }, iKey:"1a6933ad-aaaa-aaaa-aaaa-000000000000" }); + this._ai.trackEvent({ name: 'event1', properties: { "prop1": "value1" }, measurements: { "measurement1": 200 }, iKey: "1a6933ad-aaaa-aaaa-aaaa-000000000000" }); Assert.ok(this.envelopeConstructorSpy.called); const envelope = this.envelopeConstructorSpy.returnValues[0]; Assert.equal(envelope.iKey, "1a6933ad-aaaa-aaaa-aaaa-000000000000", "trackEvent iKey is replaced"); @@ -1940,7 +1994,7 @@ export class ApplicationInsightsTests extends AITestClass { this.testCase({ name: 'iKey replacement: envelope will use the config iKey if defined ikey in track method is empty', test: () => { - this._ai.trackEvent({ name: 'event1', properties: { "prop1": "value1" }, measurements: { "measurement1": 200 }, iKey:"" }); + this._ai.trackEvent({ name: 'event1', properties: { "prop1": "value1" }, measurements: { "measurement1": 200 }, iKey: "" }); Assert.ok(this.envelopeConstructorSpy.called); const envelope = this.envelopeConstructorSpy.returnValues[0]; Assert.equal(envelope.iKey, ApplicationInsightsTests._instrumentationKey, "trackEvent iKey should not be replaced"); @@ -1959,7 +2013,7 @@ export class ApplicationInsightsTests extends AITestClass { } } } - private asserts: any = (expectedCount: number, includeInit:boolean = false, doBoilerPlate:boolean = true) => [ + private asserts: any = (expectedCount: number, includeInit: boolean = false, doBoilerPlate: boolean = true) => [ () => { const message = "polling: " + new Date().toISOString(); Assert.ok(true, message); @@ -1989,21 +2043,22 @@ export class ApplicationInsightsTests extends AITestClass { if (currentCount === expectedCount && !!this._ai.context.appId()) { const payload = JSON.parse(payloadStr[0]); const baseType = payload.data.baseType; + // call the appropriate Validate depending on the baseType switch (baseType) { - case Event.dataType: + case EventDataType: return EventValidator.EventValidator.Validate(payload, baseType); - case Trace.dataType: + case TraceDataType: return TraceValidator.TraceValidator.Validate(payload, baseType); - case Exception.dataType: + case ExceptionDataType: return ExceptionValidator.ExceptionValidator.Validate(payload, baseType); - case Metric.dataType: + case MetricDataType: return MetricValidator.MetricValidator.Validate(payload, baseType); - case PageView.dataType: + case PageViewDataType: return PageViewValidator.PageViewValidator.Validate(payload, baseType); - case PageViewPerformance.dataType: + case PageViewPerformanceDataType: return PageViewPerformanceValidator.PageViewPerformanceValidator.Validate(payload, baseType); - case RemoteDependencyData.dataType: + case RemoteDependencyDataType: return RemoteDepdencyValidator.RemoteDepdencyValidator.Validate(payload, baseType); default: @@ -2020,9 +2075,9 @@ export class ApplicationInsightsTests extends AITestClass { class CustomTestError extends Error { constructor(message = "") { - super(message); - this.name = "CustomTestError"; - this.message = message + " -- test error."; + super(message); + this.name = "CustomTestError"; + this.message = message + " -- test error."; } } diff --git a/AISKU/Tests/Unit/src/sanitizer.e2e.tests.ts b/AISKU/Tests/Unit/src/sanitizer.e2e.tests.ts index 3a6048f60..6e1d55b29 100644 --- a/AISKU/Tests/Unit/src/sanitizer.e2e.tests.ts +++ b/AISKU/Tests/Unit/src/sanitizer.e2e.tests.ts @@ -3,7 +3,7 @@ import { Sender } from '@microsoft/applicationinsights-channel-js'; import { AITestClass, Assert, PollingAssert } from '@microsoft/ai-test-framework'; import { SinonSpy } from 'sinon'; import { newId } from '@microsoft/applicationinsights-core-js'; -import { BreezeChannelIdentifier } from '@microsoft/applicationinsights-common'; +import { BreezeChannelIdentifier } from '@microsoft/applicationinsights-core-js'; export class SanitizerE2ETests extends AITestClass { private readonly _instrumentationKey = 'b7170927-2d1c-44f1-acec-59f4e1751c11'; diff --git a/AISKU/Tests/Unit/src/sender.e2e.tests.ts b/AISKU/Tests/Unit/src/sender.e2e.tests.ts index b74fa774f..9d22b6d2b 100644 --- a/AISKU/Tests/Unit/src/sender.e2e.tests.ts +++ b/AISKU/Tests/Unit/src/sender.e2e.tests.ts @@ -1,6 +1,6 @@ import { ApplicationInsights, IApplicationInsights } from '../../../src/applicationinsights-web' import { Sender } from '@microsoft/applicationinsights-channel-js'; -import { BreezeChannelIdentifier, utlGetSessionStorage, utlRemoveSessionStorage } from '@microsoft/applicationinsights-common'; +import { BreezeChannelIdentifier, utlGetSessionStorage, utlRemoveSessionStorage } from '@microsoft/applicationinsights-core-js'; import { ActiveStatus, dumpObj, getJSON, isArray } from '@microsoft/applicationinsights-core-js'; import { SinonSpy } from 'sinon'; import { Assert, AITestClass, PollingAssert} from "@microsoft/ai-test-framework" diff --git a/AISKU/Tests/Unit/src/validate.e2e.tests.ts b/AISKU/Tests/Unit/src/validate.e2e.tests.ts index 95b8045a7..9adc6ddaa 100644 --- a/AISKU/Tests/Unit/src/validate.e2e.tests.ts +++ b/AISKU/Tests/Unit/src/validate.e2e.tests.ts @@ -3,7 +3,7 @@ import { Sender } from '@microsoft/applicationinsights-channel-js'; import { SinonSpy } from 'sinon'; import { AITestClass, Assert, PollingAssert } from '@microsoft/ai-test-framework'; import { dumpObj } from '@microsoft/applicationinsights-core-js'; -import { BreezeChannelIdentifier } from '@microsoft/applicationinsights-common'; +import { BreezeChannelIdentifier } from '@microsoft/applicationinsights-core-js'; export class ValidateE2ETests extends AITestClass { private readonly _instrumentationKey = 'b7170927-2d1c-44f1-acec-59f4e1751c11'; diff --git a/AISKU/Tests/es6-module-type-check/package.json b/AISKU/Tests/es6-module-type-check/package.json index 4e29d203e..b129e0fa7 100644 --- a/AISKU/Tests/es6-module-type-check/package.json +++ b/AISKU/Tests/es6-module-type-check/package.json @@ -32,7 +32,7 @@ "tslib": ">= 1.0.0" }, "dependencies": { - "@microsoft/applicationinsights-common": "3.3.11", + "@microsoft/applicationinsights-core-js": "3.3.11", "@microsoft/applicationinsights-web": "3.3.11" } } diff --git a/AISKU/Tests/es6-module-type-check/src/main.ts b/AISKU/Tests/es6-module-type-check/src/main.ts index 9d6515a9c..227ed05a5 100644 --- a/AISKU/Tests/es6-module-type-check/src/main.ts +++ b/AISKU/Tests/es6-module-type-check/src/main.ts @@ -1,2 +1,2 @@ -import { ContextTagKeys } from "@microsoft/applicationinsights-common"; +import { ContextTagKeys } from "@microsoft/applicationinsights-core-js"; import { ITelemetryItem } from "@microsoft/applicationinsights-web"; \ No newline at end of file diff --git a/AISKU/examples/span-usage-example.ts b/AISKU/examples/span-usage-example.ts new file mode 100644 index 000000000..2df4a5d10 --- /dev/null +++ b/AISKU/examples/span-usage-example.ts @@ -0,0 +1,68 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +/** + * Example showing how to use the ApplicationInsights span functionality + * with the provider pattern. + */ + +import { ApplicationInsights, eOTelSpanKind } from "@microsoft/applicationinsights-web"; + +// Initialize ApplicationInsights +const appInsights = new ApplicationInsights({ + config: { + connectionString: "YOUR_CONNECTION_STRING_HERE" + } +}); +appInsights.loadAppInsights(); + +// Example usage +function exampleSpanUsage() { + // Start a span + const span = appInsights.startSpan("example-operation", { + kind: eOTelSpanKind.CLIENT, + attributes: { + "operation.name": "example", + "user.id": "12345" + } + }); + + if (span) { + try { + // Do some work... + span.setAttribute("result", "success"); + span.setAttribute("duration", 100); + + // Create a child span + const childSpan = appInsights.startSpan("child-operation", { + kind: eOTelSpanKind.INTERNAL, + startTime: Date.now() + }, span.spanContext()); + + if (childSpan) { + // Do child work... + childSpan.setAttribute("child.data", "value"); + childSpan.end(); + } + + } catch (error: any) { + span.setAttribute("error", true); + span.setAttribute("error.message", error.message); + } finally { + span.end(); + } + } +} + +// Example of checking if trace provider is available +function checkTraceProviderAvailability() { + const provider = appInsights.getTraceProvider(); + if (provider && provider.isAvailable()) { + console.log(`Trace provider available: ${provider.getProviderId()}`); + exampleSpanUsage(); + } else { + console.log("No trace provider available"); + } +} + +export { exampleSpanUsage, checkTraceProviderAvailability }; diff --git a/AISKU/package.json b/AISKU/package.json index 500bb7531..66a67338b 100644 --- a/AISKU/package.json +++ b/AISKU/package.json @@ -67,7 +67,6 @@ "@microsoft/applicationinsights-analytics-js": "3.3.11", "@microsoft/applicationinsights-channel-js": "3.3.11", "@microsoft/applicationinsights-cfgsync-js": "3.3.11", - "@microsoft/applicationinsights-common": "3.3.11", "@microsoft/applicationinsights-core-js": "3.3.11", "@microsoft/applicationinsights-dependencies-js": "3.3.11", "@microsoft/applicationinsights-properties-js": "3.3.11", diff --git a/AISKU/scripts/publishAzReleaseToCdn.ps1 b/AISKU/scripts/publishAzReleaseToCdn.ps1 index 99aacadc3..4367b754a 100644 --- a/AISKU/scripts/publishAzReleaseToCdn.ps1 +++ b/AISKU/scripts/publishAzReleaseToCdn.ps1 @@ -139,7 +139,7 @@ elseif ($version.type -eq "dev" -or $version.type -eq "beta") { # Publish to release type folder folder PublishFiles $releaseFiles "$($version.type)" $cacheControl1Year $contentType $overwrite } -elseif ($version.type -eq "nightly" -or $version.type -eq "nightly3") { +elseif ($version.type -like "nightly*") { # Publish to release nightly folder folder PublishFiles $releaseFiles "nightly" $cacheControl1Year $contentType $overwrite } diff --git a/AISKU/scripts/setAzActiveCdnVersion.ps1 b/AISKU/scripts/setAzActiveCdnVersion.ps1 index c83e1680e..97472c382 100644 --- a/AISKU/scripts/setAzActiveCdnVersion.ps1 +++ b/AISKU/scripts/setAzActiveCdnVersion.ps1 @@ -87,7 +87,7 @@ Function Validate-Params Write-LogFailure "Container [$container] is not valid for version type [$($version.type)]" } } - elseif ($version.type -eq "nightly" -or $version.type -eq "nightly3") { + elseif ($version.type -like "nightly*") { if ("nightly" -ne $container) { Write-LogFailure "Container [$container] is not valid for version type [$($version.type)]" } diff --git a/AISKU/src/AISku.ts b/AISKU/src/AISku.ts index 5ca448464..51a55e209 100644 --- a/AISKU/src/AISku.ts +++ b/AISKU/src/AISku.ts @@ -7,19 +7,17 @@ import { AnalyticsPlugin, ApplicationInsights } from "@microsoft/applicationinsi import { CfgSyncPlugin, ICfgSyncConfig, ICfgSyncMode } from "@microsoft/applicationinsights-cfgsync-js"; import { Sender } from "@microsoft/applicationinsights-channel-js"; import { - AnalyticsPluginIdentifier, ConnectionString, DEFAULT_BREEZE_PATH, IAutoExceptionTelemetry, IConfig, IDependencyTelemetry, - IEventTelemetry, IExceptionTelemetry, IMetricTelemetry, IPageViewPerformanceTelemetry, IPageViewTelemetry, IRequestHeaders, - ITelemetryContext as Common_ITelemetryContext, IThrottleInterval, IThrottleLimit, IThrottleMgrConfig, ITraceTelemetry, - PropertiesPluginIdentifier, ThrottleMgr, parseConnectionString -} from "@microsoft/applicationinsights-common"; -import { - AppInsightsCore, FeatureOptInMode, IAppInsightsCore, IChannelControls, IConfigDefaults, IConfiguration, ICookieMgr, ICustomProperties, - IDiagnosticLogger, IDistributedTraceContext, IDynamicConfigHandler, ILoadedPlugin, INotificationManager, IPlugin, - ITelemetryInitializerHandler, ITelemetryItem, ITelemetryPlugin, ITelemetryUnloadState, IUnloadHook, UnloadHandler, WatcherFunction, + AnalyticsPluginIdentifier, AppInsightsCore, ConnectionString, DEFAULT_BREEZE_PATH, FeatureOptInMode, IAppInsightsCore, + IAutoExceptionTelemetry, IChannelControls, IConfig, IConfigDefaults, IConfiguration, ICookieMgr, ICustomProperties, IDependencyTelemetry, + IDiagnosticLogger, IDistributedTraceContext, IDynamicConfigHandler, IEventTelemetry, IExceptionTelemetry, ILoadedPlugin, + IMetricTelemetry, INotificationManager, IOTelApi, IOTelSpanOptions, IPageViewPerformanceTelemetry, IPageViewTelemetry, IPlugin, + IReadableSpan, IRequestHeaders, ISpanScope, ITelemetryContext as Common_ITelemetryContext, ITelemetryInitializerHandler, ITelemetryItem, + ITelemetryPlugin, ITelemetryUnloadState, IThrottleInterval, IThrottleLimit, IThrottleMgrConfig, ITraceApi, ITraceProvider, + ITraceTelemetry, IUnloadHook, OTelTimeInput, PropertiesPluginIdentifier, ThrottleMgr, UnloadHandler, WatcherFunction, _eInternalMessageId, _throwInternal, addPageHideEventListener, addPageUnloadEventListener, cfgDfMerge, cfgDfValidate, - createDynamicConfig, createProcessTelemetryContext, createUniqueNamespace, doPerf, eLoggingSeverity, hasDocument, hasWindow, isArray, - isFeatureEnabled, isFunction, isNullOrUndefined, isReactNative, isString, mergeEvtNamespace, onConfigChange, proxyAssign, proxyFunctions, - removePageHideEventListener, removePageUnloadEventListener + createDynamicConfig, createOTelApi, createProcessTelemetryContext, createTraceProvider, createUniqueNamespace, doPerf, eLoggingSeverity, + hasDocument, hasWindow, isArray, isFeatureEnabled, isFunction, isNullOrUndefined, isReactNative, isString, mergeEvtNamespace, + onConfigChange, parseConnectionString, proxyAssign, proxyFunctions, removePageHideEventListener, removePageUnloadEventListener, useSpan } from "@microsoft/applicationinsights-core-js"; import { AjaxPlugin as DependenciesPlugin, DependencyInitializerFunction, DependencyListenerFunction, IDependencyInitializerHandler, @@ -27,24 +25,39 @@ import { } from "@microsoft/applicationinsights-dependencies-js"; import { PropertiesPlugin } from "@microsoft/applicationinsights-properties-js"; import { IPromise, createPromise, createSyncPromise, doAwaitResponse } from "@nevware21/ts-async"; -import { arrForEach, arrIndexOf, isPromiseLike, objDefine, objForEachKey, strIndexOf, throwUnsupported } from "@nevware21/ts-utils"; +import { + ICachedValue, arrForEach, arrIndexOf, dumpObj, getDeferred, isPromiseLike, objDefine, objForEachKey, strIndexOf, throwUnsupported +} from "@nevware21/ts-utils"; import { IApplicationInsights } from "./IApplicationInsights"; import { CONFIG_ENDPOINT_URL, STR_ADD_TELEMETRY_INITIALIZER, STR_CLEAR_AUTHENTICATED_USER_CONTEXT, STR_EVT_NAMESPACE, STR_GET_COOKIE_MGR, STR_GET_PLUGIN, STR_POLL_INTERNAL_LOGS, STR_SET_AUTHENTICATED_USER_CONTEXT, STR_SNIPPET, STR_START_TRACK_EVENT, STR_START_TRACK_PAGE, STR_STOP_TRACK_EVENT, STR_STOP_TRACK_PAGE, STR_TRACK_DEPENDENCY_DATA, STR_TRACK_EVENT, STR_TRACK_EXCEPTION, STR_TRACK_METRIC, - STR_TRACK_PAGE_VIEW, STR_TRACK_TRACE + STR_TRACK_PAGE_VIEW, STR_TRACK_TRACE, UNDEFINED_VALUE } from "./InternalConstants"; import { Snippet } from "./Snippet"; +import { createTelemetryItemFromSpan } from "./internal/trace/spanUtils"; export { IRequestHeaders }; let _internalSdkSrc: string; +const STR_DEPENDENCIES = "dependencies"; +const STR_PROPERTIES = "properties"; +const STR_SNIPPET_VERSION = "_snippetVersion"; +const STR_APP_INSIGHTS_NEW = "appInsightsNew"; +const STR_GET_SKU_DEFAULTS = "getSKUDefaults"; + // This is an exclude list of properties that should not be updated during initialization // They include a combination of private and internal property names const _ignoreUpdateSnippetProperties = [ - STR_SNIPPET, "dependencies", "properties", "_snippetVersion", "appInsightsNew", "getSKUDefaults" + STR_SNIPPET, STR_DEPENDENCIES, STR_PROPERTIES, STR_SNIPPET_VERSION, STR_APP_INSIGHTS_NEW, STR_GET_SKU_DEFAULTS, "trace", "otelApi" +]; + +// This is an exclude list of properties that should not be proxied to the snippet +// They include a combination of private and internal property names +const _ignoreProxyAssignProperties = [ + STR_SNIPPET, STR_DEPENDENCIES, STR_PROPERTIES, STR_SNIPPET_VERSION, STR_APP_INSIGHTS_NEW, STR_GET_SKU_DEFAULTS ]; const IKEY_USAGE = "iKeyUsage"; @@ -52,8 +65,6 @@ const CDN_USAGE = "CdnUsage"; const SDK_LOADER_VER = "SdkLoaderVer"; const ZIP_PAYLOAD = "zipPayload"; -const UNDEFINED_VALUE: undefined = undefined; - const default_limit = { samplingRate: 100, maxSendNumber: 1 @@ -122,12 +133,25 @@ function _parseCs(config: IConfiguration & IConfig, configCs: string | IPromise< }); } +function _initOTel(sku: AppInsightsSku, traceName: string, onEnd: (span: IReadableSpan) => void, onException?: (span: IReadableSpan, exception: any, time?: OTelTimeInput) => void): ICachedValue { + let otelApi: ICachedValue = getDeferred(createOTelApi, [{ + host: sku + }]); + + // Create the initial default traceProvider + sku.core.setTraceProvider(getDeferred(() => { + return createTraceProvider(sku, traceName, otelApi.v, onEnd, onException); + })); + + return otelApi; +} + /** * Application Insights API * @group Entrypoint * @group Classes */ -export class AppInsightsSku implements IApplicationInsights { +export class AppInsightsSku implements IApplicationInsights { public snippet: Snippet; /** @@ -151,6 +175,10 @@ export class AppInsightsSku implements IApplicationInsights { */ public readonly pluginVersionString: string; + public readonly trace: ITraceApi; + + public readonly otelApi: IOTelApi; + constructor(snippet: Snippet) { // NOTE!: DON'T set default values here, instead set them in the _initDefaults() function as it is also called during teardown() let dependencies: DependenciesPlugin; @@ -167,6 +195,7 @@ export class AppInsightsSku implements IApplicationInsights { let _iKeySentMessage: boolean; let _cdnSentMessage: boolean; let _sdkVerSentMessage: boolean; + let _otelApi: ICachedValue; dynamicProto(AppInsightsSku, this, (_self) => { _initDefaults(); @@ -181,7 +210,7 @@ export class AppInsightsSku implements IApplicationInsights { objDefine(_self, key, { g: () => { if (_core) { - return _core[key]; + return (_core as any)[key]; } return null; @@ -209,12 +238,25 @@ export class AppInsightsSku implements IApplicationInsights { _sender = new Sender(); _core = new AppInsightsCore(); + objDefine(_self, "core", { g: () => { return _core; } }); + objDefine(_self, "otelApi", { + g: function() { + return _otelApi ? _otelApi.v : null; + } + }); + + objDefine(_self, "trace", { + g: function() { + return _otelApi ? _otelApi.v.trace : null; + } + }); + // Will get recalled if any referenced values are changed _addUnloadHook(onConfigChange(cfgHandler, () => { let configCs = _config.connectionString; @@ -314,8 +356,6 @@ export class AppInsightsSku implements IApplicationInsights { } }); }; - - _self.loadAppInsights = (legacyMode: boolean = false, logger?: IDiagnosticLogger, notificationManager?: INotificationManager): IApplicationInsights => { if (legacyMode) { @@ -338,8 +378,8 @@ export class AppInsightsSku implements IApplicationInsights { !isFunction(value) && field && field[0] !== "_" && // Don't copy "internal" values arrIndexOf(_ignoreUpdateSnippetProperties, field) === -1) { - if (snippet[field] !== value) { - snippet[field as string] = value; + if ((snippet as any)[field] !== value) { + (snippet as any)[field as string] = value; } } }); @@ -349,9 +389,14 @@ export class AppInsightsSku implements IApplicationInsights { doPerf(_self.core, () => "AISKU.loadAppInsights", () => { // initialize core _core.initialize(_config, [ _sender, properties, dependencies, _analyticsPlugin, _cfgSyncPlugin], logger, notificationManager); + + // Initialize the initial OTel API + _otelApi = _initOTel(_self, "aisku", _onEnd, _onException); + objDefine(_self, "context", { g: () => properties.context }); + if (!_throttleMgr){ _throttleMgr = new ThrottleMgr(_core); } @@ -402,7 +447,7 @@ export class AppInsightsSku implements IApplicationInsights { // Note: This must be called before loadAppInsights is called proxyAssign(snippet, _self, (name: string) => { // Not excluding names prefixed with "_" as we need to proxy some functions like _onError - return name && arrIndexOf(_ignoreUpdateSnippetProperties, name) === -1; + return name && arrIndexOf(_ignoreProxyAssignProperties, name) === -1; }); }; @@ -528,7 +573,21 @@ export class AppInsightsSku implements IApplicationInsights { if (!unloadDone) { unloadDone = true; + // Reset OTel API to clean up all trace state before unloading core + if (_core) { + // Clear the trace provider to stop any active spans + _core.setTraceProvider(null); + + // Reset the OTel API instances - this will be recreated on next init + if (_otelApi) { + _otelApi.v.shutdown(); + } + + _otelApi = null; + } + _initDefaults(); + unloadComplete && unloadComplete(unloadState); } } @@ -571,9 +630,16 @@ export class AppInsightsSku implements IApplicationInsights { "addPlugin", STR_EVT_NAMESPACE, "addUnloadCb", - "getTraceCtx", "updateCfg", - "onCfgChange" + "onCfgChange", + // ITraceHost Proxy + "getTraceCtx", + "setTraceCtx", + "startSpan", + "getActiveSpan", + "setActiveSpan", + "setTraceProvider", + "getTraceProvider" ]); proxyFunctions(_self, () => { @@ -583,7 +649,51 @@ export class AppInsightsSku implements IApplicationInsights { STR_SET_AUTHENTICATED_USER_CONTEXT, STR_CLEAR_AUTHENTICATED_USER_CONTEXT ]); - + + // Handle span end event - create telemetry from span data + function _onEnd(span: IReadableSpan) { + if (_otelApi && span && span.isRecording() && !_otelApi.v.cfg.traceCfg.suppressTracing) { + + // Flip this span to be the "current" span during processing, so any telemetry created during the span processing + // is associated with this span + useSpan(_core, span, () => { + try { + // Create trace telemetry for the span + let telemetryItem: ITelemetryItem = createTelemetryItemFromSpan(_core, span); + if (telemetryItem) { + _self.core.track(telemetryItem); + } + } catch (error) { + // Log any errors during trace processing but don't let them break the span lifecycle + _throwInternal(_core.logger, eLoggingSeverity.WARNING, + _eInternalMessageId.TelemetryInitializerFailed, + "Error processing span - " + dumpObj(error)); + } + }); + } + } + + function _onException(span: IReadableSpan, exception: any, time?: OTelTimeInput) { + if (_otelApi) { + // Flip this span to be the "current" span during processing, so any telemetry created during the span processing + useSpan(_core, span, () => { + try { + _self.trackException({ + exception: exception, + properties: { + time: time + } + }); + } catch (error) { + // Log any errors during exception processing but don't let them break the span lifecycle + _throwInternal(_core.logger, eLoggingSeverity.WARNING, + _eInternalMessageId.TelemetryInitializerFailed, + "Error processing exception - " + dumpObj(error)); + } + }); + } + } + // Using a function to support the dynamic adding / removal of plugins, so this will always return the current value function _getCurrentDependencies() { return dependencies; @@ -790,7 +900,7 @@ export class AppInsightsSku implements IApplicationInsights { /** * Manually trigger an immediate send of all telemetry still in the buffer using beacon Sender. * Fall back to xhr sender if beacon is not supported. - * @param [async=true] + * @param async - send data asynchronously when true, default is true */ public onunloadFlush(async: boolean = true) { // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging @@ -798,8 +908,8 @@ export class AppInsightsSku implements IApplicationInsights { /** * Initialize this instance of ApplicationInsights - * @returns {IApplicationInsights} * @param legacyMode - MUST always be false, it is no longer supported from v3.x onwards + * @returns The initialized {@link IApplicationInsights} instance */ public loadAppInsights(legacyMode: boolean = false, logger?: IDiagnosticLogger, notificationManager?: INotificationManager): IApplicationInsights { // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging @@ -930,11 +1040,18 @@ export class AppInsightsSku implements IApplicationInsights { /** * Gets the current distributed trace context for this instance if available */ - public getTraceCtx(): IDistributedTraceContext | null | undefined { + public getTraceCtx(): IDistributedTraceContext | null { // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging return null; } + /** + * Sets the current distributed trace context for this instance if available + */ + public setTraceCtx(newTraceCtx: IDistributedTraceContext | null | undefined): void { + // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging + } + /** * Watches and tracks changes for accesses to the current config, and if the accessed config changes the * handler will be recalled. @@ -945,6 +1062,72 @@ export class AppInsightsSku implements IApplicationInsights { // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging return null; } + + /** + * Start a new span with the given name and optional parent context. + * + * Note: This method only creates and returns the span. It does not automatically + * set the span as the active trace context. Context management should be handled + * separately using setTraceCtx() if needed. + * + * @param name - The name of the span + * @param options - Options for creating the span (kind, attributes, startTime) + * @param parent - Optional parent context. If not provided, uses the current active trace context + * @returns A new span instance, or null if no trace provider is available + * @since 3.4.0 + */ + public startSpan(name: string, options?: IOTelSpanOptions, parent?: IDistributedTraceContext): IReadableSpan | null { + // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging + return null; + } + + /** + * Return the current active span, if no trace provider is available null will be returned + * but when a trace provider is available a span instance will always be returned, even if + * there is no active span (in which case a non-recording span will be returned). + * @param createNew - Optional flag to create a non-recording span if no active span exists, defaults to true. + * When false, returns the existing active span or null without creating a non-recording span. + * @returns The current active span or null if no trace provider is available or if createNew is false and no active span exists + * @since 3.4.0 + */ + public getActiveSpan(createNew?: boolean): IReadableSpan | null { + // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging + return null; + } + + /** + * Set the current Active Span, if no trace provider is available the span will be not be set as the active span. + * @param span - The span to set as the active span + * @returns An ISpanScope instance that provides the current scope, the span will always be the span passed in + * even when no trace provider is available + * @since 3.4.0 + */ + public setActiveSpan(span: IReadableSpan): ISpanScope { + // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging + return null; + } + + /** + * Set the trace provider for creating spans. + * This allows different SKUs to provide their own span implementations. + * + * @param provider - The trace provider to use for span creation + * @since 3.4.0 + */ + public setTraceProvider(provider: ICachedValue): void { + // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging + } + + /** + * Get the current trace provider. + * + * @returns The current trace provider, or null if none is set + * @since 3.4.0 + */ + public getTraceProvider(): ITraceProvider | null { + // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging + return null; + } } // tslint:disable-next-line diff --git a/AISKU/src/IApplicationInsights.ts b/AISKU/src/IApplicationInsights.ts index 254d012b0..6d3cf4fc7 100644 --- a/AISKU/src/IApplicationInsights.ts +++ b/AISKU/src/IApplicationInsights.ts @@ -1,21 +1,33 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -"use strict"; - import { AnalyticsPlugin } from "@microsoft/applicationinsights-analytics-js"; import { Sender } from "@microsoft/applicationinsights-channel-js"; -import { IAppInsights, IPropertiesPlugin, IRequestHeaders } from "@microsoft/applicationinsights-common"; import { - IConfiguration, ILoadedPlugin, IPlugin, ITelemetryPlugin, ITelemetryUnloadState, UnloadHandler + IAppInsights, IConfiguration, ILoadedPlugin, IOTelApi, IPlugin, IPropertiesPlugin, IRequestHeaders, ITelemetryPlugin, + ITelemetryUnloadState, ITraceApi, ITraceHost, UnloadHandler } from "@microsoft/applicationinsights-core-js"; import { IDependenciesPlugin } from "@microsoft/applicationinsights-dependencies-js"; import { IPromise } from "@nevware21/ts-async"; export { IRequestHeaders }; -export interface IApplicationInsights extends IAppInsights, IDependenciesPlugin, IPropertiesPlugin { +export interface IApplicationInsights extends IAppInsights, IDependenciesPlugin, IPropertiesPlugin, ITraceHost { appInsights: AnalyticsPlugin; + /** + * The OpenTelemetry API instance associated with this instance + * Unlike OpenTelemetry, this API does not return a No-Op implementation and returns null if the SDK has been torn + * down or not yet initialized. + */ + readonly otelApi: IOTelApi | null; + + /** + * OpenTelemetry trace API for creating spans. + * Unlike OpenTelemetry, this API does not return a No-Op implementation and returns null if the SDK has been torn + * down or not yet initialized. + */ + readonly trace: ITraceApi | null; + /** * Attempt to flush data immediately; If executing asynchronously (the default) and * you DO NOT pass a callback function then a [IPromise](https://nevware21.github.io/ts-async/typedoc/interfaces/IPromise.html) @@ -59,7 +71,7 @@ export interface IApplicationInsights extends IAppInsights, IDependenciesPlugin, /** * Find and return the (first) plugin with the specified identifier if present - * @param pluginIdentifier + * @param pluginIdentifier - The identifier of the plugin to find */ getPlugin(pluginIdentifier: string): ILoadedPlugin; diff --git a/AISKU/src/Init.ts b/AISKU/src/Init.ts index 22c39f5ab..a831b9a5e 100644 --- a/AISKU/src/Init.ts +++ b/AISKU/src/Init.ts @@ -44,7 +44,7 @@ export { PropertiesPluginIdentifier, BreezeChannelIdentifier, AnalyticsPluginIdentifier -} from "@microsoft/applicationinsights-common"; +} from "@microsoft/applicationinsights-core-js"; // ---------------------------------------------------------------------------------------------------- // End of Exports available from the Cdn bundles @@ -82,11 +82,12 @@ export { ITraceTelemetry, IMetricTelemetry, IEventTelemetry, + IRequestTelemetry, IAppInsights, eSeverityLevel, IRequestHeaders, EventPersistence -} from "@microsoft/applicationinsights-common"; +} from "@microsoft/applicationinsights-core-js"; export { ISenderConfig } from "@microsoft/applicationinsights-channel-js"; diff --git a/AISKU/src/InternalConstants.ts b/AISKU/src/InternalConstants.ts index 9e840b4d9..94c935053 100644 --- a/AISKU/src/InternalConstants.ts +++ b/AISKU/src/InternalConstants.ts @@ -10,6 +10,8 @@ const _AUTHENTICATED_USER_CONTEXT = "AuthenticatedUserContext"; const _TRACK = "track"; + +export const UNDEFINED_VALUE: undefined = undefined; export const STR_EMPTY = ""; export const STR_SNIPPET = "snippet"; export const STR_GET_COOKIE_MGR = "getCookieMgr"; diff --git a/AISKU/src/Snippet.ts b/AISKU/src/Snippet.ts index 79955c805..45456a546 100644 --- a/AISKU/src/Snippet.ts +++ b/AISKU/src/Snippet.ts @@ -2,8 +2,7 @@ // Licensed under the MIT License. "use strict"; -import { IConfig } from "@microsoft/applicationinsights-common"; -import { IConfiguration } from "@microsoft/applicationinsights-core-js"; +import { IConfig, IConfiguration } from "@microsoft/applicationinsights-core-js"; /** * diff --git a/AISKU/src/applicationinsights-web.ts b/AISKU/src/applicationinsights-web.ts index 0ad8ca2b3..a627bf6a2 100644 --- a/AISKU/src/applicationinsights-web.ts +++ b/AISKU/src/applicationinsights-web.ts @@ -4,6 +4,9 @@ export { AppInsightsSku as ApplicationInsights } from "./AISku"; export { ApplicationInsightsContainer } from "./ApplicationInsightsContainer"; +// OpenTelemetry trace API exports (public interfaces only) +export { IOTelTracerProvider, IOTelTracer, IAttributeContainer, IOTelAttributes, IReadableSpan } from "@microsoft/applicationinsights-core-js"; + // Re-exports export { IConfiguration, @@ -30,7 +33,11 @@ export { INotificationManager, IProcessTelemetryContext, Tags, - ILoadedPlugin + ILoadedPlugin, + IOTelSpan, + eOTelSpanKind, + OTelSpanKind, + IOTelSpanOptions } from "@microsoft/applicationinsights-core-js"; export { IConfig, @@ -50,12 +57,11 @@ export { Metric, PageView, PageViewPerformance, - RemoteDependencyData, Trace, DistributedTracingModes, IRequestHeaders, EventPersistence -} from "@microsoft/applicationinsights-common"; +} from "@microsoft/applicationinsights-core-js"; export { Sender, ISenderConfig } from "@microsoft/applicationinsights-channel-js"; export { ApplicationInsights as ApplicationAnalytics, IAppInsightsInternal, IAnalyticsConfig } from "@microsoft/applicationinsights-analytics-js"; export { PropertiesPlugin } from "@microsoft/applicationinsights-properties-js"; diff --git a/AISKU/src/internal/trace/spanUtils.ts b/AISKU/src/internal/trace/spanUtils.ts new file mode 100644 index 000000000..9ba03d029 --- /dev/null +++ b/AISKU/src/internal/trace/spanUtils.ts @@ -0,0 +1,506 @@ +// Copyright (c) Microsoft Corporation. +// Licensed under the MIT License. + +import { + ATTR_CLIENT_ADDRESS, ATTR_CLIENT_PORT, ATTR_ENDUSER_ID, ATTR_ENDUSER_PSEUDO_ID, ATTR_ERROR_TYPE, ATTR_EXCEPTION_MESSAGE, + ATTR_EXCEPTION_STACKTRACE, ATTR_EXCEPTION_TYPE, ATTR_HTTP_REQUEST_METHOD, ATTR_HTTP_RESPONSE_STATUS_CODE, ATTR_HTTP_ROUTE, + ATTR_NETWORK_LOCAL_ADDRESS, ATTR_NETWORK_LOCAL_PORT, ATTR_NETWORK_PEER_ADDRESS, ATTR_NETWORK_PEER_PORT, ATTR_NETWORK_PROTOCOL_NAME, + ATTR_NETWORK_PROTOCOL_VERSION, ATTR_NETWORK_TRANSPORT, ATTR_SERVER_ADDRESS, ATTR_SERVER_PORT, ATTR_URL_FULL, ATTR_URL_PATH, + ATTR_URL_QUERY, ATTR_URL_SCHEME, ATTR_USER_AGENT_ORIGINAL, CtxTagKeys, DBSYSTEMVALUES_MONGODB, DBSYSTEMVALUES_MYSQL, + DBSYSTEMVALUES_POSTGRESQL, DBSYSTEMVALUES_REDIS, EXP_ATTR_ENDUSER_ID, EXP_ATTR_ENDUSER_PSEUDO_ID, EXP_ATTR_SYNTHETIC_TYPE, + IAppInsightsCore, IAttributeContainer, IConfiguration, IContextTagKeys, IDependencyTelemetry, IReadableSpan, IRequestTelemetry, + ITelemetryItem, MicrosoftClientIp, OTelAttributeValue, RemoteDependencyDataType, RemoteDependencyEnvelopeType, RequestDataType, + RequestEnvelopeType, SEMATTRS_DB_NAME, SEMATTRS_DB_OPERATION, SEMATTRS_DB_STATEMENT, SEMATTRS_DB_SYSTEM, SEMATTRS_ENDUSER_ID, + SEMATTRS_EXCEPTION_MESSAGE, SEMATTRS_EXCEPTION_STACKTRACE, SEMATTRS_EXCEPTION_TYPE, SEMATTRS_HTTP_CLIENT_IP, SEMATTRS_HTTP_FLAVOR, + SEMATTRS_HTTP_HOST, SEMATTRS_HTTP_METHOD, SEMATTRS_HTTP_ROUTE, SEMATTRS_HTTP_SCHEME, SEMATTRS_HTTP_STATUS_CODE, SEMATTRS_HTTP_TARGET, + SEMATTRS_HTTP_URL, SEMATTRS_HTTP_USER_AGENT, SEMATTRS_NET_HOST_IP, SEMATTRS_NET_HOST_NAME, SEMATTRS_NET_HOST_PORT, SEMATTRS_NET_PEER_IP, + SEMATTRS_NET_PEER_NAME, SEMATTRS_NET_PEER_PORT, SEMATTRS_NET_TRANSPORT, SEMATTRS_PEER_SERVICE, SEMATTRS_RPC_GRPC_STATUS_CODE, + SEMATTRS_RPC_SYSTEM, Tags, createAttributeContainer, createTelemetryItem, eDependencyTypes, eOTelSpanKind, eOTelSpanStatusCode, + fieldRedaction, getDependencyTarget, getHttpMethod, getHttpStatusCode, getHttpUrl, getLocationIp, getUrl, getUserAgent, + hrTimeToMilliseconds, isSqlDB, isSyntheticSource, urlGetPathName +} from "@microsoft/applicationinsights-core-js"; +import { ILazyValue, arrIncludes, asString, getLazy, isNullOrUndefined, strLower, strStartsWith } from "@nevware21/ts-utils"; +import { STR_EMPTY, UNDEFINED_VALUE } from "../../InternalConstants"; + +/** + * Azure SDK namespace. + * @internal + */ +const AzNamespace = "az.namespace"; +const AzResourceNamespace = "azure.resource_provider.namespace"; + +/** + * Azure SDK Eventhub. + * @internal + */ +const MicrosoftEventHub = "Microsoft.EventHub"; + +/** + * Azure SDK message bus destination. + * @internal + */ +const MessageBusDestination = "message_bus.destination"; + +/** + * AI time since enqueued attribute. + * @internal + */ +const TIME_SINCE_ENQUEUED = "timeSinceEnqueued"; + +const PORT_REGEX: ILazyValue = (/*#__PURE__*/ getLazy(() => new RegExp(/(https?)(:\/\/.*)(:\d+)(\S*)/))); +const HTTP_DOT = (/*#__PURE__*/ "http."); + +const _MS_PROCESSED_BY_METRICS_EXTRACTORS = (/* #__PURE__*/"_MS.ProcessedByMetricExtractors"); + +/** + * Legacy HTTP semantic convention values + * @internal + */ +const _ignoreSemanticValues: ILazyValue = (/* #__PURE__*/ getLazy(_initIgnoreSemanticValues)); + +/* #__NO_SIDE_EFFECTS__ */ +function _initIgnoreSemanticValues(): string[] { + return [ + // Internal Microsoft attributes + _MS_PROCESSED_BY_METRICS_EXTRACTORS, + + // Legacy HTTP semantic values + SEMATTRS_NET_PEER_IP, + SEMATTRS_NET_PEER_NAME, + SEMATTRS_NET_HOST_IP, + SEMATTRS_PEER_SERVICE, + SEMATTRS_HTTP_USER_AGENT, + SEMATTRS_HTTP_METHOD, + SEMATTRS_HTTP_URL, + SEMATTRS_HTTP_STATUS_CODE, + SEMATTRS_HTTP_ROUTE, + SEMATTRS_HTTP_HOST, + SEMATTRS_DB_SYSTEM, + SEMATTRS_DB_STATEMENT, + SEMATTRS_DB_OPERATION, + SEMATTRS_DB_NAME, + SEMATTRS_RPC_SYSTEM, + SEMATTRS_RPC_GRPC_STATUS_CODE, + SEMATTRS_EXCEPTION_TYPE, + SEMATTRS_EXCEPTION_MESSAGE, + SEMATTRS_EXCEPTION_STACKTRACE, + SEMATTRS_HTTP_SCHEME, + SEMATTRS_HTTP_TARGET, + SEMATTRS_HTTP_FLAVOR, + SEMATTRS_NET_TRANSPORT, + SEMATTRS_NET_HOST_NAME, + SEMATTRS_NET_HOST_PORT, + SEMATTRS_NET_PEER_PORT, + SEMATTRS_HTTP_CLIENT_IP, + SEMATTRS_ENDUSER_ID, + HTTP_DOT + "status_text", + + // http Semabtic conventions + ATTR_CLIENT_ADDRESS, + ATTR_CLIENT_PORT, + ATTR_SERVER_ADDRESS, + ATTR_SERVER_PORT, + ATTR_URL_FULL, + ATTR_URL_PATH, + ATTR_URL_QUERY, + ATTR_URL_SCHEME, + ATTR_ERROR_TYPE, + ATTR_NETWORK_LOCAL_ADDRESS, + ATTR_NETWORK_LOCAL_PORT, + ATTR_NETWORK_PROTOCOL_NAME, + ATTR_NETWORK_PEER_ADDRESS, + ATTR_NETWORK_PEER_PORT, + ATTR_NETWORK_PROTOCOL_VERSION, + ATTR_NETWORK_TRANSPORT, + ATTR_USER_AGENT_ORIGINAL, + ATTR_HTTP_REQUEST_METHOD, + ATTR_HTTP_RESPONSE_STATUS_CODE, + ATTR_EXCEPTION_TYPE, + ATTR_EXCEPTION_MESSAGE, + ATTR_EXCEPTION_STACKTRACE, + EXP_ATTR_ENDUSER_ID, + EXP_ATTR_ENDUSER_PSEUDO_ID, + EXP_ATTR_SYNTHETIC_TYPE + ]; +} + +function _populateTagsFromSpan(telemetryItem: ITelemetryItem, span: IReadableSpan, contextKeys: IContextTagKeys, config: IConfiguration): void { + + let tags: Tags = telemetryItem.tags = (telemetryItem.tags || [] as Tags); + let container = span.attribContainer || createAttributeContainer(span.attributes); + + tags[contextKeys.operationId] = span.spanContext().traceId; + if ((span.parentSpanContext || {}).spanId) { + tags[contextKeys.operationParentId] = span.parentSpanContext.spanId; + } + + // Map OpenTelemetry enduser attributes to Application Insights user attributes + const endUserId = container.get(ATTR_ENDUSER_ID); + if (endUserId) { + tags[contextKeys.userAuthUserId] = asString(endUserId); + } + + const endUserPseudoId = container.get(ATTR_ENDUSER_PSEUDO_ID); + if (endUserPseudoId) { + tags[contextKeys.userId] = asString(endUserPseudoId); + } + + const httpUserAgent = getUserAgent(container); + if (httpUserAgent) { + tags["ai.user.userAgent"] = String(httpUserAgent); + } + if (isSyntheticSource(container)) { + tags[contextKeys.operationSyntheticSource] = "True"; + } + + // Check for microsoft.client.ip first - this takes precedence over all other IP logic + const microsoftClientIp = container.get(MicrosoftClientIp); + if (microsoftClientIp) { + tags[contextKeys.locationIp] = asString(microsoftClientIp); + } + + if (span.kind === eOTelSpanKind.SERVER) { + const httpMethod = getHttpMethod(container); + // Only use the fallback IP logic for server spans if microsoft.client.ip is not set + if (!microsoftClientIp) { + tags[contextKeys.locationIp] = getLocationIp(container); + } + + if (httpMethod) { + const httpRoute = container.get(ATTR_HTTP_ROUTE); + const httpUrl = getHttpUrl(container); + tags[contextKeys.operationName] = span.name; // Default + if (httpRoute) { + // AiOperationName max length is 1024 + // https://github.com/MohanGsk/ApplicationInsights-Home/blob/master/EndpointSpecs/Schemas/Bond/ContextTagKeys.bond + tags[contextKeys.operationName] = httpMethod + " " + fieldRedaction(asString(httpRoute), config); + } else if (httpUrl) { + try { + const urlPathName = fieldRedaction(urlGetPathName(asString(httpUrl)), config); + tags[contextKeys.operationName] = httpMethod + " " + urlPathName; + } catch { + /* no-op */ + } + } + } else { + tags[contextKeys.operationName] = span.name; + } + } else { + let opName = container.get(contextKeys.operationName); + if (opName) { + tags[contextKeys.operationName] = opName as string; + } + } + // TODO: Location IP TBD for non server spans +} + +/** + * Check to see if the key is in the list of known properties to ignore (exclude) + * from the properties collection + * @param key - the property key to check + * @param contextKeys - The current context keys + * @returns true if the key should be ignored, false otherwise + */ +function _isIgnorePropertiesKey(key: string, contextKeys: IContextTagKeys): boolean { + let result = false; + + if (arrIncludes(_ignoreSemanticValues.v, key)) { + // The key is in set of known keys to ignore + result = true; + } else if (strStartsWith(key, "microsoft.")) { + // Ignoring all ALL keys starting with "microsoft." + result = true; + } else if (key === contextKeys.operationName) { + // Ignoring the key if it is the operation name context tag + result = true; + } + + return result; +} + +function _populatePropertiesFromAttributes(item: ITelemetryItem, contextKeys: IContextTagKeys, container: IAttributeContainer): void { + if (container) { + let baseData = item.baseData = (item.baseData || {}); + let properties: { [propertyName: string]: any } = baseData.properties = (baseData.properties || {}); + + container.forEach((key: string, value) => { + // Avoid duplication ignoring fields already mapped. + if (!_isIgnorePropertiesKey(key, contextKeys)) { + properties[key] = value; + } + }); + } +} + +function _populateHttpDependencyProperties(dependencyTelemetry: IDependencyTelemetry, container: IAttributeContainer, httpMethod: OTelAttributeValue | undefined, config: IConfiguration): boolean { + if (httpMethod) { + // HTTP Dependency + const httpUrl = getHttpUrl(container); + if (httpUrl) { + try { + const pathName = urlGetPathName(asString(httpUrl)); + if (pathName) { + dependencyTelemetry.name = httpMethod + " " + fieldRedaction(pathName, config); + } + } catch { + /* no-op */ + } + } + + dependencyTelemetry.type = eDependencyTypes.Http; + dependencyTelemetry.data = fieldRedaction(getUrl(container), config); + const httpStatusCode = getHttpStatusCode(container); + if (httpStatusCode) { + dependencyTelemetry.responseCode = +httpStatusCode; + } + + let target = getDependencyTarget(container); + if (target) { + try { + // Remove default port + const res = PORT_REGEX.v.exec(target); + if (res !== null) { + const protocol = res[1]; + const port = res[3]; + if ( + (protocol === "https" && port === ":443") || + (protocol === "http" && port === ":80") + ) { + // Drop port + target = res[1] + res[2] + res[4]; + } + } + } catch { + /* no-op */ + } + + dependencyTelemetry.target = target; + } + } + + return !!httpMethod; +} + +function _populateDbDependencyProperties(dependencyTelemetry: IDependencyTelemetry, container: IAttributeContainer, dbSystem: OTelAttributeValue | undefined): boolean { + if (dbSystem) { + // TODO: Remove special logic when Azure UX supports OpenTelemetry dbSystem + if (String(dbSystem) === DBSYSTEMVALUES_MYSQL) { + dependencyTelemetry.type = "mysql"; + } else if (String(dbSystem) === DBSYSTEMVALUES_POSTGRESQL) { + dependencyTelemetry.type = "postgresql"; + } else if (String(dbSystem) === DBSYSTEMVALUES_MONGODB) { + dependencyTelemetry.type = "mongodb"; + } else if (String(dbSystem) === DBSYSTEMVALUES_REDIS) { + dependencyTelemetry.type = "redis"; + } else if (isSqlDB(String(dbSystem))) { + dependencyTelemetry.type = "SQL"; + } else { + dependencyTelemetry.type = String(dbSystem); + } + const dbStatement = container.get(SEMATTRS_DB_STATEMENT); + const dbOperation = container.get(SEMATTRS_DB_OPERATION); + if (dbStatement) { + dependencyTelemetry.data = String(dbStatement); + } else if (dbOperation) { + dependencyTelemetry.data = String(dbOperation); + } + const target = getDependencyTarget(container); + const dbName = container.get(SEMATTRS_DB_NAME); + if (target) { + dependencyTelemetry.target = dbName ? target + "|" + asString(dbName) : target; + } else { + dependencyTelemetry.target = dbName ? asString(dbName) : asString(dbSystem); + } + } + + return !!dbSystem; +} + +function _populateRpcDependencyProperties(dependencyTelemetry: IDependencyTelemetry, container: IAttributeContainer, rpcSystem: OTelAttributeValue | undefined): boolean { + if (rpcSystem) { + if (strLower(rpcSystem) === "wcf") { + dependencyTelemetry.type = eDependencyTypes.Wcf; + } else { + dependencyTelemetry.type = eDependencyTypes.Grpc; + } + const grpcStatusCode = container.get(SEMATTRS_RPC_GRPC_STATUS_CODE); + if (!isNullOrUndefined(grpcStatusCode)) { + dependencyTelemetry.responseCode = +grpcStatusCode; + } + const target = getDependencyTarget(container); + if (target) { + dependencyTelemetry.target = asString(target); + } else { + dependencyTelemetry.target = String(rpcSystem); + } + } + + return !!rpcSystem; +} + +function createDependencyTelemetryItem(core: IAppInsightsCore, span: IReadableSpan, contextKeys: IContextTagKeys): ITelemetryItem { + let container = span.attribContainer || createAttributeContainer(span.attributes); + let dependencyType = "Dependency"; + + if (span.kind === eOTelSpanKind.PRODUCER) { + dependencyType = eDependencyTypes.QueueMessage; + } else if (span.kind === eOTelSpanKind.INTERNAL && span.parentSpanContext) { + dependencyType = eDependencyTypes.InProc; + } + + let spanCtx = span.spanContext(); + let dependencyTelemetry: IDependencyTelemetry = { + name: span.name, // Default + id: spanCtx.spanId || core.getTraceCtx().spanId, + success: (span.status || {}).code !== eOTelSpanStatusCode.ERROR, + responseCode: 0, + type: dependencyType, + duration: hrTimeToMilliseconds(span.duration), + data: STR_EMPTY, + target: STR_EMPTY, + properties: UNDEFINED_VALUE, + measurements: UNDEFINED_VALUE + }; + + // Check for HTTP Dependency + if (!_populateHttpDependencyProperties(dependencyTelemetry, container, getHttpMethod(container), core.config)) { + // Check for DB Dependency + if (!_populateDbDependencyProperties(dependencyTelemetry, container, container.get(SEMATTRS_DB_SYSTEM))) { + // Check for Rpc Dependency + _populateRpcDependencyProperties(dependencyTelemetry, container, container.get(SEMATTRS_RPC_SYSTEM)); + } + } + + return createTelemetryItem(dependencyTelemetry, RemoteDependencyDataType, RemoteDependencyEnvelopeType.replace("{0}.", ""), core.logger); +} + +function createRequestTelemetryItem(core: IAppInsightsCore, span: IReadableSpan, contextKeys: IContextTagKeys): ITelemetryItem { + let container = span.attribContainer || createAttributeContainer(span.attributes); + + let spanCtx = span.spanContext(); + const requestData: IRequestTelemetry = { + name: span.name, // Default + id: spanCtx.spanId || core.getTraceCtx().spanId, + success: + span.status.code !== eOTelSpanStatusCode.UNSET + ? span.status.code === eOTelSpanStatusCode.OK + : (Number(getHttpStatusCode(container)) || 0) < 400, + responseCode: 0, + duration: hrTimeToMilliseconds(span.duration), + source: undefined + }; + const httpMethod = getHttpMethod(container); + const grpcStatusCode = container.get(SEMATTRS_RPC_GRPC_STATUS_CODE); + if (httpMethod) { + requestData.url = fieldRedaction(getUrl(container), core.config); + const httpStatusCode = getHttpStatusCode(container); + if (httpStatusCode) { + requestData.responseCode = +httpStatusCode; + } + } else if (grpcStatusCode) { + requestData.responseCode = +grpcStatusCode; + } + + return createTelemetryItem(requestData, RequestDataType, RequestEnvelopeType.replace("{0}.", ""), core.logger); +} + +export function createTelemetryItemFromSpan(core: IAppInsightsCore, span: IReadableSpan): ITelemetryItem | null { + let telemetryItem: ITelemetryItem = null; + let container = span.attribContainer || createAttributeContainer(span.attributes); + let contextKeys: IContextTagKeys = CtxTagKeys; + let kind = span.kind; + if (kind == eOTelSpanKind.SERVER || kind == eOTelSpanKind.CONSUMER) { + // Request + telemetryItem = createRequestTelemetryItem(core, span, contextKeys); + } else if (kind == eOTelSpanKind.CLIENT || kind == eOTelSpanKind.PRODUCER || kind == eOTelSpanKind.INTERNAL) { + // RemoteDependency + telemetryItem = createDependencyTelemetryItem(core, span, contextKeys); + } else { + //diag.error(`Unsupported span kind ${span.kind}`); + } + + if (telemetryItem) { + // Set start time for the telemetry item from the event, not the time it is being processed (the default) + // The channel envelope creator uses this value when creating the envelope only when defined, otherwise it + // uses the time when the item is being processed + let baseData = telemetryItem.baseData = telemetryItem.baseData || {}; + baseData.startTime = new Date(hrTimeToMilliseconds(span.startTime)); + + // Add dt extension to the telemetry item + let ext = telemetryItem.ext = telemetryItem.ext || {}; + let dt = ext["dt"] = ext["dt"] || {}; + + // Don't overwrite any existing values + dt.spanId = dt.spanId || span.spanContext().spanId; + dt.traceId = dt.traceId || span.spanContext().traceId; + + let traceFlags = span.spanContext().traceFlags; + if (!isNullOrUndefined(traceFlags)) { + dt.traceFlags = dt.traceFlags || traceFlags; + } + + _populateTagsFromSpan(telemetryItem, span, contextKeys, core.config); + _populatePropertiesFromAttributes(telemetryItem, contextKeys, container); + + let sampleRate = container.get("microsoft.sample_rate"); + if (!isNullOrUndefined(sampleRate)) { + (telemetryItem as any).sampleRate = Number(sampleRate); + } + + // Azure SDK + let azNamespace = container.get(AzResourceNamespace) || container.get(AzNamespace); + if (azNamespace) { + if (span.kind === eOTelSpanKind.INTERNAL) { + baseData.type = eDependencyTypes.InProc + " | " + azNamespace; + } + + if (azNamespace === MicrosoftEventHub) { + _parseEventHubSpan(telemetryItem, span); + } + } + } + + return telemetryItem; +} + +/** + * Implementation of Mapping to Azure Monitor + * + * https://gist.github.com/lmolkova/e4215c0f44a49ef824983382762e6b92#mapping-to-azure-monitor-application-insights-telemetry + * @internal + */ +function _parseEventHubSpan(telemetryItem: ITelemetryItem, span: IReadableSpan): void { + let baseData = telemetryItem.baseData = telemetryItem.baseData || {}; + let container = span.attribContainer || createAttributeContainer(span.attributes); + const namespace = container.get(AzResourceNamespace) || container.get(AzNamespace); + const peerAddress = asString(container.get(SEMATTRS_NET_PEER_NAME) || container.get("peer.address") || "unknown").replace(/\/$/g, ""); // remove trailing "/" + const messageBusDestination = (container.get(MessageBusDestination) || "unknown") as string; + let baseType = baseData.type || ""; + let kind = span.kind; + + if (kind === eOTelSpanKind.CLIENT) { + baseType = namespace; + baseData.target = peerAddress + "/" + messageBusDestination; + } else if (kind === eOTelSpanKind.PRODUCER) { + baseType = "Queue Message | " + namespace; + baseData.target = peerAddress + "/" + messageBusDestination; + } else if (kind === eOTelSpanKind.CONSUMER) { + baseType = "Queue Message | " + namespace; + (baseData as any).source = peerAddress + "/" + messageBusDestination; + + let measurements = baseData.measurements = (baseData.measurements || {}); + let timeSinceEnqueued = container.get("timeSinceEnqueued"); + if (timeSinceEnqueued) { + measurements[TIME_SINCE_ENQUEUED] = Number(timeSinceEnqueued); + } else { + let enqueuedTime = parseFloat(asString(container.get("enqueuedTime"))); + if (isNaN(enqueuedTime)) { + enqueuedTime = 0; + } + + measurements[TIME_SINCE_ENQUEUED] = hrTimeToMilliseconds(span.startTime) - enqueuedTime; + } + } + + baseData.type = baseType; +} diff --git a/AISKULight/Tests/Unit/src/AISKULightSize.Tests.ts b/AISKULight/Tests/Unit/src/AISKULightSize.Tests.ts index e4b7ba884..58cec6c38 100644 --- a/AISKULight/Tests/Unit/src/AISKULightSize.Tests.ts +++ b/AISKULight/Tests/Unit/src/AISKULightSize.Tests.ts @@ -51,10 +51,10 @@ function _checkSize(checkType: string, maxSize: number, size: number, isNightly: } export class AISKULightSizeCheck extends AITestClass { - private readonly MAX_RAW_SIZE = 94; - private readonly MAX_BUNDLE_SIZE = 94; - private readonly MAX_RAW_DEFLATE_SIZE = 39; - private readonly MAX_BUNDLE_DEFLATE_SIZE = 39; + private readonly MAX_RAW_SIZE = 102; + private readonly MAX_BUNDLE_SIZE = 102; + private readonly MAX_RAW_DEFLATE_SIZE = 42; + private readonly MAX_BUNDLE_DEFLATE_SIZE = 42; private readonly rawFilePath = "../dist/es5/applicationinsights-web-basic.min.js"; private readonly currentVer = "3.3.11"; private readonly prodFilePath = `../browser/es5/aib.${this.currentVer[0]}.min.js`; diff --git a/AISKULight/Tests/Unit/src/aiskuliteunittests.ts b/AISKULight/Tests/Unit/src/aiskuliteunittests.ts index f1a132879..65eb86968 100644 --- a/AISKULight/Tests/Unit/src/aiskuliteunittests.ts +++ b/AISKULight/Tests/Unit/src/aiskuliteunittests.ts @@ -2,10 +2,12 @@ import { AISKULightSizeCheck } from "./AISKULightSize.Tests"; import { ApplicationInsightsDynamicConfigTests } from "./dynamicconfig.tests"; import { ApplicationInsightsConfigTests } from "./config.tests"; import { GlobalTestHooks } from "./GlobalTestHooks.Test"; +import { AISKULightOTelNegativeTests } from "./otelNegative.tests"; export function runTests() { new GlobalTestHooks().registerTests(); new AISKULightSizeCheck().registerTests(); new ApplicationInsightsDynamicConfigTests().registerTests(); new ApplicationInsightsConfigTests().registerTests(); + new AISKULightOTelNegativeTests().registerTests(); } \ No newline at end of file diff --git a/AISKULight/Tests/Unit/src/config.tests.ts b/AISKULight/Tests/Unit/src/config.tests.ts index e234d8ff1..2cd6c46e5 100644 --- a/AISKULight/Tests/Unit/src/config.tests.ts +++ b/AISKULight/Tests/Unit/src/config.tests.ts @@ -1,7 +1,7 @@ import { AITestClass, Assert, PollingAssert } from "@microsoft/ai-test-framework"; import { ITelemetryItem, newId } from "@microsoft/applicationinsights-core-js"; import { ApplicationInsights} from "../../../src/index"; -import { BreezeChannelIdentifier, ContextTagKeys, utlRemoveSessionStorage } from "@microsoft/applicationinsights-common"; +import { BreezeChannelIdentifier, ContextTagKeys, utlRemoveSessionStorage } from "@microsoft/applicationinsights-core-js"; import { Sender } from "@microsoft/applicationinsights-channel-js"; export class ApplicationInsightsConfigTests extends AITestClass { diff --git a/AISKULight/Tests/Unit/src/dynamicconfig.tests.ts b/AISKULight/Tests/Unit/src/dynamicconfig.tests.ts index 6bfc1e34b..1405c691c 100644 --- a/AISKULight/Tests/Unit/src/dynamicconfig.tests.ts +++ b/AISKULight/Tests/Unit/src/dynamicconfig.tests.ts @@ -1,5 +1,5 @@ import { AITestClass, Assert, PollingAssert } from "@microsoft/ai-test-framework"; -import { IConfig } from "@microsoft/applicationinsights-common"; +import { IConfig } from "@microsoft/applicationinsights-core-js"; import { IConfiguration, IPayloadData, isString, ITelemetryItem, IXHROverride, newId } from "@microsoft/applicationinsights-core-js"; import { ApplicationInsights, ISenderConfig } from "../../../src/index"; import { createAsyncResolvedPromise } from "@nevware21/ts-async"; diff --git a/AISKULight/Tests/Unit/src/otelNegative.tests.ts b/AISKULight/Tests/Unit/src/otelNegative.tests.ts new file mode 100644 index 000000000..f8fc5780e --- /dev/null +++ b/AISKULight/Tests/Unit/src/otelNegative.tests.ts @@ -0,0 +1,189 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { ApplicationInsights } from "../../../src/index"; +import { utlRemoveSessionStorage } from "@microsoft/applicationinsights-core-js"; +import { isNullOrUndefined, newId } from "@microsoft/applicationinsights-core-js"; + +/** + * Negative tests for OpenTelemetry usage in AISKU Light + * These tests ensure that no exceptions are thrown and helpers behave correctly + * when there is no trace provider or OTel support instances + */ +export class AISKULightOTelNegativeTests extends AITestClass { + private readonly _instrumentationKey = "testIkey-1234-5678-9012-3456789012"; + private _sessionPrefix: string; + + public testInitialize() { + super.testInitialize(); + this._sessionPrefix = newId(); + } + + public testCleanup() { + utlRemoveSessionStorage(null as any, "AI_sentBuffer"); + utlRemoveSessionStorage(null as any, "AI_buffer"); + utlRemoveSessionStorage(null as any, this._sessionPrefix + "_AI_sentBuffer"); + utlRemoveSessionStorage(null as any, this._sessionPrefix + "_AI_buffer"); + super.testCleanup(); + } + + public registerTests() { + this.addTraceContextWithoutProviderTests(); + this.addUnloadWithoutProviderTests(); + this.addConfigurationChangesWithoutProviderTests(); + } + + private addTraceContextWithoutProviderTests(): void { + this.testCase({ + name: "AISKULight.getTraceCtx: should return valid context without trace provider", + test: () => { + // Arrange + const config = { + instrumentationKey: this._instrumentationKey, + namePrefix: this._sessionPrefix + }; + const ai = new ApplicationInsights(config); + this.onDone(() => { + ai.unload(false); + }); + + // Act - no trace provider is set by default in AISKU Light + const ctx = ai.getTraceCtx(); + + // Assert - should return a valid context without throwing + Assert.ok(ctx !== undefined, "Should return a context (can be null)"); + + // If it returns a context, it should be valid + Assert.ok(!isNullOrUndefined(ctx?.traceId), "Context should have traceId"); + Assert.ok(!isNullOrUndefined(ctx?.spanId), "Context should have spanId"); + Assert.equal("", ctx?.spanId, "SpanId should be empty string without provider"); + } + }); + + this.testCase({ + name: "AISKULight.getTraceCtx: should not throw when called multiple times", + test: () => { + // Arrange + const config = { + instrumentationKey: this._instrumentationKey, + namePrefix: this._sessionPrefix + }; + const ai = new ApplicationInsights(config); + this.onDone(() => { + ai.unload(false); + }); + + // Act & Assert + Assert.doesNotThrow(() => { + // eslint-disable-next-line @typescript-eslint/no-unused-vars + const _ctx1 = ai.getTraceCtx(); + // eslint-disable-next-line @typescript-eslint/no-unused-vars + const _ctx2 = ai.getTraceCtx(); + // eslint-disable-next-line @typescript-eslint/no-unused-vars + const _ctx3 = ai.getTraceCtx(); + + // Multiple calls should work without issues + Assert.ok(true, "Multiple getTraceCtx calls should not throw"); + Assert.equal(_ctx1, _ctx2, "Multiple calls should return same context instance"); + Assert.equal(_ctx2, _ctx3, "Multiple calls should return same context instance"); + Assert.equal(_ctx1?.traceId, _ctx2?.traceId, "TraceId should be consistent across calls"); + Assert.equal(_ctx2?.traceId, _ctx3?.traceId, "TraceId should be consistent across calls"); + Assert.equal(_ctx1?.spanId, _ctx2?.spanId, "SpanId should be consistent across calls"); + Assert.equal(_ctx2?.spanId, _ctx3?.spanId, "SpanId should be consistent across calls"); + }, "Multiple getTraceCtx calls should be safe"); + } + }); + + this.testCase({ + name: "AISKULight: getTraceCtx should work after unload", + test: () => { + // Arrange + const config = { + instrumentationKey: this._instrumentationKey, + namePrefix: this._sessionPrefix + }; + const ai = new ApplicationInsights(config); + + // Act - unload first + ai.unload(false); + + // Assert - should not throw even after unload + Assert.doesNotThrow(() => { + // eslint-disable-next-line @typescript-eslint/no-unused-vars + const _ctx = ai.getTraceCtx(); + // Context might be null after unload, which is fine + }, "getTraceCtx should not throw after unload"); + } + }); + } + + + + private addUnloadWithoutProviderTests(): void { + this.testCase({ + name: "AISKULight: unload should work gracefully without trace provider", + test: () => { + // Arrange + const config = { + instrumentationKey: this._instrumentationKey, + namePrefix: this._sessionPrefix + }; + const ai = new ApplicationInsights(config); + + // Act & Assert + Assert.doesNotThrow(() => { + ai.unload(false); + }, "Unload should work without trace provider"); + + // Verify we can still access config after unload + Assert.ok(ai.config, "Config should still be accessible after unload"); + } + }); + + this.testCase({ + name: "AISKULight: unload with async flag should work without provider", + test: () => { + // Arrange + const config = { + instrumentationKey: this._instrumentationKey, + namePrefix: this._sessionPrefix + }; + const ai = new ApplicationInsights(config); + + // Act & Assert + Assert.doesNotThrow(() => { + ai.unload(true); + }, "Async unload should work without trace provider"); + } + }); + } + + private addConfigurationChangesWithoutProviderTests(): void { + this.testCase({ + name: "AISKULight: should handle traceCfg in config without trace provider", + test: () => { + // Arrange + const config = { + instrumentationKey: this._instrumentationKey, + namePrefix: this._sessionPrefix, + traceCfg: { + suppressTracing: false + } + }; + + // Act & Assert + Assert.doesNotThrow(() => { + const ai = new ApplicationInsights(config); + + // Verify traceCfg is present + Assert.ok(ai.config.traceCfg, "traceCfg should be accessible"); + + this.onDone(() => { + ai.unload(false); + }); + }, "Should handle traceCfg without trace provider"); + } + }); + } +} diff --git a/AISKULight/Tests/UnitTests.html b/AISKULight/Tests/UnitTests.html index e5f022e3f..389753544 100644 --- a/AISKULight/Tests/UnitTests.html +++ b/AISKULight/Tests/UnitTests.html @@ -5,9 +5,9 @@ Tests for Application Insights JavaScript AISKU Light - - - + + + - + - + - - - - - - - -
-
-
- - - \ No newline at end of file diff --git a/shared/AppInsightsCommon/Tests/tsconfig.json b/shared/AppInsightsCommon/Tests/tsconfig.json deleted file mode 100644 index d1b20db2c..000000000 --- a/shared/AppInsightsCommon/Tests/tsconfig.json +++ /dev/null @@ -1,13 +0,0 @@ -{ - "compilerOptions": { - "sourceMap": true, - "inlineSources": true, - "noImplicitAny": false, - "module": "amd", - "moduleResolution": "Node", - "target": "es5", - "alwaysStrict": true, - "declaration": true - }, - "files": [] -} \ No newline at end of file diff --git a/shared/AppInsightsCommon/package.json b/shared/AppInsightsCommon/package.json index 0df6049dc..859e42208 100644 --- a/shared/AppInsightsCommon/package.json +++ b/shared/AppInsightsCommon/package.json @@ -18,8 +18,8 @@ "build:esm": "grunt common", "build:browser": "rollup -c rollup.config.js --bundleConfigAsCjs", "rebuild": "npm run build", - "test": "grunt commontest", - "mintest": "grunt common-mintest", + "test": "", + "mintest": "", "lint": "tslint -p tsconfig.json", "dtsgen": "api-extractor run --local && node ../../scripts/dtsgen.js \"Microsoft Application Insights Common JavaScript Library\"", "sri": "node ../../tools/subResourceIntegrity/generateIntegrityFile.js", @@ -29,7 +29,6 @@ "api-docs": "typedoc" }, "devDependencies": { - "@microsoft/ai-test-framework": "0.0.1", "@microsoft/applicationinsights-rollup-plugin-uglify3-js": "1.0.0", "@microsoft/applicationinsights-rollup-es5": "1.0.2", "@microsoft/api-extractor": "^7.40.0", diff --git a/shared/AppInsightsCommon/src/Enums.ts b/shared/AppInsightsCommon/src/Enums.ts deleted file mode 100644 index a8c159c3d..000000000 --- a/shared/AppInsightsCommon/src/Enums.ts +++ /dev/null @@ -1,79 +0,0 @@ -// Copyright (c) Microsoft Corporation. All rights reserved. -// Licensed under the MIT License. - -import { createEnumStyle } from "@microsoft/applicationinsights-core-js"; - -/** - * Type of storage to differentiate between local storage and session storage - */ -export const enum eStorageType { - LocalStorage, - SessionStorage -} - -export const StorageType = (/* @__PURE__ */ createEnumStyle({ - LocalStorage: eStorageType.LocalStorage, - SessionStorage: eStorageType.SessionStorage -})); -export type StorageType = eStorageType | typeof StorageType; - -/** - * Enum is used in aiDataContract to describe how fields are serialized. - * For instance: (Fieldtype.Required | FieldType.Array) will mark the field as required and indicate it's an array - */ -export const enum FieldType { Default = 0, Required = 1, Array = 2, Hidden = 4 } - -export const enum eDistributedTracingModes { - /** - * (Default) Send Application Insights correlation headers - */ - - AI = 0, - - /** - * Send both W3C Trace Context headers and back-compatibility Application Insights headers - */ - AI_AND_W3C, - - /** - * Send W3C Trace Context headers - */ - W3C -} - -export const DistributedTracingModes = (/* @__PURE__ */ createEnumStyle({ - AI: eDistributedTracingModes.AI, - AI_AND_W3C: eDistributedTracingModes.AI_AND_W3C, - W3C: eDistributedTracingModes.W3C -})); -export type DistributedTracingModes = number | eDistributedTracingModes; - -/** - * The EventPersistence contains a set of values that specify the event's persistence. - */ -export const enum EventPersistenceValue { - /** - * Normal persistence. - */ - Normal = 1, - /** - * Critical persistence. - */ - Critical = 2 -} - -/** - * The EventPersistence contains a set of values that specify the event's persistence. - */ -export const EventPersistence = (/* @__PURE__ */ createEnumStyle({ - /** - * Normal persistence. - */ - Normal: EventPersistenceValue.Normal, - - /** - * Critical persistence. - */ - Critical: EventPersistenceValue.Critical -})); -export type EventPersistence = number | EventPersistenceValue; diff --git a/shared/AppInsightsCommon/src/HelperFuncs.ts b/shared/AppInsightsCommon/src/HelperFuncs.ts deleted file mode 100644 index ddd212fea..000000000 --- a/shared/AppInsightsCommon/src/HelperFuncs.ts +++ /dev/null @@ -1,55 +0,0 @@ -// Copyright (c) Microsoft Corporation. All rights reserved. -// Licensed under the MIT License. - -import { IPlugin, arrForEach, isString } from "@microsoft/applicationinsights-core-js"; -import { mathFloor, mathRound } from "@nevware21/ts-utils"; - -const strEmpty = ""; - -export function stringToBoolOrDefault(str: any, defaultValue = false): boolean { - if (str === undefined || str === null) { - return defaultValue; - } - - return str.toString().toLowerCase() === "true"; -} - -/** - * Convert ms to c# time span format - */ -export function msToTimeSpan(totalms: number): string { - if (isNaN(totalms) || totalms < 0) { - totalms = 0; - } - - totalms = mathRound(totalms); - - let ms = strEmpty + totalms % 1000; - let sec = strEmpty + mathFloor(totalms / 1000) % 60; - let min = strEmpty + mathFloor(totalms / (1000 * 60)) % 60; - let hour = strEmpty + mathFloor(totalms / (1000 * 60 * 60)) % 24; - const days = mathFloor(totalms / (1000 * 60 * 60 * 24)); - - ms = ms.length === 1 ? "00" + ms : ms.length === 2 ? "0" + ms : ms; - sec = sec.length < 2 ? "0" + sec : sec; - min = min.length < 2 ? "0" + min : min; - hour = hour.length < 2 ? "0" + hour : hour; - - return (days > 0 ? days + "." : strEmpty) + hour + ":" + min + ":" + sec + "." + ms; -} - -export function getExtensionByName(extensions: IPlugin[], identifier: string): IPlugin | null { - let extension: IPlugin = null; - arrForEach(extensions, (value) => { - if (value.identifier === identifier) { - extension = value; - return -1; - } - }); - - return extension; -} - -export function isCrossOriginError(message: string|Event, url: string, lineNumber: number, columnNumber: number, error: Error | Event): boolean { - return !error && isString(message) && (message === "Script error." || message === "Script error"); -} diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IEnvelope.ts b/shared/AppInsightsCommon/src/Interfaces/Contracts/IEnvelope.ts deleted file mode 100644 index 916f65649..000000000 --- a/shared/AppInsightsCommon/src/Interfaces/Contracts/IEnvelope.ts +++ /dev/null @@ -1,50 +0,0 @@ -// // Copyright (c) Microsoft Corporation. All rights reserved. -// // Licensed under the MIT License. - -// import { IBase } from "./IBase"; - -// /** -// * System variables for a telemetry item. -// */ -// export interface IEnvelope { - -// /** -// * Envelope version. For internal use only. By assigning this the default, it will not be serialized within the payload unless changed to a value other than #1. -// */ -// ver: number; /* 1 */ - -// /** -// * Type name of telemetry data item. -// */ -// name: string; - -// /** -// * Event date time when telemetry item was created. This is the wall clock time on the client when the event was generated. There is no guarantee that the client's time is accurate. This field must be formatted in UTC ISO 8601 format, with a trailing 'Z' character, as described publicly on https://en.wikipedia.org/wiki/ISO_8601#UTC. Note: the number of decimal seconds digits provided are variable (and unspecified). Consumers should handle this, i.e. managed code consumers should not use format 'O' for parsing as it specifies a fixed length. Example: 2009-06-15T13:45:30.0000000Z. -// */ -// time: string; - -// /** -// * Sampling rate used in application. This telemetry item represents 1 / sampleRate actual telemetry items. -// */ -// sampleRate: number; /* 100.0 */ - -// /** -// * Sequence field used to track absolute order of uploaded events. -// */ -// seq: string; - -// /** -// * The application's instrumentation key. The key is typically represented as a GUID, but there are cases when it is not a guid. No code should rely on iKey being a GUID. Instrumentation key is case insensitive. -// */ -// iKey: string; - -// /** -// * Key/value collection of context properties. See ContextTagKeys for information on available properties. -// */ -// tags: any; /* {} */ - -// /** -// * Telemetry data item. -// */ -// data: IBase; -// } diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/RequestData.ts b/shared/AppInsightsCommon/src/Interfaces/Contracts/RequestData.ts deleted file mode 100644 index 9b37549d4..000000000 --- a/shared/AppInsightsCommon/src/Interfaces/Contracts/RequestData.ts +++ /dev/null @@ -1,50 +0,0 @@ -// // Copyright (c) Microsoft Corporation. All rights reserved. -// // Licensed under the MIT License. - -// import { IDomain } from "./IDomain"; - -// /** -// * An instance of Request represents completion of an external request to the application to do work and contains a summary of that request execution and the results. -// */ -// export class RequestData implements IDomain { - -// /** -// * Schema version -// */ -// public ver: number = 2; - -// /** -// * Identifier of a request call instance. Used for correlation between request and other telemetry items. -// */ -// public id: string; - -// /** -// * Source of the request. Examples are the instrumentation key of the caller or the ip address of the caller. -// */ -// public source: string; - -// /** -// * Name of the request. Represents code path taken to process request. Low cardinality value to allow better grouping of requests. For HTTP requests it represents the HTTP method and URL path template like 'GET /values/{id}'. -// */ -// public name: string; - -// /** -// * Indication of successful or unsuccessful call. -// */ -// public success: boolean; - -// /** -// * Request URL with all query string parameters. -// */ -// public url: string; - -// /** -// * Collection of custom properties. -// */ -// public properties: any = {}; - -// /** -// * Collection of custom measurements. -// */ -// public measurements: any = {}; -// } diff --git a/shared/AppInsightsCommon/src/Interfaces/ICorrelationConfig.ts b/shared/AppInsightsCommon/src/Interfaces/ICorrelationConfig.ts deleted file mode 100644 index 1f1de850c..000000000 --- a/shared/AppInsightsCommon/src/Interfaces/ICorrelationConfig.ts +++ /dev/null @@ -1,100 +0,0 @@ -// Copyright (c) Microsoft Corporation. All rights reserved. -// Licensed under the MIT License. - -import { ICustomProperties } from "@microsoft/applicationinsights-core-js"; -import { DistributedTracingModes } from "../Enums"; -import { IRequestContext } from "./IRequestContext"; - -export interface ICorrelationConfig { - enableCorsCorrelation: boolean; - correlationHeaderExcludedDomains: string[]; - correlationHeaderExcludePatterns?: RegExp[]; - disableCorrelationHeaders: boolean; - distributedTracingMode: DistributedTracingModes; - maxAjaxCallsPerView: number; - disableAjaxTracking: boolean; - disableFetchTracking: boolean; - appId?: string; - enableRequestHeaderTracking?: boolean; - enableResponseHeaderTracking?: boolean; - enableAjaxErrorStatusText?: boolean; - - /** - * Flag to enable looking up and including additional browser window.performance timings - * in the reported ajax (XHR and fetch) reported metrics. - * Defaults to false. - */ - enableAjaxPerfTracking?: boolean; - - /** - * The maximum number of times to look for the window.performance timings (if available), this - * is required as not all browsers populate the window.performance before reporting the - * end of the XHR request and for fetch requests this is added after its complete - * Defaults to 3 - */ - maxAjaxPerfLookupAttempts?: number; - - /** - * The amount of time to wait before re-attempting to find the windows.performance timings - * for an ajax request, time is in milliseconds and is passed directly to setTimeout() - * Defaults to 25. - */ - ajaxPerfLookupDelay?: number; - - correlationHeaderDomains?: string[]; - - /** - * [Optional] Response and request headers to be excluded from AJAX & Fetch tracking data. - * To override or discard the default, add an array with all headers to be excluded or - * an empty array to the configuration. - * - * For example: `["Authorization", "X-API-Key", "WWW-Authenticate"]` - * - * @example - * ```js - * import { ApplicationInsights } from '@microsoft/applicationinsights-web'; - * import { AjaxPlugin } from '@microsoft/applicationinsights-dependencies-js'; - * - * const dependencyPlugin = new AjaxPlugin(); - * const appInsights = new ApplicationInsights({ - * config: { - * connectionString: 'InstrumentationKey=YOUR_INSTRUMENTATION_KEY_GOES_HERE', - * extensions: [dependencyPlugin], - * extensionConfig: { - * [dependencyPlugin.identifier]: { - * ignoreHeaders: [ - * "Authorization", - * "X-API-Key", - * "WWW-Authenticate" - * ] - * } - * } - * } - * }); - * appInsights.loadAppInsights(); - * appInsights.trackPageView(); // Manually call trackPageView to establish the current user/session/pageview - * ``` - */ - ignoreHeaders?: string[]; - - /** - * Provide a way to exclude specific route from automatic tracking for XMLHttpRequest or Fetch request. - * For an ajax / fetch request that the request url matches with the regex patterns, auto tracking is turned off. - * Default is undefined. - */ - excludeRequestFromAutoTrackingPatterns?: string[] | RegExp[]; - - /** - * Provide a way to enrich dependencies logs with context at the beginning of api call. - * Default is undefined. - */ - addRequestContext?: (requestContext?: IRequestContext) => ICustomProperties; - - /** - * [Optional] Flag to indicate whether the internal looking endpoints should be automatically - * added to the `excludeRequestFromAutoTrackingPatterns` collection. (defaults to true). - * This flag exists as the provided regex is generic and may unexpectedly match a domain that - * should not be excluded. - */ - addIntEndpoints?: boolean; -} \ No newline at end of file diff --git a/shared/AppInsightsCommon/src/Interfaces/PartAExtensions.ts b/shared/AppInsightsCommon/src/Interfaces/PartAExtensions.ts deleted file mode 100644 index 1aa523099..000000000 --- a/shared/AppInsightsCommon/src/Interfaces/PartAExtensions.ts +++ /dev/null @@ -1,14 +0,0 @@ -import { ContextTagKeys } from "./Contracts/ContextTagKeys"; - -export const Extensions = { - UserExt: "user", - DeviceExt: "device", - TraceExt: "trace", - WebExt: "web", - AppExt: "app", - OSExt: "os", - SessionExt: "ses", - SDKExt: "sdk" -}; - -export let CtxTagKeys = new ContextTagKeys(); \ No newline at end of file diff --git a/shared/AppInsightsCommon/src/applicationinsights-common.ts b/shared/AppInsightsCommon/src/applicationinsights-common.ts index 958c027c3..36be30b15 100644 --- a/shared/AppInsightsCommon/src/applicationinsights-common.ts +++ b/shared/AppInsightsCommon/src/applicationinsights-common.ts @@ -1,80 +1,125 @@ // Copyright (c) Microsoft Corporation. All rights reserved. -// Licensed under the -export { - correlationIdSetPrefix, correlationIdGetPrefix, correlationIdCanIncludeCorrelationHeader, correlationIdGetCorrelationContext, - correlationIdGetCorrelationContextValue, dateTimeUtilsNow, dateTimeUtilsDuration, isInternalApplicationInsightsEndpoint, - createDistributedTraceContextFromTrace -} from "./Util"; -export { ThrottleMgr } from "./ThrottleMgr"; -export { parseConnectionString, ConnectionStringParser } from "./ConnectionStringParser"; -export { ConnectionString } from "./Interfaces/ConnectionString"; -export { FieldType } from "./Enums"; -export { IRequestHeaders, RequestHeaders, eRequestHeaders } from "./RequestResponseHeaders"; -export { DisabledPropertyName, ProcessLegacy, SampleRate, HttpMethod, DEFAULT_BREEZE_ENDPOINT, DEFAULT_BREEZE_PATH, strNotSpecified } from "./Constants"; -export { IData as AIData } from "./Interfaces/Contracts/IData"; -export { IBase as AIBase } from "./Interfaces/Contracts/IBase"; -export { ISerializable } from "./Interfaces/Telemetry/ISerializable"; -export { IEnvelope } from "./Interfaces/Telemetry/IEnvelope"; -export { Envelope } from "./Telemetry/Common/Envelope"; -export { Event } from "./Telemetry/Event"; -export { Exception } from "./Telemetry/Exception"; -export { Metric } from "./Telemetry/Metric"; -export { PageView } from "./Telemetry/PageView"; -export { IPageViewData } from "./Interfaces/Contracts/IPageViewData"; -export { RemoteDependencyData } from "./Telemetry/RemoteDependencyData"; -export { IEventTelemetry } from "./Interfaces/IEventTelemetry"; -export { ITraceTelemetry } from "./Interfaces/ITraceTelemetry"; -export { IMetricTelemetry } from "./Interfaces/IMetricTelemetry"; -export { IDependencyTelemetry } from "./Interfaces/IDependencyTelemetry"; -export { IExceptionTelemetry, IAutoExceptionTelemetry, IExceptionInternal } from "./Interfaces/IExceptionTelemetry"; -export { IPageViewTelemetry, IPageViewTelemetryInternal } from "./Interfaces/IPageViewTelemetry"; -export { IPageViewPerformanceTelemetry, IPageViewPerformanceTelemetryInternal } from "./Interfaces/IPageViewPerformanceTelemetry"; -export { Trace } from "./Telemetry/Trace"; -export { PageViewPerformance } from "./Telemetry/PageViewPerformance"; -export { Data } from "./Telemetry/Common/Data"; -export { eSeverityLevel, SeverityLevel } from "./Interfaces/Contracts/SeverityLevel"; -export { IConfig, ConfigurationManager } from "./Interfaces/IConfig"; -export { IStorageBuffer } from "./Interfaces/IStorageBuffer"; -export { IContextTagKeys, ContextTagKeys } from "./Interfaces/Contracts/ContextTagKeys"; +// Licensed under the MIT License. + +/** + * @deprecated The @microsoft/applicationinsights-common package has been merged + * into @microsoft/applicationinsights-core-js for simplified dependency management + * and improved tree-shaking capabilities. + * + * Please migrate your imports from "@microsoft/applicationinsights-common" to + * "@microsoft/applicationinsights-core-js". + * + * This compatibility layer will be maintained through version 3.x and removed in 4.0.0. + */ + +// Re-export everything from core for backward compatibility export { + // Utility functions + correlationIdSetPrefix, correlationIdGetPrefix, correlationIdCanIncludeCorrelationHeader, + correlationIdGetCorrelationContext, correlationIdGetCorrelationContextValue, + dateTimeUtilsNow, dateTimeUtilsDuration, isInternalApplicationInsightsEndpoint, + createDistributedTraceContextFromTrace, + + // Throttle manager + ThrottleMgr, + + // Connection string parsing + parseConnectionString, ConnectionStringParser, + + // Enums + FieldType, eDistributedTracingModes, DistributedTracingModes, EventPersistence, + eSeverityLevel, SeverityLevel, + + // Request/Response + eRequestHeaders, RequestHeaders, + + // Constants + DisabledPropertyName, ProcessLegacy, SampleRate, HttpMethod, + DEFAULT_BREEZE_ENDPOINT, DEFAULT_BREEZE_PATH, strNotSpecified, + + // Contracts (as AIData and AIBase for backward compatibility) + AIData, AIBase, + + // Telemetry interfaces + ISerializable, IEnvelope, IPageViewData, IRemoteDependencyData, + IEventTelemetry, ITraceTelemetry, IMetricTelemetry, IDependencyTelemetry, + IExceptionTelemetry, IAutoExceptionTelemetry, IExceptionInternal, + IPageViewTelemetry, IPageViewTelemetryInternal, + IPageViewPerformanceTelemetry, IPageViewPerformanceTelemetryInternal, + IRequestTelemetry, + + // Telemetry classes + Envelope, Event, Exception, Metric, PageView, PageViewPerformance, + RemoteDependencyData, Trace, Data, + + // Data Sanitizer DataSanitizerValues, - dataSanitizeKeyAndAddUniqueness, dataSanitizeKey, dataSanitizeString, dataSanitizeUrl, dataSanitizeMessage, - dataSanitizeException, dataSanitizeProperties, dataSanitizeMeasurements, dataSanitizeId, dataSanitizeInput, - dsPadNumber -} from "./Telemetry/Common/DataSanitizer"; -export { TelemetryItemCreator, createTelemetryItem } from "./TelemetryItemCreator"; -export { ICorrelationConfig } from "./Interfaces/ICorrelationConfig"; -export { IAppInsights } from "./Interfaces/IAppInsights"; -export { IWeb } from "./Interfaces/Context/IWeb"; -export { CtxTagKeys, Extensions } from "./Interfaces/PartAExtensions"; -export { ISession } from "./Interfaces/Context/ISession"; -export { ISessionManager } from "./Interfaces/Context/ISessionManager"; -export { ITelemetryContext } from "./Interfaces/ITelemetryContext"; -export { IApplication } from "./Interfaces/Context/IApplication"; -export { IDevice } from "./Interfaces/Context/IDevice"; -export { IInternal } from "./Interfaces/Context/IInternal"; -export { ILocation } from "./Interfaces/Context/ILocation"; -export { ISample } from "./Interfaces/Context/ISample"; -export { IOperatingSystem } from "./Interfaces/Context/IOperatingSystem"; -export { IPropertiesPlugin } from "./Interfaces/IPropertiesPlugin"; -export { IUser, IUserContext } from "./Interfaces/Context/IUser"; -export { ITelemetryTrace, ITraceState } from "./Interfaces/Context/ITelemetryTrace"; -export { IRequestContext } from "./Interfaces/IRequestContext"; -export { eDistributedTracingModes, DistributedTracingModes, EventPersistence } from "./Enums"; -export { stringToBoolOrDefault, msToTimeSpan, getExtensionByName, isCrossOriginError } from "./HelperFuncs"; -export { - isBeaconsSupported as isBeaconApiSupported, - ITraceParent, createTraceParent, parseTraceParent, isValidTraceId, isValidSpanId, isValidTraceParent, isSampledFlag, formatTraceParent, findW3cTraceParent, findAllScripts + dataSanitizeKeyAndAddUniqueness, dataSanitizeKey, dataSanitizeString, + dataSanitizeUrl, dataSanitizeMessage, dataSanitizeException, + dataSanitizeProperties, dataSanitizeMeasurements, dataSanitizeId, + dataSanitizeInput, dsPadNumber, + + // Configuration + IConfig, ConfigurationManager, IStorageBuffer, ICorrelationConfig, + + // Context tags + IContextTagKeys, ContextTagKeys, CtxTagKeys, Extensions, + + // Data types + EventDataType, ExceptionDataType, MetricDataType, PageViewDataType, + PageViewPerformanceDataType, RemoteDependencyDataType, RequestDataType, TraceDataType, + + // Envelope types + EventEnvelopeType, ExceptionEnvelopeType, MetricEnvelopeType, PageViewEnvelopeType, + PageViewPerformanceEnvelopeType, RemoteDependencyEnvelopeType, RequestEnvelopeType, + TraceEnvelopeType, + + // Telemetry creation + TelemetryItemCreator, createTelemetryItem, + + // Application Insights interfaces + IAppInsights, ITelemetryContext, IPropertiesPlugin, IRequestContext, + + // Context interfaces + IWeb, ISession, ISessionManager, IApplication, IDevice, IInternal, + ILocation, ISample, IOperatingSystem, IUser, IUserContext, ITelemetryTrace, + + // Helper functions + stringToBoolOrDefault, msToTimeSpan, getExtensionByName, isCrossOriginError, + + // Trace parent functions (re-exported from core) + createTraceParent, parseTraceParent, isValidTraceId, isValidSpanId, + isValidTraceParent, isSampledFlag, formatTraceParent, findW3cTraceParent, + findAllScripts, isBeaconsSupported as isBeaconApiSupported, + + // DOM helpers + createDomEvent, + + // Storage helpers + utlDisableStorage, utlEnableStorage, utlCanUseLocalStorage, utlGetLocalStorage, + utlSetLocalStorage, utlRemoveStorage, utlCanUseSessionStorage, + utlGetSessionStorageKeys, utlGetSessionStorage, utlSetSessionStorage, + utlRemoveSessionStorage, utlSetStoragePrefix, + + // URL helpers + urlParseUrl, urlGetAbsoluteUrl, urlGetPathName, urlGetCompleteUrl, + urlParseHost, urlParseFullHost, + + // Throttle manager interfaces + IThrottleLimit, IThrottleInterval, IThrottleMgrConfig, + IThrottleLocalStorageObj, IThrottleResult, + + // Offline support + IOfflineListener, createOfflineListener, IOfflineState, + eOfflineValue, OfflineCallback, + + // ITraceParent interface + ITraceParent, + + // Plugin identifiers + PropertiesPluginIdentifier, BreezeChannelIdentifier, AnalyticsPluginIdentifier + } from "@microsoft/applicationinsights-core-js"; -export { createDomEvent } from "./DomHelperFuncs"; -export { - utlDisableStorage, utlEnableStorage, utlCanUseLocalStorage, utlGetLocalStorage, utlSetLocalStorage, utlRemoveStorage, - utlCanUseSessionStorage, utlGetSessionStorageKeys, utlGetSessionStorage, utlSetSessionStorage, utlRemoveSessionStorage, utlSetStoragePrefix -} from "./StorageHelperFuncs"; -export { urlParseUrl, urlGetAbsoluteUrl, urlGetPathName, urlGetCompleteUrl, urlParseHost, urlParseFullHost } from "./UrlHelperFuncs"; -export { IThrottleLimit, IThrottleInterval, IThrottleMgrConfig, IThrottleLocalStorageObj, IThrottleResult } from "./Interfaces/IThrottleMgr"; -export { IOfflineListener, createOfflineListener, IOfflineState, eOfflineValue, OfflineCallback } from "./Offline"; -export const PropertiesPluginIdentifier = "AppInsightsPropertiesPlugin"; -export const BreezeChannelIdentifier = "AppInsightsChannelPlugin"; -export const AnalyticsPluginIdentifier = "ApplicationInsightsAnalytics"; +// Type re-exports (zero runtime cost) +export type { ConnectionString, IRequestHeaders } from "@microsoft/applicationinsights-core-js"; diff --git a/shared/AppInsightsCore/README.md b/shared/AppInsightsCore/README.md index c58c205db..3b82afafb 100644 --- a/shared/AppInsightsCore/README.md +++ b/shared/AppInsightsCore/README.md @@ -8,6 +8,17 @@ Core SDK layer for next version of application insights javascript SDK. +## V3.4.0 - Common Package Merged + +As of version 3.4.0, the functionality from `@microsoft/applicationinsights-common` has been merged into this package (`@microsoft/applicationinsights-core-js`). This consolidation simplifies the dependency tree and improves tree-shaking capabilities. + +**What this means:** +- All exports previously in `applicationinsights-common` are now available from `applicationinsights-core-js` +- The `applicationinsights-common` package continues to work as a compatibility layer (re-exports from Core) +- No breaking changes to existing APIs + +See the [Migration Guide](../../docs/upgrade/MergeCommonToCore.md) for details on updating your imports. + ## V3.x Release Breaking changes - Removed ES3 / IE8 Support diff --git a/shared/AppInsightsCore/SpanImplementationSummary.md b/shared/AppInsightsCore/SpanImplementationSummary.md new file mode 100644 index 000000000..3e5815b1d --- /dev/null +++ b/shared/AppInsightsCore/SpanImplementationSummary.md @@ -0,0 +1,100 @@ +# Span Implementation Summary + +## What Was Implemented + +Successfully implemented an OpenTelemetry-like span functionality in ApplicationInsights using a provider pattern architecture. + +## Key Components Created + +### Core Package (AppInsightsCore) + +1. **Interfaces**: + - `IOTelSpan`: OpenTelemetry-like span interface (simplified) + - `IOTelSpanContext`: Span context interface + - `ITraceProvider`: Provider interface for span creation + - `SpanOptions`: Configuration options for spans + +2. **Core Integration**: + - Extended `IAppInsightsCore` with provider management methods + - Updated `AppInsightsCore` implementation to use provider pattern + - Added utility functions for span context creation + +3. **Utilities**: + - `createOTelSpanContext()`: Creates span contexts + - `isSpanContext()`: Type guard for span contexts + - `wrapDistributedTrace()`: Wraps distributed trace contexts + +### Web Package (AISKU) + +1. **Concrete Implementation**: + - `AppInsightsSpan`: ApplicationInsights-specific span implementation + - `AppInsightsTraceProvider`: Provider that creates ApplicationInsights spans + - `createSpan()`: Factory function for creating spans + +2. **Integration**: + - Exported span implementation from web package + - Provider can be registered with core SDK + +## Architecture Benefits + +### Provider Pattern Advantages + +1. **Separation of Concerns**: Core manages lifecycle, providers handle creation +2. **Flexibility**: Different SKUs can provide their own implementations +3. **Extensibility**: Easy to add new span providers for different scenarios +4. **Type Safety**: Full TypeScript support with proper interfaces + +### OpenTelemetry Compatibility + +1. **Familiar API**: Similar to OpenTelemetry span interface +2. **Standard Patterns**: Uses established tracing concepts +3. **Future-Proof**: Easy to extend with more OpenTelemetry features + +## Usage Flow + +1. **Setup**: Register a trace provider with the core SDK +2. **Creation**: Use `appInsightsCore.startSpan()` to create spans +3. **Management**: Core delegates to provider for span creation +4. **Lifecycle**: Spans follow standard OpenTelemetry patterns + +## Removed Features + +As requested, the following methods were removed from the span interface: +- `addEvent()`: Event recording functionality +- `setStatus()`: Status setting functionality + +This keeps the implementation focused on ApplicationInsights core needs. + +## Files Modified/Created + +### Core Package +- `shared/AppInsightsCore/src/OpenTelemetry/interfaces/trace/ITraceProvider.ts` (new) +- `shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IAppInsightsCore.ts` (modified) +- `shared/AppInsightsCore/src/JavaScriptSDK/AppInsightsCore.ts` (modified) +- `shared/AppInsightsCore/src/applicationinsights-core-js.ts` (modified - exports) + +### Web Package +- `AISKU/src/OpenTelemetry/trace/span.ts` (new) +- `AISKU/src/OpenTelemetry/trace/AppInsightsTraceProvider.ts` (new) +- `AISKU/src/applicationinsights-web.ts` (modified - exports) + +### Documentation +- `AISKU/SpanImplementation.md` (new) + +## Testing + +Both packages build successfully: +- ✅ Core package compiles and exports correctly +- ✅ Web package compiles with span implementation +- ✅ Provider pattern works as designed +- ✅ TypeScript types are properly exported + +## Next Steps + +1. **Testing**: Add unit tests for span implementation +2. **Integration**: Integrate with ApplicationInsights telemetry pipeline +3. **Documentation**: Add API documentation +4. **Examples**: Create more usage examples +5. **Other SKUs**: Implement providers for other ApplicationInsights packages + +The implementation is complete and follows the requested architecture of having spans in the web package managed by the core through a provider pattern similar to OpenTelemetry's TracerProvider. diff --git a/shared/AppInsightsCore/Span_Implementation_Refactoring_Summary.md b/shared/AppInsightsCore/Span_Implementation_Refactoring_Summary.md new file mode 100644 index 000000000..06a123359 --- /dev/null +++ b/shared/AppInsightsCore/Span_Implementation_Refactoring_Summary.md @@ -0,0 +1,126 @@ +# Span Implementation Refactoring Summary + +## Changes Implemented + +### 1. ✅ Removed AppInsightsTraceProvider Class +- **File**: `AISKU/src/OpenTelemetry/trace/AppInsightsTraceProvider.ts` +- **Change**: Replaced the `AppInsightsTraceProvider` class with a factory function `createTraceProvider()` +- **Reason**: The main AISKU should internally implement this during initialization rather than exporting a class + +### 2. ✅ Removed AppInsightsSpan Class +- **File**: `AISKU/src/OpenTelemetry/trace/span.ts` +- **Change**: + - Removed the `AppInsightsSpan` class + - Implemented span functionality inline in the `createSpan` factory function using closure-based approach + - Added `SpanEndCallback` type for handling span end events +- **Reason**: Eliminates class overhead and makes the implementation more internal + +### 3. ✅ Made createSpan Internal-Only +- **File**: `AISKU/src/OpenTelemetry/trace/span.ts` +- **Change**: + - Made `createSpan` function internal (not exported from package) + - Added `onSpanEnd` callback parameter to support trace provider integration + - Span `end()` function now calls back into the provided callback function +- **Reason**: Factory function should be internal and support telemetry event creation + +### 4. ✅ Updated Package Exports +- **File**: `AISKU/src/applicationinsights-web.ts` +- **Change**: Removed exports for `AppInsightsSpan`, `createSpan`, and `AppInsightsTraceProvider` +- **Reason**: These should be internal-only implementations + +### 5. ✅ Integrated Trace Provider in AISKU Initialization +- **File**: `AISKU/src/AISku.ts` +- **Change**: + - Added trace provider setup in the `loadAppInsights` initialization function + - Created span end callback that converts span data to telemetry using `trackTrace` + - Set the trace provider on the core using `_core.setTraceProvider()` +- **Reason**: Main AISKU should internally implement and configure the trace provider + +## Technical Implementation Details + +### Inline Span Implementation +The span is now implemented using closures instead of a class: + +```typescript +function createSpan(name, parent, options, onSpanEnd) { + // Private variables in closure + let _spanContext = spanContext; + let _attributes = {}; + let _name = name; + let _ended = false; + + // Return span implementation object + const span = { + spanContext: () => _spanContext, + setAttribute: (key, value) => { /* implementation */ }, + setAttributes: (attributes) => { /* implementation */ }, + updateName: (newName) => { /* implementation */ }, + end: (endTime) => { + if (!_ended) { + _ended = true; + _endTime = endTime || utcNow(); + + // Call the end callback if provided + if (onSpanEnd) { + onSpanEnd(span, _endTime); + } + } + }, + isRecording: () => !_ended + }; + + return span; +} +``` + +### Span End Callback Integration +When a span ends, it calls back to the AISKU which creates telemetry: + +```typescript +const traceProvider = createTraceProvider((span, endTime) => { + const spanData = span as any; // Access helper methods + if (spanData.getName && spanData.getAttributes) { + const name = spanData.getName(); + const attributes = spanData.getAttributes(); + + // Create trace telemetry for the span + _self.trackTrace({ + message: `Span: ${name}`, + severityLevel: 1, + properties: { + ...attributes, + spanId: span.spanContext().spanId, + traceId: span.spanContext().traceId, + startTime: spanData.getStartTime().toString(), + endTime: endTime.toString(), + duration: (endTime - spanData.getStartTime()).toString() + } + }); + } +}); +``` + +## Benefits + +1. **Simplified Architecture**: No longer exposes internal span classes +2. **Better Encapsulation**: Span implementation is truly internal to the SDK +3. **Automatic Telemetry**: Spans automatically generate telemetry events when they end +4. **Memory Efficient**: Closure-based implementation reduces object overhead +5. **Cleaner API**: Only the necessary interfaces are exposed publicly + +## Files Modified + +- ✅ `AISKU/src/OpenTelemetry/trace/AppInsightsTraceProvider.ts` - Converted to factory function +- ✅ `AISKU/src/OpenTelemetry/trace/span.ts` - Inline implementation with callback support +- ✅ `AISKU/src/applicationinsights-web.ts` - Removed exports +- ✅ `AISKU/src/AISku.ts` - Added trace provider integration +- ✅ `shared/AppInsightsCore/src/*` - Previous IOTelSpan interface reversion (completed) + +## Status + +✅ **COMPLETED** - All requested changes have been implemented: +- AppInsightsTraceProvider class removed and replaced with factory function +- AppInsightsSpan class removed and implemented inline in createSpan factory +- createSpan factory function is internal-only and includes callback for telemetry +- AISKU internally sets up trace provider during initialization +- Span end events automatically create trace telemetry through callback mechanism diff --git a/shared/AppInsightsCore/Tests/Unit/src/GlobalTestHooks.Test.ts b/shared/AppInsightsCore/Tests/Unit/src/GlobalTestHooks.Test.ts deleted file mode 100644 index 99e370c4a..000000000 --- a/shared/AppInsightsCore/Tests/Unit/src/GlobalTestHooks.Test.ts +++ /dev/null @@ -1,13 +0,0 @@ -import { Assert } from "@microsoft/ai-test-framework"; -import { _testHookMaxUnloadHooksCb } from "../../../src/JavaScriptSDK/UnloadHookContainer"; -import { dumpObj } from "@nevware21/ts-utils"; - -export class GlobalTestHooks { - - public registerTests() { - // Set a global maximum - _testHookMaxUnloadHooksCb(20, (state: string, hooks: Array) => { - Assert.ok(false, "Max unload hooks exceeded [" + hooks.length + "] - " + state + " - " + dumpObj(hooks)); - }); - } -} diff --git a/shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/commonUtils.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/commonUtils.Tests.ts new file mode 100644 index 000000000..92cf3bbad --- /dev/null +++ b/shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/commonUtils.Tests.ts @@ -0,0 +1,1031 @@ +import { Assert, AITestClass } from "@microsoft/ai-test-framework"; +import { getGlobal } from "@nevware21/ts-utils"; +import { + handleAttribError, handleSpanError, handleDebug, handleWarn, handleError, handleNotImplemented +} from "../../../../src/internal/handleErrors"; +import { getUrl, getHttpUrl } from "../../../../src/internal/commonUtils"; +import { createAttributeContainer } from "../../../../src/otel/attribute/attributeContainer"; +import { IOTelErrorHandlers } from "../../../../src/interfaces/otel/config/IOTelErrorHandlers"; +import { IOTelConfig } from "../../../../src/interfaces/otel/config/IOTelConfig"; + +export class CommonUtilsTests extends AITestClass { + + public testInitialize() { + super.testInitialize(); + // Reset console mocks before each test + this._resetConsoleMocks(); + } + + public testCleanup() { + super.testCleanup(); + // Reset console mocks after each test + this._resetConsoleMocks(); + } + + public registerTests() { + this.testCase({ + name: "handleAttribError: should call custom attribError handler when provided", + test: () => { + // Arrange + let calledMessage = ""; + let calledKey = ""; + let calledValue: any = null; + const handlers: IOTelErrorHandlers = { + attribError: (message: string, key: string, value: any) => { + calledMessage = message; + calledKey = key; + calledValue = value; + } + }; + const testMessage = "Test error message"; + const testKey = "testKey"; + const testValue = { test: "value" }; + + // Act + handleAttribError(handlers, testMessage, testKey, testValue); + + // Assert + Assert.ok(calledMessage === testMessage, "Message should match"); + Assert.ok(calledKey === testKey, "Key should match"); + Assert.ok(calledValue === testValue, "Value should match"); + } + }); + + this.testCase({ + name: "handleAttribError: should call handleWarn when no custom attribError handler provided", + test: () => { + // Arrange + let warnCalled = false; + let warnMessage = ""; + const handlers: IOTelErrorHandlers = { + warn: (message: string) => { + warnCalled = true; + warnMessage = message; + } + }; + const testMessage = "Test error"; + const testKey = "testKey"; + const testValue = "testValue"; + + // Act + handleAttribError(handlers, testMessage, testKey, testValue); + + // Assert + Assert.ok(warnCalled, "Warn should be called"); + Assert.ok(warnMessage.includes(testMessage), "Warn message should contain original message"); + Assert.ok(warnMessage.includes(testKey), "Warn message should contain key"); + Assert.ok(warnMessage.includes(testValue), "Warn message should contain value"); + } + }); + + this.testCase({ + name: "handleSpanError: should call custom spanError handler when provided", + test: () => { + // Arrange + let calledMessage = ""; + let calledSpanName = ""; + const handlers: IOTelErrorHandlers = { + spanError: (message: string, spanName: string) => { + calledMessage = message; + calledSpanName = spanName; + } + }; + const testMessage = "Span error occurred"; + const testSpanName = "testSpan"; + + // Act + handleSpanError(handlers, testMessage, testSpanName); + + // Assert + Assert.ok(calledMessage === testMessage, "Message should match"); + Assert.ok(calledSpanName === testSpanName, "Span name should match"); + } + }); + + this.testCase({ + name: "handleSpanError: should call handleWarn when no custom spanError handler provided", + test: () => { + // Arrange + let warnCalled = false; + let warnMessage = ""; + const handlers: IOTelErrorHandlers = { + warn: (message: string) => { + warnCalled = true; + warnMessage = message; + } + }; + const testMessage = "Span error"; + const testSpanName = "testSpan"; + + // Act + handleSpanError(handlers, testMessage, testSpanName); + + // Assert + Assert.ok(warnCalled, "Warn should be called"); + Assert.ok(warnMessage.includes(testMessage), "Warn message should contain original message"); + Assert.ok(warnMessage.includes(testSpanName), "Warn message should contain span name"); + } + }); + + this.testCase({ + name: "handleDebug: should call custom debug handler when provided", + test: () => { + // Arrange + let debugCalled = false; + let debugMessage = ""; + const handlers: IOTelErrorHandlers = { + debug: (message: string) => { + debugCalled = true; + debugMessage = message; + } + }; + const testMessage = "Debug message"; + + // Act + handleDebug(handlers, testMessage); + + // Assert + Assert.ok(debugCalled, "Debug should be called"); + Assert.ok(debugMessage === testMessage, "Debug message should match"); + } + }); + + this.testCase({ + name: "handleDebug: should use console.log when no custom debug handler provided", + test: () => { + // Arrange + const handlers: IOTelErrorHandlers = {}; + const testMessage = "Debug via console"; + let consoleCalled = false; + let consoleMessage = ""; + + // Mock console.log + const originalConsole = console; + const globalObj = (typeof window !== "undefined") ? window : (global || {}); + (globalObj as any).console = { + log: (message: string) => { + consoleCalled = true; + consoleMessage = message; + } + }; + + try { + // Act + handleDebug(handlers, testMessage); + + // Assert + Assert.ok(consoleCalled, "Console.log should be called"); + Assert.ok(consoleMessage === testMessage, "Console message should match"); + } finally { + // Restore console + (globalObj as any).console = originalConsole; + } + } + }); + + this.testCase({ + name: "handleWarn: should call custom warn handler when provided", + test: () => { + // Arrange + let warnCalled = false; + let warnMessage = ""; + const handlers: IOTelErrorHandlers = { + warn: (message: string) => { + warnCalled = true; + warnMessage = message; + } + }; + const testMessage = "Warning message"; + + // Act + handleWarn(handlers, testMessage); + + // Assert + Assert.ok(warnCalled, "Warn should be called"); + Assert.ok(warnMessage === testMessage, "Warn message should match"); + } + }); + + this.testCase({ + name: "handleWarn: should use console.warn when no custom warn handler provided", + test: () => { + // Arrange + const globalObj = getGlobal(); + const handlers: IOTelErrorHandlers = {}; + const testMessage = "Warning via console"; + let consoleCalled = false; + let consoleMessage = ""; + + // Mock console.warn + const originalConsole = console; + (globalObj as any).console = { + warn: (message: string) => { + consoleCalled = true; + consoleMessage = message; + } + }; + + try { + // Act + handleWarn(handlers, testMessage); + + // Assert + Assert.ok(consoleCalled, "Console.warn should be called"); + Assert.ok(consoleMessage === testMessage, "Console message should match"); + } finally { + // Restore console + (globalObj as any).console = originalConsole; + } + } + }); + + this.testCase({ + name: "handleWarn: should fallback to console.log when console.warn not available", + test: () => { + // Arrange + const globalObj = getGlobal(); + const handlers: IOTelErrorHandlers = {}; + const testMessage = "Warning fallback to log"; + let consoleCalled = false; + let consoleMessage = ""; + + // Mock console without warn + const originalConsole = console; + (globalObj as any).console = { + log: (message: string) => { + consoleCalled = true; + consoleMessage = message; + } + }; + + try { + // Act + handleWarn(handlers, testMessage); + + // Assert + Assert.ok(consoleCalled, "Console.log should be called as fallback"); + Assert.ok(consoleMessage === testMessage, "Console message should match"); + } finally { + // Restore console + (globalObj as any).console = originalConsole; + } + } + }); + + this.testCase({ + name: "handleError: should call custom error handler when provided", + test: () => { + // Arrange + let errorCalled = false; + let errorMessage = ""; + const handlers: IOTelErrorHandlers = { + error: (message: string) => { + errorCalled = true; + errorMessage = message; + } + }; + const testMessage = "Error message"; + + // Act + handleError(handlers, testMessage); + + // Assert + Assert.ok(errorCalled, "Error should be called"); + Assert.ok(errorMessage === testMessage, "Error message should match"); + } + }); + + this.testCase({ + name: "handleError: should fallback to warn handler when no custom error handler provided", + test: () => { + // Arrange + let warnCalled = false; + let warnMessage = ""; + const handlers: IOTelErrorHandlers = { + warn: (message: string) => { + warnCalled = true; + warnMessage = message; + } + }; + const testMessage = "Error fallback to warn"; + + // Act + handleError(handlers, testMessage); + + // Assert + Assert.ok(warnCalled, "Warn should be called as fallback"); + Assert.ok(warnMessage === testMessage, "Warn message should match"); + } + }); + + this.testCase({ + name: "handleError: should use console.error when no custom handlers provided", + test: () => { + // Arrange + const globalObj = getGlobal(); + const handlers: IOTelErrorHandlers = {}; + const testMessage = "Error via console"; + let consoleCalled = false; + let consoleMessage = ""; + + // Mock console.error + const originalConsole = console; + (globalObj as any).console = { + error: (message: string) => { + consoleCalled = true; + consoleMessage = message; + } + }; + + try { + // Act + handleError(handlers, testMessage); + + // Assert + Assert.ok(consoleCalled, "Console.error should be called"); + Assert.ok(consoleMessage === testMessage, "Console message should match"); + } finally { + // Restore console + (globalObj as any).console = originalConsole; + } + } + }); + + this.testCase({ + name: "handleError: should fallback through console methods when preferred not available", + test: () => { + // Arrange + const globalObj = getGlobal(); + const handlers: IOTelErrorHandlers = {}; + const testMessage = "Error fallback chain"; + let consoleCalled = false; + let consoleMessage = ""; + + // Mock console with only log available + const originalConsole = console; + (globalObj as any).console = { + log: (message: string) => { + consoleCalled = true; + consoleMessage = message; + } + }; + + try { + // Act + handleError(handlers, testMessage); + + // Assert + Assert.ok(consoleCalled, "Console.log should be called as final fallback"); + Assert.ok(consoleMessage === testMessage, "Console message should match"); + } finally { + // Restore console + (globalObj as any).console = originalConsole; + } + } + }); + + this.testCase({ + name: "handleNotImplemented: should call custom notImplemented handler when provided", + test: () => { + // Arrange + let notImplementedCalled = false; + let notImplementedMessage = ""; + const handlers: IOTelErrorHandlers = { + notImplemented: (message: string) => { + notImplementedCalled = true; + notImplementedMessage = message; + } + }; + const testMessage = "Not implemented feature"; + + // Act + handleNotImplemented(handlers, testMessage); + + // Assert + Assert.ok(notImplementedCalled, "NotImplemented should be called"); + Assert.ok(notImplementedMessage === testMessage, "NotImplemented message should match"); + } + }); + + this.testCase({ + name: "handleNotImplemented: should use console.error when no custom handler provided", + test: () => { + // Arrange + const globalObj = getGlobal(); + const handlers: IOTelErrorHandlers = {}; + const testMessage = "Not implemented via console"; + let consoleCalled = false; + let consoleMessage = ""; + + // Mock console.error + const originalConsole = console; + (globalObj as any).console = { + error: (message: string) => { + consoleCalled = true; + consoleMessage = message; + } + }; + + try { + // Act + handleNotImplemented(handlers, testMessage); + + // Assert + Assert.ok(consoleCalled, "Console.error should be called"); + Assert.ok(consoleMessage === testMessage, "Console message should match"); + } finally { + // Restore console + (globalObj as any).console = originalConsole; + } + } + }); + + this.testCase({ + name: "handleNotImplemented: should fallback to console.log when console.error not available", + test: () => { + // Arrange + const globalObj = getGlobal(); + const handlers: IOTelErrorHandlers = {}; + const testMessage = "Not implemented fallback"; + let consoleCalled = false; + let consoleMessage = ""; + + // Mock console with only log available + const originalConsole = console; + (globalObj as any).console = { + log: (message: string) => { + consoleCalled = true; + consoleMessage = message; + } + }; + + try { + // Act + handleNotImplemented(handlers, testMessage); + + // Assert + Assert.ok(consoleCalled, "Console.log should be called as fallback"); + Assert.ok(consoleMessage === testMessage, "Console message should match"); + } finally { + // Restore console + (globalObj as any).console = originalConsole; + } + } + }); + + this.testCase({ + name: "Error handlers should handle undefined console gracefully", + test: () => { + // Arrange + const globalObj = getGlobal(); + const handlers: IOTelErrorHandlers = {}; + const testMessage = "Test with no console"; + const originalConsole = console; + + try { + // Remove console + (globalObj as any).console = undefined; + + // Act & Assert - should not throw + handleDebug(handlers, testMessage); + handleWarn(handlers, testMessage); + handleError(handlers, testMessage); + handleNotImplemented(handlers, testMessage); + + // If we get here, no exceptions were thrown + Assert.ok(true, "All handlers should complete without throwing when console is undefined"); + } finally { + // Restore console + (globalObj as any).console = originalConsole; + } + } + }); + + this.addGetUrlTests(); + this.addGetHttpUrlTests(); + } + + private addGetHttpUrlTests(): void { + this.testCase({ + name: "getHttpUrl: should return undefined when container is null", + test: () => { + // Act + const result = getHttpUrl(null as any); + + // Assert + Assert.equal(result, undefined, "Should return undefined for null container"); + } + }); + + this.testCase({ + name: "getHttpUrl: should return undefined when container is undefined", + test: () => { + // Act + const result = getHttpUrl(undefined as any); + + // Assert + Assert.equal(result, undefined, "Should return undefined for undefined container"); + } + }); + + this.testCase({ + name: "getHttpUrl: should return value from url.full (stable semantic convention)", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("url.full", "https://example.com/api/users?id=123"); + + // Act + const result = getHttpUrl(container); + + // Assert + Assert.equal(result, "https://example.com/api/users?id=123", "Should return value from url.full"); + } + }); + + this.testCase({ + name: "getHttpUrl: should return value from http.url (legacy semantic convention)", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.url", "https://legacy.example.com/endpoint"); + + // Act + const result = getHttpUrl(container); + + // Assert + Assert.equal(result, "https://legacy.example.com/endpoint", "Should return value from http.url"); + } + }); + + this.testCase({ + name: "getHttpUrl: should prefer url.full over http.url when both present", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("url.full", "https://stable.example.com/path"); + container.set("http.url", "https://legacy.example.com/path"); + + // Act + const result = getHttpUrl(container); + + // Assert + Assert.equal(result, "https://stable.example.com/path", "Should prefer url.full over http.url"); + } + }); + + this.testCase({ + name: "getHttpUrl: should return undefined when neither attribute is present", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("url.scheme", "https"); + container.set("server.address", "example.com"); + + // Act + const result = getHttpUrl(container); + + // Assert + Assert.equal(result, undefined, "Should return undefined when neither url.full nor http.url is present"); + } + }); + + this.testCase({ + name: "getHttpUrl: should handle empty string values", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("url.full", ""); + + // Act + const result = getHttpUrl(container); + + // Assert + Assert.equal(result, undefined, "Should return empty string when url.full is empty"); + } + }); + + this.testCase({ + name: "getHttpUrl: should handle numeric values", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("url.full", 12345); + + // Act + const result = getHttpUrl(container); + + // Assert + Assert.equal(result, 12345, "Should return numeric value as-is"); + } + }); + + this.testCase({ + name: "getHttpUrl: should handle boolean values", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("url.full", true); + + // Act + const result = getHttpUrl(container); + + // Assert + Assert.equal(result, true, "Should return boolean value as-is"); + } + }); + + this.testCase({ + name: "getHttpUrl: should handle URLs with special characters", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + const urlWithSpecialChars = "https://example.com/api/search?q=hello%20world&filter=%7B%22type%22%3A%22test%22%7D"; + container.set("url.full", urlWithSpecialChars); + + // Act + const result = getHttpUrl(container); + + // Assert + Assert.equal(result, urlWithSpecialChars, "Should handle URLs with encoded special characters"); + } + }); + + this.testCase({ + name: "getHttpUrl: should handle relative URLs", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.url", "/api/users"); + + // Act + const result = getHttpUrl(container); + + // Assert + Assert.equal(result, "/api/users", "Should handle relative URLs"); + } + }); + } + + private addGetUrlTests(): void { + this.testCase({ + name: "getUrl: should return empty string when container is null", + test: () => { + // Act + const result = getUrl(null as any); + + // Assert + Assert.equal(result, "", "Should return empty string for null container"); + } + }); + + this.testCase({ + name: "getUrl: should return empty string when container is undefined", + test: () => { + // Act + const result = getUrl(undefined as any); + + // Assert + Assert.equal(result, "", "Should return empty string for undefined container"); + } + }); + + this.testCase({ + name: "getUrl: should return empty string when no httpMethod is present", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("url.full", "https://example.com/path"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "", "Should return empty string when httpMethod is missing"); + } + }); + + this.testCase({ + name: "getUrl: should return url from url.full (stable semantic convention)", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.request.method", "GET"); + container.set("url.full", "https://example.com/api/users"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "https://example.com/api/users", "Should return url from url.full"); + } + }); + + this.testCase({ + name: "getUrl: should return url from http.url (legacy semantic convention)", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.method", "POST"); + container.set("http.url", "https://api.example.com/data"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "https://api.example.com/data", "Should return url from http.url"); + } + }); + + this.testCase({ + name: "getUrl: should prefer url.full over http.url when both present", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.request.method", "GET"); + container.set("url.full", "https://stable.example.com/path"); + container.set("http.url", "https://legacy.example.com/path"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "https://stable.example.com/path", "Should prefer url.full over http.url"); + } + }); + + this.testCase({ + name: "getUrl: should construct url from httpScheme, httpHost, and httpTarget", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.request.method", "GET"); + container.set("url.scheme", "https"); + container.set("server.address", "example.com"); + container.set("url.path", "/api/users"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "https://example.com/api/users", "Should construct url from scheme, host, and path"); + } + }); + + this.testCase({ + name: "getUrl: should construct url from legacy http attributes", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.method", "POST"); + container.set("http.scheme", "http"); + container.set("http.host", "localhost:8080"); + container.set("http.target", "/api/data?id=123"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "http://localhost:8080/api/data?id=123", "Should construct url from legacy http attributes"); + } + }); + + this.testCase({ + name: "getUrl: should use url.query when url.path is not present", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.request.method", "GET"); + container.set("url.scheme", "https"); + container.set("server.address", "example.com"); + container.set("url.query", "?q=search"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "https://example.com?q=search", "Should use url.query when url.path not present"); + } + }); + + this.testCase({ + name: "getUrl: should construct url with netPeerName and netPeerPort when httpHost not present", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.method", "GET"); + container.set("http.scheme", "http"); + container.set("net.peer.name", "api.service.local"); + container.set("net.peer.port", 8080); + container.set("http.target", "/health"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "http://api.service.local:8080/health", "Should construct url with netPeerName and port"); + } + }); + + this.testCase({ + name: "getUrl: should construct url with client.address (stable) for netPeerName", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.request.method", "GET"); + container.set("url.scheme", "https"); + container.set("client.address", "service.example.com"); + container.set("client.port", 443); + container.set("url.path", "/api"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "https://service.example.com:443/api", "Should use client.address for peer name"); + } + }); + + this.testCase({ + name: "getUrl: should construct url with netPeerIp when netPeerName not present", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.method", "GET"); + container.set("http.scheme", "http"); + container.set("net.peer.ip", "192.168.1.100"); + container.set("net.peer.port", 3000); + container.set("http.target", "/endpoint"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "http://192.168.1.100:3000/endpoint", "Should construct url with IP address and port"); + } + }); + + this.testCase({ + name: "getUrl: should use network.peer.address (stable) for peer IP", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.request.method", "POST"); + container.set("url.scheme", "https"); + container.set("network.peer.address", "10.0.0.5"); + container.set("server.port", 8443); + container.set("url.path", "/data"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "https://10.0.0.5:8443/data", "Should use network.peer.address for IP"); + } + }); + + this.testCase({ + name: "getUrl: should return empty string when scheme and target present but no host/peer info", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.request.method", "GET"); + container.set("url.scheme", "https"); + container.set("url.path", "/api/users"); + // No host, no peer name, no peer IP + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "", "Should return empty string when no host information available"); + } + }); + + this.testCase({ + name: "getUrl: should return empty string when scheme present but target missing", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.request.method", "GET"); + container.set("url.scheme", "https"); + container.set("server.address", "example.com"); + // No target/path/query + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "", "Should return empty string when target is missing"); + } + }); + + this.testCase({ + name: "getUrl: should handle IPv6 addresses", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.method", "GET"); + container.set("http.scheme", "http"); + container.set("net.peer.ip", "::1"); + container.set("net.peer.port", 8080); + container.set("http.target", "/api"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "http://::1:8080/api", "Should handle IPv6 addresses"); + } + }); + + this.testCase({ + name: "getUrl: should prefer server.port over other port attributes", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.request.method", "GET"); + container.set("url.scheme", "https"); + container.set("client.address", "example.com"); + container.set("server.port", 9000); + container.set("client.port", 8000); + container.set("net.peer.port", 7000); + container.set("url.path", "/api"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "https://example.com:8000/api", "Should prefer server.port"); + } + }); + + this.testCase({ + name: "getUrl: should handle Unix socket paths in network.peer.address", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.request.method", "GET"); + container.set("url.scheme", "http"); + container.set("network.peer.address", "/tmp/my.sock"); + container.set("server.port", 80); + container.set("url.path", "/status"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "http:///tmp/my.sock:80/status", "Should handle Unix socket paths"); + } + }); + + this.testCase({ + name: "getUrl: should construct url with path containing query parameters", + test: () => { + // Arrange + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test"); + container.set("http.method", "GET"); + container.set("http.scheme", "https"); + container.set("http.host", "api.example.com"); + container.set("http.target", "/search?q=test&page=1"); + + // Act + const result = getUrl(container); + + // Assert + Assert.equal(result, "https://api.example.com/search?q=test&page=1", "Should handle target with query parameters"); + } + }); + } + + private _resetConsoleMocks() { + // Helper to reset any console mocks - implementation depends on your mocking strategy + // This is a placeholder that might need adjustment based on your test framework + } +} diff --git a/shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/errors.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/errors.Tests.ts new file mode 100644 index 000000000..50012728a --- /dev/null +++ b/shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/errors.Tests.ts @@ -0,0 +1,680 @@ +import { Assert, AITestClass } from "@microsoft/ai-test-framework"; +import { isFunction, isString, dumpObj } from "@nevware21/ts-utils"; +import { + OpenTelemetryError, + OpenTelemetryErrorConstructor, + getOpenTelemetryError, + throwOTelError +} from "../../../../src/otel/api/errors/OTelError"; +import { + OTelInvalidAttributeError, + throwOTelInvalidAttributeError +} from "../../../../src/otel/api/errors/OTelInvalidAttributeError"; +import { + OTelSpanError, + throwOTelSpanError +} from "../../../../src/otel/api/errors/OTelSpanError"; + +export class OpenTelemetryErrorsTests extends AITestClass { + + public testInitialize() { + super.testInitialize(); + } + + public testCleanup() { + super.testCleanup(); + } + + public registerTests() { + + // OTelError tests + this.testCase({ + name: "OTelError: getOpenTelemetryError should return a constructor", + test: () => { + // Act + const ErrorConstructor = getOpenTelemetryError(); + + // Assert + Assert.ok(ErrorConstructor, "Constructor should be defined"); + Assert.ok(isFunction(ErrorConstructor), "Constructor should be a function"); + } + }); + + this.testCase({ + name: "OTelError: should create error instance with message", + test: () => { + // Arrange + const ErrorConstructor = getOpenTelemetryError(); + const testMessage = "Test OpenTelemetry error"; + + // Act + const error = new ErrorConstructor(testMessage); + + // Assert + Assert.ok(error instanceof Error, "Should be an instance of Error"); + Assert.ok(error.message === testMessage, "Message should match"); + Assert.ok(error.name === "OpenTelemetryError", "Name should be OpenTelemetryError"); + } + }); + + this.testCase({ + name: "OTelError: should create error instance without message", + test: () => { + // Arrange + const ErrorConstructor = getOpenTelemetryError(); + + // Act + const error = new ErrorConstructor(); + + // Assert + Assert.ok(error instanceof Error, "Should be an instance of Error"); + Assert.ok(error.name === "OpenTelemetryError", "Name should be OpenTelemetryError"); + } + }); + + this.testCase({ + name: "OTelError: throwOTelError should throw OpenTelemetryError", + test: () => { + // Arrange + const testMessage = "Test throw error"; + let caughtError: any = null; + + // Act & Assert + try { + throwOTelError(testMessage); + Assert.ok(false, "Should have thrown an error"); + } catch (error) { + caughtError = error; + } + + Assert.ok(caughtError, "Error should have been caught"); + Assert.ok(caughtError instanceof Error, "Should be an instance of Error"); + Assert.ok(caughtError.name === "OpenTelemetryError", "Name should be OpenTelemetryError"); + Assert.ok(caughtError.message === testMessage, "Message should match"); + } + }); + + this.testCase({ + name: "OTelError: should return same constructor instance on multiple calls", + test: () => { + // Act + const constructor1 = getOpenTelemetryError(); + const constructor2 = getOpenTelemetryError(); + + // Assert + Assert.ok(constructor1 === constructor2, "Should return same constructor instance"); + } + }); + + // OTelInvalidAttributeError tests + this.testCase({ + name: "OTelInvalidAttributeError: should throw with message, attribName, and value", + test: () => { + // Arrange + const testMessage = "Invalid attribute error"; + const testAttribName = "invalidAttr"; + const testValue = { invalid: true }; + let caughtError: any = null; + + // Act & Assert + try { + throwOTelInvalidAttributeError(testMessage, testAttribName, testValue); + Assert.ok(false, "Should have thrown an error"); + } catch (error) { + caughtError = error; + } + + Assert.ok(caughtError, "Error should have been caught"); + Assert.ok(caughtError instanceof Error, "Should be an instance of Error"); + Assert.ok(caughtError.name === "OTelInvalidAttributeError", "Error name should be OTelInvalidAttributeError"); + Assert.ok(caughtError.message === testMessage, "Message should match"); + Assert.ok((caughtError as OTelInvalidAttributeError).attribName === testAttribName, "Attribute name should match"); + Assert.ok((caughtError as OTelInvalidAttributeError).value === testValue, "Attribute value should match"); + } + }); + + this.testCase({ + name: "OTelInvalidAttributeError: should handle empty parameters", + test: () => { + // Arrange + const testMessage = "Empty attribute error"; + let caughtError: any = null; + + // Act & Assert + try { + throwOTelInvalidAttributeError(testMessage, "", ""); + Assert.ok(false, "Should have thrown an error"); + } catch (error) { + caughtError = error; + } + + Assert.ok(caughtError, "Error should have been caught"); + Assert.ok(caughtError.name === "OTelInvalidAttributeError", "Error name should be OTelInvalidAttributeError"); + Assert.ok(caughtError.message === testMessage, "Message should match"); + Assert.ok((caughtError as OTelInvalidAttributeError).attribName === "", "Attribute name should be empty string"); + Assert.ok((caughtError as OTelInvalidAttributeError).value === "", "Attribute value should be empty string"); + } + }); + + this.testCase({ + name: "OTelInvalidAttributeError: should handle undefined parameters", + test: () => { + // Arrange + const testMessage = "Undefined attribute error"; + let caughtError: any = null; + + // Act & Assert + try { + throwOTelInvalidAttributeError(testMessage, undefined as any, undefined); + Assert.ok(false, "Should have thrown an error"); + } catch (error) { + caughtError = error; + } + + Assert.ok(caughtError, "Error should have been caught"); + Assert.ok(caughtError.name === "OTelInvalidAttributeError", "Error name should be OTelInvalidAttributeError"); + Assert.ok(caughtError.message === testMessage, "Message should match"); + Assert.ok((caughtError as OTelInvalidAttributeError).attribName === undefined, "Attribute name should be undefined when explicitly passed undefined"); + Assert.ok((caughtError as OTelInvalidAttributeError).value === undefined, "Attribute value should be undefined when explicitly passed undefined"); + } + }); + + this.testCase({ + name: "OTelInvalidAttributeError: should inherit from OpenTelemetryError", + test: () => { + // Arrange + const testMessage = "Inheritance test"; + const OpenTelemetryErrorConstructor = getOpenTelemetryError(); + let caughtError: any = null; + + // Act & Assert + try { + throwOTelInvalidAttributeError(testMessage, "test", "value"); + Assert.ok(false, "Should have thrown an error"); + } catch (error) { + caughtError = error; + } + + Assert.ok(caughtError instanceof Error, "Should be instance of Error"); + Assert.ok(caughtError instanceof OpenTelemetryErrorConstructor, "Should be instance of OpenTelemetryError"); + Assert.ok(caughtError.name === "OTelInvalidAttributeError", "Should have correct error name"); + } + }); + + this.testCase({ + name: "OTelInvalidAttributeError: attribName property should not conflict with Error.name", + test: () => { + // Arrange + const testMessage = "Property conflict test"; + const testAttribName = "customAttributeName"; + let caughtError: any = null; + + // Act & Assert + try { + throwOTelInvalidAttributeError(testMessage, testAttribName, "value"); + Assert.ok(false, "Should have thrown an error"); + } catch (error) { + caughtError = error; + } + + // Assert that both properties exist and have different values + Assert.ok(caughtError.name === "OTelInvalidAttributeError", "Error.name should be the error type name"); + Assert.ok((caughtError as OTelInvalidAttributeError).attribName === testAttribName, "attribName should be the custom attribute name"); + Assert.ok(caughtError.name !== (caughtError as OTelInvalidAttributeError).attribName, "Error.name and attribName should be different"); + } + }); + + this.testCase({ + name: "OTelInvalidAttributeError: should handle different value types", + test: () => { + // Test different value types + const testCases = [ + { value: null, description: "null value" }, + { value: undefined, description: "undefined value" }, + { value: 0, description: "zero number" }, + { value: false, description: "false boolean" }, + { value: [], description: "empty array" }, + { value: {}, description: "empty object" }, + { value: "string", description: "string value" }, + { value: 123, description: "number value" }, + { value: true, description: "boolean value" } + ]; + + testCases.forEach(testCase => { + let caughtError: any = null; + try { + throwOTelInvalidAttributeError("Test message", "testAttr", testCase.value); + Assert.ok(false, `Should have thrown an error for ${testCase.description}`); + } catch (error) { + caughtError = error; + } + + Assert.ok(caughtError, `Error should have been caught for ${testCase.description}`); + Assert.ok((caughtError as OTelInvalidAttributeError).value === testCase.value, + `Value should match for ${testCase.description}`); + }); + } + }); + + // OTelSpanError tests + this.testCase({ + name: "OTelSpanError: should throw with message and span name", + test: () => { + // Arrange + const testMessage = "Span error occurred"; + const testSpanName = "test-span"; + let caughtError: any = null; + + // Act & Assert + try { + throwOTelSpanError(testMessage, testSpanName); + Assert.ok(false, "Should have thrown an error"); + } catch (error) { + caughtError = error; + } + + Assert.ok(caughtError, "Error should have been caught"); + Assert.ok(caughtError instanceof Error, "Should be an instance of Error"); + Assert.ok(caughtError.name === "OTelSpanError", "Error name should be OTelSpanError"); + Assert.ok(caughtError.message === testMessage, "Message should match"); + Assert.ok((caughtError as OTelSpanError).spanName === testSpanName, "Span name should match"); + } + }); + + this.testCase({ + name: "OTelSpanError: should handle empty span name", + test: () => { + // Arrange + const testMessage = "Span error with empty name"; + const testSpanName = ""; + let caughtError: any = null; + + // Act & Assert + try { + throwOTelSpanError(testMessage, testSpanName); + Assert.ok(false, "Should have thrown an error"); + } catch (error) { + caughtError = error; + } + + Assert.ok(caughtError, "Error should have been caught"); + Assert.ok(caughtError.message === testMessage, "Message should match"); + Assert.ok((caughtError as OTelSpanError).spanName === testSpanName, "Span name should be empty string"); + } + }); + + this.testCase({ + name: "OTelSpanError: should handle undefined span name", + test: () => { + // Arrange + const testMessage = "Span error with undefined name"; + let caughtError: any = null; + + // Act & Assert + try { + throwOTelSpanError(testMessage, undefined as any); + Assert.ok(false, "Should have thrown an error"); + } catch (error) { + caughtError = error; + } + + Assert.ok(caughtError, "Error should have been caught"); + Assert.ok(caughtError.message === testMessage, "Message should match"); + Assert.ok((caughtError as OTelSpanError).spanName === undefined, "Span name should be undefined when explicitly passed undefined"); + } + }); + + this.testCase({ + name: "OTelSpanError: should handle various span name types", + test: () => { + // Test different span name values + const testCases = [ + { spanName: "", expected: "", description: "empty string" }, + { spanName: "valid-span-name", expected: "valid-span-name", description: "valid span name" }, + { spanName: "span with spaces", expected: "span with spaces", description: "span name with spaces" }, + { spanName: "span-with-special-chars!@#", expected: "span-with-special-chars!@#", description: "span name with special characters" } + ]; + + testCases.forEach(testCase => { + let caughtError: any = null; + try { + throwOTelSpanError("Test message", testCase.spanName); + Assert.ok(false, `Should have thrown an error for ${testCase.description}`); + } catch (error) { + caughtError = error; + } + + Assert.ok(caughtError, `Error should have been caught for ${testCase.description}`); + Assert.ok((caughtError as OTelSpanError).spanName === testCase.expected, + `Span name should be '${testCase.expected}' for ${testCase.description}`); + }); + } + }); + + this.testCase({ + name: "OTelSpanError: should inherit from OpenTelemetryError", + test: () => { + // Arrange + const testMessage = "Inheritance test"; + const OpenTelemetryErrorConstructor = getOpenTelemetryError(); + let caughtError: any = null; + + // Act & Assert + try { + throwOTelSpanError(testMessage, "test-span"); + Assert.ok(false, "Should have thrown an error"); + } catch (error) { + caughtError = error; + } + + Assert.ok(caughtError instanceof OpenTelemetryErrorConstructor, "Should be instance of OpenTelemetryError"); + } + }); + + // Integration tests + this.testCase({ + name: "All error types should have unique names", + test: () => { + // Arrange + let errors: any[] = []; + + // Act - Create instances of all error types + try { + throwOTelError("base error"); + } catch (e) { + errors.push(e); + } + + try { + throwOTelInvalidAttributeError("invalid attr", "key", "value"); + } catch (e) { + errors.push(e); + } + + try { + throwOTelSpanError("span error", "span"); + } catch (e) { + errors.push(e); + } + + // Assert - Check that all error types have unique names + Assert.ok(errors.length === 3, "Should have caught 3 errors"); + Assert.ok(errors[0].name === "OpenTelemetryError", "First error should be OpenTelemetryError"); + Assert.ok(errors[1].name === "OTelInvalidAttributeError", "Second error should be OTelInvalidAttributeError"); + Assert.ok(errors[2].name === "OTelSpanError", "Third error should be OTelSpanError"); + + // Verify all names are unique + const names = errors.map(e => e.name); + const uniqueNames = [...new Set(names)]; + Assert.ok(uniqueNames.length === 3, "All error names should be unique"); + } + }); + + this.testCase({ + name: "Error constructors should be reusable", + test: () => { + // Arrange & Act - Create multiple errors of the same type + let errors: any[] = []; + + try { + throwOTelInvalidAttributeError("first error", "key1", "value1"); + } catch (e) { + errors.push(e); + } + + try { + throwOTelInvalidAttributeError("second error", "key2", "value2"); + } catch (e) { + errors.push(e); + } + + // Assert + Assert.ok(errors.length === 2, "Should have caught 2 errors"); + Assert.ok(errors[0].message === "first error", "First error message should match"); + Assert.ok(errors[1].message === "second error", "Second error message should match"); + Assert.ok(errors[0].name === "OTelInvalidAttributeError", "First error name should be OTelInvalidAttributeError"); + Assert.ok(errors[1].name === "OTelInvalidAttributeError", "Second error name should be OTelInvalidAttributeError"); + Assert.ok(errors[0].attribName === "key1", "First error attribName should match"); + Assert.ok(errors[1].attribName === "key2", "Second error attribName should match"); + Assert.ok(errors[0].value === "value1", "First error value should match"); + Assert.ok(errors[1].value === "value2", "Second error value should match"); + } + }); + + this.testCase({ + name: "Error stack traces should be preserved", + test: () => { + // Arrange + let caughtError: any = null; + + // Act + try { + throwOTelError("Stack trace test"); + } catch (error) { + caughtError = error; + } + + // Assert + Assert.ok(caughtError, "Error should have been caught"); + Assert.ok(caughtError.stack, "Error should have a stack trace"); + Assert.ok(isString(caughtError.stack), "Stack trace should be a string"); + Assert.ok(caughtError.stack.indexOf("OpenTelemetryError") !== -1, "Stack trace should contain error name"); + } + }); + + this.testCase({ + name: "OTelError: dumpObj should contain essential error properties", + test: () => { + // Arrange & Act + const error = new (getOpenTelemetryError())("Test error message"); + const errorDump = dumpObj(error); + + // Assert + Assert.ok(errorDump, "dumpObj should return a string representation"); + Assert.ok(errorDump.indexOf("Test error message") !== -1, `Error dump should contain message: ${errorDump}`); + Assert.ok(errorDump.indexOf("name") !== -1, `Error dump should contain name property: ${errorDump}`); + Assert.ok(errorDump.indexOf("OpenTelemetryError") !== -1, `Error dump should contain error type: ${errorDump}`); + } + }); + + this.testCase({ + name: "OTelInvalidAttributeError: dumpObj should show attribute-specific properties", + test: () => { + // Arrange & Act + let error: any = null; + try { + throwOTelInvalidAttributeError("Invalid attribute test", "testAttribute", { complex: "value" }); + } catch (e) { + error = e; + } + const errorDump = dumpObj(error); + + // Assert + Assert.ok(error, "Error should have been caught"); + Assert.ok(errorDump, "dumpObj should return a string representation"); + Assert.ok(errorDump.indexOf("Invalid attribute test") !== -1, `Error dump should contain message: ${errorDump}`); + Assert.ok(errorDump.indexOf("OTelInvalidAttributeError") !== -1, `Error dump should contain error type: ${errorDump}`); + + // Check properties directly on the error object since dumpObj might not serialize custom properties + Assert.ok(error.attribName === "testAttribute", `Error should have attribName property: ${error.attribName}`); + Assert.ok(error.value !== undefined, `Error should have value property: ${dumpObj(error.value)}`); + + // Ensure the error dump contains meaningful content even if custom properties aren't serialized + Assert.ok(errorDump.length > 50, `Error dump should be substantial: ${errorDump}`); + } + }); + + this.testCase({ + name: "OTelSpanError: dumpObj should show span-specific properties", + test: () => { + // Arrange & Act + let error: any = null; + try { + throwOTelSpanError("Span operation failed", "testSpan"); + } catch (e) { + error = e; + } + const errorDump = dumpObj(error); + + // Assert + Assert.ok(error, "Error should have been caught"); + Assert.ok(errorDump, "dumpObj should return a string representation"); + Assert.ok(errorDump.indexOf("Span operation failed") !== -1, `Error dump should contain message: ${errorDump}`); + Assert.ok(errorDump.indexOf("OTelSpanError") !== -1, `Error dump should contain error type: ${errorDump}`); + + // Check properties directly on the error object since dumpObj might not serialize custom properties + Assert.ok(error.spanName === "testSpan", `Error should have spanName property: ${error.spanName}`); + + // Ensure the error dump contains meaningful content even if custom properties aren't serialized + Assert.ok(errorDump.length > 50, `Error dump should be substantial: ${errorDump}`); + } + }); + + this.testCase({ + name: "Error serialization: dumpObj should handle complex attribute values", + test: () => { + // Arrange & Act + const complexValue = { + nested: { deep: "value" }, + array: [1, 2, 3], + number: 42, + boolean: true, + nullValue: null, + undefinedValue: undefined + }; + let error: any = null; + try { + throwOTelInvalidAttributeError("Complex value test", "complexAttr", complexValue); + } catch (e) { + error = e; + } + const errorDump = dumpObj(error); + + // Assert + Assert.ok(error, "Error should have been caught"); + Assert.ok(errorDump, "dumpObj should return a string representation"); + Assert.ok(errorDump.indexOf("Complex value test") !== -1, `Error dump should contain message: ${errorDump}`); + + // Check the actual property values directly + Assert.ok(error.attribName === "complexAttr", `Error should have correct attribName: ${error.attribName}`); + Assert.ok(error.value === complexValue, `Error should have correct value reference: ${dumpObj(error.value)}`); + + // Dump should be meaningful even if it doesn't contain all custom properties + Assert.ok(errorDump.length > 50, `Error dump should be substantial: ${errorDump}`); + } + }); + + this.testCase({ + name: "Error serialization: dumpObj should be different for different error types", + test: () => { + // Arrange & Act + const baseError = new (getOpenTelemetryError())("Base error"); + + let invalidAttrError: any = null; + try { + throwOTelInvalidAttributeError("Invalid attr", "attr", "value"); + } catch (e) { + invalidAttrError = e; + } + + let spanError: any = null; + try { + throwOTelSpanError("Span error", "span"); + } catch (e) { + spanError = e; + } + + const baseDump = dumpObj(baseError); + const invalidAttrDump = dumpObj(invalidAttrError); + const spanDump = dumpObj(spanError); + + // Assert - Each dump should be unique based on error type and message + Assert.notEqual(baseDump, invalidAttrDump, "Base and InvalidAttribute dumps should be different"); + Assert.notEqual(baseDump, spanDump, "Base and Span dumps should be different"); + Assert.notEqual(invalidAttrDump, spanDump, "InvalidAttribute and Span dumps should be different"); + + // Each should contain their specific error type names + Assert.ok(baseDump.indexOf("OpenTelemetryError") !== -1, `Base error dump should contain type: ${baseDump}`); + Assert.ok(invalidAttrDump.indexOf("OTelInvalidAttributeError") !== -1, `InvalidAttribute dump should contain type: ${invalidAttrDump}`); + Assert.ok(spanDump.indexOf("OTelSpanError") !== -1, `Span dump should contain type: ${spanDump}`); + + // Verify error objects have the expected custom properties (beyond what dumpObj shows) + Assert.ok(invalidAttrError.attribName === "attr", `InvalidAttribute error should have attribName: ${invalidAttrError.attribName}`); + Assert.ok(invalidAttrError.value === "value", `InvalidAttribute error should have value: ${invalidAttrError.value}`); + Assert.ok(spanError.spanName === "span", `Span error should have spanName: ${spanError.spanName}`); + } + }); + + this.testCase({ + name: "Error serialization: dumpObj should handle empty and null values gracefully", + test: () => { + // Arrange & Act + const emptyMsgError = new (getOpenTelemetryError())(""); + + let nullAttrError: any = null; + try { + throwOTelInvalidAttributeError("Test", null as any, null); + } catch (e) { + nullAttrError = e; + } + + let undefinedSpanError: any = null; + try { + throwOTelSpanError("Test", undefined as any); + } catch (e) { + undefinedSpanError = e; + } + + const emptyDump = dumpObj(emptyMsgError); + const nullDump = dumpObj(nullAttrError); + const undefinedDump = dumpObj(undefinedSpanError); + + // Assert - Should not throw and should contain valid representations + Assert.ok(emptyDump, "Empty message error should have valid dump"); + Assert.ok(nullDump, "Null attribute error should have valid dump"); + Assert.ok(undefinedDump, "Undefined span error should have valid dump"); + + // Should contain error type names even with empty values + Assert.ok(emptyDump.indexOf("OpenTelemetryError") !== -1, `Empty dump should contain error type: ${emptyDump}`); + Assert.ok(nullDump.indexOf("OTelInvalidAttributeError") !== -1, `Null dump should contain error type: ${nullDump}`); + Assert.ok(undefinedDump.indexOf("OTelSpanError") !== -1, `Undefined dump should contain error type: ${undefinedDump}`); + + // Verify actual property values (not assuming they're converted to empty strings) + Assert.ok(nullAttrError.attribName === null, `Null attribute error should have null attribName: ${nullAttrError.attribName}`); + Assert.ok(nullAttrError.value === null, `Null attribute error should have null value: ${nullAttrError.value}`); + Assert.ok(undefinedSpanError.spanName === undefined, `Undefined span error should have undefined spanName: ${undefinedSpanError.spanName}`); + } + }); + + this.testCase({ + name: "Error serialization: dumpObj should preserve inheritance chain information", + test: () => { + // Arrange & Act + let invalidAttrError: any = null; + try { + throwOTelInvalidAttributeError("Test", "attr", "val"); + } catch (e) { + invalidAttrError = e; + } + const errorDump = dumpObj(invalidAttrError); + + // Assert - Should show inheritance information + Assert.ok(invalidAttrError, "Error should have been caught"); + Assert.ok(errorDump, "Error dump should exist"); + Assert.ok(invalidAttrError instanceof Error, "Should be instance of Error"); + Assert.ok(invalidAttrError instanceof (getOpenTelemetryError()), "Should be instance of OpenTelemetryError"); + + // Dump should reflect the actual error type, not just base Error + Assert.ok(errorDump.indexOf("OTelInvalidAttributeError") !== -1, `Dump should show specific error type: ${errorDump}`); + + // Should have standard Error properties + Assert.ok(errorDump.indexOf("message") !== -1 || errorDump.indexOf("Test") !== -1, `Dump should contain message content: ${errorDump}`); + Assert.ok(errorDump.indexOf("name") !== -1 || errorDump.indexOf("OTelInvalidAttributeError") !== -1, `Dump should contain name/type information: ${errorDump}`); + + // Verify custom properties exist on the error object + Assert.ok(invalidAttrError.attribName === "attr", `Error should have attribName property: ${invalidAttrError.attribName}`); + Assert.ok(invalidAttrError.value === "val", `Error should have value property: ${invalidAttrError.value}`); + } + }); + } +} diff --git a/shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/otelNegative.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/otelNegative.Tests.ts new file mode 100644 index 000000000..32967a14a --- /dev/null +++ b/shared/AppInsightsCore/Tests/Unit/src/OpenTelemetry/otelNegative.Tests.ts @@ -0,0 +1,690 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { AppInsightsCore } from "../../../../src/index"; +import { IAppInsightsCore } from "../../../../src/interfaces/ai/IAppInsightsCore"; +import { createOTelApi } from "../../../../src/otel/api/OTelApi"; +import { IOTelApi } from "../../../../src/interfaces/otel/IOTelApi"; +import { IOTelApiCtx } from "../../../../src/interfaces/otel/IOTelApiCtx"; +import { createNonRecordingSpan, wrapSpanContext, withSpan, useSpan, isTracingSuppressed, suppressTracing, unsuppressTracing } from "../../../../src/otel/api/trace/utils"; +import { createDistributedTraceContext } from "../../../../src/core/TelemetryHelpers"; +import { IChannelControls } from "../../../../src/interfaces/ai/IChannelControls"; + +/** + * Negative tests for OpenTelemetry SDK helpers in AppInsights Core + * These tests ensure that no exceptions are thrown and helpers behave correctly + * when there is no trace provider or support instances + */ +export class OTelNegativeTests extends AITestClass { + private _core: IAppInsightsCore = null as any; + + public testInitialize() { + // Create a minimal mock channel to satisfy core initialization requirements + const mockChannel: IChannelControls = { + pause: () => {}, + resume: () => {}, + flush: () => {}, + teardown: () => {}, + processTelemetry: () => {}, + initialize: () => {}, + identifier: "mockChannel", + priority: 1001 + } as any; + + this._core = new AppInsightsCore(); + this._core.initialize({ + instrumentationKey: "00000000-0000-0000-0000-000000000000", + disableInstrumentationKeyValidation: true, + traceCfg: {}, + errorHandlers: {} + }, [mockChannel]); + } + + public testCleanup() { + if (this._core) { + this._core.unload(false); + this._core = null as any; + } + } + + public registerTests() { + this.addCoreWithoutTraceProviderTests(); + this.addTraceApiWithNullCoreTests(); + this.addOTelApiWithInvalidContextTests(); + this.addUtilsWithNullParametersTests(); + this.addSpanOperationsWithoutProviderTests(); + this.addTraceCfgNullHandlingTests(); + } + + private addCoreWithoutTraceProviderTests(): void { + this.testCase({ + name: "Core.startSpan: should return null when no trace provider is set", + test: () => { + // Act - core initialized but no trace provider set + const span = this._core.startSpan("test-span"); + + // Assert - should return null gracefully without throwing + Assert.equal(span, null, "Should return null when no trace provider is available"); + } + }); + + this.testCase({ + name: "Core.getActiveSpan: should return null when no trace provider is set", + test: () => { + // Act + const activeSpan = this._core.getActiveSpan(); + + // Assert + Assert.equal(activeSpan, null, "Should return null when no trace provider is available"); + } + }); + + this.testCase({ + name: "Core.getActiveSpan: should return null when createNew is false without trace provider", + test: () => { + // Act + const activeSpan = this._core.getActiveSpan(false); + + // Assert + Assert.equal(activeSpan, null, "Should return null when createNew is false and no trace provider is available"); + } + }); + + this.testCase({ + name: "Core.getActiveSpan: should return null when createNew is true without trace provider", + test: () => { + // Act + const activeSpan = this._core.getActiveSpan(true); + + // Assert + Assert.equal(activeSpan, null, "Should return null when no trace provider is available regardless of createNew value"); + } + }); + + this.testCase({ + name: "Core.setActiveSpan: should handle gracefully when no trace provider is set", + test: () => { + // Arrange + const mockSpan = { + name: "test", + spanContext: () => createDistributedTraceContext() + } as any; + + // Act - should not throw even without trace provider + const scope = this._core.setActiveSpan(mockSpan); + const activeSpan = this._core.getActiveSpan(); + + // Assert + Assert.ok(scope, "Should return a scope object"); + Assert.equal(scope.host, this._core, "Scope should reference the core"); + Assert.equal(scope.span, mockSpan, "Scope should reference the span"); + Assert.equal(activeSpan, mockSpan, "GetGetGetGetGetGetGetGetGetActiveSpan() should return the same span object"); + + // Restore should be callable without throwing + Assert.doesNotThrow(() => { + scope.restore(); + }, "Restore should not throw when no trace provider is available"); + + // Multiple restore calls should be safe + Assert.doesNotThrow(() => { + scope.restore(); + scope.restore(); + }, "Multiple restore calls should not throw"); + } + }); + + this.testCase({ + name: "Core.getTraceProvider: should return null when no trace provider is set", + test: () => { + // Act + const provider = this._core.getTraceProvider(); + + // Assert + Assert.equal(provider, null, "Should return null when no trace provider is set"); + } + }); + + this.testCase({ + name: "Core.setTraceCtx: should not throw when setting context without trace provider", + test: () => { + // Arrange + const ctx = createDistributedTraceContext(); + ctx.traceId = "12345678901234567890123456789012"; + ctx.spanId = "1234567890123456"; + + // Act & Assert + Assert.doesNotThrow(() => { + this._core.setTraceCtx(ctx); + }, "setTraceCtx should not throw without trace provider"); + + // Verify context was set + const retrievedCtx = this._core.getTraceCtx(); + Assert.equal(retrievedCtx, ctx, "Should retrieve the same context that was set"); + } + }); + + this.testCase({ + name: "Core.setTraceCtx: should handle null context gracefully", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + this._core.setTraceCtx(null); + }, "setTraceCtx should handle null gracefully"); + + const ctx = this._core.getTraceCtx(false); + Assert.equal(ctx, null, "Should return null when set to null and createNew is false"); + } + }); + } + + private addTraceApiWithNullCoreTests(): void { + this.testCase({ + name: "TraceApi: should create otelApi with initialized core", + test: () => { + // Arrange + const otelApiCtx: IOTelApiCtx = { + host: this._core + }; + + let otelApi: IOTelApi; + + // Act & Assert - creation should not throw + Assert.doesNotThrow(() => { + otelApi = createOTelApi(otelApiCtx); + }, "createOTelApi should not throw with initialized core"); + + Assert.ok(otelApi, "Should create otelApi with initialized core"); + } + }); + + this.testCase({ + name: "TraceApi.getActiveSpan: should not return a span without trace provider", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + const traceApi = otelApi.trace; + + // Act + const activeSpan = traceApi.getActiveSpan(); + + // Assert + Assert.ok(!activeSpan, "Should not return a span"); + } + }); + + this.testCase({ + name: "TraceApi.setActiveSpan: should handle null span without throwing", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + const traceApi = otelApi.trace; + + // Act & Assert - should not throw with null + Assert.doesNotThrow(() => { + traceApi.setActiveSpan(null); + }, "setActiveSpan should handle null gracefully"); + + // Should not throw with undefined + Assert.doesNotThrow(() => { + traceApi.setActiveSpan(undefined); + }, "setActiveSpan should handle undefined gracefully"); + } + }); + + this.testCase({ + name: "TraceApi.setActiveSpan: should handle span without spanContext method", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + const traceApi = otelApi.trace; + + const invalidSpan = { + name: "test" + // Missing spanContext method + } as any; + + // Act & Assert - should handle gracefully + let scope: any; + Assert.doesNotThrow(() => { + scope = traceApi.setActiveSpan(invalidSpan); + }, "Should handle span without spanContext method"); + + // Validate scope was returned and references the span + Assert.ok(scope, "Scope should be returned"); + Assert.equal(scope.span, invalidSpan, "Scope.span should equal the invalid span"); + + // Validate activeSpan returns the span + const activeSpan = this._core.getActiveSpan(); + Assert.equal(activeSpan, invalidSpan, "GetGetGetGetGetGetGetGetGetActiveSpan() should return the invalid span"); + } + }); + + this.testCase({ + name: "TraceApi.setActiveSpan: should handle span with legacy context() method", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + const traceApi = otelApi.trace; + + const legacySpan = { + name: "legacy", + context: () => ({ + traceId: "12345678901234567890123456789012", + spanId: "1234567890123456", + traceFlags: 1 + }) + } as any; + + // Act & Assert - should handle legacy API + let scope: any; + Assert.doesNotThrow(() => { + scope = traceApi.setActiveSpan(legacySpan); + }, "Should handle legacy span with context() method"); + + // Validate scope was returned and references the span + Assert.ok(scope, "Scope should be returned"); + Assert.equal(scope.span, legacySpan, "Scope.span should equal the legacy span"); + + // Validate activeSpan returns the span + const activeSpan = this._core.getActiveSpan(); + Assert.equal(activeSpan, legacySpan, "GetGetGetGetGetGetGetGetGetActiveSpan() should return the legacy span"); + } + }); + } + + private addOTelApiWithInvalidContextTests(): void { + this.testCase({ + name: "TracerProvider.getTracer: should return tracer without trace provider", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + + // Act & Assert - should not throw when getting tracer + Assert.doesNotThrow(() => { + const tracer = otelApi.getTracer("test-component"); + Assert.ok(tracer, "Should return a tracer instance"); + }, "getTracer should not throw without trace provider"); + } + }); + + this.testCase({ + name: "Tracer.startSpan: should return null without trace provider", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + const tracer = otelApi.getTracer("test"); + + // Act + const span = tracer.startSpan("operation"); + + // Assert + Assert.equal(span, null, "Should return null when trace provider is not available"); + } + }); + + this.testCase({ + name: "Tracer.startActiveSpan: should handle missing trace provider", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + const tracer = otelApi.getTracer("test"); + let callbackExecuted = false; + + // Act + const result = tracer.startActiveSpan("operation", (span) => { + callbackExecuted = true; + return "result"; + }); + + // Assert - callback should not execute when span creation fails + Assert.ok(callbackExecuted, "Callback should execute even when span is null"); + Assert.equal(result, "result", "Should return the result from the callback even when no span created"); + } + }); + + this.testCase({ + name: "TracerProvider.shutdown: should not throw without trace provider", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + + // Act & Assert - shutdown should not throw + Assert.doesNotThrow(() => { + otelApi.shutdown(); + }, "shutdown should not throw without trace provider"); + } + }); + + this.testCase({ + name: "TracerProvider.forceFlush: should not throw without trace provider", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + + // Act & Assert + Assert.doesNotThrow(() => { + otelApi.forceFlush(); + }, "forceFlush should not throw without trace provider"); + } + }); + } + + private addUtilsWithNullParametersTests(): void { + this.testCase({ + name: "createNonRecordingSpan: should handle null spanContext gracefully", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + + // Act & Assert + Assert.doesNotThrow(() => { + const span = createNonRecordingSpan(otelApi, "test", null as any); + Assert.ok(span, "Should create span even with null context"); + Assert.ok(!span.isRecording(), "Should be non-recording"); + }, "createNonRecordingSpan should handle null context"); + } + }); + + this.testCase({ + name: "wrapSpanContext: should handle minimal context object", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + const minimalContext = { + traceId: "12345678901234567890123456789012", + spanId: "1234567890123456" + } as any; + + // Act & Assert + Assert.doesNotThrow(() => { + const wrapped = wrapSpanContext(otelApi, minimalContext); + Assert.ok(wrapped, "Should wrap minimal context"); + Assert.ok(!wrapped.isRecording(), "Should be non-recording"); + }, "wrapSpanContext should handle minimal context"); + } + }); + + this.testCase({ + name: "withSpan: should execute callback with core", + test: () => { + // Arrange + const mockSpan = { + name: "test", + spanContext: () => createDistributedTraceContext() + } as any; + + let callbackExecuted = false; + const callback = () => { + callbackExecuted = true; + return "result"; + }; + + // Act & Assert + Assert.doesNotThrow(() => { + const result = withSpan(this._core, mockSpan, callback); + Assert.ok(callbackExecuted, "Callback should execute"); + Assert.equal(result, "result", "Should return callback result"); + }, "withSpan should work without trace provider"); + } + }); + + this.testCase({ + name: "useSpan: should provide scope to callback", + test: () => { + // Arrange + const mockSpan = { + name: "test" + } as any; + + let scopeReceived = false; + const callback = (scope: any) => { + scopeReceived = scope !== undefined; + return "result"; + }; + + // Act & Assert + Assert.doesNotThrow(() => { + const result = useSpan(this._core, mockSpan, callback); + Assert.ok(scopeReceived, "Callback should receive scope"); + Assert.equal(result, "result", "Should return callback result"); + }, "useSpan should work without trace provider"); + } + }); + + this.testCase({ + name: "withSpan: should handle callback throwing exception", + test: () => { + // Arrange + const ctx = createDistributedTraceContext(); + ctx.traceId = "12345678901234567890123456789012"; + ctx.spanId = "1234567890123456"; + + const mockSpan = { + name: "test", + spanContext: () => ctx + } as any; + + const callback = () => { + throw new Error("Test error"); + }; + + // Act & Assert + Assert.throws(() => { + withSpan(this._core, mockSpan, callback); + }, (err: Error) => err.message === "Test error", "Should propagate callback exception"); + } + }); + + this.testCase({ + name: "useSpan: should restore scope even when callback throws", + test: () => { + // Arrange + const mockSpan = { name: "test" } as any; + const callback = () => { + throw new Error("Test error"); + }; + + // Act & Assert - should throw but ensure cleanup happens + Assert.throws(() => { + useSpan(this._core, mockSpan, callback); + }, (err: Error) => err.message === "Test error", "Should propagate exception"); + } + }); + } + + private addSpanOperationsWithoutProviderTests(): void { + this.testCase({ + name: "NonRecordingSpan: operations should not throw", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + const ctx = createDistributedTraceContext(); + ctx.traceId = "12345678901234567890123456789012"; + ctx.spanId = "1234567890123456"; + const span = createNonRecordingSpan(otelApi, "test", ctx); + + // Act & Assert - all operations should be safe + Assert.doesNotThrow(() => { + span.setAttribute("key", "value"); + span.setAttributes({ key1: "value1", key2: "value2" }); + span.setStatus({ code: 0 }); + span.updateName("new-name"); + span.recordException(new Error("test")); + span.end(); + }, "Non-recording span operations should not throw"); + } + }); + + this.testCase({ + name: "NonRecordingSpan: multiple end calls should be safe", + test: () => { + // Arrange + const otelApi = createOTelApi({ host: this._core }); + const ctx = createDistributedTraceContext(); + ctx.traceId = "12345678901234567890123456789012"; + ctx.spanId = "1234567890123456"; + const span = createNonRecordingSpan(otelApi, "test", ctx); + + // Act & Assert + Assert.doesNotThrow(() => { + span.end(); + span.end(); + span.end(); + }, "Multiple end calls should be safe"); + } + }); + + this.testCase({ + name: "Span: operations after end should log errors but not throw", + test: () => { + // Arrange + const otelApi = createOTelApi({ + host: this._core + }); + + const ctx = createDistributedTraceContext(); + ctx.traceId = "12345678901234567890123456789012"; + ctx.spanId = "1234567890123456"; + const span = createNonRecordingSpan(otelApi, "test", ctx); + span.end(); + + // Act & Assert - operations after end should not throw + Assert.doesNotThrow(() => { + span.setAttribute("key", "value"); + span.setAttributes({ key: "value" }); + }, "Operations after end should not throw"); + } + }); + } + + private addTraceCfgNullHandlingTests(): void { + this.testCase({ + name: "suppressTracing: should handle null context gracefully", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const result = suppressTracing(null as any); + Assert.equal(result, null, "Should return the input context"); + }, "suppressTracing should handle null gracefully"); + } + }); + + this.testCase({ + name: "unsuppressTracing: should handle null context gracefully", + test: () => { + // Act & Assert + Assert.doesNotThrow(() => { + const result = unsuppressTracing(null as any); + Assert.equal(result, null, "Should return the input context"); + }, "unsuppressTracing should handle null gracefully"); + } + }); + + this.testCase({ + name: "isTracingSuppressed: should return false for null context", + test: () => { + // Act + const result = isTracingSuppressed(null as any); + + // Assert + Assert.equal(result, false, "Should return false when context is null"); + } + }); + + this.testCase({ + name: "suppressTracing: should handle context without traceCfg", + test: () => { + // Arrange + const contextWithoutTraceCfg = {} as any; + + // Act & Assert + Assert.doesNotThrow(() => { + suppressTracing(contextWithoutTraceCfg); + }, "Should handle context without traceCfg"); + + // Should not affect the context when traceCfg is missing + Assert.equal(isTracingSuppressed(contextWithoutTraceCfg), false, + "Should return false when traceCfg is missing"); + } + }); + + this.testCase({ + name: "suppressTracing: should work with core instance", + test: () => { + // Arrange + this._core.config.traceCfg = {}; + + // Act + suppressTracing(this._core); + + // Assert + Assert.equal(isTracingSuppressed(this._core), true, + "Should suppress tracing on core instance"); + + // Cleanup + unsuppressTracing(this._core); + Assert.equal(isTracingSuppressed(this._core), false, + "Should unsuppress tracing on core instance"); + } + }); + + this.testCase({ + name: "suppressTracing: should work with otelApi instance", + test: () => { + // Arrange + this._core.config.traceCfg = {}; + const otelApi = createOTelApi({ host: this._core }); + + // Act + suppressTracing(otelApi); + + // Assert + Assert.equal(isTracingSuppressed(otelApi), true, + "Should suppress tracing on otelApi instance"); + + // Cleanup + unsuppressTracing(otelApi); + } + }); + + this.testCase({ + name: "suppressTracing: should work with config instance", + test: () => { + // Arrange + const config = { + instrumentationKey: "test", + traceCfg: {} + } as any; + + // Act + suppressTracing(config); + + // Assert + Assert.equal(isTracingSuppressed(config), true, + "Should suppress tracing on config instance"); + + // Cleanup + unsuppressTracing(config); + } + }); + + this.testCase({ + name: "isTracingSuppressed: should handle undefined traceCfg properties", + test: () => { + // Arrange + const config = { + traceCfg: { + // suppressTracing is undefined + } + } as any; + + // Act + const result = isTracingSuppressed(config); + + // Assert + Assert.equal(result, false, "Should return false when suppressTracing is undefined"); + } + }); + } +} diff --git a/shared/AppInsightsCommon/Tests/Unit/src/AppInsightsCommon.tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/AppInsightsCommon.tests.ts similarity index 98% rename from shared/AppInsightsCommon/Tests/Unit/src/AppInsightsCommon.tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/AppInsightsCommon.tests.ts index 16e034509..19a432098 100644 --- a/shared/AppInsightsCommon/Tests/Unit/src/AppInsightsCommon.tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/AppInsightsCommon.tests.ts @@ -1,7 +1,8 @@ import { strRepeat } from "@nevware21/ts-utils"; import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { DiagnosticLogger, IConfiguration } from "@microsoft/applicationinsights-core-js"; -import { dataSanitizeInput, dataSanitizeKey, dataSanitizeMessage, DataSanitizerValues, dataSanitizeString, dataSanitizeUrl } from "../../../src/Telemetry/Common/DataSanitizer"; +import { DiagnosticLogger } from "../../../../src/diagnostics/DiagnosticLogger"; +import { IConfiguration } from "../../../../src/interfaces/ai/IConfiguration"; +import { dataSanitizeInput, dataSanitizeKey, dataSanitizeMessage, DataSanitizerValues, dataSanitizeString, dataSanitizeUrl } from "../../../../src/telemetry/ai/Common/DataSanitizer"; export class ApplicationInsightsTests extends AITestClass { diff --git a/shared/AppInsightsCore/Tests/Unit/src/AppInsightsCoreSize.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/AppInsightsCoreSize.Tests.ts similarity index 94% rename from shared/AppInsightsCore/Tests/Unit/src/AppInsightsCoreSize.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/AppInsightsCoreSize.Tests.ts index 87bcfa7f6..98af1328b 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/AppInsightsCoreSize.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/AppInsightsCoreSize.Tests.ts @@ -43,19 +43,19 @@ function _loadPackageJson(cb:(isNightly: boolean, packageJson: any) => IPromise< } function _checkSize(checkType: string, maxSize: number, size: number, isNightly: boolean): void { - if (isNightly) { + if (isNightly) { maxSize += .5; } Assert.ok(size <= maxSize, `exceed ${maxSize} KB, current ${checkType} size is: ${size} KB`); -} +} export class AppInsightsCoreSizeCheck extends AITestClass { - private readonly MAX_RAW_SIZE = 68; - private readonly MAX_BUNDLE_SIZE = 68; - private readonly MAX_RAW_DEFLATE_SIZE = 29; - private readonly MAX_BUNDLE_DEFLATE_SIZE = 29; - private readonly rawFilePath = "../dist/es5/applicationinsights-core-js.min.js"; + private readonly MAX_RAW_SIZE = 124; + private readonly MAX_BUNDLE_SIZE = 124; + private readonly MAX_RAW_DEFLATE_SIZE = 49; + private readonly MAX_BUNDLE_DEFLATE_SIZE = 49; + private readonly rawFilePath = "../dist/es5/index.min.js"; private readonly prodFilePath = "../browser/es5/applicationinsights-core-js.min.js"; public testInitialize() { diff --git a/shared/AppInsightsCore/Tests/Unit/src/ApplicationInsightsCore.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/ApplicationInsightsCore.Tests.ts similarity index 99% rename from shared/AppInsightsCore/Tests/Unit/src/ApplicationInsightsCore.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/ApplicationInsightsCore.Tests.ts index 1e173596d..15826fc36 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/ApplicationInsightsCore.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/ApplicationInsightsCore.Tests.ts @@ -2,15 +2,15 @@ import { Assert, AITestClass, PollingAssert } from "@microsoft/ai-test-framework import { IConfiguration, ITelemetryPlugin, ITelemetryItem, IPlugin, IAppInsightsCore, normalizeJsName, random32, mwcRandomSeed, newId, randomValue, mwcRandom32, isNullOrUndefined, SenderPostManager, - OnCompleteCallback, IPayloadData, _ISenderOnComplete, TransportType, _ISendPostMgrConfig, fieldRedaction, - isString -} from "../../../src/applicationinsights-core-js" -import { AppInsightsCore } from "../../../src/JavaScriptSDK/AppInsightsCore"; -import { IChannelControls } from "../../../src/JavaScriptSDK.Interfaces/IChannelControls"; -import { _eInternalMessageId, LoggingSeverity } from "../../../src/JavaScriptSDK.Enums/LoggingEnums"; -import { _InternalLogMessage, DiagnosticLogger } from "../../../src/JavaScriptSDK/DiagnosticLogger"; -import { ActiveStatus } from "../../../src/JavaScriptSDK.Enums/InitActiveStatusEnum"; -import { createAsyncPromise, createAsyncRejectedPromise, createAsyncResolvedPromise, createPromise, createTimeoutPromise, doAwait, doAwaitResponse } from "@nevware21/ts-async"; + OnCompleteCallback, IPayloadData, _ISenderOnComplete, TransportType, _ISendPostMgrConfig, fieldRedaction +} from "../../../../src/index" +import { AppInsightsCore } from "../../../../src/core/AppInsightsCore"; +import { IChannelControls } from "../../../../src/interfaces/ai/IChannelControls"; +import { _eInternalMessageId, LoggingSeverity } from "../../../../src/enums/ai/LoggingEnums"; +import { _InternalLogMessage, DiagnosticLogger } from "../../../../src/diagnostics/DiagnosticLogger"; +import { ActiveStatus } from "../../../../src/enums/ai/InitActiveStatusEnum"; +import { createAsyncPromise, createAsyncRejectedPromise, createAsyncResolvedPromise, createTimeoutPromise, doAwaitResponse } from "@nevware21/ts-async"; +import { setBypassLazyCache } from "@nevware21/ts-utils"; const AIInternalMessagePrefix = "AITR_"; const MaxInt32 = 0xFFFFFFFF; @@ -21,6 +21,7 @@ export class ApplicationInsightsCoreTests extends AITestClass { public testInitialize() { super.testInitialize(); this.ctx = {}; + setBypassLazyCache(true); } public testCleanup() { diff --git a/shared/AppInsightsCommon/Tests/Unit/src/ConnectionStringParser.tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/ConnectionStringParser.tests.ts similarity index 97% rename from shared/AppInsightsCommon/Tests/Unit/src/ConnectionStringParser.tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/ConnectionStringParser.tests.ts index 866110abc..06ff93cfc 100644 --- a/shared/AppInsightsCommon/Tests/Unit/src/ConnectionStringParser.tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/ConnectionStringParser.tests.ts @@ -1,6 +1,6 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { ConnectionStringParser } from "../../../src/ConnectionStringParser"; -import * as Constants from "../../../src/Constants"; +import { ConnectionStringParser } from "../../../../src/telemetry/ConnectionStringParser"; +import * as Constants from "../../../../src/constants/Constants"; const runTest = (options: { connectionString: string, diff --git a/shared/AppInsightsCore/Tests/Unit/src/CookieManager.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/CookieManager.Tests.ts similarity index 99% rename from shared/AppInsightsCore/Tests/Unit/src/CookieManager.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/CookieManager.Tests.ts index 684b7d2de..a07f2af40 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/CookieManager.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/CookieManager.Tests.ts @@ -1,7 +1,8 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { AppInsightsCore, createCookieMgr, IAppInsightsCore, IConfiguration, ICookieMgrConfig, IPlugin, ITelemetryItem, newId, objExtend } from "../../../src/applicationinsights-core-js" -import { _eInternalMessageId } from "../../../src/JavaScriptSDK.Enums/LoggingEnums"; -import { _InternalLogMessage } from "../../../src/JavaScriptSDK/DiagnosticLogger"; +import { AppInsightsCore, createCookieMgr, IAppInsightsCore, IConfiguration, ICookieMgrConfig, IPlugin, ITelemetryItem, newId, objExtend } from "../../../../src/index" +import { _eInternalMessageId } from "../../../../src/enums/ai/LoggingEnums"; +import { _InternalLogMessage } from "../../../../src/diagnostics/DiagnosticLogger"; +import { setBypassLazyCache } from "@nevware21/ts-utils"; export class CookieManagerTests extends AITestClass { private _cookieMgrCfg: ICookieMgrConfig; @@ -15,6 +16,7 @@ export class CookieManagerTests extends AITestClass { public testInitialize() { let _self = this; super.testInitialize(); + setBypassLazyCache(true); _self._testCookies = {}; _self._cookieMgrCfg = { getCookie: (name) => { diff --git a/shared/AppInsightsCore/Tests/Unit/src/EventHelper.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/EventHelper.Tests.ts similarity index 96% rename from shared/AppInsightsCore/Tests/Unit/src/EventHelper.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/EventHelper.Tests.ts index 879809b69..91233c3b1 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/EventHelper.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/EventHelper.Tests.ts @@ -1,13 +1,15 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { addEventHandler, createUniqueNamespace, removeEventHandler } from "../../../src/applicationinsights-core-js"; -import { _eInternalMessageId } from "../../../src/JavaScriptSDK.Enums/LoggingEnums"; -import { _InternalLogMessage } from "../../../src/JavaScriptSDK/DiagnosticLogger"; -import { mergeEvtNamespace, __getRegisteredEvents } from "../../../src/JavaScriptSDK/EventHelpers"; +import { addEventHandler, createUniqueNamespace, removeEventHandler } from "../../../../src/index"; +import { _eInternalMessageId } from "../../../../src/enums/ai/LoggingEnums"; +import { _InternalLogMessage } from "../../../../src/diagnostics/DiagnosticLogger"; +import { mergeEvtNamespace, __getRegisteredEvents } from "../../../../src/internal/EventHelpers"; +import { setBypassLazyCache } from "@nevware21/ts-utils"; export class EventHelperTests extends AITestClass { public testInitialize() { super.testInitialize(); + setBypassLazyCache(true); } public testCleanup() { @@ -283,15 +285,15 @@ export class EventHelperTests extends AITestClass { function _checkRegisteredAddEventHandler(name: string, expected: number) { let registered = __getRegisteredEvents(window, name); - Assert.equal(expected, registered.length, "Check that window event was registered"); + Assert.equal(expected, registered.length, "Check that window event was registered for " + name); - if (window["body"]) { + if (window && window["body"]) { registered = __getRegisteredEvents(window["body"], name); - Assert.equal(expected, registered.length, "Check that window.body event was registered"); + Assert.equal(expected, registered.length, "Check that window.body event was registered for " + name); } registered = __getRegisteredEvents(document, name); - Assert.equal(expected, registered.length, "Check that document event was registered"); + Assert.equal(expected, registered.length, "Check that document event was registered for " + name); } } } \ No newline at end of file diff --git a/shared/AppInsightsCore/Tests/Unit/src/EventsDiscardedReason.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/EventsDiscardedReason.Tests.ts similarity index 99% rename from shared/AppInsightsCore/Tests/Unit/src/EventsDiscardedReason.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/EventsDiscardedReason.Tests.ts index 63faef72c..b48486d99 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/EventsDiscardedReason.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/EventsDiscardedReason.Tests.ts @@ -1,5 +1,5 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { eEventsDiscardedReason, EventsDiscardedReason } from "../../../src/JavaScriptSDK.Enums/EventsDiscardedReason"; +import { eEventsDiscardedReason, EventsDiscardedReason } from "../../../../src/enums/ai/EventsDiscardedReason"; export class EventsDiscardedReasonTests extends AITestClass { diff --git a/shared/AppInsightsCommon/Tests/Unit/src/Exception.tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/Exception.tests.ts similarity index 98% rename from shared/AppInsightsCommon/Tests/Unit/src/Exception.tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/Exception.tests.ts index 495c3c7e2..c15dcabed 100644 --- a/shared/AppInsightsCommon/Tests/Unit/src/Exception.tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/Exception.tests.ts @@ -1,9 +1,10 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { DataSanitizerValues, Exception } from "../../../src/applicationinsights-common" -import { DiagnosticLogger } from "@microsoft/applicationinsights-core-js"; -import { IExceptionInternal, IExceptionDetailsInternal, IExceptionStackFrameInternal } from "../../../src/Interfaces/IExceptionTelemetry"; -import { _createExceptionDetails, _createExDetailsFromInterface, _extractStackFrame, _parsedFrameToInterface, _IParsedStackFrame } from "../../../src/Telemetry/Exception"; -import { IStackFrame } from "../../../src/Interfaces/Contracts/IStackFrame"; +import { DataSanitizerValues } from "../../../../src/telemetry/ai/Common/DataSanitizer"; +import { Exception } from "../../../../src/telemetry/ai/Exception"; +import { DiagnosticLogger } from "../../../../src/diagnostics/DiagnosticLogger"; +import { IExceptionInternal, IExceptionDetailsInternal, IExceptionStackFrameInternal } from "../../../../src/interfaces/ai/IExceptionTelemetry"; +import { _createExceptionDetails, _createExDetailsFromInterface, _extractStackFrame, _parsedFrameToInterface, _IParsedStackFrame } from "../../../../src/telemetry/ai/Exception"; +import { IStackFrame } from "../../../../src/interfaces/ai/contracts/IStackFrame"; function _checkExpectedFrame(expectedFrame: IStackFrame, actualFrame: IStackFrame, index: number) { Assert.equal(expectedFrame.assembly, actualFrame.assembly, index + ") Assembly is not as expected"); @@ -530,4 +531,4 @@ export class ExceptionTests extends AITestClass { } }); } -} \ No newline at end of file +} diff --git a/shared/AppInsightsCommon/Tests/Unit/src/GlobalTestHooks.Test.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/GlobalTestHooks.Test.ts similarity index 82% rename from shared/AppInsightsCommon/Tests/Unit/src/GlobalTestHooks.Test.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/GlobalTestHooks.Test.ts index b67bf2fe7..1b1f56955 100644 --- a/shared/AppInsightsCommon/Tests/Unit/src/GlobalTestHooks.Test.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/GlobalTestHooks.Test.ts @@ -1,5 +1,5 @@ import { Assert } from "@microsoft/ai-test-framework"; -import { _testHookMaxUnloadHooksCb } from "@microsoft/applicationinsights-core-js"; +import { _testHookMaxUnloadHooksCb } from "../../../../src/core/UnloadHookContainer"; import { dumpObj } from "@nevware21/ts-utils"; export class GlobalTestHooks { diff --git a/shared/AppInsightsCore/Tests/Unit/src/HelperFunc.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/HelperFunc.Tests.ts similarity index 98% rename from shared/AppInsightsCore/Tests/Unit/src/HelperFunc.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/HelperFunc.Tests.ts index 899443c7f..2b157734e 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/HelperFunc.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/HelperFunc.Tests.ts @@ -1,10 +1,10 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { _eInternalMessageId } from "../../../src/JavaScriptSDK.Enums/LoggingEnums"; -import { _InternalLogMessage } from "../../../src/JavaScriptSDK/DiagnosticLogger"; -import { normalizeJsName, objExtend, _getObjProto, isFeatureEnabled } from "../../../src/JavaScriptSDK/HelperFuncs"; -import { AppInsightsCore } from "../../../src/JavaScriptSDK/AppInsightsCore"; +import { _eInternalMessageId } from "../../../../src/enums/ai/LoggingEnums"; +import { _InternalLogMessage } from "../../../../src/diagnostics/DiagnosticLogger"; +import { normalizeJsName, objExtend, _getObjProto, isFeatureEnabled } from "../../../../src/utils/HelperFuncs"; +import { AppInsightsCore } from "../../../../src/core/AppInsightsCore"; import { isArray, isObject, objKeys, strEndsWith, strStartsWith, isPlainObject, utcNow } from "@nevware21/ts-utils"; -import { FeatureOptInMode, IConfiguration, IFeatureOptInDetails, dumpObj } from "../../../src/applicationinsights-core-js"; +import { FeatureOptInMode, IConfiguration, IFeatureOptInDetails, dumpObj } from "../../../../src/index"; diff --git a/shared/AppInsightsCore/Tests/Unit/src/LoggingEnum.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/LoggingEnum.Tests.ts similarity index 98% rename from shared/AppInsightsCore/Tests/Unit/src/LoggingEnum.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/LoggingEnum.Tests.ts index 7cbf4d1bc..1608fe7f6 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/LoggingEnum.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/LoggingEnum.Tests.ts @@ -1,7 +1,7 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; import { isString, objForEachKey } from "@nevware21/ts-utils"; -import { LoggingSeverity, eLoggingSeverity, _eInternalMessageId } from "../../../src/JavaScriptSDK.Enums/LoggingEnums"; -import { _InternalLogMessage } from "../../../src/JavaScriptSDK/DiagnosticLogger"; +import { LoggingSeverity, eLoggingSeverity, _eInternalMessageId } from "../../../../src/enums/ai/LoggingEnums"; +import { _InternalLogMessage } from "../../../../src/diagnostics/DiagnosticLogger"; type NoRepeats = { [M in keyof T]: { [N in keyof T]: N extends M ? never : T[M] extends T[N] ? unknown : never diff --git a/shared/AppInsightsCommon/Tests/Unit/src/RequestHeaders.tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/RequestHeaders.tests.ts similarity index 97% rename from shared/AppInsightsCommon/Tests/Unit/src/RequestHeaders.tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/RequestHeaders.tests.ts index f178549d1..7f339442a 100644 --- a/shared/AppInsightsCommon/Tests/Unit/src/RequestHeaders.tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/RequestHeaders.tests.ts @@ -1,5 +1,5 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { eRequestHeaders, RequestHeaders } from "../../../src/RequestResponseHeaders"; +import { eRequestHeaders, RequestHeaders } from "../../../../src/telemetry/RequestResponseHeaders"; export class RequestHeadersTests extends AITestClass { @@ -48,4 +48,4 @@ export class RequestHeadersTests extends AITestClass { } }); } -} \ No newline at end of file +} diff --git a/shared/AppInsightsCore/Tests/Unit/src/SendPostManager.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/SendPostManager.Tests.ts similarity index 97% rename from shared/AppInsightsCore/Tests/Unit/src/SendPostManager.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/SendPostManager.Tests.ts index fad029b30..40c0ca24f 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/SendPostManager.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/SendPostManager.Tests.ts @@ -1,8 +1,8 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { _eInternalMessageId } from "../../../src/JavaScriptSDK.Enums/LoggingEnums"; -import { _InternalLogMessage } from "../../../src/JavaScriptSDK/DiagnosticLogger"; -import { SenderPostManager } from "../../../src/JavaScriptSDK/SenderPostManager"; -import { IPayloadData } from "../../../src/JavaScriptSDK.Interfaces/IXHROverride"; +import { _eInternalMessageId } from "../../../../src/enums/ai/LoggingEnums"; +import { _InternalLogMessage } from "../../../../src/diagnostics/DiagnosticLogger"; +import { SenderPostManager } from "../../../../src/core/SenderPostManager"; +import { IPayloadData } from "../../../../src/interfaces/ai/IXHROverride"; import { getInst, isFunction, mathRandom } from "@nevware21/ts-utils"; import { createPromise, doAwaitResponse } from "@nevware21/ts-async"; diff --git a/shared/AppInsightsCommon/Tests/Unit/src/SeverityLevel.tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/SeverityLevel.tests.ts similarity index 97% rename from shared/AppInsightsCommon/Tests/Unit/src/SeverityLevel.tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/SeverityLevel.tests.ts index c278df8a0..5c7b38a4b 100644 --- a/shared/AppInsightsCommon/Tests/Unit/src/SeverityLevel.tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/SeverityLevel.tests.ts @@ -1,5 +1,5 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { eSeverityLevel, SeverityLevel } from "../../../src/applicationinsights-common"; +import { eSeverityLevel, SeverityLevel } from "../../../../src/interfaces/ai/contracts/SeverityLevel"; export class SeverityLevelTests extends AITestClass { @@ -55,4 +55,4 @@ export class SeverityLevelTests extends AITestClass { } }); } -} \ No newline at end of file +} diff --git a/shared/AppInsightsCore/Tests/Unit/src/StatsBeat.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/StatsBeat.Tests.ts similarity index 94% rename from shared/AppInsightsCore/Tests/Unit/src/StatsBeat.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/StatsBeat.Tests.ts index 3ba1eb90c..fef4337fa 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/StatsBeat.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/StatsBeat.Tests.ts @@ -1,16 +1,16 @@ // import * as sinon from "sinon"; // import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -// import { IPayloadData } from "../../../src/JavaScriptSDK.Interfaces/IXHROverride"; -// import { IStatsMgr } from "../../../src/JavaScriptSDK.Interfaces/IStatsMgr"; -// import { AppInsightsCore } from "../../../src/JavaScriptSDK/AppInsightsCore"; -// import { IConfiguration } from "../../../src/JavaScriptSDK.Interfaces/IConfiguration"; -// import { createStatsMgr } from "../../../src/JavaScriptSDK/StatsBeat"; -// import { IStatsBeatState } from "../../../src/JavaScriptSDK.Interfaces/IStatsBeat"; -// import { eStatsType } from "../../../src/JavaScriptSDK.Enums/StatsType"; -// import { ITelemetryItem } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryItem"; -// import { IPlugin } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryPlugin"; -// import { IAppInsightsCore } from "../../../src/JavaScriptSDK.Interfaces/IAppInsightsCore"; -// import { FeatureOptInMode } from "../../../src/JavaScriptSDK.Enums/FeatureOptInEnums"; +// import { IPayloadData } from "../../../../src/interfaces/ai/IXHROverride"; +// import { IStatsMgr } from "../../../../src/JavaScriptSDK.Interfaces/IStatsMgr"; +// import { AppInsightsCore } from "../../../../src/core/AppInsightsCore"; +// import { IConfiguration } from "../../../../src/interfaces/ai/IConfiguration"; +// import { createStatsMgr } from "../../../../src/JavaScriptSDK/StatsBeat"; +// import { IStatsBeatState } from "../../../../src/JavaScriptSDK.Interfaces/IStatsBeat"; +// import { eStatsType } from "../../../../src/JavaScriptSDK.Enums/StatsType"; +// import { ITelemetryItem } from "../../../../src/interfaces/ai/ITelemetryItem"; +// import { IPlugin } from "../../../../src/interfaces/ai/ITelemetryPlugin"; +// import { IAppInsightsCore } from "../../../../src/interfaces/ai/IAppInsightsCore"; +// import { FeatureOptInMode } from "../../../../src/JavaScriptSDK.Enums/FeatureOptInEnums"; // const STATS_COLLECTION_SHORT_INTERVAL: number = 900; // 15 minutes diff --git a/shared/AppInsightsCore/Tests/Unit/src/TestPlugins.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/TestPlugins.ts similarity index 84% rename from shared/AppInsightsCore/Tests/Unit/src/TestPlugins.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/TestPlugins.ts index 78b07c556..f3277c0f6 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/TestPlugins.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/TestPlugins.ts @@ -1,16 +1,16 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { _eInternalMessageId } from "../../../src/JavaScriptSDK.Enums/LoggingEnums"; -import { _InternalLogMessage } from "../../../src/JavaScriptSDK/DiagnosticLogger"; -import { ITelemetryItem } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryItem"; -import { IProcessTelemetryContext, IProcessTelemetryUpdateContext } from "../../../src/JavaScriptSDK.Interfaces/IProcessTelemetryContext"; -import { TelemetryUpdateReason } from "../../../src/JavaScriptSDK.Enums/TelemetryUpdateReason"; -import { IConfiguration } from "../../../src/JavaScriptSDK.Interfaces/IConfiguration"; -import { IPlugin, ITelemetryPlugin } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryPlugin"; -import { BaseTelemetryPlugin } from "../../../src/JavaScriptSDK/BaseTelemetryPlugin"; -import { IAppInsightsCore } from "../../../src/JavaScriptSDK.Interfaces/IAppInsightsCore"; -import { ITelemetryPluginChain } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryPluginChain"; -import { ITelemetryUpdateState } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryUpdateState"; -import { IChannelControls } from "../../../src/JavaScriptSDK.Interfaces/IChannelControls"; +import { _eInternalMessageId } from "../../../../src/enums/ai/LoggingEnums"; +import { _InternalLogMessage } from "../../../../src/diagnostics/DiagnosticLogger"; +import { ITelemetryItem } from "../../../../src/interfaces/ai/ITelemetryItem"; +import { IProcessTelemetryContext, IProcessTelemetryUpdateContext } from "../../../../src/interfaces/ai/IProcessTelemetryContext"; +import { TelemetryUpdateReason } from "../../../../src/enums/ai/TelemetryUpdateReason"; +import { IConfiguration } from "../../../../src/interfaces/ai/IConfiguration"; +import { IPlugin, ITelemetryPlugin } from "../../../../src/interfaces/ai/ITelemetryPlugin"; +import { BaseTelemetryPlugin } from "../../../../src/core/BaseTelemetryPlugin"; +import { IAppInsightsCore } from "../../../../src/interfaces/ai/IAppInsightsCore"; +import { ITelemetryPluginChain } from "../../../../src/interfaces/ai/ITelemetryPluginChain"; +import { ITelemetryUpdateState } from "../../../../src/interfaces/ai/ITelemetryUpdateState"; +import { IChannelControls } from "../../../../src/interfaces/ai/IChannelControls"; export class TestPlugin implements IPlugin { diff --git a/shared/AppInsightsCommon/Tests/Unit/src/ThrottleMgr.tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/ThrottleMgr.tests.ts similarity index 99% rename from shared/AppInsightsCommon/Tests/Unit/src/ThrottleMgr.tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/ThrottleMgr.tests.ts index 5d6a39462..efde4a2f6 100644 --- a/shared/AppInsightsCommon/Tests/Unit/src/ThrottleMgr.tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/ThrottleMgr.tests.ts @@ -1,10 +1,14 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { AppInsightsCore, IAppInsightsCore, IConfiguration, IPlugin, _eInternalMessageId } from "@microsoft/applicationinsights-core-js"; -import { ThrottleMgr} from "../../../src/ThrottleMgr"; +import { AppInsightsCore } from "../../../../src/core/AppInsightsCore"; +import { IAppInsightsCore } from "../../../../src/interfaces/ai/IAppInsightsCore"; +import { IConfiguration } from "../../../../src/interfaces/ai/IConfiguration"; +import { IPlugin } from "../../../../src/interfaces/ai/ITelemetryPlugin"; +import { _eInternalMessageId } from "../../../../src/enums/ai/LoggingEnums"; +import { ThrottleMgr } from "../../../../src/diagnostics/ThrottleMgr"; import { SinonSpy } from "sinon"; -import { utlCanUseLocalStorage } from "../../../src/StorageHelperFuncs"; -import { IThrottleInterval, IThrottleLimit, IThrottleMgrConfig, IThrottleResult } from "../../../src/Interfaces/IThrottleMgr"; -import { IConfig } from "../../../types/applicationinsights-common"; +import { utlCanUseLocalStorage } from "../../../../src/utils/StorageHelperFuncs"; +import { IThrottleInterval, IThrottleLimit, IThrottleMgrConfig, IThrottleResult } from "../../../../src/interfaces/ai/IThrottleMgr"; +import { IConfig } from "../../../../src/interfaces/ai/IConfig"; const daysInMonth = [ 31, // Jan @@ -1673,4 +1677,4 @@ class ChannelPlugin implements IPlugin { private _processTelemetry(env: any) { } -} \ No newline at end of file +} diff --git a/shared/AppInsightsCore/Tests/Unit/src/UpdateConfig.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/UpdateConfig.Tests.ts similarity index 98% rename from shared/AppInsightsCore/Tests/Unit/src/UpdateConfig.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/UpdateConfig.Tests.ts index 45603fb90..d6fce3c81 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/UpdateConfig.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/UpdateConfig.Tests.ts @@ -1,9 +1,9 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; import { dumpObj } from "@nevware21/ts-utils"; -import { AppInsightsCore } from "../../../src/JavaScriptSDK/AppInsightsCore"; -import { _eInternalMessageId } from "../../../src/JavaScriptSDK.Enums/LoggingEnums"; -import { _InternalLogMessage } from "../../../src/JavaScriptSDK/DiagnosticLogger"; -import { IConfiguration } from "../../../src/JavaScriptSDK.Interfaces/IConfiguration"; +import { AppInsightsCore } from "../../../../src/core/AppInsightsCore"; +import { _eInternalMessageId } from "../../../../src/enums/ai/LoggingEnums"; +import { _InternalLogMessage } from "../../../../src/diagnostics/DiagnosticLogger"; +import { IConfiguration } from "../../../../src/interfaces/ai/IConfiguration"; import { OldTrackPlugin, TestChannelPlugin, TestPlugin, TestSamplingPlugin, TrackPlugin } from "./TestPlugins"; const AIInternalMessagePrefix = "AITR_"; diff --git a/shared/AppInsightsCommon/Tests/Unit/src/Util.tests.ts b/shared/AppInsightsCore/Tests/Unit/src/ai/Util.tests.ts similarity index 98% rename from shared/AppInsightsCommon/Tests/Unit/src/Util.tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/ai/Util.tests.ts index 57479d3a9..4e9769c12 100644 --- a/shared/AppInsightsCommon/Tests/Unit/src/Util.tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/ai/Util.tests.ts @@ -1,9 +1,11 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { ICorrelationConfig } from "../../../src/Interfaces/ICorrelationConfig"; -import { getIEVersion, strStartsWith, uaDisallowsSameSiteNone } from "@microsoft/applicationinsights-core-js"; -import { correlationIdCanIncludeCorrelationHeader } from "../../../src/Util"; -import { createDomEvent } from "../../../src/DomHelperFuncs"; -import { urlParseFullHost, urlParseHost, urlParseUrl } from "../../../src/UrlHelperFuncs"; +import { setBypassLazyCache, strStartsWith } from "@nevware21/ts-utils"; +import { ICorrelationConfig } from "../../../../src/interfaces/ai/ICorrelationConfig"; +import { getIEVersion } from "../../../../src/utils/EnvUtils"; +import { uaDisallowsSameSiteNone } from "../../../../src/core/CookieMgr"; +import { correlationIdCanIncludeCorrelationHeader } from "../../../../src/utils/Util"; +import { createDomEvent } from "../../../../src/utils/DomHelperFuncs"; +import { urlParseFullHost, urlParseHost, urlParseUrl } from "../../../../src/utils/UrlHelperFuncs"; export class UtilTests extends AITestClass { private testRegexLists = (config: ICorrelationConfig, exp: boolean, host: string) => { @@ -15,9 +17,13 @@ export class UtilTests extends AITestClass { }; public testInitialize() { + super.testInitialize(); + setBypassLazyCache(true); } - public testCleanup() {} + public testCleanup() { + super.testCleanup(); + } public registerTests() { this.testCase({ @@ -171,7 +177,7 @@ export class UtilTests extends AITestClass { this.testRegexLists(config, false, "test"); this.testRegexLists(config, true, "portal.azure.com"); this.testRegexLists(config, true, "azure.com"); - this.testRegexLists(config, false, "localhost"); + this.testRegexLists(config, false, "localhost:9001"); this.testRegexLists(config, false, "bing.com"); this.testRegexLists(config, true, "prefix.bing.com"); } @@ -187,7 +193,7 @@ export class UtilTests extends AITestClass { this.testRegexLists(config, false, "test"); this.testRegexLists(config, false, "portal.azure.com"); this.testRegexLists(config, true, "azure.com"); - this.testRegexLists(config, true, "localhost"); + this.testRegexLists(config, true, "localhost:9001"); this.testRegexLists(config, true, "bing.com"); this.testRegexLists(config, true, "prefix.bing.com"); } @@ -204,7 +210,7 @@ export class UtilTests extends AITestClass { this.testRegexLists(config, false, "test"); this.testRegexLists(config, false, "portal.azure.com"); this.testRegexLists(config, true, "azure.com"); - this.testRegexLists(config, false, "localhost"); + this.testRegexLists(config, false, "localhost:9001"); this.testRegexLists(config, false, "bing.com"); this.testRegexLists(config, true, "prefix.bing.com"); this.testRegexLists(config, false, "ignore.microsoft.com"); diff --git a/shared/AppInsightsCore/Tests/Unit/src/aiunittests.ts b/shared/AppInsightsCore/Tests/Unit/src/aiunittests.ts index 613a53421..90ae0a7da 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/aiunittests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/aiunittests.ts @@ -1,18 +1,36 @@ -import '@microsoft/applicationinsights-shims'; -import { ApplicationInsightsCoreTests } from "./ApplicationInsightsCore.Tests"; -import { CookieManagerTests } from "./CookieManager.Tests"; -import { GlobalTestHooks } from "./GlobalTestHooks.Test"; -import { HelperFuncTests } from './HelperFunc.Tests'; -import { AppInsightsCoreSizeCheck } from "./AppInsightsCoreSize.Tests"; -import { EventHelperTests } from "./EventHelper.Tests"; -import { LoggingEnumTests } from "./LoggingEnum.Tests"; -import { DynamicTests } from "./Dynamic.Tests"; -import { UpdateConfigTests } from "./UpdateConfig.Tests"; -import { EventsDiscardedReasonTests } from "./EventsDiscardedReason.Tests"; -import { W3cTraceParentTests } from "./W3cTraceParentTests"; -import { DynamicConfigTests } from "./DynamicConfig.Tests"; -import { SendPostManagerTests } from './SendPostManager.Tests'; -// import { StatsBeatTests } from './StatsBeat.Tests'; +import "@microsoft/applicationinsights-shims"; +import { ApplicationInsightsCoreTests } from "./ai/ApplicationInsightsCore.Tests"; +import { CookieManagerTests } from "./ai/CookieManager.Tests"; +import { GlobalTestHooks } from "./ai/GlobalTestHooks.Test"; +import { HelperFuncTests } from "./ai/HelperFunc.Tests"; +import { AppInsightsCoreSizeCheck } from "./ai/AppInsightsCoreSize.Tests"; +import { EventHelperTests } from "./ai/EventHelper.Tests"; +import { LoggingEnumTests } from "./ai/LoggingEnum.Tests"; +import { DynamicTests } from "./config/Dynamic.Tests"; +import { UpdateConfigTests } from "./ai/UpdateConfig.Tests"; +import { EventsDiscardedReasonTests } from "./ai/EventsDiscardedReason.Tests"; +import { W3cTraceParentTests } from "./trace/W3cTraceParentTests"; +import { DynamicConfigTests } from "./config/DynamicConfig.Tests"; +import { SendPostManagerTests } from "./ai/SendPostManager.Tests"; +// import { StatsBeatTests } from "./StatsBeat.Tests"; +import { OTelTraceApiTests } from "./trace/traceState.Tests"; +import { CommonUtilsTests } from "./OpenTelemetry/commonUtils.Tests"; +import { OpenTelemetryErrorsTests } from "./OpenTelemetry/errors.Tests"; +import { SpanTests } from "./trace/span.Tests"; +import { AttributeContainerTests } from "./attribute/attributeContainer.Tests"; +import { W3cTraceStateTests } from "./trace/W3cTraceState.Tests"; +import { OTelNegativeTests } from "./OpenTelemetry/otelNegative.Tests"; +import { TraceUtilsTests } from "./trace/traceUtils.Tests"; + +// Application Insights Common tests (merged from AppInsightsCommon) +import { ApplicationInsightsTests } from "./ai/AppInsightsCommon.tests"; +import { ConnectionStringParserTests } from "./ai/ConnectionStringParser.tests"; +import { ExceptionTests } from "./ai/Exception.tests"; +import { RequestHeadersTests } from "./ai/RequestHeaders.tests"; +import { SeverityLevelTests } from "./ai/SeverityLevel.tests"; +import { ThrottleMgrTest } from "./ai/ThrottleMgr.tests"; +import { UtilTests } from "./ai/Util.tests"; +import { W3CTraceStateModesTests } from "./trace/W3CTraceStateModes.tests"; export function runTests() { new GlobalTestHooks().registerTests(); @@ -28,7 +46,25 @@ export function runTests() { new UpdateConfigTests().registerTests(); new EventsDiscardedReasonTests().registerTests(); new W3cTraceParentTests().registerTests(); + new OTelTraceApiTests().registerTests(); + new CommonUtilsTests().registerTests(); + new OpenTelemetryErrorsTests().registerTests(); + new SpanTests().registerTests(); + new AttributeContainerTests().registerTests(); + new W3cTraceStateTests().registerTests(); + new TraceUtilsTests().registerTests(); + new OTelNegativeTests().registerTests(); // new StatsBeatTests(false).registerTests(); // new StatsBeatTests(true).registerTests(); new SendPostManagerTests().registerTests(); + + // Application Insights Common tests (merged from AppInsightsCommon) + new ApplicationInsightsTests().registerTests(); + new ConnectionStringParserTests().registerTests(); + new ExceptionTests().registerTests(); + new RequestHeadersTests().registerTests(); + new SeverityLevelTests().registerTests(); + new ThrottleMgrTest().registerTests(); + new UtilTests().registerTests(); + new W3CTraceStateModesTests().registerTests(); } diff --git a/shared/AppInsightsCore/Tests/Unit/src/attribute/attributeContainer.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/attribute/attributeContainer.Tests.ts new file mode 100644 index 000000000..256f3ea6c --- /dev/null +++ b/shared/AppInsightsCore/Tests/Unit/src/attribute/attributeContainer.Tests.ts @@ -0,0 +1,3412 @@ +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { objKeys } from "@nevware21/ts-utils"; +import { createAttributeContainer, addAttributes, isAttributeContainer, createAttributeSnapshot } from "../../../../src/otel/attribute/attributeContainer"; +import { eAttributeFilter, IAttributeChangeInfo } from "../../../../src/interfaces/otel/attribute/IAttributeContainer"; +import { eAttributeChangeOp } from "../../../../src/enums/otel/eAttributeChangeOp"; +import { IOTelConfig } from "../../../../src/interfaces/otel/config/IOTelConfig"; +import { IOTelAttributes } from "../../../../src/interfaces/otel/IOTelAttributes"; + +export class AttributeContainerTests extends AITestClass { + + public testInitialize() { + super.testInitialize(); + } + + public testCleanup() { + super.testCleanup(); + } + + public registerTests() { + this.testCase({ + name: "AttributeContainer: Basic functionality", + test: () => { + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test-container"); + + // Test initial state + Assert.equal(0, container.size, "Initial size should be 0"); + Assert.equal(0, container.droppedAttributes, "Initial dropped attributes should be 0"); + + // Test set/get + Assert.ok(container.set("key1", "value1"), "Should successfully set attribute"); + Assert.equal("value1", container.get("key1"), "Should retrieve correct value"); + Assert.equal(1, container.size, "Size should be 1 after adding one attribute"); + + // Test has + Assert.ok(container.has("key1"), "Should return true for existing key"); + Assert.ok(!container.has("nonexistent"), "Should return false for non-existent key"); + + // Test clear + container.clear(); + Assert.equal(0, container.size, "Size should be 0 after clear"); + Assert.ok(!container.has("key1"), "Should not have key after clear"); + Assert.equal(undefined, container.get("key1"), "Should return undefined after clear"); + } + }); + + this.testCase({ + name: "AttributeContainer: Hierarchical keys (dotted notation)", + test: () => { + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test-container"); + + // Test hierarchical keys + Assert.ok(container.set("parent.child", "value1"), "Should set hierarchical key"); + Assert.ok(container.set("parent.child2", "value2"), "Should set second child"); + Assert.equal("value1", container.get("parent.child"), "Should get hierarchical value"); + Assert.equal("value2", container.get("parent.child2"), "Should get second hierarchical value"); + Assert.equal(2, container.size, "Should count hierarchical keys correctly"); + } + }); + + this.testCase({ + name: "AttributeContainer: Iterator methods", + test: () => { + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test-container"); + + container.set("key1", "value1"); + container.set("key2", "value2"); + container.set("parent.child", "value3"); + + // Test keys iterator + const keys: string[] = []; + const keysIter = container.keys(); + let keysResult = keysIter.next(); + while (!keysResult.done) { + keys.push(keysResult.value); + keysResult = keysIter.next(); + } + Assert.equal(3, keys.length, "Should have 3 keys"); + Assert.ok(keys.includes("key1"), "Should include key1"); + Assert.ok(keys.includes("key2"), "Should include key2"); + Assert.ok(keys.includes("parent.child"), "Should include hierarchical key"); + + // Test entries iterator + const entries: [string, any, eAttributeFilter][] = []; + const entriesIter = container.entries(); + let entriesResult = entriesIter.next(); + while (!entriesResult.done) { + entries.push(entriesResult.value); + entriesResult = entriesIter.next(); + } + Assert.equal(3, entries.length, "Should have 3 entries"); + + // Test values iterator + const values: any[] = []; + const valuesIter = container.values(); + let valuesResult = valuesIter.next(); + while (!valuesResult.done) { + values.push(valuesResult.value); + valuesResult = valuesIter.next(); + } + Assert.equal(3, values.length, "Should have 3 values"); + Assert.ok(values.includes("value1"), "Should include value1"); + Assert.ok(values.includes("value2"), "Should include value2"); + Assert.ok(values.includes("value3"), "Should include value3"); + + // Test forEach + const forEachResults: { [key: string]: any } = {}; + container.forEach((key, value) => { + forEachResults[key] = value; + }); + Assert.equal(3, objKeys(forEachResults).length, "forEach should iterate over all items"); + Assert.equal("value1", forEachResults["key1"], "forEach should provide correct key-value pairs"); + } + }); + + this.testCase({ + name: "AttributeContainer: attributes accessor property", + test: () => { + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test-container"); + + container.set("key1", "value1"); + container.set("key2", 42); + container.set("parent.child", true); + + const attributes = container.attributes; + Assert.equal("value1", attributes["key1"], "Should include simple string attribute"); + Assert.equal(42, attributes["key2"], "Should include number attribute"); + Assert.equal(true, attributes["parent.child"], "Should include hierarchical boolean attribute"); + Assert.equal(3, objKeys(attributes).length, "Should have correct number of attributes"); + } + }); + + this.testCase({ + name: "AttributeContainer: Container ID validation with name parameter", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Test container with no name + const containerNoName = createAttributeContainer(otelCfg); + Assert.ok(containerNoName.id, "Container should have an ID even without name"); + Assert.ok(containerNoName.id.includes("."), "Container ID should include dot separator"); + + // Test container with custom name + const containerWithName = createAttributeContainer(otelCfg, "custom-container"); + Assert.ok(containerWithName.id, "Container with name should have an ID"); + Assert.ok(containerWithName.id.startsWith("custom-container."), "Container ID should start with provided name"); + + // Test container with descriptive name + const containerDescriptive = createAttributeContainer(otelCfg, "span-attributes"); + Assert.ok(containerDescriptive.id.startsWith("span-attributes."), "Container ID should include descriptive name"); + + // Test that different containers have different IDs + const container1 = createAttributeContainer(otelCfg, "test-1"); + const container2 = createAttributeContainer(otelCfg, "test-2"); + const container3 = createAttributeContainer(otelCfg, "test-1"); // Same name, different instance + + Assert.notEqual(container1.id, container2.id, "Different containers should have different IDs"); + Assert.notEqual(container1.id, container3.id, "Containers with same name should have different IDs"); + Assert.notEqual(container2.id, container3.id, "All containers should have unique IDs"); + + // Test ID format consistency + Assert.ok(container1.id.includes("test-1."), "Container 1 should include its name"); + Assert.ok(container2.id.includes("test-2."), "Container 2 should include its name"); + Assert.ok(container3.id.includes("test-1."), "Container 3 should include its name"); + } + }); + + this.testCase({ + name: "AttributeContainer: Snapshot ID validation includes source container details", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create source container with name + const sourceContainer = createAttributeContainer(otelCfg, "source-container"); + sourceContainer.set("key1", "value1"); + sourceContainer.set("key2", "value2"); + + // Create snapshot from container + const snapshot = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + Assert.ok(snapshot.id, "Snapshot should have an ID"); + Assert.ok(snapshot.id.includes("<-@["), "Snapshot ID should indicate it's a snapshot"); + Assert.ok(snapshot.id.includes(sourceContainer.id), "Snapshot ID should include source container ID"); + Assert.ok(snapshot.id.includes("]"), "Snapshot ID should close the snapshot notation"); + + // Test snapshot from container without explicit name + const unnamedContainer = createAttributeContainer(otelCfg); + unnamedContainer.set("test", "value"); + const unnamedSnapshot = createAttributeSnapshot(otelCfg, "unnamed-snapshot", unnamedContainer); + + Assert.ok(unnamedSnapshot.id.includes("<-@["), "Unnamed container snapshot should indicate it's a snapshot"); + Assert.ok(unnamedSnapshot.id.includes(unnamedContainer.id), "Unnamed snapshot should include source container ID"); + + // Test snapshot from plain object (should not include source container ID) + const plainAttributes = { "plain": "value", "another": 42 }; + const plainSnapshot = createAttributeSnapshot(otelCfg, "plain-snapshot", plainAttributes); + + // Plain object snapshots get a different naming pattern since there's no source container + Assert.ok(plainSnapshot.id, "Plain object snapshot should have an ID"); + Assert.notEqual(plainSnapshot.id, snapshot.id, "Plain snapshot should have different ID than container snapshot"); + } + }); + + this.testCase({ + name: "AttributeContainer: Snapshot ID uniqueness and inheritance details", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create multiple containers with the same name + const container1 = createAttributeContainer(otelCfg, "parent-container"); + const container2 = createAttributeContainer(otelCfg, "parent-container"); + + container1.set("key1", "value1"); + container2.set("key2", "value2"); + + // Create snapshots from both containers + const snapshot1 = createAttributeSnapshot(otelCfg, "snapshot1", container1); + const snapshot2 = createAttributeSnapshot(otelCfg, "snapshot2", container2); + + Assert.notEqual(snapshot1.id, snapshot2.id, "Snapshots from different containers should have different IDs"); + + // Both should reference their respective source container IDs + Assert.ok(snapshot1.id.includes(container1.id), "Snapshot 1 should reference container 1 ID"); + Assert.ok(snapshot2.id.includes(container2.id), "Snapshot 2 should reference container 2 ID"); + + // Test nested snapshot scenario + const childContainer = createAttributeContainer(otelCfg, "child-container", container1); + const childSnapshot = createAttributeSnapshot(otelCfg, "child-snapshot", childContainer); + + Assert.ok(childSnapshot.id.includes("<-@["), "Child snapshot should be identified as snapshot"); + Assert.ok(childSnapshot.id.includes(childContainer.id), "Child snapshot should reference child container ID"); + Assert.notEqual(childSnapshot.id, snapshot1.id, "Child snapshot should have different ID than parent snapshot"); + + // Verify the child container ID includes its name + Assert.ok(childContainer.id.includes("child-container."), "Child container ID should include its name"); + } + }); + + this.testCase({ + name: "AttributeContainer: Inheritance - basic functionality", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "parent.key2": "parent_value2", + "shared.key": "parent_shared" + }; + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + + // Test inherited attributes are accessible + Assert.equal("parent_value1", container.get("parent.key1"), "Should get inherited attribute"); + Assert.equal("parent_value2", container.get("parent.key2"), "Should get second inherited attribute"); + Assert.equal("parent_shared", container.get("shared.key"), "Should get shared inherited attribute"); + Assert.ok(container.has("parent.key1"), "Should report inherited attribute as existing"); + Assert.equal(3, container.size, "Should count inherited attributes in size"); + } + }); + + this.testCase({ + name: "AttributeContainer: Inheritance - override behavior", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "shared.key": "parent_shared", + "override.me": "parent_override" + }; + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + + // Test initial inherited state + Assert.equal(3, container.size, "Should start with 3 inherited attributes"); + Assert.equal("parent_shared", container.get("shared.key"), "Should get inherited value initially"); + + // Test overriding inherited attribute + Assert.ok(container.set("shared.key", "child_shared"), "Should successfully override inherited attribute"); + Assert.equal("child_shared", container.get("shared.key"), "Should get overridden value"); + Assert.equal(3, container.size, "Size should remain 3 after override"); + + // Test adding new attribute alongside inherited ones + Assert.ok(container.set("child.key", "child_value"), "Should add new attribute"); + Assert.equal("child_value", container.get("child.key"), "Should get new attribute value"); + Assert.equal("parent_value1", container.get("parent.key1"), "Should still get inherited attribute"); + Assert.equal(4, container.size, "Size should be 4 after adding new attribute"); + } + }); + + this.testCase({ + name: "AttributeContainer: Inheritance - iterator methods include inherited attributes", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "parent.key2": "parent_value2", + "shared.key": "parent_shared" + }; + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + + // Add some child attributes + container.set("child.key", "child_value"); + container.set("shared.key", "child_shared"); // Override inherited + + // Test keys iterator includes both inherited and child keys + const keys: string[] = []; + const keysIter = container.keys(); + let keysResult = keysIter.next(); + while (!keysResult.done) { + keys.push(keysResult.value); + keysResult = keysIter.next(); + } + Assert.equal(4, keys.length, "Should have 4 total keys (2 child + 2 non-overridden inherited)"); + Assert.ok(keys.includes("child.key"), "Should include child key"); + Assert.ok(keys.includes("shared.key"), "Should include overridden key"); + Assert.ok(keys.includes("parent.key1"), "Should include inherited key1"); + Assert.ok(keys.includes("parent.key2"), "Should include inherited key2"); + + // Test entries iterator + const entries: [string, any, eAttributeFilter][] = []; + const entriesIter = container.entries(); + let entriesResult = entriesIter.next(); + while (!entriesResult.done) { + entries.push(entriesResult.value); + entriesResult = entriesIter.next(); + } + Assert.equal(4, entries.length, "Should have 4 total entries"); + + // Convert to map by extracting key-value pairs (ignoring source) + const entryMap = new Map(entries.map(entry => [entry[0], entry[1]])); + Assert.equal("child_value", entryMap.get("child.key"), "Should have child entry"); + Assert.equal("child_shared", entryMap.get("shared.key"), "Should have overridden value in entries"); + Assert.equal("parent_value1", entryMap.get("parent.key1"), "Should have inherited entry"); + + // Test forEach + const forEachResults: { [key: string]: any } = {}; + container.forEach((key, value) => { + forEachResults[key] = value; + }); + Assert.equal(4, objKeys(forEachResults).length, "forEach should iterate over all items including inherited"); + Assert.equal("child_shared", forEachResults["shared.key"], "forEach should use overridden value"); + Assert.equal("parent_value1", forEachResults["parent.key1"], "forEach should include inherited value"); + } + }); + + this.testCase({ + name: "AttributeContainer: Inheritance - attributes includes inherited", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "parent.key2": 42, + "shared.key": "parent_shared" + }; + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + + container.set("child.key", "child_value"); + container.set("shared.key", "child_shared"); // Override + + const attributes = container.attributes; + Assert.equal("parent_value1", attributes["parent.key1"], "Should include inherited string"); + Assert.equal(42, attributes["parent.key2"], "Should include inherited number"); + Assert.equal("child_value", attributes["child.key"], "Should include child attribute"); + Assert.equal("child_shared", attributes["shared.key"], "Should use overridden value"); + Assert.equal(4, objKeys(attributes).length, "Should have correct total count"); + } + }); + + this.testCase({ + name: "AttributeContainer: Inheritance - clear removes inherited attributes", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "parent.key2": "parent_value2" + }; + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + + // Verify inherited attributes are present + Assert.equal(2, container.size, "Should start with 2 inherited attributes"); + Assert.equal("parent_value1", container.get("parent.key1"), "Should access inherited attribute"); + + // Add child attribute + container.set("child.key", "child_value"); + Assert.equal(3, container.size, "Should have 3 total attributes"); + + // Clear should remove everything including inherited + container.clear(); + Assert.equal(0, container.size, "Should have 0 attributes after clear"); + Assert.ok(!container.has("parent.key1"), "Should not have inherited attribute after clear"); + Assert.ok(!container.has("child.key"), "Should not have child attribute after clear"); + Assert.equal(undefined, container.get("parent.key1"), "Should return undefined for inherited attribute after clear"); + } + }); + + this.testCase({ + name: "AttributeContainer: Inheritance - empty parent attributes", + test: () => { + const otelCfg: IOTelConfig = {}; + const emptyParent: IOTelAttributes = {}; + const container = createAttributeContainer(otelCfg, "test-empty-parent", emptyParent); + + // Should behave like container without inheritance + Assert.equal(0, container.size, "Should start with 0 attributes"); + container.set("key1", "value1"); + Assert.equal(1, container.size, "Should have 1 attribute after adding"); + Assert.equal("value1", container.get("key1"), "Should get attribute value"); + } + }); + + this.testCase({ + name: "AttributeContainer: Inheritance - null/undefined parent attributes", + test: () => { + const otelCfg: IOTelConfig = {}; + const containerNull = createAttributeContainer(otelCfg, "test-null", null as any); + const containerUndefined = createAttributeContainer(otelCfg, "test-undefined", undefined as any); + + // Both should behave like containers without inheritance + Assert.equal(0, containerNull.size, "Null parent should start with 0 attributes"); + Assert.equal(0, containerUndefined.size, "Undefined parent should start with 0 attributes"); + + containerNull.set("key1", "value1"); + containerUndefined.set("key2", "value2"); + + Assert.equal(1, containerNull.size, "Null parent container should have 1 after adding"); + Assert.equal(1, containerUndefined.size, "Undefined parent container should have 1 after adding"); + } + }); + + this.testCase({ + name: "AttributeContainer: addAttributes function with inheritance", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "parent.key": "parent_value" + }; + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + + const attributesToAdd: IOTelAttributes = { + "added.key1": "added_value1", + "added.key2": 42, + "parent.key": "overridden_value" // Override inherited + }; + + addAttributes(container, attributesToAdd); + + Assert.equal(3, container.size, "Should have 3 attributes after adding"); + Assert.equal("added_value1", container.get("added.key1"), "Should have added attribute 1"); + Assert.equal(42, container.get("added.key2"), "Should have added attribute 2"); + Assert.equal("overridden_value", container.get("parent.key"), "Should override inherited attribute"); + } + }); + + this.testCase({ + name: "AttributeContainer: Complex inheritance scenarios", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "service.name": "parent-service", + "service.version": "1.0.0", + "deployment.environment": "staging", + "telemetry.sdk.name": "opentelemetry", + "common.attribute": "from_parent" + }; + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + + // Add child-specific attributes + container.set("span.kind", "client"); + container.set("http.method", "GET"); + container.set("http.url", "https://api.example.com"); + + // Override some parent attributes + container.set("service.version", "2.0.0"); // Override + container.set("deployment.environment", "production"); // Override + + // Verify final state + Assert.equal(8, container.size, "Should have 8 total attributes (5 parent + 5 child - 2 overrides)"); + + // Check overridden values + Assert.equal("2.0.0", container.get("service.version"), "Should use child version"); + Assert.equal("production", container.get("deployment.environment"), "Should use child environment"); + + // Check inherited values + Assert.equal("parent-service", container.get("service.name"), "Should inherit service name"); + Assert.equal("opentelemetry", container.get("telemetry.sdk.name"), "Should inherit SDK name"); + Assert.equal("from_parent", container.get("common.attribute"), "Should inherit common attribute"); + + // Check child-only values + Assert.equal("client", container.get("span.kind"), "Should have child span kind"); + Assert.equal("GET", container.get("http.method"), "Should have child HTTP method"); + + const finalAttributes = container.attributes; + Assert.equal(8, objKeys(finalAttributes).length, "attributes should return correct count"); + Assert.equal("2.0.0", finalAttributes["service.version"], "attributes should use overridden value"); + Assert.equal("parent-service", finalAttributes["service.name"], "attributes should include inherited value"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Valid container identification", + test: () => { + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test-container"); + + // Test with valid container + Assert.ok(isAttributeContainer(container), "Should identify valid container"); + + // Test with inherited attributes + const inheritedAttribs: IOTelAttributes = { "parent.key": "value" }; + const containerWithInheritance = createAttributeContainer(otelCfg, "test-inherited", inheritedAttribs); + Assert.ok(isAttributeContainer(containerWithInheritance), "Should identify container with inheritance"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Child containers identification", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentContainer = createAttributeContainer(otelCfg, "parent-container"); + parentContainer.set("parent.key", "parent-value"); + + // Test child container created with child() method + const childContainer = parentContainer.child("child-container"); + Assert.ok(isAttributeContainer(childContainer), "Should identify child container as valid"); + + // Test snapshot child container created with child() method + const snapshotContainer = parentContainer.child("snapshot-container", true); + Assert.ok(isAttributeContainer(snapshotContainer), "Should identify snapshot child container as valid"); + + // Verify that child containers have the required methods and properties + Assert.ok(typeof childContainer.child === "function", "Child container should have child method"); + Assert.ok(typeof childContainer.listen === "function", "Child container should have listen method"); + Assert.ok("id" in childContainer, "Child container should have id property"); + Assert.ok("size" in childContainer, "Child container should have size property"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Invalid object identification - null and undefined", + test: () => { + // Test null and undefined + Assert.ok(!isAttributeContainer(null), "Should return false for null"); + Assert.ok(!isAttributeContainer(undefined), "Should return false for undefined"); + Assert.ok(!isAttributeContainer(void 0), "Should return false for void 0"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Invalid object identification - primitive types", + test: () => { + // Test primitive types + Assert.ok(!isAttributeContainer("string"), "Should return false for string"); + Assert.ok(!isAttributeContainer(123), "Should return false for number"); + Assert.ok(!isAttributeContainer(true), "Should return false for boolean"); + Assert.ok(!isAttributeContainer(false), "Should return false for boolean false"); + Assert.ok(!isAttributeContainer(Symbol("test")), "Should return false for symbol"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Invalid object identification - arrays and other objects", + test: () => { + // Test arrays and other objects + Assert.ok(!isAttributeContainer([]), "Should return false for empty array"); + Assert.ok(!isAttributeContainer([1, 2, 3]), "Should return false for array with values"); + Assert.ok(!isAttributeContainer({}), "Should return false for empty object"); + Assert.ok(!isAttributeContainer({ key: "value" }), "Should return false for plain object"); + Assert.ok(!isAttributeContainer(new Date()), "Should return false for Date object"); + Assert.ok(!isAttributeContainer(/regex/), "Should return false for RegExp object"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Missing required properties", + test: () => { + // Test objects missing size property + const missingSize = { + droppedAttributes: 0, + attributes: {}, + id: "test-id", + clear: () => {}, + get: () => {}, + has: () => {}, + set: () => {}, + del: () => {}, + keys: () => {}, + entries: () => {}, + forEach: () => {}, + values: () => {}, + child: () => {}, + listen: () => {} + }; + Assert.ok(!isAttributeContainer(missingSize), "Should return false when missing size property"); + + // Test objects missing droppedAttributes property + const missingDroppedAttributes = { + size: 0, + attributes: {}, + id: "test-id", + clear: () => {}, + get: () => {}, + has: () => {}, + set: () => {}, + del: () => {}, + keys: () => {}, + entries: () => {}, + forEach: () => {}, + values: () => {}, + child: () => {}, + listen: () => {} + }; + Assert.ok(!isAttributeContainer(missingDroppedAttributes), "Should return false when missing droppedAttributes property"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Missing required methods", + test: () => { + const baseObj = { + size: 0, + droppedAttributes: 0, + attributes: {} + }; + + // Test missing each required method + const requiredMethods = ["clear", "get", "has", "set", "del", "keys", "entries", "forEach", "values", "child", "listen"]; + + requiredMethods.forEach(methodName => { + const objMissingMethod = { ...baseObj }; + // Add all methods except the one we're testing + requiredMethods.forEach(method => { + if (method !== methodName) { + (objMissingMethod as any)[method] = () => {}; + } + }); + + Assert.ok(!isAttributeContainer(objMissingMethod), `Should return false when missing ${methodName} method`); + }); + } + }); + + this.testCase({ + name: "isAttributeContainer: Wrong property types", + test: () => { + const baseObj = { + clear: () => {}, + get: () => {}, + has: () => {}, + set: () => {}, + del: () => {}, + keys: () => {}, + entries: () => {}, + forEach: () => {}, + values: () => {}, + child: () => {}, + listen: () => {} + }; + + // Test size property with wrong type + const wrongSizeType = { + ...baseObj, + size: "not-a-number", + droppedAttributes: 0, + attributes: {} + }; + Assert.ok(!isAttributeContainer(wrongSizeType), "Should return false when size is not a number"); + + // Test droppedAttributes property with wrong type + const wrongDroppedAttributesType = { + ...baseObj, + size: 0, + droppedAttributes: "not-a-number" + }; + Assert.ok(!isAttributeContainer(wrongDroppedAttributesType), "Should return false when droppedAttributes is not a number"); + + // Test both properties with wrong types + const bothWrongTypes = { + ...baseObj, + size: true, + droppedAttributes: [] + }; + Assert.ok(!isAttributeContainer(bothWrongTypes), "Should return false when both properties have wrong types"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Wrong method types", + test: () => { + const baseObj = { + size: 0, + droppedAttributes: 0, + attributes: {} + }; + + const requiredMethods = ["clear", "get", "has", "set", "del", "keys", "entries", "forEach", "values", "child", "listen"]; + + requiredMethods.forEach(methodName => { + const objWithWrongMethodType = { ...baseObj }; + // Add all methods as functions except the one we're testing + requiredMethods.forEach(method => { + if (method !== methodName) { + (objWithWrongMethodType as any)[method] = () => {}; + } else { + (objWithWrongMethodType as any)[method] = "not-a-function"; + } + }); + + Assert.ok(!isAttributeContainer(objWithWrongMethodType), `Should return false when ${methodName} is not a function`); + }); + } + }); + + this.testCase({ + name: "isAttributeContainer: Objects with getter properties", + test: () => { + // Test object with lazy properties (similar to how attributeContainer implements size) + const objWithGetters = {}; + + // Define getters for size and droppedAttributes + Object.defineProperty(objWithGetters, "size", { + get: () => 5, + enumerable: true + }); + + Object.defineProperty(objWithGetters, "droppedAttributes", { + get: () => 2, + enumerable: true + }); + + Object.defineProperties(objWithGetters, { + attributes: { + get: () => ({}), + enumerable: true + } + }); + + // Add required methods + const requiredMethods = ["clear", "get", "has", "set", "del", "keys", "entries", "forEach", "values", "child", "listen"]; + requiredMethods.forEach(method => { + (objWithGetters as any)[method] = () => {}; + }); + + // Add required id property + (objWithGetters as any).id = "test-id"; + + Assert.ok(isAttributeContainer(objWithGetters), "Should correctly identify object with getter properties"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Partial interface implementation", + test: () => { + // Test object that has some but not all required properties/methods + const partialImplementation = { + size: 0, + droppedAttributes: 0, + clear: () => {}, + get: () => {}, + has: () => {}, + set: () => {} + // Missing: del, keys, entries, forEach, values, child, listen, id, attributes + }; + + Assert.ok(!isAttributeContainer(partialImplementation), "Should return false for partial implementation"); + + // Test object with all methods but wrong property types + const wrongPropertyTypes = { + size: null, + droppedAttributes: undefined, + attributes: null, + clear: [], + get: () => {}, + has: () => {}, + set: () => {}, + del: () => {}, + keys: () => {}, + entries: () => {}, + forEach: () => {}, + values: () => {}, + child: () => {}, + listen: () => {} + }; + + Assert.ok(!isAttributeContainer(wrongPropertyTypes), "Should return false when properties are null/undefined"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Edge cases and complex objects", + test: () => { + // Test function object (functions are objects in JavaScript) + const func = function() {}; + func.size = 0; + func.droppedAttributes = 0; + func.attributes = {}; + func.id = "test-id"; + func.clear = () => {}; + func.get = () => {}; + func.has = () => {}; + func.set = () => {}; + func.del = () => {}; + func.keys = () => {}; + func.entries = () => {}; + func.forEach = () => {}; + func.values = () => {}; + func.child = () => {}; + func.listen = () => {}; + + Assert.ok(isAttributeContainer(func), "Should identify function object with all required properties"); + + // Test class instance + class MockContainer { + size = 10; + droppedAttributes = 1; + attributes = {}; + id = "mock-id"; + clear() {} + get() {} + has() {} + set() {} + del() {} + keys() {} + entries() {} + forEach() {} + values() {} + child() {} + listen() {} + } + + const mockInstance = new MockContainer(); + Assert.ok(isAttributeContainer(mockInstance), "Should identify class instance with all required properties"); + + // Test object with prototype chain + const prototypeObj = Object.create({ + clear: () => {}, + get: () => {}, + has: () => {} + }); + prototypeObj.size = 0; + prototypeObj.droppedAttributes = 0; + prototypeObj.attributes = {}; + prototypeObj.id = "proto-id"; + prototypeObj.set = () => {}; + prototypeObj.del = () => {}; + prototypeObj.keys = () => {}; + prototypeObj.entries = () => {}; + prototypeObj.forEach = () => {}; + prototypeObj.values = () => {}; + prototypeObj.child = () => {}; + prototypeObj.listen = () => {}; + + Assert.ok(isAttributeContainer(prototypeObj), "Should identify object with methods in prototype chain"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Type guard functionality", + test: () => { + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test-container"); + const notContainer = { some: "object" }; + + // Test type narrowing works correctly + function processContainer(obj: any) { + if (isAttributeContainer(obj)) { + // In this block, TypeScript should know obj is IAttributeContainer + return obj.size; // This should compile without errors + } + return -1; + } + + Assert.equal(0, processContainer(container), "Should return container size for valid container"); + Assert.equal(-1, processContainer(notContainer), "Should return -1 for invalid container"); + Assert.equal(-1, processContainer(null), "Should return -1 for null"); + Assert.equal(-1, processContainer("string"), "Should return -1 for string"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Real container usage scenarios", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Test container after operations + const container = createAttributeContainer(otelCfg, "test-container"); + container.set("key1", "value1"); + container.set("key2", 42); + + Assert.ok(isAttributeContainer(container), "Should identify container after adding attributes"); + + // Test container after clear + container.clear(); + Assert.ok(isAttributeContainer(container), "Should still identify container after clear"); + + // Test container with inheritance + const inheritedAttribs: IOTelAttributes = { + "parent.key1": "value1", + "parent.key2": "value2" + }; + const containerWithInheritance = createAttributeContainer(otelCfg, "test-inherited", inheritedAttribs); + Assert.ok(isAttributeContainer(containerWithInheritance), "Should identify container with inheritance"); + + // Test container after inheritance operations + containerWithInheritance.set("child.key", "child_value"); + Assert.ok(isAttributeContainer(containerWithInheritance), "Should identify container after inheritance operations"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Does not access lazy properties", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create a mock object that has all the required methods and properties + // but will throw if the lazy properties are accessed + const mockContainer = { + id: "mock-container-id", // Add required id property + clear: () => {}, + get: () => undefined, + has: () => false, + set: () => true, + del: () => false, + keys: () => { + return { + next: () => ({ done: true, value: undefined }) + } as Iterator; + }, + entries: () => { + return { + next: () => ({ done: true, value: undefined }) + } as Iterator<[string, any, any]>; + }, + forEach: () => {}, + values: () => { + return { + next: () => ({ done: true, value: undefined }) + } as Iterator; + }, + child: () => ({}), // Add missing child method + listen: () => ({ rm: () => {} }) + }; + + // Add properties that will throw if accessed via getter + let sizeAccessed = false; + let droppedAttributesAccessed = false; + let attributesAccessed = false; + + Object.defineProperty(mockContainer, "size", { + get: () => { + sizeAccessed = true; + throw new Error("size property should not be accessed in isAttributeContainer"); + }, + enumerable: false, + configurable: true + }); + + Object.defineProperty(mockContainer, "droppedAttributes", { + get: () => { + droppedAttributesAccessed = true; + throw new Error("droppedAttributes property should not be accessed in isAttributeContainer"); + }, + enumerable: false, + configurable: true + }); + + Object.defineProperty(mockContainer, "attributes", { + get: () => { + attributesAccessed = true; + throw new Error("attributes property should not be accessed in isAttributeContainer"); + }, + enumerable: false, + configurable: true + }); + + // Test that isAttributeContainer does not access the lazy properties + let result: boolean; + let errorThrown = false; + try { + result = isAttributeContainer(mockContainer); + } catch (error) { + errorThrown = true; + // If an error was thrown, it means a lazy property was accessed + } + + // Verify no error was thrown (meaning no lazy properties were accessed) + Assert.ok(!errorThrown, "isAttributeContainer should not access lazy properties"); + + // Verify the function works correctly + Assert.ok(result!, "Should correctly identify as attribute container"); + + // Verify that none of the lazy properties were accessed + Assert.ok(!sizeAccessed, "size property should not be accessed during isAttributeContainer check"); + Assert.ok(!droppedAttributesAccessed, "droppedAttributes property should not be accessed during isAttributeContainer check"); + Assert.ok(!attributesAccessed, "attributes property should not be accessed during isAttributeContainer check"); + } + }); + + this.testCase({ + name: "isAttributeContainer: Lazy properties behavior verification", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create a real container + const container = createAttributeContainer(otelCfg, "test-container"); + container.set("test.key", "test_value"); + + // Test that the lazy properties exist (using 'in' operator which doesn't trigger getters) + Assert.ok("size" in container, "Container should have size property"); + Assert.ok("droppedAttributes" in container, "Container should have droppedAttributes property"); + Assert.ok("attributes" in container, "Container should have attributes property"); + + // Verify isAttributeContainer works with real container + Assert.ok(isAttributeContainer(container), "Should identify real container correctly"); + + // Now verify the properties actually work when accessed + Assert.equal(1, container.size, "Size should be 1"); + Assert.equal(0, container.droppedAttributes, "DroppedAttributes should be 0"); + Assert.equal(1, Object.keys(container.attributes).length, "Attributes should have 1 key"); + } + }); + + this.testCase({ + name: "AttributeContainer: Container-to-container inheritance - basic functionality", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create parent container with some attributes + const parentContainer = createAttributeContainer(otelCfg, "test-container"); + parentContainer.set("parent.key1", "parent_value1"); + parentContainer.set("parent.key2", "parent_value2"); + parentContainer.set("shared.key", "parent_shared"); + + // Create child container with parent container as inheritance source + const childContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + + // Test inherited attributes are accessible from parent container + Assert.equal("parent_value1", childContainer.get("parent.key1"), "Should get inherited attribute from parent container"); + Assert.equal("parent_value2", childContainer.get("parent.key2"), "Should get second inherited attribute from parent container"); + Assert.equal("parent_shared", childContainer.get("shared.key"), "Should get shared inherited attribute from parent container"); + Assert.ok(childContainer.has("parent.key1"), "Should report inherited attribute from parent container as existing"); + Assert.equal(3, childContainer.size, "Should count inherited attributes from parent container in size"); + } + }); + + this.testCase({ + name: "AttributeContainer: Container-to-container inheritance - override behavior", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create parent container + const parentContainer = createAttributeContainer(otelCfg, "test-container"); + parentContainer.set("parent.key1", "parent_value1"); + parentContainer.set("shared.key", "parent_shared"); + parentContainer.set("override.me", "parent_override"); + + // Create child container with parent container as inheritance source + const childContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + + // Test initial inherited state + Assert.equal(3, childContainer.size, "Should start with 3 inherited attributes from parent container"); + Assert.equal("parent_shared", childContainer.get("shared.key"), "Should get inherited value from parent container initially"); + + // Test overriding inherited attribute from parent container + Assert.ok(childContainer.set("shared.key", "child_shared"), "Should successfully override inherited attribute from parent container"); + Assert.equal("child_shared", childContainer.get("shared.key"), "Should get overridden value instead of parent container value"); + Assert.equal(3, childContainer.size, "Size should remain 3 after override of parent container attribute"); + + // Test adding new attribute alongside inherited ones from parent container + Assert.ok(childContainer.set("child.key", "child_value"), "Should add new attribute to child container"); + Assert.equal("child_value", childContainer.get("child.key"), "Should get new child attribute value"); + Assert.equal("parent_value1", childContainer.get("parent.key1"), "Should still get inherited attribute from parent container"); + Assert.equal(4, childContainer.size, "Size should be 4 after adding new attribute to child container"); + } + }); + + this.testCase({ + name: "AttributeContainer: Container-to-container inheritance - iterator methods", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create parent container + const parentContainer = createAttributeContainer(otelCfg, "test-container"); + parentContainer.set("parent.key1", "parent_value1"); + parentContainer.set("parent.key2", "parent_value2"); + parentContainer.set("shared.key", "parent_shared"); + + // Create child container with parent container as inheritance source + const childContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + + // Add some child attributes + childContainer.set("child.key", "child_value"); + childContainer.set("shared.key", "child_shared"); // Override inherited from parent container + + // Test keys iterator includes both inherited from parent container and child keys + const keys: string[] = []; + const keysIter = childContainer.keys(); + let keysResult = keysIter.next(); + while (!keysResult.done) { + keys.push(keysResult.value); + keysResult = keysIter.next(); + } + Assert.equal(4, keys.length, "Should have 4 total keys (2 child + 2 non-overridden from parent container)"); + Assert.ok(keys.includes("child.key"), "Should include child key"); + Assert.ok(keys.includes("shared.key"), "Should include overridden key"); + Assert.ok(keys.includes("parent.key1"), "Should include inherited key1 from parent container"); + Assert.ok(keys.includes("parent.key2"), "Should include inherited key2 from parent container"); + + // Test entries iterator with parent container inheritance + const entries: [string, any, eAttributeFilter][] = []; + const entriesIter = childContainer.entries(); + let entriesResult = entriesIter.next(); + while (!entriesResult.done) { + entries.push(entriesResult.value); + entriesResult = entriesIter.next(); + } + Assert.equal(4, entries.length, "Should have 4 total entries including parent container attributes"); + + // Convert to map by extracting key-value pairs (ignoring source) + const entryMap = new Map(entries.map(entry => [entry[0], entry[1]])); + Assert.equal("child_value", entryMap.get("child.key"), "Should have child entry"); + Assert.equal("child_shared", entryMap.get("shared.key"), "Should have overridden value in entries"); + Assert.equal("parent_value1", entryMap.get("parent.key1"), "Should have inherited entry from parent container"); + + // Test forEach with parent container inheritance + const forEachResults: { [key: string]: any } = {}; + childContainer.forEach((key, value) => { + forEachResults[key] = value; + }); + Assert.equal(4, objKeys(forEachResults).length, "forEach should iterate over all items including parent container attributes"); + Assert.equal("child_shared", forEachResults["shared.key"], "forEach should use overridden value"); + Assert.equal("parent_value1", forEachResults["parent.key1"], "forEach should include inherited value from parent container"); + } + }); + + this.testCase({ + name: "AttributeContainer: Container-to-container inheritance - attributes includes parent container", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create parent container + const parentContainer = createAttributeContainer(otelCfg, "test-container"); + parentContainer.set("parent.key1", "parent_value1"); + parentContainer.set("parent.key2", 42); + parentContainer.set("shared.key", "parent_shared"); + + // Create child container with parent container as inheritance source + const childContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + + childContainer.set("child.key", "child_value"); + childContainer.set("shared.key", "child_shared"); // Override parent container value + + const attributes = childContainer.attributes; + Assert.equal("parent_value1", attributes["parent.key1"], "Should include inherited string from parent container"); + Assert.equal(42, attributes["parent.key2"], "Should include inherited number from parent container"); + Assert.equal("child_value", attributes["child.key"], "Should include child attribute"); + Assert.equal("child_shared", attributes["shared.key"], "Should use overridden value instead of parent container value"); + Assert.equal(4, objKeys(attributes).length, "Should have correct total count including parent container attributes"); + } + }); + + this.testCase({ + name: "AttributeContainer: Container-to-container inheritance - clear behavior", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create parent container + const parentContainer = createAttributeContainer(otelCfg, "test-container"); + parentContainer.set("parent.key1", "parent_value1"); + parentContainer.set("parent.key2", "parent_value2"); + + // Create child container with parent container as inheritance source + const childContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + + // Verify inherited attributes from parent container are present + Assert.equal(2, childContainer.size, "Should start with 2 inherited attributes from parent container"); + Assert.equal("parent_value1", childContainer.get("parent.key1"), "Should access inherited attribute from parent container"); + + // Add child attribute + childContainer.set("child.key", "child_value"); + Assert.equal(3, childContainer.size, "Should have 3 total attributes"); + + // Clear should remove everything including inherited from parent container + childContainer.clear(); + Assert.equal(0, childContainer.size, "Should have 0 attributes after clear"); + Assert.ok(!childContainer.has("parent.key1"), "Should not have inherited attribute from parent container after clear"); + Assert.ok(!childContainer.has("child.key"), "Should not have child attribute after clear"); + Assert.equal(undefined, childContainer.get("parent.key1"), "Should return undefined for inherited attribute from parent container after clear"); + + // Verify parent container is unaffected by child clear + Assert.equal(2, parentContainer.size, "Parent container should still have its attributes after child clear"); + Assert.equal("parent_value1", parentContainer.get("parent.key1"), "Parent container should still have its values after child clear"); + } + }); + + this.testCase({ + name: "AttributeContainer: Multi-level container inheritance - container with IOTelAttributes parent", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create grandparent attributes (IOTelAttributes) + const grandparentAttribs: IOTelAttributes = { + "grandparent.key1": "grandparent_value1", + "grandparent.key2": "grandparent_value2", + "shared.all": "grandparent_shared" + }; + + // Create parent container with grandparent attributes as inheritance + const parentContainer = createAttributeContainer(otelCfg, "test-grandparent", grandparentAttribs); + parentContainer.set("parent.key", "parent_value"); + parentContainer.set("shared.all", "parent_shared"); // Override grandparent + + // Create child container with parent container as inheritance source + const childContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + childContainer.set("child.key", "child_value"); + childContainer.set("shared.all", "child_shared"); // Override parent (and grandparent) + + // Test access to all levels + Assert.equal("grandparent_value1", childContainer.get("grandparent.key1"), "Should access grandparent attribute through parent container"); + Assert.equal("grandparent_value2", childContainer.get("grandparent.key2"), "Should access second grandparent attribute through parent container"); + Assert.equal("parent_value", childContainer.get("parent.key"), "Should access parent attribute"); + Assert.equal("child_value", childContainer.get("child.key"), "Should access child attribute"); + Assert.equal("child_shared", childContainer.get("shared.all"), "Should get child override value"); + + // Test size includes all levels + Assert.equal(5, childContainer.size, "Should count attributes from all levels (2 grandparent + 1 parent + 1 child + 1 override)"); + + // Test iterator includes all levels + const allKeys: string[] = []; + childContainer.forEach((key, value) => { + allKeys.push(key); + }); + Assert.equal(5, allKeys.length, "forEach should iterate over all levels"); + Assert.ok(allKeys.includes("grandparent.key1"), "Should include grandparent key1"); + Assert.ok(allKeys.includes("grandparent.key2"), "Should include grandparent key2"); + Assert.ok(allKeys.includes("parent.key"), "Should include parent key"); + Assert.ok(allKeys.includes("child.key"), "Should include child key"); + Assert.ok(allKeys.includes("shared.all"), "Should include overridden key"); + } + }); + + this.testCase({ + name: "AttributeContainer: Multi-level container inheritance - container with container parent", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create great-grandparent container + const greatGrandparentContainer = createAttributeContainer(otelCfg, "test-container"); + greatGrandparentContainer.set("great.key", "great_value"); + greatGrandparentContainer.set("shared.multi", "great_shared"); + + // Create grandparent container inheriting from great-grandparent + const grandparentContainer = createAttributeContainer(otelCfg, "test-grandparent", greatGrandparentContainer); + grandparentContainer.set("grandparent.key", "grandparent_value"); + grandparentContainer.set("shared.multi", "grandparent_shared"); // Override great-grandparent + + // Create parent container inheriting from grandparent + const parentContainer = createAttributeContainer(otelCfg, "test-parent", grandparentContainer); + parentContainer.set("parent.key", "parent_value"); + parentContainer.set("shared.multi", "parent_shared"); // Override grandparent + + // Create child container inheriting from parent + const childContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + childContainer.set("child.key", "child_value"); + + // Test access through the inheritance chain + Assert.equal("great_value", childContainer.get("great.key"), "Should access great-grandparent through inheritance chain"); + Assert.equal("grandparent_value", childContainer.get("grandparent.key"), "Should access grandparent through inheritance chain"); + Assert.equal("parent_value", childContainer.get("parent.key"), "Should access parent through inheritance chain"); + Assert.equal("child_value", childContainer.get("child.key"), "Should access child attribute"); + Assert.equal("parent_shared", childContainer.get("shared.multi"), "Should get most recent override in chain"); + + // Test size counts inheritance chain correctly + Assert.equal(5, childContainer.size, "Should count all unique attributes in inheritance chain"); + + // Test that modifications to ancestor containers are reflected + greatGrandparentContainer.set("new.great.key", "new_great_value"); + Assert.equal("new_great_value", childContainer.get("new.great.key"), "Should access newly added great-grandparent attribute"); + Assert.equal(6, childContainer.size, "Size should increase after ancestor modification"); + } + }); + + this.testCase({ + name: "AttributeContainer: Mixed inheritance types - IOTelAttributes and container combinations", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Test 1: IOTelAttributes -> Container -> Container + const baseAttribs: IOTelAttributes = { + "base.key1": "base_value1", + "base.key2": "base_value2", + "shared.mixed": "base_shared" + }; + + const intermediateContainer = createAttributeContainer(otelCfg, "test-base", baseAttribs); + intermediateContainer.set("intermediate.key", "intermediate_value"); + intermediateContainer.set("shared.mixed", "intermediate_shared"); + + const finalContainer = createAttributeContainer(otelCfg, "test-final", intermediateContainer); + finalContainer.set("final.key", "final_value"); + + Assert.equal("base_value1", finalContainer.get("base.key1"), "Should access base IOTelAttributes through container"); + Assert.equal("base_value2", finalContainer.get("base.key2"), "Should access second base attribute through container"); + Assert.equal("intermediate_value", finalContainer.get("intermediate.key"), "Should access intermediate container attribute"); + Assert.equal("final_value", finalContainer.get("final.key"), "Should access final container attribute"); + Assert.equal("intermediate_shared", finalContainer.get("shared.mixed"), "Should get intermediate override of base"); + Assert.equal(5, finalContainer.size, "Should count all attributes through mixed inheritance"); + + // Test 2: Container -> IOTelAttributes (should not work - IOTelAttributes can't inherit from container) + // This tests that addAttributes works with containers + const sourceContainer = createAttributeContainer(otelCfg, "test-container"); + sourceContainer.set("source.key1", "source_value1"); + sourceContainer.set("source.key2", "source_value2"); + + const targetContainer = createAttributeContainer(otelCfg, "test-container"); + addAttributes(targetContainer, sourceContainer); + + Assert.equal("source_value1", targetContainer.get("source.key1"), "Should copy container attributes via addAttributes"); + Assert.equal("source_value2", targetContainer.get("source.key2"), "Should copy second container attribute via addAttributes"); + Assert.equal(2, targetContainer.size, "Should have correct size after adding container attributes"); + } + }); + + this.testCase({ + name: "AttributeContainer: Container inheritance edge cases", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Test empty parent container + const emptyParentContainer = createAttributeContainer(otelCfg, "test-container"); + const childOfEmpty = createAttributeContainer(otelCfg, "test-child-empty", emptyParentContainer); + + Assert.equal(0, childOfEmpty.size, "Should have 0 size with empty parent container"); + childOfEmpty.set("child.key", "child_value"); + Assert.equal(1, childOfEmpty.size, "Should have 1 size after adding to child with empty parent container"); + Assert.equal("child_value", childOfEmpty.get("child.key"), "Should get child attribute with empty parent container"); + + // Test parent container that gets modified after child creation + const dynamicParent = createAttributeContainer(otelCfg, "test-container"); + const dynamicChild = createAttributeContainer(otelCfg, "test-dynamic-child", dynamicParent); + + Assert.equal(0, dynamicChild.size, "Should start with 0 size"); + + // Modify parent after child creation + dynamicParent.set("dynamic.key", "dynamic_value"); + Assert.equal("dynamic_value", dynamicChild.get("dynamic.key"), "Should access dynamically added parent attribute"); + Assert.equal(1, dynamicChild.size, "Should reflect parent modifications in size"); + + // Test parent container that gets cleared after child creation + dynamicParent.clear(); + Assert.equal(undefined, dynamicChild.get("dynamic.key"), "Should not access cleared parent attributes"); + Assert.equal(0, dynamicChild.size, "Should reflect parent clear in size"); + + // Test circular reference prevention (child shouldn't be able to inherit from itself) + const selfContainer = createAttributeContainer(otelCfg, "test-container"); + selfContainer.set("self.key", "self_value"); + // Note: This doesn't create actual circular reference as the inheritance is captured at creation time + const pseudoCircular = createAttributeContainer(otelCfg, "test-pseudo-circular", selfContainer); + pseudoCircular.set("pseudo.key", "pseudo_value"); + + Assert.equal("self_value", pseudoCircular.get("self.key"), "Should access parent attribute"); + Assert.equal("pseudo_value", pseudoCircular.get("pseudo.key"), "Should access own attribute"); + Assert.equal(2, pseudoCircular.size, "Should have correct size with pseudo-circular setup"); + } + }); + + this.testCase({ + name: "AttributeContainer: Container inheritance with addAttributes function", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create source container with inheritance + const baseAttribs: IOTelAttributes = { + "base.key": "base_value" + }; + const sourceContainer = createAttributeContainer(otelCfg, "test-base", baseAttribs); + sourceContainer.set("source.key1", "source_value1"); + sourceContainer.set("source.key2", "source_value2"); + + // Create target container + const targetContainer = createAttributeContainer(otelCfg, "test-container"); + targetContainer.set("target.existing", "existing_value"); + + // Use addAttributes to copy from source container (which has inheritance) + addAttributes(targetContainer, sourceContainer); + + // Verify all attributes from source container (including inherited) are copied + Assert.equal("base_value", targetContainer.get("base.key"), "Should copy inherited attribute from source container"); + Assert.equal("source_value1", targetContainer.get("source.key1"), "Should copy source container attribute 1"); + Assert.equal("source_value2", targetContainer.get("source.key2"), "Should copy source container attribute 2"); + Assert.equal("existing_value", targetContainer.get("target.existing"), "Should preserve existing target attributes"); + Assert.equal(4, targetContainer.size, "Should have correct size after adding container with inheritance"); + + // Test addAttributes with container that has container inheritance + const grandparentContainer = createAttributeContainer(otelCfg, "test-container"); + grandparentContainer.set("grandparent.key", "grandparent_value"); + + const parentContainer = createAttributeContainer(otelCfg, "test-parent", grandparentContainer); + parentContainer.set("parent.key", "parent_value"); + + const newTargetContainer = createAttributeContainer(otelCfg, "test-container"); + addAttributes(newTargetContainer, parentContainer); + + Assert.equal("grandparent_value", newTargetContainer.get("grandparent.key"), "Should copy multi-level inherited attribute"); + Assert.equal("parent_value", newTargetContainer.get("parent.key"), "Should copy parent container attribute"); + Assert.equal(2, newTargetContainer.size, "Should have correct size after adding multi-level container"); + } + }); + + // ===== createSnapshotAttributes Tests ===== + + this.testCase({ + name: "createSnapshotAttributes: Invalid arguments - null and undefined", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Test null source + const nullResult = createAttributeSnapshot(otelCfg, "null-test", null as any); + Assert.ok(isAttributeContainer(nullResult), "Should return container for null source"); + Assert.equal(0, nullResult.size, "Should return empty container for null source"); + + // Test undefined source + const undefinedResult = createAttributeSnapshot(otelCfg, "undefined-test", undefined as any); + Assert.ok(isAttributeContainer(undefinedResult), "Should return container for undefined source"); + Assert.equal(0, undefinedResult.size, "Should return empty container for undefined source"); + + // Verify containers are functional + Assert.ok(nullResult.set("test.key", "test_value"), "Null result container should be functional"); + Assert.equal("test_value", nullResult.get("test.key"), "Null result container should store values"); + + Assert.ok(undefinedResult.set("test.key", "test_value"), "Undefined result container should be functional"); + Assert.equal("test_value", undefinedResult.get("test.key"), "Undefined result container should store values"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: Invalid arguments - primitive types", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Test with primitive types (should create empty containers) + const stringResult = createAttributeSnapshot(otelCfg, "string-test", "string" as any); + Assert.ok(isAttributeContainer(stringResult), "Should return container for string"); + Assert.equal(0, stringResult.size, "Should return empty container for string"); + + const numberResult = createAttributeSnapshot(otelCfg, "number-test", 123 as any); + Assert.ok(isAttributeContainer(numberResult), "Should return container for number"); + Assert.equal(0, numberResult.size, "Should return empty container for number"); + + const booleanResult = createAttributeSnapshot(otelCfg, "boolean-test", true as any); + Assert.ok(isAttributeContainer(booleanResult), "Should return container for boolean"); + Assert.equal(0, booleanResult.size, "Should return empty container for boolean"); + + // Verify containers are functional + Assert.ok(stringResult.set("test.key", "test_value"), "String result container should be functional"); + Assert.ok(numberResult.set("test.key", "test_value"), "Number result container should be functional"); + Assert.ok(booleanResult.set("test.key", "test_value"), "Boolean result container should be functional"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: IOTelAttributes - immediate deep copy", + test: () => { + const otelCfg: IOTelConfig = {}; + const sourceAttribs: IOTelAttributes = { + "service.name": "test-service", + "service.version": "1.0.0", + "deployment.environment": "production", + "request.id": "req-123", + "user.authenticated": true, + "request.size": 1024 + }; + + const snapshotContainer = createAttributeSnapshot(otelCfg, "iotel-snapshot", sourceAttribs); + + // Verify all attributes are copied + Assert.ok(isAttributeContainer(snapshotContainer), "Should return attribute container"); + Assert.equal(6, snapshotContainer.size, "Should have all source attributes"); + Assert.equal("test-service", snapshotContainer.get("service.name"), "Should copy string attribute"); + Assert.equal("1.0.0", snapshotContainer.get("service.version"), "Should copy version string"); + Assert.equal("production", snapshotContainer.get("deployment.environment"), "Should copy environment"); + Assert.equal("req-123", snapshotContainer.get("request.id"), "Should copy request ID"); + Assert.equal(true, snapshotContainer.get("user.authenticated"), "Should copy boolean attribute"); + Assert.equal(1024, snapshotContainer.get("request.size"), "Should copy number attribute"); + + // Verify immutability - changes to source don't affect copy + sourceAttribs["service.name"] = "modified-service"; + sourceAttribs["new.key"] = "new_value"; + delete sourceAttribs["request.id"]; + + Assert.equal("test-service", snapshotContainer.get("service.name"), "Should remain unchanged after source modification"); + Assert.equal(undefined, snapshotContainer.get("new.key"), "Should not have new key added to source"); + Assert.equal("req-123", snapshotContainer.get("request.id"), "Should still have deleted key"); + Assert.equal(6, snapshotContainer.size, "Size should remain unchanged after source modifications"); + + // Verify the snapshot container can be modified independently + Assert.ok(snapshotContainer.set("snapshot.key", "snapshot_value"), "Should be able to add to snapshot container"); + Assert.equal("snapshot_value", snapshotContainer.get("snapshot.key"), "Should get newly added attribute"); + Assert.equal(7, snapshotContainer.size, "Size should increase after adding to snapshot container"); + Assert.equal(undefined, sourceAttribs["snapshot.key"], "Source should not have snapshot container additions"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: IOTelAttributes - empty and single attribute", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Test empty attributes + const emptyAttribs: IOTelAttributes = {}; + const emptySnapshot = createAttributeSnapshot(otelCfg, "empty-snapshot", emptyAttribs); + Assert.equal(0, emptySnapshot.size, "Should create empty container for empty attributes"); + + // Verify independence + emptyAttribs["added.later"] = "added_value"; + Assert.equal(0, emptySnapshot.size, "Empty snapshot should remain empty after source modification"); + Assert.equal(undefined, emptySnapshot.get("added.later"), "Empty snapshot should not have added attributes"); + + // Test single attribute + const singleAttrib: IOTelAttributes = { "single.key": "single_value" }; + const singleSnapshot = createAttributeSnapshot(otelCfg, "single-snapshot", singleAttrib); + Assert.equal(1, singleSnapshot.size, "Should have single attribute"); + Assert.equal("single_value", singleSnapshot.get("single.key"), "Should get single attribute value"); + + // Verify independence + singleAttrib["single.key"] = "modified_value"; + singleAttrib["second.key"] = "second_value"; + Assert.equal("single_value", singleSnapshot.get("single.key"), "Single snapshot should retain original value"); + Assert.equal(undefined, singleSnapshot.get("second.key"), "Single snapshot should not have added attributes"); + Assert.equal(1, singleSnapshot.size, "Single snapshot size should remain 1"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: IOTelAttributes - complex hierarchical keys", + test: () => { + const otelCfg: IOTelConfig = {}; + const complexAttribs: IOTelAttributes = { + "http.request.method": "GET", + "http.request.url": "https://api.example.com/users", + "http.request.headers.user-agent": "Test-Agent/1.0", + "http.request.headers.authorization": "Bearer token123", + "http.response.status_code": 200, + "http.response.headers.content-type": "application/json", + "span.attributes.custom.nested.deep.value": "deep_nested", + "user.session.id": "session-456", + "user.session.start_time": "2023-01-01T00:00:00Z" + }; + + const snapshotContainer = createAttributeSnapshot(otelCfg, "complex-snapshot", complexAttribs); + + // Verify all hierarchical attributes are copied correctly + Assert.equal(9, snapshotContainer.size, "Should have all complex attributes"); + Assert.equal("GET", snapshotContainer.get("http.request.method"), "Should copy HTTP method"); + Assert.equal("https://api.example.com/users", snapshotContainer.get("http.request.url"), "Should copy HTTP URL"); + Assert.equal("Test-Agent/1.0", snapshotContainer.get("http.request.headers.user-agent"), "Should copy deeply nested header"); + Assert.equal("Bearer token123", snapshotContainer.get("http.request.headers.authorization"), "Should copy auth header"); + Assert.equal(200, snapshotContainer.get("http.response.status_code"), "Should copy status code"); + Assert.equal("application/json", snapshotContainer.get("http.response.headers.content-type"), "Should copy content type"); + Assert.equal("deep_nested", snapshotContainer.get("span.attributes.custom.nested.deep.value"), "Should copy very deep nested value"); + Assert.equal("session-456", snapshotContainer.get("user.session.id"), "Should copy session ID"); + Assert.equal("2023-01-01T00:00:00Z", snapshotContainer.get("user.session.start_time"), "Should copy session start time"); + + // Test immutability with complex modifications + complexAttribs["http.request.method"] = "POST"; + complexAttribs["http.request.headers.new-header"] = "new_value"; + delete complexAttribs["user.session.start_time"]; + + Assert.equal("GET", snapshotContainer.get("http.request.method"), "Should retain original HTTP method"); + Assert.equal(undefined, snapshotContainer.get("http.request.headers.new-header"), "Should not have new header"); + Assert.equal("2023-01-01T00:00:00Z", snapshotContainer.get("user.session.start_time"), "Should retain deleted attribute"); + Assert.equal(9, snapshotContainer.size, "Complex snapshot size should remain unchanged"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: IAttributeContainer - basic copy-on-change", + test: () => { + const otelCfg: IOTelConfig = {}; + const sourceContainer = createAttributeContainer(otelCfg, "test-container"); + + // Set up initial state + sourceContainer.set("initial.key1", "initial_value1"); + sourceContainer.set("initial.key2", "initial_value2"); + sourceContainer.set("shared.key", "original_shared"); + + const snapshotContainer = createAttributeSnapshot(otelCfg, "basic-container-snapshot", sourceContainer); + + // Verify initial state is copied + Assert.ok(isAttributeContainer(snapshotContainer), "Should return attribute container"); + Assert.equal(3, snapshotContainer.size, "Should have all initial attributes"); + Assert.equal("initial_value1", snapshotContainer.get("initial.key1"), "Should have initial key1"); + Assert.equal("initial_value2", snapshotContainer.get("initial.key2"), "Should have initial key2"); + Assert.equal("original_shared", snapshotContainer.get("shared.key"), "Should have shared key"); + + // Modify source container after snapshot creation + sourceContainer.set("shared.key", "modified_shared"); + sourceContainer.set("new.key", "new_value"); + + // Verify snapshot container shows pre-change values (lazy copy-on-change) + Assert.equal("original_shared", snapshotContainer.get("shared.key"), "Should retain original shared value due to lazy copy-on-change"); + Assert.equal(undefined, snapshotContainer.get("new.key"), "Should not have new key added to source"); + Assert.equal(3, snapshotContainer.size, "Size should remain unchanged after source modifications"); + + // Verify snapshot container can be modified independently + Assert.ok(snapshotContainer.set("snapshot.only", "snapshot_value"), "Should be able to modify snapshot container"); + Assert.equal("snapshot_value", snapshotContainer.get("snapshot.only"), "Should get snapshot-only value"); + Assert.equal(4, snapshotContainer.size, "Snapshot size should increase"); + Assert.equal(undefined, sourceContainer.get("snapshot.only"), "Source should not have snapshot additions"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: IAttributeContainer - with inheritance", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create source container with inheritance + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "parent.key2": "parent_value2", + "shared.inherited": "parent_shared" + }; + const sourceContainer = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + sourceContainer.set("source.key", "source_value"); + sourceContainer.set("shared.inherited", "source_override"); + + const snapshotContainer = createAttributeSnapshot(otelCfg, "inheritance-snapshot", sourceContainer); + + // Verify all attributes including inherited are captured + Assert.equal(4, snapshotContainer.size, "Should capture all attributes including inherited"); + Assert.equal("parent_value1", snapshotContainer.get("parent.key1"), "Should capture inherited key1"); + Assert.equal("parent_value2", snapshotContainer.get("parent.key2"), "Should capture inherited key2"); + Assert.equal("source_value", snapshotContainer.get("source.key"), "Should capture source key"); + Assert.equal("source_override", snapshotContainer.get("shared.inherited"), "Should capture overridden value"); + + // Modify source container (both own and inherited attributes) + sourceContainer.set("parent.key1", "modified_parent1"); + sourceContainer.set("source.key", "modified_source"); + sourceContainer.set("new.source.key", "new_source_value"); + + // Verify snapshot container retains original state + Assert.equal("parent_value1", snapshotContainer.get("parent.key1"), "Should retain original inherited value"); + Assert.equal("source_value", snapshotContainer.get("source.key"), "Should retain original source value"); + Assert.equal("source_override", snapshotContainer.get("shared.inherited"), "Should retain original override"); + Assert.equal(undefined, snapshotContainer.get("new.source.key"), "Should not have new source attributes"); + Assert.equal(4, snapshotContainer.size, "Size should remain unchanged"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: IAttributeContainer - complex inheritance chain", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create a complex inheritance chain + const grandparentAttribs: IOTelAttributes = { + "grandparent.key1": "grandparent_value1", + "grandparent.key2": "grandparent_value2", + "shared.multi": "grandparent_shared" + }; + + const parentContainer = createAttributeContainer(otelCfg, "test-grandparent", grandparentAttribs); + parentContainer.set("parent.key", "parent_value"); + parentContainer.set("shared.multi", "parent_override"); + + const sourceContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + sourceContainer.set("source.key", "source_value"); + sourceContainer.set("shared.multi", "source_override"); + + const snapshotContainer = createAttributeSnapshot(otelCfg, "complex-inheritance-snapshot", sourceContainer); + + // Verify full inheritance chain is captured + Assert.equal(5, snapshotContainer.size, "Should capture entire inheritance chain"); + Assert.equal("grandparent_value1", snapshotContainer.get("grandparent.key1"), "Should capture grandparent key1"); + Assert.equal("grandparent_value2", snapshotContainer.get("grandparent.key2"), "Should capture grandparent key2"); + Assert.equal("parent_value", snapshotContainer.get("parent.key"), "Should capture parent key"); + Assert.equal("source_value", snapshotContainer.get("source.key"), "Should capture source key"); + Assert.equal("source_override", snapshotContainer.get("shared.multi"), "Should capture final override"); + + // Modify multiple levels of the inheritance chain + parentContainer.set("parent.key", "modified_parent"); + parentContainer.set("new.parent.key", "new_parent_value"); + sourceContainer.set("source.key", "modified_source"); + sourceContainer.set("grandparent.key1", "modified_grandparent"); // Override inherited + sourceContainer.set("new.source.key", "new_source_value"); + + // Verify snapshot container retains original state from all levels + Assert.equal("grandparent_value1", snapshotContainer.get("grandparent.key1"), "Should retain original grandparent value"); + Assert.equal("parent_value", snapshotContainer.get("parent.key"), "Should retain original parent value"); + Assert.equal("source_value", snapshotContainer.get("source.key"), "Should retain original source value"); + Assert.equal(undefined, snapshotContainer.get("new.parent.key"), "Should not have new parent attributes"); + Assert.equal(undefined, snapshotContainer.get("new.source.key"), "Should not have new source attributes"); + Assert.equal(5, snapshotContainer.size, "Size should remain unchanged after complex modifications"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: IAttributeContainer - empty container", + test: () => { + const otelCfg: IOTelConfig = {}; + const emptyContainer = createAttributeContainer(otelCfg, "test-container"); + + const snapshotContainer = createAttributeSnapshot(otelCfg, "empty-container-snapshot", emptyContainer); + + // Verify empty state + Assert.equal(0, snapshotContainer.size, "Should create empty snapshot container"); + Assert.equal(undefined, snapshotContainer.get("any.key"), "Should not have any attributes"); + + // Modify source after snapshot creation + emptyContainer.set("added.later", "added_value"); + emptyContainer.set("another.key", "another_value"); + + // Verify snapshot remains empty + Assert.equal(0, snapshotContainer.size, "Snapshot should remain empty"); + Assert.equal(undefined, snapshotContainer.get("added.later"), "Should not have later additions"); + Assert.equal(undefined, snapshotContainer.get("another.key"), "Should not have later additions"); + + // Verify snapshot can be modified independently + Assert.ok(snapshotContainer.set("snapshot.key", "snapshot_value"), "Should be able to modify empty snapshot"); + Assert.equal("snapshot_value", snapshotContainer.get("snapshot.key"), "Should get snapshot value"); + Assert.equal(1, snapshotContainer.size, "Snapshot size should increase"); + Assert.equal(undefined, emptyContainer.get("snapshot.key"), "Source should not have snapshot additions"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: IAttributeContainer - source container clear operation", + test: () => { + const otelCfg: IOTelConfig = {}; + const sourceContainer = createAttributeContainer(otelCfg, "test-container"); + + // Set up initial state + sourceContainer.set("key1", "value1"); + sourceContainer.set("key2", "value2"); + sourceContainer.set("key3", "value3"); + + const snapshotContainer = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + // Verify initial state + Assert.equal(3, snapshotContainer.size, "Should have all initial attributes"); + Assert.equal("value1", snapshotContainer.get("key1"), "Should have key1"); + Assert.equal("value2", snapshotContainer.get("key2"), "Should have key2"); + Assert.equal("value3", snapshotContainer.get("key3"), "Should have key3"); + + // Clear source container + sourceContainer.clear(); + + // Verify snapshot container preserves pre-clear state + Assert.equal(3, snapshotContainer.size, "Should retain all attributes after source clear"); + Assert.equal("value1", snapshotContainer.get("key1"), "Should retain key1 after source clear"); + Assert.equal("value2", snapshotContainer.get("key2"), "Should retain key2 after source clear"); + Assert.equal("value3", snapshotContainer.get("key3"), "Should retain key3 after source clear"); + + // Add new attributes to cleared source + sourceContainer.set("new.key", "new_value"); + sourceContainer.set("key1", "new_value1"); // Same key, different value + + // Verify snapshot container is unaffected by post-clear additions + Assert.equal(3, snapshotContainer.size, "Size should remain unchanged after post-clear additions"); + Assert.equal("value1", snapshotContainer.get("key1"), "Should retain original key1 value"); + Assert.equal(undefined, snapshotContainer.get("new.key"), "Should not have post-clear additions"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: IAttributeContainer - with inherited container that changes", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create parent container that will be modified + const parentContainer = createAttributeContainer(otelCfg, "test-container"); + parentContainer.set("parent.key1", "parent_value1"); + parentContainer.set("parent.key2", "parent_value2"); + + // Create source container with parent inheritance + const sourceContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + sourceContainer.set("source.key", "source_value"); + sourceContainer.set("parent.key1", "source_override"); // Override parent + + const snapshotContainer = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + // Verify initial state includes inheritance + Assert.equal(3, snapshotContainer.size, "Should have all attributes including inherited"); + Assert.equal("source_override", snapshotContainer.get("parent.key1"), "Should have source override"); + Assert.equal("parent_value2", snapshotContainer.get("parent.key2"), "Should have inherited parent key2"); + Assert.equal("source_value", snapshotContainer.get("source.key"), "Should have source key"); + + // Modify parent container after snapshot creation + parentContainer.set("parent.key1", "modified_parent1"); + parentContainer.set("parent.key2", "modified_parent2"); + parentContainer.set("new.parent.key", "new_parent_value"); + + // Modify source container + sourceContainer.set("source.key", "modified_source"); + sourceContainer.set("new.source.key", "new_source_value"); + + // Verify snapshot container retains original state despite parent and source changes + Assert.equal("source_override", snapshotContainer.get("parent.key1"), "Should retain original override"); + Assert.equal("parent_value2", snapshotContainer.get("parent.key2"), "Should retain original inherited value"); + Assert.equal("source_value", snapshotContainer.get("source.key"), "Should retain original source value"); + Assert.equal(undefined, snapshotContainer.get("new.parent.key"), "Should not have new parent attributes"); + Assert.equal(undefined, snapshotContainer.get("new.source.key"), "Should not have new source attributes"); + Assert.equal(3, snapshotContainer.size, "Size should remain unchanged"); + + // Clear parent container + parentContainer.clear(); + + // Verify snapshot container still retains original state + Assert.equal("source_override", snapshotContainer.get("parent.key1"), "Should retain override after parent clear"); + Assert.equal("parent_value2", snapshotContainer.get("parent.key2"), "Should retain inherited value after parent clear"); + Assert.equal("source_value", snapshotContainer.get("source.key"), "Should retain source value after parent clear"); + Assert.equal(3, snapshotContainer.size, "Size should remain unchanged after parent clear"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: IAttributeContainer - multiple snapshot copies", + test: () => { + const otelCfg: IOTelConfig = {}; + const sourceContainer = createAttributeContainer(otelCfg, "test-container"); + + // Set up initial state + sourceContainer.set("shared.key", "initial_value"); + sourceContainer.set("common.attr", "common_value"); + + // Create first snapshot copy + const snapshot1 = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + // Modify source + sourceContainer.set("shared.key", "modified_value"); + sourceContainer.set("new.key", "new_value"); + + // Create second snapshot copy after modification + const snapshot2 = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + // Verify both snapshot copies are independent + Assert.equal("initial_value", snapshot1.get("shared.key"), "First snapshot should have initial value"); + Assert.equal("modified_value", snapshot2.get("shared.key"), "Second snapshot should have modified value"); + + Assert.equal(undefined, snapshot1.get("new.key"), "First snapshot should not have new key"); + Assert.equal("new_value", snapshot2.get("new.key"), "Second snapshot should have new key"); + + Assert.equal(2, snapshot1.size, "First snapshot should have 2 attributes"); + Assert.equal(3, snapshot2.size, "Second snapshot should have 3 attributes"); + + // Modify both snapshot copies independently + snapshot1.set("snapshot1.key", "snapshot1_value"); + snapshot2.set("snapshot2.key", "snapshot2_value"); + + // Verify independence + Assert.equal("snapshot1_value", snapshot1.get("snapshot1.key"), "First snapshot should have its own addition"); + Assert.equal(undefined, snapshot2.get("snapshot1.key"), "Second snapshot should not have first's addition"); + Assert.equal(undefined, snapshot1.get("snapshot2.key"), "First snapshot should not have second's addition"); + Assert.equal("snapshot2_value", snapshot2.get("snapshot2.key"), "Second snapshot should have its own addition"); + + Assert.equal(3, snapshot1.size, "First snapshot size should be 3"); + Assert.equal(4, snapshot2.size, "Second snapshot size should be 4"); + + // Verify source container is unaffected by snapshot modifications + Assert.equal(undefined, sourceContainer.get("snapshot1.key"), "Source should not have first snapshot's addition"); + Assert.equal(undefined, sourceContainer.get("snapshot2.key"), "Source should not have second snapshot's addition"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: IAttributeContainer - iterators work correctly", + test: () => { + const otelCfg: IOTelConfig = {}; + const sourceContainer = createAttributeContainer(otelCfg, "test-container"); + + // Set up diverse attribute types + sourceContainer.set("string.attr", "string_value"); + sourceContainer.set("number.attr", 42); + sourceContainer.set("boolean.attr", true); + sourceContainer.set("nested.deep.attr", "nested_value"); + + const snapshotContainer = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + // Test keys iterator + const keys: string[] = []; + const keysIter = snapshotContainer.keys(); + let keyResult = keysIter.next(); + while (!keyResult.done) { + keys.push(keyResult.value); + keyResult = keysIter.next(); + } + Assert.equal(4, keys.length, "Should iterate over all keys"); + Assert.ok(keys.includes("string.attr"), "Should include string attribute key"); + Assert.ok(keys.includes("number.attr"), "Should include number attribute key"); + Assert.ok(keys.includes("boolean.attr"), "Should include boolean attribute key"); + Assert.ok(keys.includes("nested.deep.attr"), "Should include nested attribute key"); + + // Test entries iterator + const entries: [string, any, eAttributeFilter][] = []; + const entriesIter = snapshotContainer.entries(); + let entryResult = entriesIter.next(); + while (!entryResult.done) { + entries.push(entryResult.value); + entryResult = entriesIter.next(); + } + Assert.equal(4, entries.length, "Should iterate over all entries"); + + // Convert to map by extracting key-value pairs (ignoring source) + const entryMap = new Map(entries.map(entry => [entry[0], entry[1]])); + Assert.equal("string_value", entryMap.get("string.attr"), "Should have correct string entry"); + Assert.equal(42, entryMap.get("number.attr"), "Should have correct number entry"); + Assert.equal(true, entryMap.get("boolean.attr"), "Should have correct boolean entry"); + Assert.equal("nested_value", entryMap.get("nested.deep.attr"), "Should have correct nested entry"); + + // Test values iterator + const values: any[] = []; + const valuesIter = snapshotContainer.values(); + let valueResult = valuesIter.next(); + while (!valueResult.done) { + values.push(valueResult.value); + valueResult = valuesIter.next(); + } + Assert.equal(4, values.length, "Should iterate over all values"); + Assert.ok(values.includes("string_value"), "Should include string value"); + Assert.ok(values.includes(42), "Should include number value"); + Assert.ok(values.includes(true), "Should include boolean value"); + Assert.ok(values.includes("nested_value"), "Should include nested value"); + + // Test forEach + const forEachResults: { [key: string]: any } = {}; + snapshotContainer.forEach((key, value) => { + forEachResults[key] = value; + }); + Assert.equal(4, objKeys(forEachResults).length, "forEach should process all attributes"); + Assert.equal("string_value", forEachResults["string.attr"], "forEach should provide correct string value"); + Assert.equal(42, forEachResults["number.attr"], "forEach should provide correct number value"); + Assert.equal(true, forEachResults["boolean.attr"], "forEach should provide correct boolean value"); + Assert.equal("nested_value", forEachResults["nested.deep.attr"], "forEach should provide correct nested value"); + + // Test attributes + const attributes = snapshotContainer.attributes; + Assert.equal(4, objKeys(attributes).length, "attributes should include all attributes"); + Assert.equal("string_value", attributes["string.attr"], "attributes should have correct string attribute"); + Assert.equal(42, attributes["number.attr"], "attributes should have correct number attribute"); + Assert.equal(true, attributes["boolean.attr"], "attributes should have correct boolean attribute"); + Assert.equal("nested_value", attributes["nested.deep.attr"], "attributes should have correct nested attribute"); + + // Modify source and verify snapshot iterators are unaffected + sourceContainer.set("string.attr", "modified_string"); + sourceContainer.set("new.attr", "new_value"); + + const postModifyKeys: string[] = []; + const postModifyKeysIter = snapshotContainer.keys(); + let postModifyKeyResult = postModifyKeysIter.next(); + while (!postModifyKeyResult.done) { + postModifyKeys.push(postModifyKeyResult.value); + postModifyKeyResult = postModifyKeysIter.next(); + } + Assert.equal(4, postModifyKeys.length, "Should still have 4 keys after source modification"); + Assert.ok(!postModifyKeys.includes("new.attr"), "Should not include new source attribute"); + + const postModifyAttributes = snapshotContainer.attributes; + Assert.equal("string_value", postModifyAttributes["string.attr"], "Should retain original string value"); + Assert.equal(undefined, postModifyAttributes["new.attr"], "Should not have new source attribute"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: Mixed types - IOTelAttributes vs IAttributeContainer behavior", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create equivalent data in both forms + const sourceAttribs: IOTelAttributes = { + "service.name": "test-service", + "service.version": "1.0.0", + "shared.key": "shared_value" + }; + + const sourceContainer = createAttributeContainer(otelCfg, "test-container"); + sourceContainer.set("service.name", "test-service"); + sourceContainer.set("service.version", "1.0.0"); + sourceContainer.set("shared.key", "shared_value"); + + // Create snapshot versions + const snapshotFromAttribs = createAttributeSnapshot(otelCfg, "test-snapshot", sourceAttribs); + const snapshotFromContainer = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + // Verify both have equivalent initial state + Assert.equal(3, snapshotFromAttribs.size, "IOTelAttributes snapshot should have 3 attributes"); + Assert.equal(3, snapshotFromContainer.size, "IAttributeContainer snapshot should have 3 attributes"); + + Assert.equal("test-service", snapshotFromAttribs.get("service.name"), "IOTelAttributes snapshot should have service name"); + Assert.equal("test-service", snapshotFromContainer.get("service.name"), "IAttributeContainer snapshot should have service name"); + + Assert.equal("shared_value", snapshotFromAttribs.get("shared.key"), "IOTelAttributes snapshot should have shared key"); + Assert.equal("shared_value", snapshotFromContainer.get("shared.key"), "IAttributeContainer snapshot should have shared key"); + + // Modify sources differently + sourceAttribs["service.name"] = "modified-attribs-service"; + sourceAttribs["attribs.only"] = "attribs_only_value"; + + sourceContainer.set("service.name", "modified-container-service"); + sourceContainer.set("container.only", "container_only_value"); + + // Verify different immutability behaviors + + // IOTelAttributes snapshot should be unaffected (immediate copy) + Assert.equal("test-service", snapshotFromAttribs.get("service.name"), "IOTelAttributes snapshot should retain original value"); + Assert.equal(undefined, snapshotFromAttribs.get("attribs.only"), "IOTelAttributes snapshot should not have new attribs"); + Assert.equal(3, snapshotFromAttribs.size, "IOTelAttributes snapshot size should remain 3"); + + // IAttributeContainer snapshot should be unaffected (copy-on-change) + Assert.equal("test-service", snapshotFromContainer.get("service.name"), "IAttributeContainer snapshot should retain original value"); + Assert.equal(undefined, snapshotFromContainer.get("container.only"), "IAttributeContainer snapshot should not have new container attrs"); + Assert.equal(3, snapshotFromContainer.size, "IAttributeContainer snapshot size should remain 3"); + + // Both snapshot containers should function identically for modifications + snapshotFromAttribs.set("snapshot.attribs", "attribs_snapshot_value"); + snapshotFromContainer.set("snapshot.container", "container_snapshot_value"); + + Assert.equal("attribs_snapshot_value", snapshotFromAttribs.get("snapshot.attribs"), "IOTelAttributes snapshot should store new value"); + Assert.equal("container_snapshot_value", snapshotFromContainer.get("snapshot.container"), "IAttributeContainer snapshot should store new value"); + + Assert.equal(4, snapshotFromAttribs.size, "IOTelAttributes snapshot size should increase to 4"); + Assert.equal(4, snapshotFromContainer.size, "IAttributeContainer snapshot size should increase to 4"); + + // Verify sources are unaffected by snapshot modifications + Assert.equal(undefined, sourceAttribs["snapshot.attribs"], "Source attribs should not have snapshot addition"); + Assert.equal(undefined, sourceContainer.get("snapshot.container"), "Source container should not have snapshot addition"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: Stress test - large attribute sets", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create large attribute set + let largeAttribs = createAttributeContainer(otelCfg, "largeAttribs"); + for (let i = 0; i < 100; i++) { + largeAttribs.set(`attr.${i}.key`, `value_${i}`); + largeAttribs.set(`nested.level1.level2.attr${i}`, i); + largeAttribs.set(`boolean.attr.${i}`, i % 2 === 0); + } + + const snapshotContainer = createAttributeSnapshot(otelCfg, "large-snapshot", largeAttribs); + + // Verify all attributes are captured + Assert.equal(128, snapshotContainer.size, "Should have all 128 large attributes"); + Assert.equal(172, snapshotContainer.droppedAttributes, "As the size should hanve been limited the dropped count should reflect the excess attributes"); + + // Spot check various types + Assert.equal("value_0", snapshotContainer.get("attr.0.key"), "Should have first string attribute"); + Assert.equal("value_42", snapshotContainer.get("attr.42.key"), "Should have last string attribute"); + Assert.equal(0, snapshotContainer.get("nested.level1.level2.attr0"), "Should have first number attribute"); + Assert.equal(42, snapshotContainer.get("nested.level1.level2.attr42"), "Should have last number attribute"); + Assert.equal(true, snapshotContainer.get("boolean.attr.0"), "Should have first boolean (true)"); + Assert.equal(false, snapshotContainer.get("boolean.attr.1"), "Should have second boolean (false)"); + Assert.equal(false, snapshotContainer.get("boolean.attr.41"), "Should have second-to-last boolean (false)"); + + // Modify large source + for (let i = 0; i < 100; i++) { + largeAttribs.set(`attr.${i}.key`, `modified_value_${i}`); + largeAttribs.set(`new.attr.${i}`, `new_value_${i}`); + } + + // Verify snapshot container retains original values + Assert.equal(128, snapshotContainer.size, "Size should remain 128 after large source modification"); + Assert.equal(329, snapshotContainer.droppedAttributes, "As the size should hanve been limited the dropped count should reflect the excess attributes"); + Assert.equal("value_0", snapshotContainer.get("attr.0.key"), "Should retain original first value"); + Assert.equal("value_42", snapshotContainer.get("attr.42.key"), "Should retain original last value"); + Assert.equal(undefined, snapshotContainer.get("new.attr.0"), "Should not have new attributes"); + Assert.equal(undefined, snapshotContainer.get("new.attr.99"), "Should not have new attributes"); + + // Test iterator performance with large set + let keyCount = 0; + snapshotContainer.forEach((key, value) => { + keyCount++; + }); + Assert.equal(128, keyCount, "forEach should iterate over all 300 attributes"); + + const attributes = snapshotContainer.attributes; + Assert.equal(128, objKeys(attributes).length, "attributes should return all 300 attributes"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: Edge cases - malformed and special inputs", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Test with attribute objects that have non-string keys (should be handled gracefully) + const weirdAttribs = { + "normal.key": "normal_value", + 123: "numeric_key_value", // This may or may not be included depending on implementation + "": "empty_key_value", + " ": "space_key_value", + "key with spaces": "spaces_value", + "key.with.many.dots.in.it": "many_dots_value" + } as any; + + const snapshotContainer = createAttributeSnapshot(otelCfg, "weird-snapshot", weirdAttribs); + + // Verify normal attributes are handled + Assert.equal("normal_value", snapshotContainer.get("normal.key"), "Should handle normal key"); + Assert.equal("spaces_value", snapshotContainer.get("key with spaces"), "Should handle keys with spaces"); + Assert.equal("many_dots_value", snapshotContainer.get("key.with.many.dots.in.it"), "Should handle keys with many dots"); + + // Test edge case values + const edgeValueAttribs: IOTelAttributes = { + "null.value": null as any, + "undefined.value": undefined as any, + "zero.number": 0, + "false.boolean": false, + "empty.string": "", + "very.long.string": "a".repeat(1000), + "negative.number": -42, + "float.number": 3.14159 + }; + + const edgeSnapshotContainer = createAttributeSnapshot(otelCfg, "edge-snapshot", edgeValueAttribs); + + // Verify edge case values are handled correctly + Assert.equal(null, edgeSnapshotContainer.get("null.value"), "Should handle null value"); + Assert.equal(undefined, edgeSnapshotContainer.get("undefined.value"), "Should handle undefined value"); + Assert.equal(0, edgeSnapshotContainer.get("zero.number"), "Should handle zero value"); + Assert.equal(false, edgeSnapshotContainer.get("false.boolean"), "Should handle false value"); + Assert.equal("", edgeSnapshotContainer.get("empty.string"), "Should handle empty string"); + Assert.equal("a".repeat(1000), edgeSnapshotContainer.get("very.long.string"), "Should handle very long string"); + Assert.equal(-42, edgeSnapshotContainer.get("negative.number"), "Should handle negative number"); + Assert.equal(3.14159, edgeSnapshotContainer.get("float.number"), "Should handle float number"); + + // Test with array-like object (should create empty container) + const arrayLike = ["item1", "item2", "item3"] as any; + const arraySnapshotContainer = createAttributeSnapshot(otelCfg, "array-snapshot", arrayLike); + Assert.equal(0, arraySnapshotContainer.size, "Should create empty container for array-like input"); + + // Test with function object which has properties assigned + const funcObj = function() { return "test"; } as any; + funcObj.customProp = "custom_value"; + const funcSnapshotContainer = createAttributeSnapshot(otelCfg, "func-snapshot", funcObj); + Assert.equal(1, funcSnapshotContainer.size, "Should create container for function input"); + Assert.equal("custom_value", funcSnapshotContainer.get("customProp"), "Should include function property"); + + // Test with function object (should create a container with it's properties) + const funcObj2 = function() { return "test"; } as any; + const func2SnapshotContainer = createAttributeSnapshot(otelCfg, "func2-snapshot", funcObj2); + Assert.equal(0, func2SnapshotContainer.size, "Should create empty container for function input"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: Lazy copy-on-change with eAttributeFilter behavior", + test: () => { + const otelCfg: IOTelConfig = {}; + const sourceContainer = createAttributeContainer(otelCfg, "test-container"); + + // Set up initial state + sourceContainer.set("original.key1", "original_value1"); + sourceContainer.set("shared.key", "original_shared"); + + const snapshotContainer = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + // Before any source changes, snapshot should access inherited values + Assert.equal("original_value1", snapshotContainer.get("original.key1"), "Should get inherited value"); + Assert.equal("original_shared", snapshotContainer.get("shared.key"), "Should get inherited shared value"); + + // Test eAttributeFilter behavior before changes + Assert.ok(snapshotContainer.has("original.key1"), "Should have key (default - includes inherited)"); + Assert.ok(!snapshotContainer.has("original.key1", eAttributeFilter.Local), "Should NOT have key locally before source change"); + Assert.ok(snapshotContainer.has("original.key1", eAttributeFilter.Inherited), "Should have key as inherited before source change"); + + // Now modify the source - this should trigger lazy copy-on-change + sourceContainer.set("shared.key", "modified_shared"); + sourceContainer.set("new.key", "new_value"); + + // After source change, snapshot should preserve original values via local copies + Assert.equal("original_shared", snapshotContainer.get("shared.key"), "Should retain original shared value after source change"); + Assert.equal(undefined, snapshotContainer.get("new.key"), "Should not have new key added after snapshot"); + + // Test eAttributeFilter behavior after changes + Assert.ok(snapshotContainer.has("shared.key"), "Should have shared key (default - includes both local and inherited)"); + Assert.ok(snapshotContainer.has("shared.key", eAttributeFilter.Local), "Should have shared key locally after source change (lazy copy)"); + Assert.ok(snapshotContainer.has("shared.key", eAttributeFilter.Inherited), "Should still have an inherited shared key even after copied locally"); + Assert.equal("original_shared", snapshotContainer.get("shared.key"), "Should return the local copy of shared.key"); + Assert.equal("original_shared", snapshotContainer.get("shared.key", eAttributeFilter.Local), "Should return the local copy of shared.key"); + Assert.equal("original_shared", snapshotContainer.get("shared.key", eAttributeFilter.LocalOrDeleted), "Should return the local copy of shared.key"); + Assert.equal("modified_shared", snapshotContainer.get("shared.key", eAttributeFilter.Inherited), "Should return the inherited copy of shared.key"); + + // Keys that weren't changed should still be inherited + Assert.ok(snapshotContainer.has("original.key1"), "Should still have unchanged key"); + Assert.ok(!snapshotContainer.has("original.key1", eAttributeFilter.Local), "Unchanged key should still NOT be local"); + Assert.ok(snapshotContainer.has("original.key1", eAttributeFilter.Inherited), "Unchanged key should still be inherited"); + + // Test clear operation behavior + sourceContainer.clear(); + + // After source clear, all original keys should be copied locally to preserve snapshot + Assert.equal("original_value1", snapshotContainer.get("original.key1"), "Should still have original key1 after source clear"); + Assert.equal("original_shared", snapshotContainer.get("shared.key"), "Should still have original shared after source clear"); + + // All keys should now be local (copied due to clear operation) + Assert.ok(snapshotContainer.has("original.key1", eAttributeFilter.Local), "Key should be local after source clear"); + Assert.ok(!snapshotContainer.has("original.key1", eAttributeFilter.Inherited), "Key should NOT be inherited after source clear"); + Assert.ok(snapshotContainer.has("shared.key", eAttributeFilter.Local), "Shared key should be local after source clear"); + Assert.ok(!snapshotContainer.has("shared.key", eAttributeFilter.Inherited), "Shared key should NOT be inherited after source clear"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: Iterator behavior with source tracking", + test: () => { + const otelCfg: IOTelConfig = {}; + const sourceContainer = createAttributeContainer(otelCfg, "test-container"); + + sourceContainer.set("inherited.key1", "inherited_value1"); + sourceContainer.set("will.change", "original_value"); + + const snapshotContainer = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + // Add some local keys to snapshot + snapshotContainer.set("local.key1", "local_value1"); + + // Trigger lazy copy by changing source + sourceContainer.set("will.change", "changed_value"); + + // Collect entries with source information + let entries: [string, any, number][] = []; + let entriesIter = snapshotContainer.entries(); + let entryResult = entriesIter.next(); + while (!entryResult.done) { + entries.push(entryResult.value); + entryResult = entriesIter.next(); + } + + // Should have 3 entries: 1 inherited + 1 local copy due to change + 1 purely local + Assert.equal(3, entries.length, "Should have 3 entries total"); + + // Find each entry and verify source + let inheritedEntry = entries.find(e => e[0] === "inherited.key1"); + let changedEntry = entries.find(e => e[0] === "will.change"); + let localEntry = entries.find(e => e[0] === "local.key1"); + + Assert.ok(inheritedEntry, "Should have inherited entry"); + Assert.equal(eAttributeFilter.Inherited, inheritedEntry![2], "inherited.key1 should be marked as inherited"); + + Assert.ok(changedEntry, "Should have changed entry"); + Assert.equal(eAttributeFilter.Local, changedEntry![2], "will.change should be marked as local (lazy copied)"); + Assert.equal("original_value", changedEntry![1], "will.change should have original value"); + + Assert.ok(localEntry, "Should have local entry"); + Assert.equal(eAttributeFilter.Local, localEntry![2], "local.key1 should be marked as local"); + } + }); + + // ===== Delete Functionality Tests ===== + + this.testCase({ + name: "AttributeContainer: Delete functionality - basic delete", + test: () => { + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test-container"); + + // Set some attributes + Assert.ok(container.set("key1", "value1"), "Should set key1"); + Assert.ok(container.set("key2", "value2"), "Should set key2"); + Assert.equal(2, container.size, "Should have 2 attributes"); + + // Delete one attribute + Assert.ok(container.del("key1"), "Should delete key1"); + Assert.ok(!container.has("key1"), "key1 should not exist after delete"); + Assert.equal(undefined, container.get("key1"), "key1 should return undefined after delete"); + Assert.equal("value2", container.get("key2"), "key2 should still exist"); + Assert.equal(1, container.size, "Should have 1 attribute after delete"); + + // Try to delete non-existent key + Assert.ok(!container.del("nonexistent"), "Should return false for non-existent key"); + + // Try to delete already deleted key + Assert.ok(!container.del("key1"), "Should return false for already deleted key"); + } + }); + + this.testCase({ + name: "AttributeContainer: Delete functionality - delete inherited attribute", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "parent.key2": "parent_value2", + "shared.key": "parent_shared" + }; + + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + container.set("local.key", "local_value"); + + // Verify initial state + Assert.equal(4, container.size, "Should have 4 attributes (3 inherited + 1 local)"); + Assert.equal("parent_value1", container.get("parent.key1"), "Should get inherited value"); + Assert.ok(container.has("parent.key1"), "Should have inherited key"); + + // Delete inherited attribute + Assert.ok(container.del("parent.key1"), "Should delete inherited attribute"); + Assert.ok(!container.has("parent.key1"), "Inherited key should not exist after delete"); + Assert.equal(undefined, container.get("parent.key1"), "Deleted inherited key should return undefined"); + Assert.equal(3, container.size, "Size should decrease after deleting inherited key"); + + // Other inherited attributes should still be accessible + Assert.equal("parent_value2", container.get("parent.key2"), "Other inherited attributes should remain"); + Assert.equal("local_value", container.get("local.key"), "Local attributes should remain"); + + // Delete should only affect this container, not the source + const newContainer = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + Assert.equal("parent_value1", newContainer.get("parent.key1"), "Source attributes should be unaffected"); + } + }); + + this.testCase({ + name: "AttributeContainer: Delete functionality - with container inheritance", + test: () => { + const otelCfg: IOTelConfig = {}; + + const parentContainer = createAttributeContainer(otelCfg, "test-container"); + parentContainer.set("parent.key1", "parent_value1"); + parentContainer.set("parent.key2", "parent_value2"); + + const childContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + childContainer.set("child.key", "child_value"); + + // Verify initial state + Assert.equal(3, childContainer.size, "Should have 3 attributes"); + Assert.equal("parent_value1", childContainer.get("parent.key1"), "Should get inherited from parent container"); + + // Delete inherited attribute from parent container + Assert.ok(childContainer.del("parent.key1"), "Should delete inherited attribute from parent container"); + Assert.ok(!childContainer.has("parent.key1"), "Deleted inherited key should not exist"); + Assert.equal(undefined, childContainer.get("parent.key1"), "Deleted inherited key should return undefined"); + Assert.equal(2, childContainer.size, "Size should decrease"); + + // Parent container should still have the attribute + Assert.equal("parent_value1", parentContainer.get("parent.key1"), "Parent container should be unaffected"); + Assert.equal(2, parentContainer.size, "Parent container size should be unchanged"); + + // Child can still access other inherited attributes + Assert.equal("parent_value2", childContainer.get("parent.key2"), "Other inherited attributes should remain"); + } + }); + + this.testCase({ + name: "AttributeContainer: Delete functionality - iterator excludes deleted inherited keys", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "parent.key2": "parent_value2", + "parent.key3": "parent_value3" + }; + + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + container.set("local.key", "local_value"); + + // Delete one inherited key + Assert.ok(container.del("parent.key2"), "Should delete inherited key"); + + // Collect all keys via iterator + const keys: string[] = []; + container.forEach((key, value) => { + keys.push(key); + }); + + Assert.equal(3, keys.length, "Should iterate over 3 keys (2 inherited + 1 local, excluding deleted)"); + Assert.ok(keys.includes("parent.key1"), "Should include non-deleted inherited key1"); + Assert.ok(!keys.includes("parent.key2"), "Should not include deleted inherited key2"); + Assert.ok(keys.includes("parent.key3"), "Should include non-deleted inherited key3"); + Assert.ok(keys.includes("local.key"), "Should include local key"); + + // Test entries iterator + const entries: [string, any, eAttributeFilter][] = []; + const entriesIter = container.entries(); + let result = entriesIter.next(); + while (!result.done) { + entries.push(result.value); + result = entriesIter.next(); + } + + Assert.equal(3, entries.length, "entries() should return 3 entries"); + const entryKeys = entries.map(e => e[0]); + Assert.ok(!entryKeys.includes("parent.key2"), "entries() should not include deleted inherited key"); + } + }); + + this.testCase({ + name: "AttributeContainer: Delete functionality - clear doesn't affect deleted key tracking", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "parent.key2": "parent_value2" + }; + + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + container.set("local.key", "local_value"); + + // Delete inherited key + Assert.ok(container.del("parent.key1"), "Should delete inherited key"); + Assert.ok(!container.has("parent.key1"), "Deleted key should not exist"); + + // Clear container + container.clear(); + Assert.equal(0, container.size, "Size should be 0 after clear"); + + // Deleted key tracking should be reset after clear + // Re-adding parent attributes should make the key available again + const newContainer = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + Assert.equal("parent_value1", newContainer.get("parent.key1"), "Key should be available in new container"); + } + }); + + // ===== Change Operation Type Tests ===== + + this.testCase({ + name: "AttributeContainer: Change operations - Add vs Change operations", + test: () => { + const otelCfg: IOTelConfig = {}; + const container = createAttributeContainer(otelCfg, "test-container"); + + let lastChangeInfo: IAttributeChangeInfo | null = null; + const unloadHook = container.listen((changeInfo) => { + lastChangeInfo = changeInfo; + }); + + // Test Add operation (new key) + Assert.ok(container.set("new.key", "new_value"), "Should add new key"); + Assert.ok(lastChangeInfo, "Should have change info"); + Assert.equal(eAttributeChangeOp.Add, lastChangeInfo!.op, "Should be Add operation for new key"); + Assert.equal("new.key", lastChangeInfo!.k, "Should have correct key"); + Assert.equal("new_value", lastChangeInfo!.val, "Should have new value"); + Assert.equal(undefined, lastChangeInfo!.prev, "Should have no previous value for new key"); + + // Test Change operation (existing key) + lastChangeInfo = null; + Assert.ok(container.set("new.key", "updated_value"), "Should update existing key"); + Assert.ok(lastChangeInfo, "Should have change info for update"); + Assert.equal(eAttributeChangeOp.Set, lastChangeInfo!.op, "Should be Set operation for existing key"); + Assert.equal("new.key", lastChangeInfo!.k, "Should have correct key"); + Assert.equal("updated_value", lastChangeInfo!.val, "Should have updated value"); + Assert.equal("new_value", lastChangeInfo!.prev, "Should have previous value"); + + // Test Delete operation + lastChangeInfo = null; + Assert.ok(container.del("new.key"), "Should delete key"); + Assert.ok(lastChangeInfo, "Should have change info for delete"); + Assert.equal(eAttributeChangeOp.Delete, lastChangeInfo!.op, "Should be Delete operation"); + Assert.equal("new.key", lastChangeInfo!.k, "Should have correct key for delete"); + Assert.equal("updated_value", lastChangeInfo!.prev, "Should have previous value for delete"); + Assert.equal(undefined, lastChangeInfo!.val, "Should have no new value for delete"); + + // Test Clear operation + container.set("key1", "value1"); + container.set("key2", "value2"); + lastChangeInfo = null; + container.clear(); + Assert.ok(lastChangeInfo, "Should have change info for clear"); + Assert.equal(eAttributeChangeOp.Clear, lastChangeInfo!.op, "Should be Clear operation"); + Assert.equal(undefined, lastChangeInfo!.k, "Should have no key for clear"); + + unloadHook.rm(); + } + }); + + this.testCase({ + name: "AttributeContainer: Change operations - with inherited attributes", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "inherited.key": "inherited_value" + }; + + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + + let lastChangeInfo: IAttributeChangeInfo | null = null; + const unloadHook = container.listen((changeInfo) => { + lastChangeInfo = changeInfo; + }); + + // Test overriding inherited key (should be Add operation since it's new locally) + Assert.ok(container.set("inherited.key", "override_value"), "Should override inherited key"); + Assert.ok(lastChangeInfo, "Should have change info"); + Assert.equal(eAttributeChangeOp.Set, lastChangeInfo!.op, "Should be Add operation when overriding inherited key"); + Assert.equal("inherited.key", lastChangeInfo!.k, "Should have correct key"); + Assert.equal("override_value", lastChangeInfo!.val, "Should have override value"); + Assert.equal("inherited_value", lastChangeInfo!.prev, "Should have no previous local value"); + + // Test changing the local override (should be Change operation) + lastChangeInfo = null; + Assert.ok(container.set("inherited.key", "new_override_value"), "Should change local override"); + Assert.ok(lastChangeInfo, "Should have change info for change"); + Assert.equal(eAttributeChangeOp.Set, lastChangeInfo!.op, "Should be Change operation for existing local key"); + Assert.equal("inherited.key", lastChangeInfo!.k, "Should have correct key"); + Assert.equal("new_override_value", lastChangeInfo!.val, "Should have new override value"); + Assert.equal("override_value", lastChangeInfo!.prev, "Should have previous local value"); + + unloadHook.rm(); + } + }); + + // ===== Snapshot Container with Delete Tests ===== + + this.testCase({ + name: "createSnapshotAttributes: Delete operations - source container delete after snapshot", + test: () => { + const otelCfg: IOTelConfig = {}; + const sourceContainer = createAttributeContainer(otelCfg, "test-container"); + + // Set up initial state + sourceContainer.set("key1", "value1"); + sourceContainer.set("key2", "value2"); + sourceContainer.set("key3", "value3"); + + const snapshotContainer = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + // Verify initial state + Assert.equal(3, snapshotContainer.size, "Should have all initial attributes"); + Assert.equal("value1", snapshotContainer.get("key1"), "Should have key1"); + Assert.equal("value2", snapshotContainer.get("key2"), "Should have key2"); + + // Delete from source container after snapshot creation + Assert.ok(sourceContainer.del("key1"), "Should delete key1 from source"); + + // Verify snapshot container preserves pre-delete state (lazy copy-on-change) + Assert.equal("value1", snapshotContainer.get("key1"), "Should retain original key1 value due to lazy copy-on-change"); + Assert.ok(snapshotContainer.has("key1"), "Should still have key1 due to lazy copy-on-change"); + Assert.equal(3, snapshotContainer.size, "Size should remain unchanged after source delete"); + + // Verify source container shows the deletion + Assert.equal(undefined, sourceContainer.get("key1"), "Source should not have deleted key"); + Assert.equal(2, sourceContainer.size, "Source size should decrease after delete"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: Delete operations - snapshot container delete", + test: () => { + const otelCfg: IOTelConfig = {}; + const sourceContainer = createAttributeContainer(otelCfg, "test-container"); + + // Set up initial state + sourceContainer.set("key1", "value1"); + sourceContainer.set("key2", "value2"); + + const snapshotContainer = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + // Verify initial state + Assert.equal(2, snapshotContainer.size, "Should have all initial attributes"); + + // Delete from snapshot container + Assert.ok(snapshotContainer.del("key1"), "Should delete key1 from snapshot"); + Assert.equal(undefined, snapshotContainer.get("key1"), "Snapshot should not have deleted key"); + Assert.ok(!snapshotContainer.has("key1"), "Snapshot should not have deleted key"); + Assert.equal(1, snapshotContainer.size, "Snapshot size should decrease after delete"); + + // Verify source container is unaffected + Assert.equal("value1", sourceContainer.get("key1"), "Source should still have key1"); + Assert.equal(2, sourceContainer.size, "Source size should be unchanged"); + } + }); + + this.testCase({ + name: "createSnapshotAttributes: Delete operations - with inherited attributes", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "parent.key2": "parent_value2" + }; + + const sourceContainer = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + sourceContainer.set("source.key", "source_value"); + + const snapshotContainer = createAttributeSnapshot(otelCfg, "test-snapshot", sourceContainer); + + // Verify initial state with inheritance + Assert.equal(3, snapshotContainer.size, "Should have all attributes including inherited"); + Assert.equal("parent_value1", snapshotContainer.get("parent.key1"), "Should access inherited attribute"); + + // Delete inherited attribute from source after snapshot + Assert.ok(sourceContainer.del("parent.key1"), "Should delete inherited attribute from source"); + + // Verify snapshot preserves the inherited attribute + Assert.equal("parent_value1", snapshotContainer.get("parent.key1"), "Snapshot should preserve inherited attribute"); + Assert.equal(3, snapshotContainer.size, "Snapshot size should remain unchanged"); + + // Delete inherited attribute from snapshot directly + Assert.ok(snapshotContainer.del("parent.key2"), "Should delete inherited attribute from snapshot"); + Assert.equal(undefined, snapshotContainer.get("parent.key2"), "Snapshot should not have deleted inherited attribute"); + Assert.equal(2, snapshotContainer.size, "Snapshot size should decrease"); + + // Source should still have the inherited attribute + Assert.equal("parent_value2", sourceContainer.get("parent.key2"), "Source should still have inherited attribute"); + } + }); + + // ===== Edge Case Tests ===== + + this.testCase({ + name: "AttributeContainer: Delete edge case - inherited key added after deletion", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create parent container that starts empty + const parentContainer = createAttributeContainer(otelCfg, "test-container"); + + // Create child container with empty parent + const childContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + + // Create snapshot of child before any keys exist + const snapshotContainer = createAttributeSnapshot(otelCfg, "child-snapshot", childContainer); + + // Delete a key that doesn't exist yet (preemptive deletion) + Assert.ok(!childContainer.del("future.key"), "Should not be able to delete non-existent key"); + Assert.ok(!childContainer.has("future.key"), "Child should not have the key after deletion"); + + // Now add the key to the parent (this is the edge case) + parentContainer.set("future.key", "parent_value"); + + // Child should still not see the key because it was preemptively deleted + Assert.ok(childContainer.has("future.key"), "Child should still have the key after parent addition as the deletion failed"); + Assert.equal("parent_value", childContainer.get("future.key"), "Child should not get the inherited value"); + + // Snapshot should also not see the key + Assert.ok(!snapshotContainer.has("future.key"), "Snapshot should not have the key as it didn't exist at the point of creation"); + Assert.equal(undefined, snapshotContainer.get("future.key"), "Snapshot should not get the inherited value"); + + // Parent should have the key + Assert.ok(parentContainer.has("future.key"), "Parent should have the key"); + Assert.equal("parent_value", parentContainer.get("future.key"), "Parent should get the value"); + + // New child container should see the inherited key + const newChildContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + Assert.ok(newChildContainer.has("future.key"), "New child should have inherited key"); + Assert.equal("parent_value", newChildContainer.get("future.key"), "New child should get inherited value"); + } + }); + + this.testCase({ + name: "AttributeContainer: Iterator performance - no extra _find calls", + test: () => { + const otelCfg: IOTelConfig = {}; + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "parent.key2": "parent_value2", + "parent.key3": "parent_value3" + }; + + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + container.set("local.key1", "local_value1"); + + // Delete some inherited keys + Assert.ok(container.del("parent.key2"), "Should delete inherited key"); + + // Iterate through all keys - this should use the optimized deletedKeys tracking + const keys: string[] = []; + let iterationCount = 0; + container.forEach((key, value) => { + keys.push(key); + iterationCount++; + }); + + // Should have 3 keys: parent.key1, parent.key3, local.key1 (excluding deleted parent.key2) + Assert.equal(3, keys.length, "Should iterate over correct number of keys"); + Assert.equal(3, iterationCount, "Should iterate exactly 3 times"); + Assert.ok(keys.includes("parent.key1"), "Should include non-deleted inherited key1"); + Assert.ok(!keys.includes("parent.key2"), "Should not include deleted inherited key2"); + Assert.ok(keys.includes("parent.key3"), "Should include non-deleted inherited key3"); + Assert.ok(keys.includes("local.key1"), "Should include local key"); + + // Test entries iterator as well + const entries: [string, any, eAttributeFilter][] = []; + const entriesIter = container.entries(); + let result = entriesIter.next(); + while (!result.done) { + entries.push(result.value); + result = entriesIter.next(); + } + + Assert.equal(3, entries.length, "entries() should return correct number of entries"); + const entryKeys = entries.map(e => e[0]); + Assert.ok(!entryKeys.includes("parent.key2"), "entries() should not include deleted inherited key"); + } + }); + + // ===== Comprehensive Multi-Level Inheritance Tests ===== + + this.testCase({ + name: "AttributeContainer: Deep inheritance chain - 5 levels with delete operations", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create 5-level inheritance chain + const level5Container = createAttributeContainer(otelCfg, "level5-container"); + level5Container.set("level5.key", "level5_value"); + level5Container.set("shared.key", "level5_shared"); + + const level4Container = createAttributeContainer(otelCfg, "level4-container", level5Container); + level4Container.set("level4.key", "level4_value"); + level4Container.set("shared.key", "level4_shared"); // Override level5 + + const level3Container = createAttributeContainer(otelCfg, "level3-container", level4Container); + level3Container.set("level3.key", "level3_value"); + + const level2Container = createAttributeContainer(otelCfg, "level2-container", level3Container); + level2Container.set("level2.key", "level2_value"); + level2Container.set("shared.key", "level2_shared"); // Override level4 + + const level1Container = createAttributeContainer(otelCfg, "level1-container", level2Container); + level1Container.set("level1.key", "level1_value"); + + // Verify full inheritance chain + Assert.equal(6, level1Container.size, "Should have 6 attributes from all levels"); + Assert.equal("level5_value", level1Container.get("level5.key"), "Should access level5 attribute"); + Assert.equal("level4_value", level1Container.get("level4.key"), "Should access level4 attribute"); + Assert.equal("level3_value", level1Container.get("level3.key"), "Should access level3 attribute"); + Assert.equal("level2_value", level1Container.get("level2.key"), "Should access level2 attribute"); + Assert.equal("level1_value", level1Container.get("level1.key"), "Should access level1 attribute"); + Assert.equal("level2_shared", level1Container.get("shared.key"), "Should get level2 override of shared key"); + + // Test delete operations across inheritance chain + Assert.ok(level1Container.del("level4.key"), "Should delete inherited level4 key"); + Assert.ok(!level1Container.has("level4.key"), "level4.key should not exist after delete"); + Assert.equal("level4_value", level4Container.get("level4.key"), "level4 container should still have the key"); + Assert.equal(5, level1Container.size, "Size should decrease after delete"); + + // Delete shared key that's overridden at current level + Assert.ok(level2Container.del("shared.key"), "Should delete overridden shared key"); + Assert.ok(!level2Container.has("shared.key"), "shared.key should not exist after delete"); + Assert.equal("level4_shared", level3Container.get("shared.key"), "level3 should now see level4 value"); + Assert.equal(undefined, level1Container.get("shared.key"), "level1 should not see the level4 value through inheritance as it was deleted at level 2"); + + // Test iterator with deleted keys + const keys: string[] = []; + level1Container.forEach((key, value) => { + keys.push(key); + }); + Assert.equal(4, keys.length, "Should iterate over 4 remaining keys"); + Assert.ok(!keys.includes("level4.key"), "Should not include deleted level4 key"); + Assert.ok(keys.includes("level5.key"), "Should include level5 key"); + } + }); + + this.testCase({ + name: "AttributeContainer: Complex inheritance with snapshot containers", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create inheritance chain + const parentContainer = createAttributeContainer(otelCfg, "test-container"); + parentContainer.set("parent.key1", "parent_value1"); + parentContainer.set("parent.key2", "parent_value2"); + parentContainer.set("shared.key", "parent_shared"); + + const childContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + childContainer.set("child.key", "child_value"); + childContainer.set("shared.key", "child_shared"); // Override parent + + // Create snapshot of child (which includes inheritance) + const snapshot = createAttributeSnapshot(otelCfg, "complex-inheritance-snapshot", childContainer); + + // Verify snapshot captures full inheritance + Assert.equal(4, snapshot.size, "Snapshot should have all inherited attributes"); + Assert.equal("parent_value1", snapshot.get("parent.key1"), "Snapshot should have parent key1"); + Assert.equal("child_shared", snapshot.get("shared.key"), "Snapshot should have child override"); + + // Delete inherited key from child + Assert.ok(childContainer.del("parent.key1"), "Should delete inherited key from child"); + + // Snapshot should preserve original state via lazy copy + Assert.equal("parent_value1", snapshot.get("parent.key1"), "Snapshot should preserve deleted inherited key"); + Assert.ok(snapshot.has("parent.key1", eAttributeFilter.Local), "Deleted key should be copied locally in snapshot"); + + // Test source filtering in snapshot after delete + Assert.ok(!snapshot.has("parent.key1", eAttributeFilter.Inherited), "Should not appear as inherited after lazy copy"); + Assert.ok(snapshot.has("parent.key2", eAttributeFilter.Inherited), "Non-deleted keys should remain inherited"); + + // Modify parent after snapshot - snapshot should copy lazily + parentContainer.set("parent.key2", "modified_parent2"); + Assert.equal("parent_value2", snapshot.get("parent.key2"), "Snapshot should preserve original value"); + Assert.ok(snapshot.has("parent.key2", eAttributeFilter.Local), "Modified inherited key should be copied locally"); + } + }); + + this.testCase({ + name: "AttributeContainer: Circular reference prevention and edge cases", + test: () => { + const otelCfg: IOTelConfig = {}; + + const container1 = createAttributeContainer(otelCfg, "test-container"); + container1.set("container1.key", "value1"); + + // Test with null/undefined inheritance + const containerWithNull = createAttributeContainer(otelCfg, "test-null", null as any); + Assert.equal(0, containerWithNull.size, "Should handle null inheritance"); + + const containerWithUndefined = createAttributeContainer(otelCfg, "test-undefined", undefined as any); + Assert.equal(0, containerWithUndefined.size, "Should handle undefined inheritance"); + + // Test with empty inheritance object + const emptyAttribs: IOTelAttributes = {}; + const containerWithEmpty = createAttributeContainer(otelCfg, "test-empty-attribs", emptyAttribs); + Assert.equal(0, containerWithEmpty.size, "Should handle empty inheritance object"); + + // Test with inheritance object containing undefined/null values + const attribsWithNulls: IOTelAttributes = { + "valid.key": "valid_value", + "null.key": null as any, + "undefined.key": undefined as any, + "empty.string": "", + "zero.value": 0 + }; + + const containerWithNulls = createAttributeContainer(otelCfg, "test-with-nulls", attribsWithNulls); + Assert.equal(5, containerWithNulls.size, "Should include all inheritance attributes including null/undefined"); + Assert.equal("valid_value", containerWithNulls.get("valid.key"), "Should get valid inherited value"); + Assert.equal(null, containerWithNulls.get("null.key"), "Should get null inherited value"); + Assert.equal(undefined, containerWithNulls.get("undefined.key"), "Should get undefined inherited value"); + Assert.equal("", containerWithNulls.get("empty.string"), "Should get empty string"); + Assert.equal(0, containerWithNulls.get("zero.value"), "Should get zero value"); + } + }); + + this.testCase({ + name: "AttributeContainer: Performance with large inheritance chains", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create a large inheritance chain + let currentContainer = createAttributeContainer(otelCfg, "test-container"); + currentContainer.set("root.key", "root_value"); + + // Build 10-level inheritance chain + for (let i = 1; i <= 10; i++) { + const newContainer = createAttributeContainer(otelCfg, `level-${i}-container`, currentContainer); + newContainer.set(`level${i}.key`, `level${i}_value`); + newContainer.set(`shared.level${i}`, `shared_value_${i}`); + currentContainer = newContainer; + } + + // Final container should have access to all levels + Assert.equal(21, currentContainer.size, "Should have all attributes from 10-level chain"); + Assert.equal("root_value", currentContainer.get("root.key"), "Should access root attribute"); + Assert.equal("level5_value", currentContainer.get("level5.key"), "Should access middle level"); + Assert.equal("level10_value", currentContainer.get("level10.key"), "Should access top level"); + + // Test performance of iteration over large inheritance + let count = 0; + currentContainer.forEach((key, value) => { + count++; + }); + Assert.equal(21, count, "Should iterate over all inherited attributes"); + + // Test delete performance in large chain + Assert.ok(currentContainer.del("level5.key"), "Should delete from middle of chain"); + Assert.equal(20, currentContainer.size, "Size should decrease after delete"); + + // Verify iteration still works correctly after delete + count = 0; + currentContainer.forEach((key, value) => { + count++; + }); + Assert.equal(20, count, "Should iterate correctly after delete"); + } + }); + + this.testCase({ + name: "AttributeContainer: Change listener propagation through inheritance", + test: () => { + const otelCfg: IOTelConfig = {}; + + const parentContainer = createAttributeContainer(otelCfg, "test-container"); + parentContainer.set("parent.key", "parent_value"); + + const childContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + childContainer.set("child.key", "child_value"); + + let childChangeInfo: IAttributeChangeInfo | null = null; + const childUnloadHook = childContainer.listen((changeInfo) => { + childChangeInfo = changeInfo; + }); + + let parentChangeInfo: IAttributeChangeInfo | null = null; + const parentUnloadHook = parentContainer.listen((changeInfo) => { + parentChangeInfo = changeInfo; + }); + + // Test parent change propagation to child + parentContainer.set("parent.key", "modified_parent"); + Assert.ok(parentChangeInfo, "Parent should receive change notification"); + Assert.equal(eAttributeChangeOp.Set, parentChangeInfo!.op, "Parent should see Set operation"); + + // Child should be notified about parent changes that don't conflict with local keys + parentContainer.set("new.parent.key", "new_parent_value"); + Assert.ok(childChangeInfo, "Child should receive notification for non-conflicting parent change"); + + // Test parent clear propagation + childChangeInfo = null; + parentContainer.clear(); + Assert.ok(childChangeInfo, "Child should receive clear notification"); + Assert.equal(eAttributeChangeOp.Clear, childChangeInfo!.op, "Child should see Clear operation"); + + // Test delete operations + parentContainer.set("parent.key", "restored_value"); + childChangeInfo = null; + Assert.ok(parentContainer.del("parent.key"), "Should delete parent key"); + Assert.ok(childChangeInfo, "Child should receive delete notification"); + Assert.equal(eAttributeChangeOp.Delete, childChangeInfo!.op, "Child should see Delete operation"); + + childUnloadHook.rm(); + parentUnloadHook.rm(); + } + }); + + this.testCase({ + name: "AttributeContainer: Edge case - delete non-existent keys with inheritance", + test: () => { + const otelCfg: IOTelConfig = {}; + + const parentAttribs: IOTelAttributes = { + "parent.key1": "parent_value1", + "parent.key2": "parent_value2" + }; + + const container = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + + // Test deleting non-existent key that doesn't exist anywhere + Assert.ok(!container.del("completely.nonexistent"), "Should return false for completely non-existent key"); + + // Test deleting inherited key (should return true and mark as deleted) + Assert.ok(container.del("parent.key1"), "Should return true for deleting inherited key"); + Assert.ok(!container.has("parent.key1"), "Inherited key should not exist after delete"); + + // Test deleting already deleted inherited key + Assert.ok(!container.del("parent.key1"), "Should return false for already deleted inherited key"); + + // Test deleting key that might be added to parent later (edge case) + Assert.ok(!container.del("future.key"), "Should return false for future key that doesn't exist yet"); + + // Add the key to parent after deletion mark + const parentContainer = createAttributeContainer(otelCfg, "test-with-inheritance", parentAttribs); + const newChildContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + + // Delete a key that doesn't exist yet + Assert.ok(!newChildContainer.del("future.added.key"), "Should return false for non-existent key"); + + // Add the key to parent after child marked it as deleted + parentContainer.set("future.added.key", "future_value"); + + // Child should still not see the key due to pre-deletion + Assert.ok(newChildContainer.has("future.added.key"), "Child should see key added even though we attempted to delete prior to it existing"); + Assert.equal("future_value", newChildContainer.get("future.added.key"), "Child should return value for prior deleted key"); + + // But new containers should see it + const anotherChildContainer = createAttributeContainer(otelCfg, "test-child", parentContainer); + Assert.equal("future_value", anotherChildContainer.get("future.added.key"), "New child should see parent key"); + } + }); + + this.testCase({ + name: "AttributeContainer: Complex snapshot interactions with deep inheritance", + test: () => { + const otelCfg: IOTelConfig = {}; + + // Create complex inheritance + const greatGrandparent = createAttributeContainer(otelCfg, "test-container"); + greatGrandparent.set("ggp.key", "ggp_value"); + greatGrandparent.set("shared.deep", "ggp_shared"); + + const grandparent = createAttributeContainer(otelCfg, "grandparent-container", greatGrandparent); + grandparent.set("gp.key", "gp_value"); + grandparent.set("shared.deep", "gp_shared"); + + const parent = createAttributeContainer(otelCfg, "parent-container", grandparent); + parent.set("p.key", "p_value"); + + const child = createAttributeContainer(otelCfg, "child-container", parent); + child.set("c.key", "c_value"); + child.set("shared.deep", "c_shared"); + + // Create snapshot of deep inheritance + const snapshot = createAttributeSnapshot(otelCfg, "deep-inheritance-snapshot", child); + + // Verify snapshot captures full inheritance chain + Assert.equal(5, snapshot.size, "Snapshot should capture all levels"); + Assert.equal("ggp_value", snapshot.get("ggp.key"), "Should capture great-grandparent"); + Assert.equal("gp_value", snapshot.get("gp.key"), "Should capture grandparent"); + Assert.equal("p_value", snapshot.get("p.key"), "Should capture parent"); + Assert.equal("c_value", snapshot.get("c.key"), "Should capture child"); + Assert.equal("c_shared", snapshot.get("shared.deep"), "Should capture child override"); + + // Delete at various levels after snapshot + Assert.ok(child.del("shared.deep"), "Child should delete its override"); + Assert.equal(undefined, child.get("shared.deep"), "Child still won't see the grand-parent as it's been deleted"); + Assert.ok(grandparent.del("gp.key"), "Grandparent should delete its key"); + greatGrandparent.clear(); // Clear great-grandparent + + // Snapshot should preserve all original values through lazy copying + Assert.equal("ggp_value", snapshot.get("ggp.key"), "Should preserve ggp key after clear"); + Assert.equal("gp_value", snapshot.get("gp.key"), "Should preserve gp key after delete"); + Assert.equal("c_shared", snapshot.get("shared.deep"), "Should preserve child override after delete"); + Assert.equal(5, snapshot.size, "Snapshot size should remain unchanged"); + + // Test that child now sees different inheritance chain + Assert.equal(undefined, child.get("shared.deep"), "Child should still won't see grandparent value"); + Assert.equal(undefined, child.get("gp.key"), "Child should not see deleted grandparent key"); + Assert.equal(undefined, child.get("ggp.key"), "Child should not see cleared great-grandparent key"); + + // Verify source tracking in snapshot after complex changes + Assert.ok(snapshot.has("shared.deep", eAttributeFilter.Local), "shared.deep should be local in snapshot"); + Assert.ok(snapshot.has("ggp.key", eAttributeFilter.Local), "ggp.key should be copied locally in snapshot"); + Assert.ok(snapshot.has("gp.key", eAttributeFilter.Local), "gp.key should be copied locally in snapshot"); + } + }); + + this.testCase({ + name: "AttributeContainer: Child function - basic functionality without snapshot", + test: () => { + const otelCfg: IOTelConfig = {}; + const parent = createAttributeContainer(otelCfg, "parent-container"); + + // Add some attributes to parent + parent.set("parent.key", "parent-value"); + parent.set("shared.key", "parent-shared"); + + // Create child without snapshot + const child = parent.child("test-child", false); + + // Verify child container basics + Assert.ok(isAttributeContainer(child), "Child should be valid attribute container"); + Assert.notEqual(parent.id, child.id, "Child should have different ID"); + Assert.ok(child.id.includes("<=["), "Child ID should indicate it's a child"); + Assert.ok(child.id.includes(parent.id), "Child ID should reference parent ID"); + Assert.ok(child.id.includes("test-child"), "Child ID should include provided name"); + + // Verify inheritance + Assert.equal("parent-value", child.get("parent.key"), "Child should inherit parent values"); + Assert.equal("parent-shared", child.get("shared.key"), "Child should inherit shared values"); + Assert.equal(2, child.size, "Child size should include inherited attributes"); + + // Verify child can override parent values + Assert.ok(child.set("shared.key", "child-shared"), "Child should be able to override parent"); + Assert.equal("child-shared", child.get("shared.key"), "Child should return overridden value"); + Assert.equal("parent-shared", parent.get("shared.key"), "Parent should retain original value"); + + // Verify child can add new attributes + Assert.ok(child.set("child.key", "child-value"), "Child should be able to add new attributes"); + Assert.equal("child-value", child.get("child.key"), "Child should return new attribute"); + Assert.equal(undefined, parent.get("child.key"), "Parent should not see child-only attributes"); + + // Verify size calculations + Assert.equal(3, child.size, "Child size should include parent + child attributes"); + Assert.equal(2, parent.size, "Parent size should remain unchanged"); + } + }); + + this.testCase({ + name: "AttributeContainer: Child function - basic functionality with snapshot", + test: () => { + const otelCfg: IOTelConfig = {}; + const parent = createAttributeContainer(otelCfg, "parent-container"); + + // Add some attributes to parent + parent.set("parent.key", "parent-value"); + parent.set("shared.key", "parent-shared"); + + // Create child with snapshot + const child = parent.child("test-child", true); + + // Verify child container basics + Assert.ok(isAttributeContainer(child), "Child snapshot should be valid attribute container"); + Assert.notEqual(parent.id, child.id, "Child snapshot should have different ID"); + Assert.ok(child.id.includes("<-@["), "Child ID should indicate it's a snapshot child"); + Assert.ok(child.id.includes(parent.id), "Child ID should reference parent ID"); + Assert.ok(child.id.includes("test-child"), "Child ID should include provided name"); + + // Verify initial inheritance (snapshot captures current state) + Assert.equal("parent-value", child.get("parent.key"), "Child snapshot should have parent values"); + Assert.equal("parent-shared", child.get("shared.key"), "Child snapshot should have shared values"); + Assert.equal(2, child.size, "Child snapshot size should include captured attributes"); + + // Verify child can modify without affecting parent + Assert.ok(child.set("shared.key", "child-shared"), "Child snapshot should allow modifications"); + Assert.equal("child-shared", child.get("shared.key"), "Child snapshot should return modified value"); + Assert.equal("parent-shared", parent.get("shared.key"), "Parent should retain original value"); + + // Verify parent changes don't affect snapshot child + parent.set("parent.key", "parent-modified"); + parent.set("new.key", "new-value"); + Assert.equal("parent-value", child.get("parent.key"), "Child snapshot should retain original parent value"); + Assert.equal(undefined, child.get("new.key"), "Child snapshot should not see new parent attributes"); + Assert.equal(2, child.size, "Child snapshot size should remain stable"); + } + }); + + this.testCase({ + name: "AttributeContainer: Child function - inheritance behavior differences", + test: () => { + const otelCfg: IOTelConfig = {}; + const parent = createAttributeContainer(otelCfg, "parent-container"); + + // Setup initial parent state + parent.set("base.key", "base-value"); + parent.set("dynamic.key", "dynamic-initial"); + + // Create both types of children + const regularChild = parent.child("regular", false); + const snapshotChild = parent.child("snapshot", true); + + // Both should see initial state + Assert.equal("base-value", regularChild.get("base.key"), "Regular child should see initial base value"); + Assert.equal("base-value", snapshotChild.get("base.key"), "Snapshot child should see initial base value"); + Assert.equal("dynamic-initial", regularChild.get("dynamic.key"), "Regular child should see initial dynamic value"); + Assert.equal("dynamic-initial", snapshotChild.get("dynamic.key"), "Snapshot child should see initial dynamic value"); + + // Modify parent after children created + parent.set("dynamic.key", "dynamic-modified"); + parent.set("new.key", "new-value"); + + // Regular child should see changes, snapshot should not + Assert.equal("dynamic-modified", regularChild.get("dynamic.key"), "Regular child should see parent modifications"); + Assert.equal("new-value", regularChild.get("new.key"), "Regular child should see new parent attributes"); + Assert.equal(3, regularChild.size, "Regular child size should include new parent attributes"); + + Assert.equal("dynamic-initial", snapshotChild.get("dynamic.key"), "Snapshot child should retain original value"); + Assert.equal(undefined, snapshotChild.get("new.key"), "Snapshot child should not see new parent attributes"); + Assert.equal(2, snapshotChild.size, "Snapshot child size should remain stable"); + + // Delete from parent + Assert.ok(parent.del("base.key"), "Should be able to delete from parent"); + Assert.equal(undefined, regularChild.get("base.key"), "Regular child should see parent deletions"); + Assert.equal("base-value", snapshotChild.get("base.key"), "Snapshot child should retain deleted values"); + Assert.equal(2, regularChild.size, "Regular child size should reflect parent deletions"); + Assert.equal(2, snapshotChild.size, "Snapshot child size should remain stable"); + } + }); + + this.testCase({ + name: "AttributeContainer: Child function - nested children behavior", + test: () => { + const otelCfg: IOTelConfig = {}; + const grandparent = createAttributeContainer(otelCfg, "grandparent"); + + // Setup inheritance chain + grandparent.set("gp.key", "gp-value"); + grandparent.set("shared.key", "gp-shared"); + + // Create regular child from grandparent + const parent = grandparent.child("parent", false); + parent.set("parent.key", "parent-value"); + parent.set("shared.key", "parent-shared"); // Override grandparent + + // Create both types of grandchildren + const regularGrandchild = parent.child("regular-gc", false); + const snapshotGrandchild = parent.child("snapshot-gc", true); + + // Both grandchildren should see the inheritance chain + Assert.equal("gp-value", regularGrandchild.get("gp.key"), "Regular grandchild should see grandparent values"); + Assert.equal("parent-value", regularGrandchild.get("parent.key"), "Regular grandchild should see parent values"); + Assert.equal("parent-shared", regularGrandchild.get("shared.key"), "Regular grandchild should see parent override"); + + Assert.equal("gp-value", snapshotGrandchild.get("gp.key"), "Snapshot grandchild should see grandparent values"); + Assert.equal("parent-value", snapshotGrandchild.get("parent.key"), "Snapshot grandchild should see parent values"); + Assert.equal("parent-shared", snapshotGrandchild.get("shared.key"), "Snapshot grandchild should see parent override"); + + // Modify grandparent after grandchildren created + grandparent.set("gp.key", "gp-modified"); + grandparent.set("gp.new", "gp-new-value"); + + // Regular grandchild should see changes, snapshot should not + Assert.equal("gp-modified", regularGrandchild.get("gp.key"), "Regular grandchild should see grandparent changes"); + Assert.equal("gp-new-value", regularGrandchild.get("gp.new"), "Regular grandchild should see new grandparent attributes"); + + Assert.equal("gp-value", snapshotGrandchild.get("gp.key"), "Snapshot grandchild should retain original grandparent value"); + Assert.equal(undefined, snapshotGrandchild.get("gp.new"), "Snapshot grandchild should not see new grandparent attributes"); + + // Modify parent after grandchildren created + parent.set("parent.key", "parent-modified"); + + // Regular grandchild should see parent changes, snapshot should not + Assert.equal("parent-modified", regularGrandchild.get("parent.key"), "Regular grandchild should see parent changes"); + Assert.equal("parent-value", snapshotGrandchild.get("parent.key"), "Snapshot grandchild should retain original parent value"); + } + }); + + this.testCase({ + name: "AttributeContainer: Child function - child creation from children", + test: () => { + const otelCfg: IOTelConfig = {}; + const root = createAttributeContainer(otelCfg, "root"); + + root.set("root.key", "root-value"); + + // Create first level children + const regularChild = root.child("regular", false); + const snapshotChild = root.child("snapshot", true); + + regularChild.set("regular.key", "regular-value"); + snapshotChild.set("snapshot.key", "snapshot-value"); + + // Create second level children from each type + const regularFromRegular = regularChild.child("reg-from-reg", false); + const snapshotFromRegular = regularChild.child("snap-from-reg", true); + const regularFromSnapshot = snapshotChild.child("reg-from-snap", false); + const snapshotFromSnapshot = snapshotChild.child("snap-from-snap", true); + + // All should see root value + Assert.equal("root-value", regularFromRegular.get("root.key"), "Regular from regular should see root"); + Assert.equal("root-value", snapshotFromRegular.get("root.key"), "Snapshot from regular should see root"); + Assert.equal("root-value", regularFromSnapshot.get("root.key"), "Regular from snapshot should see root"); + Assert.equal("root-value", snapshotFromSnapshot.get("root.key"), "Snapshot from snapshot should see root"); + + // Only children of regular parent should see regular parent's values + Assert.equal("regular-value", regularFromRegular.get("regular.key"), "Regular from regular should see regular parent"); + Assert.equal("regular-value", snapshotFromRegular.get("regular.key"), "Snapshot from regular should see regular parent"); + Assert.equal(undefined, regularFromSnapshot.get("regular.key"), "Regular from snapshot should not see regular parent"); + Assert.equal(undefined, snapshotFromSnapshot.get("regular.key"), "Snapshot from snapshot should not see regular parent"); + + // Only children of snapshot parent should see snapshot parent's values + Assert.equal(undefined, regularFromRegular.get("snapshot.key"), "Regular from regular should not see snapshot parent"); + Assert.equal(undefined, snapshotFromRegular.get("snapshot.key"), "Snapshot from regular should not see snapshot parent"); + Assert.equal("snapshot-value", regularFromSnapshot.get("snapshot.key"), "Regular from snapshot should see snapshot parent"); + Assert.equal("snapshot-value", snapshotFromSnapshot.get("snapshot.key"), "Snapshot from snapshot should see snapshot parent"); + + // Modify root after all children created + root.set("root.key", "root-modified"); + + // Regular children should see changes, snapshot children should not + Assert.equal("root-modified", regularFromRegular.get("root.key"), "Regular from regular should see root changes"); + Assert.equal("root-value", snapshotFromRegular.get("root.key"), "Snapshot from regular should retain original root"); + Assert.equal("root-value", regularFromSnapshot.get("root.key"), "Regular from snapshot should see root changes"); + Assert.equal("root-value", snapshotFromSnapshot.get("root.key"), "Snapshot from snapshot should retain original root"); + } + }); + + this.testCase({ + name: "AttributeContainer: Child function - clear and delete operations", + test: () => { + const otelCfg: IOTelConfig = {}; + const parent = createAttributeContainer(otelCfg, "parent"); + + parent.set("parent.key1", "value1"); + parent.set("parent.key2", "value2"); + parent.set("shared.key", "parent-shared"); + + const regularChild = parent.child("regular", false); + const snapshotChild = parent.child("snapshot", true); + + // Both children should see all parent attributes + Assert.equal(3, regularChild.size, "Regular child should see all parent attributes"); + Assert.equal(3, snapshotChild.size, "Snapshot child should see all parent attributes"); + + // Add child-specific attributes + regularChild.set("regular.key", "regular-value"); + snapshotChild.set("snapshot.key", "snapshot-value"); + + Assert.equal(4, regularChild.size, "Regular child size should include child attribute"); + Assert.equal(4, snapshotChild.size, "Snapshot child size should include child attribute"); + + // Clear parent + parent.clear(); + + // After parent.clear(), children still maintain parent connection and see the empty parent + Assert.equal(1, regularChild.size, "Regular child should only have own attributes after parent clear"); + Assert.equal("regular-value", regularChild.get("regular.key"), "Regular child should retain own attributes"); + Assert.equal(undefined, regularChild.get("parent.key1"), "Regular child should not see cleared parent attributes"); + + // Snapshot child preserves attributes at time of snapshot creation via lazy copying + Assert.equal(4, snapshotChild.size, "Snapshot child should retain all attributes after parent clear"); + Assert.equal("value1", snapshotChild.get("parent.key1"), "Snapshot child should retain parent attributes"); + Assert.equal("snapshot-value", snapshotChild.get("snapshot.key"), "Snapshot child should retain own attributes"); + + // Clear children - this breaks THEIR parent connections + regularChild.clear(); + snapshotChild.clear(); + + // Both should be empty and lose parent connections + Assert.equal(0, regularChild.size, "Regular child should be empty after clear"); + Assert.equal(0, snapshotChild.size, "Snapshot child should be empty after clear"); + + // Re-add to parent - children should NOT see it since THEIR clear() removed parent connections + parent.set("parent.new", "new-value"); + + // Regular child should NOT see new parent attribute after its own clear() removed parent connection + Assert.equal(0, regularChild.size, "Regular child should not see new parent attribute after its own clear"); + Assert.equal(undefined, regularChild.get("parent.new"), "Regular child should not get new parent value after its own clear"); + + Assert.equal(0, snapshotChild.size, "Snapshot child should remain empty"); + Assert.equal(undefined, snapshotChild.get("parent.new"), "Snapshot child should not see new parent attributes"); + } + }); + + this.testCase({ + name: "AttributeContainer: Child function - listener propagation", + test: () => { + const otelCfg: IOTelConfig = {}; + const parent = createAttributeContainer(otelCfg, "parent"); + + parent.set("parent.key", "parent-value"); + + const regularChild = parent.child("regular", false); + const snapshotChild = parent.child("snapshot", true); + + let regularChildChanges: IAttributeChangeInfo[] = []; + let snapshotChildChanges: IAttributeChangeInfo[] = []; + + // Add listeners to children + const regularHook = regularChild.listen((change) => regularChildChanges.push(change)); + const snapshotHook = snapshotChild.listen((change) => snapshotChildChanges.push(change)); + + // Modify parent + parent.set("parent.key", "parent-modified"); + parent.set("parent.new", "parent-new"); + + // Both regular and snapshot children should receive change notifications + // Regular children propagate parent changes directly + // Snapshot children listen to maintain snapshot state + Assert.equal(2, regularChildChanges.length, "Regular child should receive parent change notifications"); + // Snapshots get an extra notification if a parent changes a value and the local snapshot doesn't have the value or + // it's different from the value being changed. + Assert.equal(4, snapshotChildChanges.length, "Snapshot child should receive parent change notifications to maintain snapshot state"); + + // Verify change details for regular child + Assert.equal("parent.key", regularChildChanges[0].k, "First change should be parent.key"); + Assert.equal(eAttributeChangeOp.Set, regularChildChanges[0].op, "First change should be Set operation"); + Assert.equal("parent.new", regularChildChanges[1].k, "Second change should be parent.new"); + Assert.equal(eAttributeChangeOp.Add, regularChildChanges[1].op, "Second change should be Add operation"); + + // Verify change details for snapshot child (receives notifications to manage snapshot state) + Assert.equal("parent.key", snapshotChildChanges[0].k, "Snapshot child should receive parent.key change"); + Assert.equal(eAttributeChangeOp.Set, snapshotChildChanges[0].op, "Snapshot child should see Set operation"); + Assert.equal("parent.key", snapshotChildChanges[1].k, "Snapshot child should receive 2nd notification for parent.key change"); + Assert.equal(eAttributeChangeOp.Set, snapshotChildChanges[1].op, "Snapshot child should see 2nd Set operation"); + // It won't receive any notifications for "parent.new" as it didn't exist prior to the snapshot + + // Modify children directly + regularChild.set("regular.key", "regular-value"); + snapshotChild.set("snapshot.key", "snapshot-value"); + + // Each child should receive its own change notification + Assert.equal(3, regularChildChanges.length, "Regular child should receive own change"); + Assert.equal(5, snapshotChildChanges.length, "Snapshot child should receive own change"); + + // Clean up listeners + regularHook.rm(); + snapshotHook.rm(); + + // No more changes should be received after listeners removed + parent.set("parent.final", "final-value"); + Assert.equal(3, regularChildChanges.length, "Regular child should not receive changes after listener removed"); + Assert.equal(5, snapshotChildChanges.length, "Snapshot child should not receive changes after listener removed"); + } + }); + + this.testCase({ + name: "AttributeContainer: Child function - clear behavior differences (parent vs child clear)", + test: () => { + const otelCfg: IOTelConfig = {}; + const parent = createAttributeContainer(otelCfg, "parent"); + + parent.set("parent.key", "parent-value"); + + // Test 1: Parent clear - children maintain connection + const regularChild1 = parent.child("regular1", false); + const snapshotChild1 = parent.child("snapshot1", true); + + // Both should see parent initially + Assert.equal(1, regularChild1.size, "Regular child should see parent attribute"); + Assert.equal(1, snapshotChild1.size, "Snapshot child should see parent attribute"); + + // Parent clears - children still have connection but see empty parent + parent.clear(); + Assert.equal(0, regularChild1.size, "Regular child should see empty parent after parent clear"); + Assert.equal(1, snapshotChild1.size, "Snapshot child should preserve attributes after parent clear"); + + // Add new attribute to parent - regular child should see it (connection maintained) + parent.set("new.key", "new-value"); + Assert.equal(1, regularChild1.size, "Regular child should see new parent attribute (connection maintained)"); + Assert.equal("new-value", regularChild1.get("new.key"), "Regular child should get new parent value"); + Assert.equal(1, snapshotChild1.size, "Snapshot child should not see new parent attribute"); + + // Test 2: Child clear - child loses parent connection + const regularChild2 = parent.child("regular2", false); + const snapshotChild2 = parent.child("snapshot2", true); + + // Both should see parent + Assert.equal(1, regularChild2.size, "Regular child2 should see parent attribute"); + Assert.equal(1, snapshotChild2.size, "Snapshot child2 should see parent attribute"); + + // Children clear themselves - they lose parent connection + regularChild2.clear(); + snapshotChild2.clear(); + + Assert.equal(0, regularChild2.size, "Regular child2 should be empty after its own clear"); + Assert.equal(0, snapshotChild2.size, "Snapshot child2 should be empty after its own clear"); + + // Add another attribute to parent - cleared children should NOT see it (connection lost) + parent.set("another.key", "another-value"); + Assert.equal(0, regularChild2.size, "Regular child2 should not see parent additions after its own clear"); + Assert.equal(0, snapshotChild2.size, "Snapshot child2 should not see parent additions after its own clear"); + + // But the first regular child should still see it (it didn't clear itself) + Assert.equal(2, regularChild1.size, "Regular child1 should see all parent attributes (it didn't clear itself)"); + Assert.equal("another-value", regularChild1.get("another.key"), "Regular child1 should get the latest parent value"); + } + }); + } +} + + diff --git a/shared/AppInsightsCore/Tests/Unit/src/Dynamic.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/config/Dynamic.Tests.ts similarity index 98% rename from shared/AppInsightsCore/Tests/Unit/src/Dynamic.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/config/Dynamic.Tests.ts index fe5c0ff06..916234bb2 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/Dynamic.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/config/Dynamic.Tests.ts @@ -1,16 +1,16 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; import { dumpObj } from "@nevware21/ts-utils"; -import { AppInsightsCore } from "../../../src/JavaScriptSDK/AppInsightsCore"; -import { _eInternalMessageId } from "../../../src/JavaScriptSDK.Enums/LoggingEnums"; -import { _InternalLogMessage } from "../../../src/JavaScriptSDK/DiagnosticLogger"; -import { IConfiguration } from "../../../src/JavaScriptSDK.Interfaces/IConfiguration"; -import { IPlugin, ITelemetryPlugin } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryPlugin"; -import { ITelemetryItem } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryItem"; -import { BaseTelemetryPlugin } from "../../../src/JavaScriptSDK/BaseTelemetryPlugin"; -import { IAppInsightsCore } from "../../../src/JavaScriptSDK.Interfaces/IAppInsightsCore"; -import { ITelemetryPluginChain } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryPluginChain"; -import { IProcessTelemetryContext } from "../../../src/JavaScriptSDK.Interfaces/IProcessTelemetryContext"; -import { OldTrackPlugin, TestChannelPlugin, TestPlugin } from "./TestPlugins"; +import { AppInsightsCore } from "../../../../src/core/AppInsightsCore"; +import { _eInternalMessageId } from "../../../../src/enums/ai/LoggingEnums"; +import { _InternalLogMessage } from "../../../../src/diagnostics/DiagnosticLogger"; +import { IConfiguration } from "../../../../src/interfaces/ai/IConfiguration"; +import { IPlugin, ITelemetryPlugin } from "../../../../src/interfaces/ai/ITelemetryPlugin"; +import { ITelemetryItem } from "../../../../src/interfaces/ai/ITelemetryItem"; +import { BaseTelemetryPlugin } from "../../../../src/core/BaseTelemetryPlugin"; +import { IAppInsightsCore } from "../../../../src/interfaces/ai/IAppInsightsCore"; +import { ITelemetryPluginChain } from "../../../../src/interfaces/ai/ITelemetryPluginChain"; +import { IProcessTelemetryContext } from "../../../../src/interfaces/ai/IProcessTelemetryContext"; +import { OldTrackPlugin, TestChannelPlugin, TestPlugin } from "../ai/TestPlugins"; export class DynamicTests extends AITestClass { diff --git a/shared/AppInsightsCore/Tests/Unit/src/DynamicConfig.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/config/DynamicConfig.Tests.ts similarity index 98% rename from shared/AppInsightsCore/Tests/Unit/src/DynamicConfig.Tests.ts rename to shared/AppInsightsCore/Tests/Unit/src/config/DynamicConfig.Tests.ts index 78b07e4a3..9e1b2ee03 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/DynamicConfig.Tests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/config/DynamicConfig.Tests.ts @@ -1,21 +1,21 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; -import { eLoggingSeverity, _eInternalMessageId, LoggingSeverity, _InternalMessageId } from "../../../src/JavaScriptSDK.Enums/LoggingEnums"; -import { _InternalLogMessage } from "../../../src/JavaScriptSDK/DiagnosticLogger"; -import { IConfigDefaults } from "../../../src/Config/IConfigDefaults"; -import { IConfiguration } from "../../../src/JavaScriptSDK.Interfaces/IConfiguration"; -import { blockDynamicConversion, forceDynamicConversion, getDynamicConfigHandler } from "../../../src/Config/DynamicSupport"; -import { createDynamicConfig, onConfigChange } from "../../../src/Config/DynamicConfig"; +import { eLoggingSeverity, _eInternalMessageId, LoggingSeverity, _InternalMessageId } from "../../../../src/enums/ai/LoggingEnums"; +import { _InternalLogMessage } from "../../../../src/diagnostics/DiagnosticLogger"; +import { IConfigDefaults } from "../../../../src/interfaces/config/IConfigDefaults"; +import { IConfiguration } from "../../../../src/interfaces/ai/IConfiguration"; +import { blockDynamicConversion, forceDynamicConversion, getDynamicConfigHandler } from "../../../../src/config/DynamicSupport"; +import { createDynamicConfig, onConfigChange } from "../../../../src/config/DynamicConfig"; import { arrForEach, dumpObj, isArray, isFunction, objForEachKey, objKeys, isPlainObject, objHasOwn, objDeepFreeze, objDefineProps, strContains } from "@nevware21/ts-utils"; -import { IAppInsightsCore } from "../../../src/JavaScriptSDK.Interfaces/IAppInsightsCore"; -import { INotificationManager } from "../../../src/JavaScriptSDK.Interfaces/INotificationManager"; -import { IPerfManager } from "../../../src/JavaScriptSDK.Interfaces/IPerfManager"; -import { AppInsightsCore, DiagnosticLogger, IDiagnosticLogger, IProcessTelemetryContext } from "../../../src/applicationinsights-core-js"; -import { ITelemetryItem } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryItem"; -import { ITelemetryPluginChain } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryPluginChain"; -import { ITelemetryPlugin } from "../../../src/JavaScriptSDK.Interfaces/ITelemetryPlugin"; -import { IChannelControls } from "../../../src/JavaScriptSDK.Interfaces/IChannelControls"; -import { TestPlugin, TestSamplingPlugin, TrackPlugin } from "./TestPlugins"; -import { STR_CHANNELS, STR_CREATE_PERF_MGR, STR_EXTENSIONS, STR_EXTENSION_CONFIG, UNDEFINED_VALUE } from "../../../src/JavaScriptSDK/InternalConstants"; +import { IAppInsightsCore } from "../../../../src/interfaces/ai/IAppInsightsCore"; +import { INotificationManager } from "../../../../src/interfaces/ai/INotificationManager"; +import { IPerfManager } from "../../../../src/interfaces/ai/IPerfManager"; +import { AppInsightsCore, DiagnosticLogger, IDiagnosticLogger, IProcessTelemetryContext } from "../../../../src/index"; +import { ITelemetryItem } from "../../../../src/interfaces/ai/ITelemetryItem"; +import { ITelemetryPluginChain } from "../../../../src/interfaces/ai/ITelemetryPluginChain"; +import { ITelemetryPlugin } from "../../../../src/interfaces/ai/ITelemetryPlugin"; +import { IChannelControls } from "../../../../src/interfaces/ai/IChannelControls"; +import { TestPlugin, TestSamplingPlugin, TrackPlugin } from "../ai/TestPlugins"; +import { STR_CHANNELS, STR_CREATE_PERF_MGR, STR_EXTENSIONS, STR_EXTENSION_CONFIG, UNDEFINED_VALUE } from "../../../../src/constants/InternalConstants"; const coreDefaultConfig: IConfigDefaults = objDeepFreeze({ cookieCfg: {}, diff --git a/shared/AppInsightsCore/Tests/Unit/src/trace/W3CTraceStateModes.tests.ts b/shared/AppInsightsCore/Tests/Unit/src/trace/W3CTraceStateModes.tests.ts new file mode 100644 index 000000000..de4b0d8f8 --- /dev/null +++ b/shared/AppInsightsCore/Tests/Unit/src/trace/W3CTraceStateModes.tests.ts @@ -0,0 +1,224 @@ +import { Assert, AITestClass } from "@microsoft/ai-test-framework"; +import { eDistributedTracingModes } from "../../../../src/enums/ai/Enums"; + +/** + * Helper function to check if a mode should include tracestate header + * This matches the internal implementation in the SDK + */ +function _checkTraceStateBit(distributedTracingMode: eDistributedTracingModes): boolean { + // Use the proper bitwise check for the _W3CTraceState flag (which is an actual bit flag) + return (distributedTracingMode & eDistributedTracingModes._W3CTraceState) === eDistributedTracingModes._W3CTraceState; +} + +/** + * Tests for W3C TraceState Configuration with different distributed tracing modes + */ +export class W3CTraceStateModesTests extends AITestClass { + + public testInitialize() { + // No special setup required + } + + public testCleanup() { + // No special cleanup required + } + + public registerTests() { + this.testCase({ + name: "W3CTraceStateModes: Bit masks work correctly for trace modes", + test: () => { + // Test AI_AND_W3C_TRACE mode + var tracingMode = eDistributedTracingModes.AI_AND_W3C_TRACE; + + // Get the base mode using _BaseMask + var baseMode = tracingMode & eDistributedTracingModes._BaseMask; + + // Assert base mode equals AI_AND_W3C + Assert.equal(baseMode, eDistributedTracingModes.AI_AND_W3C, + "AI_AND_W3C_TRACE base mode should be AI_AND_W3C"); + + // Assert W3C_TRACE bit is set (testing the _W3CTraceState bit) + Assert.equal(true, _checkTraceStateBit(tracingMode), + "AI_AND_W3C_TRACE should include _W3CTraceState bit"); + + // Test W3C_TRACE mode + tracingMode = eDistributedTracingModes.W3C_TRACE; + + // Get the base mode using _BaseMask + baseMode = tracingMode & eDistributedTracingModes._BaseMask; + + // Assert base mode equals W3C + Assert.equal(baseMode, eDistributedTracingModes.W3C, + "W3C_TRACE base mode should be W3C"); + + // Assert it's not equal to AI mode + Assert.notEqual(baseMode, eDistributedTracingModes.AI, + "W3C_TRACE base mode should NOT be AI"); + + // Assert W3C_TRACE bit is set + Assert.equal(true, _checkTraceStateBit(tracingMode), + "W3C_TRACE should include _W3CTraceState bit"); + + // Test AI_AND_W3C mode (without the trace bit) + tracingMode = eDistributedTracingModes.AI_AND_W3C; + + // Get base mode - for AI_AND_W3C it's already the base mode + baseMode = tracingMode & eDistributedTracingModes._BaseMask; + + // Assert it equals AI_AND_W3C + Assert.equal(baseMode, eDistributedTracingModes.AI_AND_W3C, + "AI_AND_W3C should be its own base mode"); + + // Assert it's not equal to just AI mode + Assert.notEqual(baseMode, eDistributedTracingModes.AI, + "AI_AND_W3C should NOT be AI"); + + // Assert W3C_TRACE bit is NOT set + Assert.equal(false, _checkTraceStateBit(tracingMode), + "AI_AND_W3C should NOT include _W3CTraceState bit"); + + // Test W3C mode (without trace bit) + tracingMode = eDistributedTracingModes.W3C; + + // Get the base mode - for W3C it's already the base mode + baseMode = tracingMode & eDistributedTracingModes._BaseMask; + + // Assert it equals W3C + Assert.equal(baseMode, eDistributedTracingModes.W3C, + "W3C should be its own base mode"); + + // Assert it's not equal to AI mode + Assert.notEqual(baseMode, eDistributedTracingModes.AI, + "W3C should NOT be AI"); + + // Assert W3C_TRACE bit is NOT set + Assert.equal(false, _checkTraceStateBit(tracingMode), + "W3C should NOT include _W3CTraceState bit"); + } + }); + + this.testCase({ + name: "W3CTraceStateModes: Changing mode dynamically updates tracestate header behavior", + test: () => { + // First set mode without tracestate bit + var tracingMode = eDistributedTracingModes.AI_AND_W3C; + + // Verify no tracestate initially + Assert.equal(false, _checkTraceStateBit(tracingMode), + "AI_AND_W3C should NOT include _W3CTraceState bit"); + + // Change to mode with tracestate bit + tracingMode = eDistributedTracingModes.AI_AND_W3C_TRACE; + + // Verify tracestate bit is now set + Assert.equal(true, _checkTraceStateBit(tracingMode), + "AI_AND_W3C_TRACE should include _W3CTraceState bit"); + + // Change to W3C_TRACE mode + tracingMode = eDistributedTracingModes.W3C_TRACE; + + // Verify tracestate bit is still set + Assert.equal(true, _checkTraceStateBit(tracingMode), + "W3C_TRACE should include _W3CTraceState bit"); + + // Get the base mode + var baseMode2 = tracingMode & eDistributedTracingModes._BaseMask; + + // Assert base mode is W3C and not AI + Assert.equal(baseMode2, eDistributedTracingModes.W3C, + "W3C_TRACE base mode should be W3C"); + Assert.notEqual(baseMode2, eDistributedTracingModes.AI, + "W3C_TRACE base mode should NOT be AI"); + + // Verify bitmask calculations work correctly + Assert.equal(eDistributedTracingModes.W3C_TRACE, + eDistributedTracingModes.W3C | eDistributedTracingModes._W3CTraceState, + "W3C_TRACE should equal W3C | _W3CTraceState"); + + Assert.equal(eDistributedTracingModes.AI_AND_W3C_TRACE, + eDistributedTracingModes.AI_AND_W3C | eDistributedTracingModes._W3CTraceState, + "AI_AND_W3C_TRACE should equal AI_AND_W3C | _W3CTraceState"); + } + }); + + this.testCase({ + name: "W3CTraceStateModes: _BaseMask correctly isolates base mode from tracestate bit", + test: () => { + // Test AI_AND_W3C_TRACE mode with _BaseMask + var tracingMode = eDistributedTracingModes.AI_AND_W3C_TRACE; + var baseMode = tracingMode & eDistributedTracingModes._BaseMask; + + // Assert base mode equals AI_AND_W3C + Assert.equal(baseMode, eDistributedTracingModes.AI_AND_W3C, + "Base mode of AI_AND_W3C_TRACE should be AI_AND_W3C"); + + // Test W3C_TRACE mode with _BaseMask + tracingMode = eDistributedTracingModes.W3C_TRACE; + baseMode = tracingMode & eDistributedTracingModes._BaseMask; + + // Assert base mode equals W3C + Assert.equal(baseMode, eDistributedTracingModes.W3C, + "Base mode of W3C_TRACE should be W3C"); + + // Test that masking doesn't affect modes without the trace bit + tracingMode = eDistributedTracingModes.AI; + baseMode = tracingMode & eDistributedTracingModes._BaseMask; + + // Assert base mode equals AI + Assert.equal(baseMode, eDistributedTracingModes.AI, + "Base mode of AI should still be AI after applying mask"); + } + }); + + this.testCase({ + name: "W3CTraceStateModes: Enable and disable tracestate bit dynamically", + test: () => { + // Start with AI_AND_W3C mode + var baseMode = eDistributedTracingModes.AI_AND_W3C; + + // Verify no tracestate initially + Assert.equal(false, _checkTraceStateBit(baseMode), + "AI_AND_W3C should NOT include _W3CTraceState bit initially"); + + // Add the tracestate bit (|= operation) + var updatedMode = baseMode | eDistributedTracingModes._W3CTraceState; + + // Verify it equals AI_AND_W3C_TRACE + Assert.equal(updatedMode, eDistributedTracingModes.AI_AND_W3C_TRACE, + "Adding _W3CTraceState bit should result in AI_AND_W3C_TRACE"); + + // Verify tracestate bit is now set + Assert.equal(true, _checkTraceStateBit(updatedMode), + "Updated mode should include _W3CTraceState bit"); + + // Remove the tracestate bit (& ~operation) + var restoredMode = updatedMode & ~eDistributedTracingModes._W3CTraceState; + + // Verify it equals the original mode + Assert.equal(restoredMode, baseMode, + "Removing _W3CTraceState bit should restore original mode"); + + // Verify tracestate bit is no longer set + Assert.equal(false, _checkTraceStateBit(restoredMode), + "Restored mode should NOT include _W3CTraceState bit"); + + // Start with W3C mode + baseMode = eDistributedTracingModes.W3C; + + // Add the tracestate bit + updatedMode = baseMode | eDistributedTracingModes._W3CTraceState; + + // Verify it equals W3C_TRACE + Assert.equal(updatedMode, eDistributedTracingModes.W3C_TRACE, + "Adding _W3CTraceState bit to W3C should result in W3C_TRACE"); + + // Remove the tracestate bit + restoredMode = updatedMode & ~eDistributedTracingModes._W3CTraceState; + + // Verify it equals original W3C mode + Assert.equal(restoredMode, eDistributedTracingModes.W3C, + "Removing _W3CTraceState bit should restore W3C mode"); + } + }); + } +} diff --git a/shared/AppInsightsCore/Tests/Unit/src/W3cTraceParentTests.ts b/shared/AppInsightsCore/Tests/Unit/src/trace/W3cTraceParentTests.ts similarity index 98% rename from shared/AppInsightsCore/Tests/Unit/src/W3cTraceParentTests.ts rename to shared/AppInsightsCore/Tests/Unit/src/trace/W3cTraceParentTests.ts index 7187ed51f..06e5411dd 100644 --- a/shared/AppInsightsCore/Tests/Unit/src/W3cTraceParentTests.ts +++ b/shared/AppInsightsCore/Tests/Unit/src/trace/W3cTraceParentTests.ts @@ -1,8 +1,8 @@ import { Assert, AITestClass } from "@microsoft/ai-test-framework"; import { utcNow } from "@nevware21/ts-utils"; -import { ITraceParent } from "../../../src/JavaScriptSDK.Interfaces/ITraceParent"; -import { generateW3CId, newGuid } from "../../../src/JavaScriptSDK/CoreUtils"; -import { formatTraceParent, isSampledFlag, isValidSpanId, isValidTraceId, isValidTraceParent, parseTraceParent } from "../../../src/JavaScriptSDK/W3cTraceParent"; +import { ITraceParent } from "../../../../src/interfaces/ai/ITraceParent"; +import { generateW3CId, newGuid } from "../../../../src/utils/CoreUtils"; +import { formatTraceParent, isSampledFlag, isValidSpanId, isValidTraceId, isValidTraceParent, parseTraceParent } from "../../../../src/utils/TraceParent"; export class W3cTraceParentTests extends AITestClass { diff --git a/shared/AppInsightsCore/Tests/Unit/src/trace/W3cTraceState.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/trace/W3cTraceState.Tests.ts new file mode 100644 index 000000000..711ef1880 --- /dev/null +++ b/shared/AppInsightsCore/Tests/Unit/src/trace/W3cTraceState.Tests.ts @@ -0,0 +1,1606 @@ +import { Assert, AITestClass } from "@microsoft/ai-test-framework"; +import { asString, strRepeat } from "@nevware21/ts-utils"; +import { createW3cTraceState, isW3cTraceState, snapshotW3cTraceState } from "../../../../src/telemetry/W3cTraceState"; + +export class W3cTraceStateTests extends AITestClass { + + public testInitialize() { + super.testInitialize(); + } + + public testCleanup() { + super.testCleanup(); + } + + public registerTests() { + + this.testCase({ + name: "W3cTraceState: default", + test: () => { + const traceState = createW3cTraceState(); + Assert.equal(traceState.keys.length, 0); + Assert.equal(traceState.hdrs().length, 0); + Assert.equal(asString(traceState), ""); + Assert.equal(traceState.isEmpty, true, "Default trace state should be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: handle empty string", + test: () => { + const traceState = createW3cTraceState(""); + Assert.equal(traceState.keys.length, 0); + Assert.equal(traceState.hdrs().length, 0); + Assert.equal(asString(traceState), ""); + Assert.equal(traceState.isEmpty, true, "Empty string trace state should be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: handle null", + test: () => { + const traceState = createW3cTraceState(null as any); + Assert.equal(traceState.keys.length, 0); + Assert.equal(traceState.hdrs().length, 0); + Assert.equal(asString(traceState), ""); + Assert.equal(traceState.isEmpty, true, "Null trace state should be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: handle undefined", + test: () => { + const traceState = createW3cTraceState(undefined); + Assert.equal(traceState.keys.length, 0); + Assert.equal(traceState.hdrs().length, 0); + Assert.equal(asString(traceState), ""); + Assert.equal(traceState.isEmpty, true, "Undefined trace state should be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: toString", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.isEmpty, false, "Trace state with values should not be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: handle invalid input", + test: () => { + const traceState = createW3cTraceState({} as any); + Assert.equal(traceState.keys.length, 0); + Assert.equal(traceState.hdrs().length, 0); + Assert.equal(asString(traceState), ""); + Assert.equal(traceState.isEmpty, true, "Invalid trace state should be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: new / updated keys are added to the front", + test: () => { + let traceState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + + traceState.set("d", "4"); + Assert.equal(asString(traceState), "d=4,a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 4); + Assert.deepEqual(["d", "a", "b", "c"], traceState.keys); + + traceState.set("a", "5"); + Assert.equal(asString(traceState), "a=5,d=4,b=2,c=3"); + Assert.equal(traceState.keys.length, 4); + Assert.deepEqual(["a", "d", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: must create new instances for each state", + test: () => { + let traceState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + + let traceState2 = createW3cTraceState(asString(traceState)); + Assert.equal(traceState2.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState2.keys); + traceState2.set("d", "4"); + Assert.equal(traceState2.keys.length, 4); + Assert.deepEqual(["d", "a", "b", "c"], traceState2.keys); + + Assert.notEqual(traceState, traceState2); + Assert.notDeepEqual(traceState, traceState2); + + Assert.equal(asString(traceState), "a=1,b=2,c=3", "Actual: " + asString(traceState) + " expected a=1,b=2,c=3"); + Assert.equal(asString(traceState2), "d=4,a=1,b=2,c=3", "Actual: " + asString(traceState2) + " expected d=4,a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "W3cTraceState: del", + test: () => { + let traceState = createW3cTraceState("a=4,b=5,c=6"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + Assert.equal(traceState.isEmpty, false, "Trace state should not be empty initially"); + + traceState.del("b"); + Assert.equal(asString(traceState), "a=4,c=6"); + Assert.equal(traceState.keys.length, 2); + Assert.deepEqual(["a", "c"], traceState.keys); + Assert.equal(traceState.isEmpty, false, "Trace state should not be empty after deleting one key"); + + traceState.del("a"); + Assert.equal(asString(traceState), "c=6"); + Assert.equal(traceState.keys.length, 1); + Assert.deepEqual(["c"], traceState.keys); + Assert.equal(traceState.isEmpty, false, "Trace state should not be empty with one key remaining"); + + traceState.del("c"); + Assert.equal(asString(traceState), ""); + Assert.equal(traceState.keys.length, 0); + Assert.deepEqual([], traceState.keys); + Assert.equal(traceState.isEmpty, true, "Trace state should be empty after deleting all keys"); + } + }); + + this.testCase({ + name: "W3cTraceState: get", + test: () => { + let traceState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), "2"); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(traceState.get("d"), undefined); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: decode with spaces", + test: () => { + const traceState = createW3cTraceState("a=1, b=2, c=3"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: decode with tabs", + test: () => { + const traceState = createW3cTraceState("a=1\t,b=2\t,c=3"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: decode with newlines", + test: () => { + const traceState = createW3cTraceState("a=1\n,b=2\n,c=3"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: decode with multiple commas", + test: () => { + const traceState = createW3cTraceState("a=1,,b=2,,c=3"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: decode with multiple equals", + test: () => { + const traceState = createW3cTraceState("a==1,b==2,c==3"); + Assert.equal(asString(traceState), ""); + Assert.equal(traceState.keys.length, 0); + Assert.deepEqual([], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: decode with multiple spaces", + test: () => { + const traceState = createW3cTraceState("a=1 , b=2 , c=3"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: decode with multiple tabs", + test: () => { + const traceState = createW3cTraceState("a=1\t\t,b=2\t\t,c=3"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: decode with multiple newlines", + test: () => { + const traceState = createW3cTraceState("a=1\n\n,b=2\n\n,c=3"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: decode with multiple commas and spaces", + test: () => { + const traceState = createW3cTraceState("a=1, ,b=2, ,c=3"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle unsetting non-existent keys", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.del("d"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting empty key", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set("", "4"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting empty value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set("d", ""); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting empty string value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set("d", " "); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting empty key and value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set("", ""); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting null key", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set(null as any, "4"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting null value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set("d", null as any); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting null string value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set("d", "null"); + Assert.equal(asString(traceState), "d=null,a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 4); + Assert.deepEqual(["d", "a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting null key and value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set(null as any, null as any); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting undefined key", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set(undefined as any, "4"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + this.testCase({ + name: "W3cTraceState: handle setting undefined value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set("d", undefined as any); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting undefined key and value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set(undefined as any, undefined as any); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting invalid key", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set({} as any, "4"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting invalid value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set("d", {} as any); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting invalid key and value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set({} as any, {} as any); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting invalid key and valid value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set({} as any, "4"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle setting valid key and invalid value", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set("d", {} as any); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle dropping states when the max number of members limit is reached", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + traceState.set("d", "4"); + traceState.set("e", "5"); + traceState.set("f", "6"); + traceState.set("g", "7"); + traceState.set("h", "8"); + traceState.set("i", "9"); + traceState.set("j", "10"); + traceState.set("k", "11"); + traceState.set("l", "12"); + traceState.set("m", "13"); + traceState.set("n", "14"); + traceState.set("o", "15"); + traceState.set("p", "16"); + traceState.set("q", "17"); + traceState.set("r", "18"); + traceState.set("s", "19"); + traceState.set("t", "20"); + traceState.set("u", "21"); + traceState.set("v", "22"); + traceState.set("w", "23"); + traceState.set("x", "24"); + traceState.set("y", "25"); + traceState.set("z", "26"); + traceState.set("aa", "27"); + traceState.set("ab", "28"); + traceState.set("ac", "29"); + traceState.set("ad", "30"); + traceState.set("ae", "31"); + traceState.set("af", "32"); + Assert.equal(asString(traceState), "af=32,ae=31,ad=30,ac=29,ab=28,aa=27,z=26,y=25,x=24,w=23,v=22,u=21,t=20,s=19,r=18,q=17,p=16,o=15,n=14,m=13,l=12,k=11,j=10,i=9,h=8,g=7,f=6,e=5,d=4,a=1,b=2,c=3"); + traceState.set("ag", "33"); + Assert.equal(asString(traceState), "ag=33,af=32,ae=31,ad=30,ac=29,ab=28,aa=27,z=26,y=25,x=24,w=23,v=22,u=21,t=20,s=19,r=18,q=17,p=16,o=15,n=14,m=13,l=12,k=11,j=10,i=9,h=8,g=7,f=6,e=5,d=4,a=1,b=2"); + Assert.equal(traceState.keys.length, 33); + Assert.deepEqual(["ag","af","ae","ad","ac","ab","aa","z","y","x","w","v","u","t","s","r","q","p","o","n","m","l","k","j","i","h","g","f","e","d","a","b","c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: drop states when the items are too long", + test: () => { + const traceState = createW3cTraceState("a=" + strRepeat("b", 512)); + Assert.equal(traceState.get("a"), undefined); + Assert.equal(asString(traceState), ""); + Assert.equal(traceState.keys.length, 0); + Assert.deepEqual([], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: drop items that are invalid", + test: () => { + const traceState = createW3cTraceState("a=1,b,c=3"); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), undefined); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(asString(traceState), "a=1,c=3"); + Assert.equal(traceState.keys.length, 2); + Assert.deepEqual(["a", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: drop items that are invalid with spaces", + test: () => { + const traceState = createW3cTraceState("a=1, b, c=3"); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), undefined); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(asString(traceState), "a=1,c=3"); + Assert.equal(traceState.keys.length, 2); + Assert.deepEqual(["a", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: drop items that are invalid with tabs", + test: () => { + const traceState = createW3cTraceState("a=1\t,b\t,c=3"); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), undefined); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(asString(traceState), "a=1,c=3"); + Assert.equal(traceState.keys.length, 2); + Assert.deepEqual(["a", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: drop items that have a single value with an '=' sign", + test: () => { + const traceState = createW3cTraceState("a=1,b=2=,c=3,d="); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), undefined); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(traceState.get("d"), undefined); + Assert.equal(asString(traceState), "a=1,c=3"); + Assert.equal(traceState.keys.length, 2); + Assert.deepEqual(["a", "c"], traceState.keys); + Assert.equal(traceState.isEmpty, false, "Trace state with valid keys should not be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: isEmpty with all invalid items", + test: () => { + const traceState = createW3cTraceState("b=2=,d="); + Assert.equal(traceState.get("b"), undefined); + Assert.equal(traceState.get("d"), undefined); + Assert.equal(asString(traceState), ""); + Assert.equal(traceState.keys.length, 0); + Assert.deepEqual([], traceState.keys); + Assert.equal(traceState.isEmpty, true, "Trace state with only invalid items should be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: must handle valid state key ranges", + test: () => { + const traceState = createW3cTraceState("a-b=1,c/d=2,e_f=3,g*h=4"); + Assert.equal(traceState.get("a-b"), "1"); + Assert.equal(traceState.get("c/d"), "2"); + Assert.equal(traceState.get("e_f"), "3"); + Assert.equal(traceState.get("g*h"), "4"); + Assert.equal(asString(traceState), "a-b=1,c/d=2,e_f=3,g*h=4"); + Assert.equal(traceState.keys.length, 4); + Assert.deepEqual(["a-b", "c/d", "e_f", "g*h"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle values with embedded spaces", + test: () => { + const traceState = createW3cTraceState("a=1 b,c=2 d,e=3 f"); + Assert.equal(traceState.get("a"), "1 b"); + Assert.equal(traceState.get("c"), "2 d"); + Assert.equal(traceState.get("e"), "3 f"); + Assert.equal(asString(traceState), "a=1 b,c=2 d,e=3 f"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "c", "e"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle inherited state values with no new values", + test: () => { + const parentState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(parentState.get("a"), "1"); + Assert.equal(parentState.get("b"), "2"); + Assert.equal(parentState.get("c"), "3"); + Assert.equal(asString(parentState), "a=1,b=2,c=3"); + Assert.equal(parentState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], parentState.keys); + + const traceState = createW3cTraceState("", parentState); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), "2"); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle inherited state values with new values", + test: () => { + const parentState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(parentState.get("a"), "1"); + Assert.equal(parentState.get("b"), "2"); + Assert.equal(parentState.get("c"), "3"); + Assert.equal(asString(parentState), "a=1,b=2,c=3"); + Assert.equal(parentState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], parentState.keys); + + const traceState = createW3cTraceState("d=4,e=5,f=6", parentState); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), "2"); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(traceState.get("d"), "4"); + Assert.equal(traceState.get("e"), "5"); + Assert.equal(traceState.get("f"), "6"); + Assert.equal(asString(traceState), "d=4,e=5,f=6,a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 6); + Assert.deepEqual(["d", "e", "f", "a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle inherited state values with new values that are too long", + test: () => { + const parentState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(parentState.get("a"), "1"); + Assert.equal(parentState.get("b"), "2"); + Assert.equal(parentState.get("c"), "3"); + Assert.equal(asString(parentState), "a=1,b=2,c=3"); + Assert.equal(parentState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], parentState.keys); + + const traceState = createW3cTraceState("d=" + strRepeat("e", 512), parentState); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), "2"); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(traceState.get("d"), undefined); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle deleting an inherited key in the child", + test: () => { + const parentState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(parentState.get("a"), "1"); + Assert.equal(parentState.get("b"), "2"); + Assert.equal(parentState.get("c"), "3"); + Assert.equal(asString(parentState), "a=1,b=2,c=3"); + Assert.equal(parentState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], parentState.keys); + + const traceState = createW3cTraceState("", parentState); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), "2"); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + + traceState.del("b"); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), undefined); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(asString(traceState), "a=1,c=3"); + Assert.equal(traceState.keys.length, 2); + Assert.deepEqual(["a", "c"], traceState.keys); + + // Should not affect the original state + Assert.equal(parentState.get("a"), "1"); + Assert.equal(parentState.get("b"), "2"); + Assert.equal(parentState.get("c"), "3"); + Assert.equal(asString(parentState), "a=1,b=2,c=3"); + Assert.equal(parentState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], parentState.keys); + } + }); + + this.testCase({ + name: "W3cTraceState: handle deleting an inherited key in the parent", + test: () => { + const parentState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(parentState.get("a"), "1"); + Assert.equal(parentState.get("b"), "2"); + Assert.equal(parentState.get("c"), "3"); + Assert.equal(asString(parentState), "a=1,b=2,c=3"); + Assert.equal(parentState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], parentState.keys); + + const traceState = createW3cTraceState("", parentState); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), "2"); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(asString(traceState), "a=1,b=2,c=3"); + Assert.equal(traceState.keys.length, 3); + Assert.deepEqual(["a", "b", "c"], traceState.keys); + + parentState.del("b"); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), undefined); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(asString(traceState), "a=1,c=3"); + Assert.equal(traceState.keys.length, 2); + Assert.deepEqual(["a", "c"], traceState.keys); + + // Should not affect the original state + Assert.equal(parentState.get("a"), "1"); + Assert.equal(parentState.get("b"), undefined); + Assert.equal(parentState.get("c"), "3"); + Assert.equal(asString(parentState), "a=1,c=3"); + Assert.equal(parentState.keys.length, 2); + Assert.deepEqual(["a", "c"], parentState.keys); + } + }); + + // isEmpty property specific tests + this.testCase({ + name: "W3cTraceState: isEmpty with trace state values", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(traceState.isEmpty, false, "Trace state with values should not be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: isEmpty after deleting all keys", + test: () => { + const traceState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(traceState.isEmpty, false, "Trace state with values should not be empty"); + + traceState.del("a"); + Assert.equal(traceState.isEmpty, false, "Trace state should not be empty after deleting one key"); + + traceState.del("b"); + Assert.equal(traceState.isEmpty, false, "Trace state should not be empty after deleting two keys"); + + traceState.del("c"); + Assert.equal(traceState.isEmpty, true, "Trace state should be empty after deleting all keys"); + } + }); + + this.testCase({ + name: "W3cTraceState: isEmpty with parent state", + test: () => { + const parentState = createW3cTraceState("a=1,b=2,c=3"); + Assert.equal(parentState.isEmpty, false, "Parent trace state with values should not be empty"); + + const childState = createW3cTraceState("", parentState); + Assert.equal(childState.isEmpty, false, "Child trace state with parent values should not be empty"); + + const emptyChildState = createW3cTraceState("", createW3cTraceState()); + Assert.equal(emptyChildState.isEmpty, true, "Child trace state with empty parent should be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: isEmpty after deleting inherited keys", + test: () => { + const parentState = createW3cTraceState("a=1,b=2,c=3"); + const childState = createW3cTraceState("", parentState); + Assert.equal(childState.isEmpty, false, "Child trace state with parent values should not be empty"); + + childState.del("a"); + Assert.equal(childState.isEmpty, false, "Child trace state should not be empty after deleting one key"); + + childState.del("b"); + Assert.equal(childState.isEmpty, false, "Child trace state should not be empty after deleting two keys"); + + childState.del("c"); + Assert.equal(childState.isEmpty, true, "Child trace state should be empty after deleting all keys"); + + // Parent state should remain unchanged + Assert.equal(parentState.isEmpty, false, "Parent trace state should remain unchanged"); + } + }); + + this.testCase({ + name: "W3cTraceState: isEmpty after parent deletes keys", + test: () => { + const parentState = createW3cTraceState("a=1,b=2,c=3"); + const childState = createW3cTraceState("", parentState); + Assert.equal(childState.isEmpty, false, "Child trace state with parent values should not be empty"); + + parentState.del("a"); + Assert.equal(childState.isEmpty, false, "Child trace state should not be empty after parent deletes one key"); + + parentState.del("b"); + Assert.equal(childState.isEmpty, false, "Child trace state should not be empty after parent deletes two keys"); + + parentState.del("c"); + Assert.equal(childState.isEmpty, true, "Child trace state should be empty after parent deletes all keys"); + } + }); + + this.testCase({ + name: "W3cTraceState: isEmpty with mixed operations", + test: () => { + const traceState = createW3cTraceState(); + Assert.equal(traceState.isEmpty, true, "Empty trace state should be empty"); + + traceState.set("a", "1"); + Assert.equal(traceState.isEmpty, false, "Trace state should not be empty after adding a key"); + + traceState.del("a"); + Assert.equal(traceState.isEmpty, true, "Trace state should be empty after deleting the key"); + + traceState.set("b", "2"); + traceState.set("c", "3"); + Assert.equal(traceState.isEmpty, false, "Trace state should not be empty after adding multiple keys"); + + traceState.del("b"); + Assert.equal(traceState.isEmpty, false, "Trace state should not be empty with remaining keys"); + + traceState.del("c"); + Assert.equal(traceState.isEmpty, true, "Trace state should be empty after deleting all keys"); + } + }); + + this.testCase({ + name: "W3cTraceState: isEmpty with changing parent values", + test: () => { + const parentState = createW3cTraceState(); + const childState = createW3cTraceState("", parentState); + Assert.equal(parentState.isEmpty, true, "Empty parent trace state should be empty"); + Assert.equal(childState.isEmpty, true, "Child of empty parent should be empty"); + + parentState.set("a", "1"); + Assert.equal(parentState.isEmpty, false, "Parent trace state should not be empty after adding a key"); + Assert.equal(childState.isEmpty, false, "Child trace state should not be empty after parent adds a key"); + + childState.del("a"); + Assert.equal(parentState.isEmpty, false, "Parent trace state should remain unchanged after child deletes key"); + Assert.equal(childState.isEmpty, true, "Child trace state should be empty after deleting parent's key"); + + childState.set("b", "2"); + Assert.equal(childState.isEmpty, false, "Child trace state should not be empty after adding own key"); + + parentState.del("a"); + Assert.equal(parentState.isEmpty, true, "Parent trace state should be empty after deleting its only key"); + Assert.equal(childState.isEmpty, false, "Child trace state should not be empty with its own keys"); + } + }); + + // Tests for the child() function + this.testCase({ + name: "W3cTraceState: child() creates a new child trace state instance", + test: () => { + const parentState = createW3cTraceState("a=1,b=2,c=3"); + const childState = parentState.child(); + + Assert.notEqual(parentState, childState, "Child should be a new instance"); + Assert.equal(childState.get("a"), "1", "Child should inherit parent values"); + Assert.equal(childState.get("b"), "2", "Child should inherit parent values"); + Assert.equal(childState.get("c"), "3", "Child should inherit parent values"); + Assert.equal(asString(childState), "a=1,b=2,c=3", "Child should have same string representation as parent"); + Assert.equal(childState.keys.length, 3, "Child should have same number of keys as parent"); + Assert.deepEqual(childState.keys, ["a", "b", "c"], "Child should have same keys as parent"); + Assert.equal(childState.isEmpty, false, "Child should not be empty if parent is not empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() from empty parent", + test: () => { + const parentState = createW3cTraceState(); + const childState = parentState.child(); + + Assert.notEqual(parentState, childState, "Child should be a new instance"); + Assert.equal(childState.keys.length, 0, "Child of empty parent should have no keys"); + Assert.equal(asString(childState), "", "Child of empty parent should have empty string representation"); + Assert.equal(childState.isEmpty, true, "Child of empty parent should be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() with modifications to parent", + test: () => { + const parentState = createW3cTraceState("a=1,b=2"); + const childState = parentState.child(); + + Assert.equal(asString(childState), "a=1,b=2", "Child should initially match parent"); + + // Modify parent after creating child + parentState.set("c", "3"); + Assert.equal(asString(parentState), "c=3,a=1,b=2", "Parent should have new value"); + Assert.equal(asString(childState), "c=3,a=1,b=2", "Child should reflect parent changes"); + Assert.equal(childState.get("c"), "3", "Child should get updated value from parent"); + + // Delete from parent + parentState.del("a"); + Assert.equal(asString(parentState), "c=3,b=2", "Parent should have key removed"); + Assert.equal(asString(childState), "c=3,b=2", "Child should reflect parent deletions"); + Assert.equal(childState.get("a"), undefined, "Child should not have deleted key"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() with modifications to child", + test: () => { + const parentState = createW3cTraceState("a=1,b=2"); + const childState = parentState.child(); + + // Modify child + childState.set("c", "3"); + Assert.equal(asString(childState), "c=3,a=1,b=2", "Child should have new value"); + Assert.equal(asString(parentState), "a=1,b=2", "Parent should not be affected by child changes"); + Assert.equal(parentState.get("c"), undefined, "Parent should not have child's new key"); + + // Modify existing value in child + childState.set("a", "4"); + Assert.equal(asString(childState), "a=4,c=3,b=2", "Child should have updated value"); + Assert.equal(asString(parentState), "a=1,b=2", "Parent should not be affected by child modifications"); + Assert.equal(parentState.get("a"), "1", "Parent should keep original value"); + Assert.equal(childState.get("a"), "4", "Child should have updated value"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() with deletions in child", + test: () => { + const parentState = createW3cTraceState("a=1,b=2,c=3"); + const childState = parentState.child(); + + // Delete key in child + childState.del("b"); + Assert.equal(asString(childState), "a=1,c=3", "Child should have key removed"); + Assert.equal(asString(parentState), "a=1,b=2,c=3", "Parent should not be affected by child deletions"); + Assert.equal(parentState.get("b"), "2", "Parent should keep deleted key"); + Assert.equal(childState.get("b"), undefined, "Child should not have deleted key"); + + // Delete all keys in child + childState.del("a"); + childState.del("c"); + Assert.equal(asString(childState), "", "Child should have all keys removed"); + Assert.equal(childState.isEmpty, true, "Child should be empty after deleting all keys"); + Assert.equal(asString(parentState), "a=1,b=2,c=3", "Parent should remain unchanged"); + Assert.equal(parentState.isEmpty, false, "Parent should not be empty"); + } + }); + + this.testCase({ + name: "W3cTraceState: nested children creation", + test: () => { + const parentState = createW3cTraceState("a=1,b=2"); + const childState = parentState.child(); + const grandchildState = childState.child(); + + Assert.notEqual(parentState, childState, "Child should be a new instance"); + Assert.notEqual(childState, grandchildState, "Grandchild should be a new instance"); + Assert.notEqual(parentState, grandchildState, "Grandchild should be a new instance different from parent"); + + Assert.equal(asString(grandchildState), "a=1,b=2", "Grandchild should inherit all values"); + + // Modify at each level + parentState.set("p", "parent"); + childState.set("c", "child"); + grandchildState.set("g", "grandchild"); + + Assert.equal(asString(parentState), "p=parent,a=1,b=2", "Parent should have its own values"); + Assert.equal(asString(childState), "c=child,p=parent,a=1,b=2", "Child should have its values and parent's"); + Assert.equal(asString(grandchildState), "g=grandchild,c=child,p=parent,a=1,b=2", "Grandchild should have all values"); + + // Override values at different levels + childState.set("a", "child-a"); + grandchildState.set("c", "grandchild-c"); + + Assert.equal(parentState.get("a"), "1", "Parent should keep original value"); + Assert.equal(childState.get("a"), "child-a", "Child should have overridden value"); + Assert.equal(grandchildState.get("a"), "child-a", "Grandchild should inherit child's value"); + Assert.equal(childState.get("c"), "child", "Child should keep its value"); + Assert.equal(grandchildState.get("c"), "grandchild-c", "Grandchild should have overridden value"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() with modifications affecting isEmpty", + test: () => { + const parentState = createW3cTraceState("a=1,b=2"); + const childState = parentState.child(); + + Assert.equal(parentState.isEmpty, false, "Parent should not be empty"); + Assert.equal(childState.isEmpty, false, "Child should not be empty"); + + // Delete all parent keys from child + childState.del("a"); + childState.del("b"); + + Assert.equal(parentState.isEmpty, false, "Parent should still not be empty"); + Assert.equal(childState.isEmpty, true, "Child should be empty after deleting all keys"); + + // Add key to parent + parentState.set("c", "3"); + Assert.equal(childState.isEmpty, false, "Child should not be empty after parent adds key"); + Assert.equal(childState.get("c"), "3", "Child should have parent's new key"); + + // Delete from parent + parentState.del("a"); + parentState.del("b"); + parentState.del("c"); + + Assert.equal(parentState.isEmpty, true, "Parent should be empty after deleting all keys"); + Assert.equal(childState.isEmpty, true, "Child should be empty when parent is empty and child has deleted parent keys"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() preserves key ordering", + test: () => { + const parentState = createW3cTraceState("a=1,b=2,c=3"); + const childState = parentState.child(); + + // Add keys to child, which should go to the front + childState.set("d", "4"); + childState.set("e", "5"); + + Assert.deepEqual(childState.keys, ["e", "d", "a", "b", "c"], "Child should have keys in correct order"); + + // Update existing key in child, which should move to front + childState.set("b", "new-b"); + + Assert.deepEqual(childState.keys, ["b", "e", "d", "a", "c"], "Updated key should move to front"); + Assert.equal(childState.get("b"), "new-b", "Should get updated value"); + + // Add new key to parent, should be inherited by child + parentState.set("f", "6"); + + // Child keys should include parent's new key but maintain child's order + Assert.deepEqual(childState.keys, ["b", "e", "d", "f", "a", "c"], "Child should include parent's new key"); + + // Update existing key in parent, which should be reflected in child unless overridden + parentState.set("a", "new-a"); + + Assert.equal(childState.get("a"), "new-a", "Child should see parent's updated value"); + + // Override parent key that was already overridden + childState.set("a", "child-a"); + + Assert.equal(parentState.get("a"), "new-a", "Parent should keep its value"); + Assert.equal(childState.get("a"), "child-a", "Child should have its own value"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() handles hdrs() correctly", + test: () => { + const parentState = createW3cTraceState("a=1,b=2"); + const childState = parentState.child(); + + Assert.deepEqual(childState.hdrs(), ["a=1,b=2"], "Child headers should match parent initially"); + + childState.set("c", "3"); + Assert.deepEqual(childState.hdrs(), ["c=3,a=1,b=2"], "Child headers should include own values"); + Assert.deepEqual(parentState.hdrs(), ["a=1,b=2"], "Parent headers should remain unchanged"); + + // Test with maxHeaders parameter + const longParent = createW3cTraceState("a=" + strRepeat("x", 250) + ",b=" + strRepeat("y", 250)); + const longChild = longParent.child(); + longChild.set("c", strRepeat("z", 250)); + + // Should split into multiple headers due to length + Assert.equal(longChild.hdrs().length > 1, true, "Long values should split into multiple headers"); + + // Test with maxKeys parameter + Assert.deepEqual(longChild.hdrs(undefined, 1), ["c=" + strRepeat("z", 250)], "Should respect maxKeys parameter"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() with invalid values in parent", + test: () => { + // Create parent with some invalid entries that will be filtered out + const parentState = createW3cTraceState("a=1,invalid,b=2,c="); + Assert.equal(asString(parentState), "a=1,b=2", "Parent should only have valid entries"); + + const childState = parentState.child(); + Assert.equal(asString(childState), "a=1,b=2", "Child should inherit only valid entries"); + + // Add an invalid entry to parent after child creation + parentState.set("d", ""); + Assert.equal(asString(parentState), "a=1,b=2", "Parent should not add invalid entry"); + Assert.equal(asString(childState), "a=1,b=2", "Child should reflect parent's valid state"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() with multiple children from same parent", + test: () => { + const parentState = createW3cTraceState("a=1,b=2"); + + // Create multiple children + const child1 = parentState.child(); + const child2 = parentState.child(); + const child3 = parentState.child(); + + Assert.notEqual(child1, child2, "Each child should be a distinct instance"); + Assert.notEqual(child2, child3, "Each child should be a distinct instance"); + + // Modify each child differently + child1.set("c", "3"); + child2.set("d", "4"); + child3.set("e", "5"); + + // Each child should have its own modifications plus parent's values + Assert.equal(asString(child1), "c=3,a=1,b=2", "Child 1 should have its own changes"); + Assert.equal(asString(child2), "d=4,a=1,b=2", "Child 2 should have its own changes"); + Assert.equal(asString(child3), "e=5,a=1,b=2", "Child 3 should have its own changes"); + + // Modify parent, all children should reflect the change + parentState.set("f", "6"); + + Assert.equal(asString(child1), "c=3,f=6,a=1,b=2", "Child 1 should see parent's new value"); + Assert.equal(asString(child2), "d=4,f=6,a=1,b=2", "Child 2 should see parent's new value"); + Assert.equal(asString(child3), "e=5,f=6,a=1,b=2", "Child 3 should see parent's new value"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() with limit on trace state members", + test: () => { + // Create a parent with more than the maximum allowed entries for hdrs + const parentState = createW3cTraceState(); + for (let i = 0; i < 40; i++) { + parentState.set("key" + i, "value" + i); + } + + // In memory, we can exceed 32 keys + Assert.equal(parentState.keys.length, 40, "Parent should have all 40 keys in memory"); + Assert.equal(parentState.get("key0"), "value0", "First key should be accessible"); + Assert.equal(parentState.get("key39"), "value39", "Last key should be accessible"); + + // When converted to headers, it's limited to 32 keys + const headers = parentState.hdrs(); + let headersStr = headers.join(","); + // First keys (0-7) should not be in headers as they're the oldest + Assert.equal(headersStr.indexOf("key0=value0"), -1, "Oldest keys should not be in headers"); + Assert.equal(headersStr.indexOf("key7=value7"), -1, "Oldest keys should not be in headers"); + // Most recent keys (8-39) should be in headers + Assert.notEqual(headersStr.indexOf("key8=value8"), -1, "More recent keys should be in headers"); + Assert.notEqual(headersStr.indexOf("key39=value39"), -1, "Most recent keys should be in headers"); + + // Create a child state + const childState = parentState.child(); + + // Child should inherit all keys in memory + Assert.equal(childState.keys.length, 40, "Child should inherit all keys in memory"); + Assert.equal(childState.get("key0"), "value0", "Child should have access to all parent keys"); + Assert.equal(childState.get("key39"), "value39", "Child should have access to all parent keys"); + + // Add more keys to child + for (let i = 0; i < 10; i++) { + childState.set("child-key" + i, "child-value" + i); + } + + // In memory, child should have all keys + Assert.equal(childState.keys.length, 50, "Child should have all 50 keys in memory"); + + // When converted to headers and back, we should only get 32 keys + const childHeaders = childState.hdrs(); + const roundTripState = createW3cTraceState(childHeaders.join(",")); + + Assert.equal(roundTripState.keys.length, 32, "Round trip state should have max 32 keys"); + + // Most recent keys should be preserved (child keys and most recent parent keys) + Assert.equal(roundTripState.get("child-key9"), "child-value9", "Most recent child key should be preserved"); + Assert.equal(roundTripState.get("child-key0"), "child-value0", "First child key should be preserved"); + + // Some parent keys should be dropped in the round trip + Assert.equal(roundTripState.get("key0"), undefined, "Oldest parent keys should be dropped in round trip"); + Assert.equal(roundTripState.get("key10"), undefined, "Older parent keys should be dropped in round trip"); + + // Make sure both original states are unaffected + Assert.equal(parentState.keys.length, 40, "Original parent should still have 40 keys"); + Assert.equal(childState.keys.length, 50, "Original child should still have 50 keys"); + } + }); + + this.testCase({ + name: "W3cTraceState: chain of multiple child() calls", + test: () => { + // Create a chain of trace states using child() + const state1 = createW3cTraceState("a=1"); + const state2 = state1.child(); + state2.set("b", "2"); + const state3 = state2.child(); + state3.set("c", "3"); + const state4 = state3.child(); + state4.set("d", "4"); + const state5 = state4.child(); + state5.set("e", "5"); + + // Each state should have its own keys plus all ancestor keys + Assert.equal(asString(state1), "a=1", "State 1 should have its own key"); + Assert.equal(asString(state2), "b=2,a=1", "State 2 should have its key and parent's"); + Assert.equal(asString(state3), "c=3,b=2,a=1", "State 3 should have its key and ancestors'"); + Assert.equal(asString(state4), "d=4,c=3,b=2,a=1", "State 4 should have its key and ancestors'"); + Assert.equal(asString(state5), "e=5,d=4,c=3,b=2,a=1", "State 5 should have its key and ancestors'"); + + // Modify an ancestor, changes should propagate down the chain + state2.set("b", "new-2"); + + Assert.equal(state3.get("b"), "new-2", "State 3 should see updated ancestor value"); + Assert.equal(state4.get("b"), "new-2", "State 4 should see updated ancestor value"); + Assert.equal(state5.get("b"), "new-2", "State 5 should see updated ancestor value"); + + // Delete from an ancestor, deletion should propagate + state1.del("a"); + + Assert.equal(state2.get("a"), undefined, "State 2 should see parent deletion"); + Assert.equal(state3.get("a"), undefined, "State 3 should see ancestor deletion"); + Assert.equal(state4.get("a"), undefined, "State 4 should see ancestor deletion"); + Assert.equal(state5.get("a"), undefined, "State 5 should see ancestor deletion"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() and toString consistency", + test: () => { + const parentState = createW3cTraceState("a=1,b=2"); + const childState = parentState.child(); + childState.set("c", "3"); + + // toString and asString should be equivalent + Assert.equal(asString(childState), childState.toString(), "toString and asString should be consistent"); + Assert.equal(childState.toString(), "c=3,a=1,b=2", "toString should reflect the current state"); + + // toString should update when state changes + childState.set("d", "4"); + Assert.equal(childState.toString(), "d=4,c=3,a=1,b=2", "toString should reflect updated state"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() with isW3cTraceState check", + test: () => { + const parentState = createW3cTraceState("a=1"); + const childState = parentState.child(); + + // The isW3cTraceState function should recognize child instances + Assert.equal(isW3cTraceState(childState), true, "Child should be recognized as W3cTraceState"); + + // Compare with non-trace state objects + Assert.equal(isW3cTraceState({}), false, "Empty object is not a trace state"); + Assert.equal(isW3cTraceState({ keys: [] }), false, "Object with only keys is not a trace state"); + Assert.equal(isW3cTraceState(null), false, "Null is not a trace state"); + Assert.equal(isW3cTraceState(undefined), false, "Undefined is not a trace state"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() after deleting parent", + test: () => { + // Create a parent and child + const parentState = createW3cTraceState("a=1,b=2"); + const childState = parentState.child(); + + // Verify initial state + Assert.equal(childState.get("a"), "1", "Child should have parent's key"); + + // Simulate "deleting" the parent by removing all references + // (Note: we can't truly delete objects in JavaScript, but we can simulate by removing all references) + // Set all parent keys to null to simulate the parent being "gone" + parentState.del("a"); + parentState.del("b"); + + // Child should still function properly + Assert.equal(childState.get("a"), undefined, "Child should see parent's deletion"); + Assert.equal(childState.get("b"), undefined, "Child should see parent's deletion"); + + // Child should still be able to set its own values + childState.set("c", "3"); + Assert.equal(childState.get("c"), "3", "Child should be able to set new values"); + Assert.equal(asString(childState), "c=3", "Child should have correct string representation"); + } + }); + + this.testCase({ + name: "W3cTraceState: child() with special characters in values", + test: () => { + // Create parent with special characters in values + const parentState = createW3cTraceState("a=value with spaces,b=value-with-dashes,c=value_with_underscores"); + const childState = parentState.child(); + + // Child should inherit these values correctly + Assert.equal(childState.get("a"), "value with spaces", "Child should inherit values with spaces"); + Assert.equal(childState.get("b"), "value-with-dashes", "Child should inherit values with dashes"); + Assert.equal(childState.get("c"), "value_with_underscores", "Child should inherit values with underscores"); + + // Child should be able to set new values with special characters + childState.set("d", "value!with@special#chars"); + Assert.equal(childState.get("d"), "value!with@special#chars", "Child should set value with special chars"); + + // Check string representation + Assert.equal(asString(childState), + "d=value!with@special#chars,a=value with spaces,b=value-with-dashes,c=value_with_underscores", + "Child should have correct string representation with special chars"); + } + }); + + this.testCase({ + name: "W3cTraceState: serialization limits with parent and child", + test: () => { + // Create a parent with many keys + const parentState = createW3cTraceState(); + for (let i = 0; i < 20; i++) { + parentState.set("p" + i, "parent" + i); + } + + // Create a child and add many keys + const childState = parentState.child(); + for (let i = 0; i < 20; i++) { + childState.set("c" + i, "child" + i); + } + + // In memory, both should have their respective keys + Assert.equal(parentState.keys.length, 20, "Parent should have 20 keys in memory"); + Assert.equal(childState.keys.length, 40, "Child should have 20 own keys + 20 parent keys"); + + // Get the string representation of the child + const childString = asString(childState); + + // Create a new state from the string representation + const roundTripState = createW3cTraceState(childString); + + // Only 32 keys should survive the round trip + Assert.equal(roundTripState.keys.length, 32, "Round-trip state should have 32 keys max"); + + // The most recent keys should be preserved + // Child keys should all be preserved (since they're more recent) + Assert.equal(roundTripState.get("c19"), "child19", "Most recent child keys should be preserved"); + Assert.equal(roundTripState.get("c0"), "child0", "All child keys should be preserved"); + + // Only the most recent parent keys should survive + Assert.equal(roundTripState.get("p19"), "parent19", "Most recent parent keys should be preserved"); + Assert.equal(roundTripState.get("p8"), "parent8", "Recent parent keys should be preserved"); + Assert.equal(roundTripState.get("p7"), undefined, "Older parent keys should be dropped"); + Assert.equal(roundTripState.get("p0"), undefined, "Oldest parent keys should be dropped"); + + // Test hdrs with explicit maxKeys parameter + const limitedHeaders = childState.hdrs(undefined, 15); + const limitedState = createW3cTraceState(limitedHeaders.join(",")); + + Assert.equal(limitedState.keys.length, 15, "State from limited headers should respect maxKeys"); + Assert.equal(limitedState.get("c19"), "child19", "Most recent keys should be preserved with limit"); + Assert.equal(limitedState.get("c5"), "child5", "Recent keys should be preserved with limit"); + Assert.equal(limitedState.get("c4"), undefined, "Older keys should be dropped with limit"); + Assert.equal(limitedState.get("p19"), undefined, "Parent keys should be dropped with stricter limit"); + } + }); + + // Tests for snapshotW3cTraceState helper function + this.testCase({ + name: "snapshotW3cTraceState: handle null input", + test: () => { + const snapshot = snapshotW3cTraceState(null as any); + + Assert.ok(snapshot, "Should return a valid trace state instance"); + Assert.equal(snapshot.keys.length, 0, "Snapshot should have no keys"); + Assert.equal(asString(snapshot), "", "Snapshot should have empty string representation"); + Assert.equal(snapshot.isEmpty, true, "Snapshot should be empty"); + } + }); + + this.testCase({ + name: "snapshotW3cTraceState: handle undefined input", + test: () => { + const snapshot = snapshotW3cTraceState(undefined as any); + + Assert.ok(snapshot, "Should return a valid trace state instance"); + Assert.equal(snapshot.keys.length, 0, "Snapshot should have no keys"); + Assert.equal(asString(snapshot), "", "Snapshot should have empty string representation"); + Assert.equal(snapshot.isEmpty, true, "Snapshot should be empty"); + } + }); + + this.testCase({ + name: "snapshotW3cTraceState: handle empty trace state", + test: () => { + const original = createW3cTraceState(); + const snapshot = snapshotW3cTraceState(original); + + Assert.ok(snapshot, "Should return a valid trace state instance"); + Assert.notEqual(snapshot, original, "Should return a different instance"); + Assert.equal(snapshot.keys.length, 0, "Snapshot should have no keys"); + Assert.equal(asString(snapshot), "", "Snapshot should have empty string representation"); + Assert.equal(snapshot.isEmpty, true, "Snapshot should be empty"); + } + }); + + this.testCase({ + name: "snapshotW3cTraceState: capture simple trace state", + test: () => { + const original = createW3cTraceState("a=1,b=2,c=3"); + const snapshot = snapshotW3cTraceState(original); + + Assert.ok(snapshot, "Should return a valid trace state instance"); + Assert.notEqual(snapshot, original, "Should return a different instance"); + Assert.equal(snapshot.keys.length, 3, "Snapshot should have 3 keys"); + Assert.deepEqual(snapshot.keys, ["a", "b", "c"], "Snapshot should have correct keys"); + Assert.equal(snapshot.get("a"), "1", "Snapshot should have correct value for 'a'"); + Assert.equal(snapshot.get("b"), "2", "Snapshot should have correct value for 'b'"); + Assert.equal(snapshot.get("c"), "3", "Snapshot should have correct value for 'c'"); + Assert.equal(asString(snapshot), "a=1,b=2,c=3", "Snapshot should have correct string representation"); + Assert.equal(snapshot.isEmpty, false, "Snapshot should not be empty"); + } + }); + + this.testCase({ + name: "snapshotW3cTraceState: snapshot is independent from original", + test: () => { + const original = createW3cTraceState("a=1,b=2,c=3"); + const snapshot = snapshotW3cTraceState(original); + + // Initial state should be identical + Assert.equal(asString(snapshot), asString(original), "Initial state should be identical"); + + // Modify original after snapshot + original.set("d", "4"); + original.set("a", "modified"); + original.del("b"); + + // Snapshot should remain unchanged + Assert.equal(snapshot.keys.length, 3, "Snapshot should still have 3 keys"); + Assert.equal(snapshot.get("a"), "1", "Snapshot should have original value for 'a'"); + Assert.equal(snapshot.get("b"), "2", "Snapshot should still have 'b' key"); + Assert.equal(snapshot.get("c"), "3", "Snapshot should have original value for 'c'"); + Assert.equal(snapshot.get("d"), undefined, "Snapshot should not have new 'd' key"); + Assert.equal(asString(snapshot), "a=1,b=2,c=3", "Snapshot should have original string representation"); + + // Original should have changes + Assert.equal(original.keys.length, 3, "Original should have 3 keys after modification"); + Assert.equal(original.get("a"), "modified", "Original should have modified value for 'a'"); + Assert.equal(original.get("b"), undefined, "Original should not have 'b' key after deletion"); + Assert.equal(original.get("c"), "3", "Original should have original value for 'c'"); + Assert.equal(original.get("d"), "4", "Original should have new 'd' key"); + Assert.equal(asString(original), "a=modified,d=4,c=3", "Original should have modified string representation"); + } + }); + + this.testCase({ + name: "snapshotW3cTraceState: snapshot with parent trace state", + test: () => { + const parentState = createW3cTraceState("parent1=value1,parent2=value2"); + const childState = createW3cTraceState("child1=cvalue1,child2=cvalue2", parentState); + + // Verify child state includes parent values + Assert.equal(childState.keys.length, 4, "Child should have 4 keys"); + Assert.equal(childState.get("parent1"), "value1", "Child should have parent1 value"); + Assert.equal(childState.get("parent2"), "value2", "Child should have parent2 value"); + Assert.equal(childState.get("child1"), "cvalue1", "Child should have child1 value"); + Assert.equal(childState.get("child2"), "cvalue2", "Child should have child2 value"); + + // Take snapshot of child state + const snapshot = snapshotW3cTraceState(childState); + + Assert.ok(snapshot, "Should return a valid trace state instance"); + Assert.notEqual(snapshot, childState, "Should return a different instance"); + Assert.equal(snapshot.keys.length, 4, "Snapshot should have 4 keys"); + Assert.equal(snapshot.get("parent1"), "value1", "Snapshot should have parent1 value"); + Assert.equal(snapshot.get("parent2"), "value2", "Snapshot should have parent2 value"); + Assert.equal(snapshot.get("child1"), "cvalue1", "Snapshot should have child1 value"); + Assert.equal(snapshot.get("child2"), "cvalue2", "Snapshot should have child2 value"); + + // Verify snapshot string representation includes all values + const snapshotStr = asString(snapshot); + Assert.ok(snapshotStr.includes("parent1=value1"), "Snapshot string should include parent1"); + Assert.ok(snapshotStr.includes("parent2=value2"), "Snapshot string should include parent2"); + Assert.ok(snapshotStr.includes("child1=cvalue1"), "Snapshot string should include child1"); + Assert.ok(snapshotStr.includes("child2=cvalue2"), "Snapshot string should include child2"); + + // Check that snapshot has no parent (is independent) + Assert.equal((snapshot as any)._p, null, "Snapshot should have no parent"); + } + }); + + this.testCase({ + name: "snapshotW3cTraceState: snapshot remains unchanged when parent/child modified", + test: () => { + const parentState = createW3cTraceState("parent1=value1,parent2=value2"); + const childState = createW3cTraceState("child1=cvalue1", parentState); + + // Take snapshot before modifications + const snapshot = snapshotW3cTraceState(childState); + + // Initial verification + Assert.equal(snapshot.keys.length, 3, "Snapshot should have 3 keys initially"); + Assert.equal(snapshot.get("parent1"), "value1", "Snapshot should have parent1 value"); + Assert.equal(snapshot.get("parent2"), "value2", "Snapshot should have parent2 value"); + Assert.equal(snapshot.get("child1"), "cvalue1", "Snapshot should have child1 value"); + + // Modify parent state + parentState.set("parent1", "modified_parent1"); + parentState.set("parent3", "new_parent3"); + parentState.del("parent2"); + + // Modify child state + childState.set("child1", "modified_child1"); + childState.set("child2", "new_child2"); + + // Snapshot should remain unchanged + Assert.equal(snapshot.keys.length, 3, "Snapshot should still have 3 keys"); + Assert.equal(snapshot.get("parent1"), "value1", "Snapshot should have original parent1 value"); + Assert.equal(snapshot.get("parent2"), "value2", "Snapshot should have original parent2 value"); + Assert.equal(snapshot.get("child1"), "cvalue1", "Snapshot should have original child1 value"); + Assert.equal(snapshot.get("parent3"), undefined, "Snapshot should not have new parent3"); + Assert.equal(snapshot.get("child2"), undefined, "Snapshot should not have new child2"); + + // Child state should reflect changes + Assert.equal(childState.get("parent1"), "modified_parent1", "Child should have modified parent1"); + Assert.equal(childState.get("parent2"), undefined, "Child should not have deleted parent2"); + Assert.equal(childState.get("parent3"), "new_parent3", "Child should have new parent3"); + Assert.equal(childState.get("child1"), "modified_child1", "Child should have modified child1"); + Assert.equal(childState.get("child2"), "new_child2", "Child should have new child2"); + } + }); + + this.testCase({ + name: "snapshotW3cTraceState: snapshot with overridden parent values", + test: () => { + const parentState = createW3cTraceState("a=parent_a,b=parent_b,c=parent_c"); + const childState = createW3cTraceState("a=child_a,d=child_d", parentState); + + // Verify child state has overridden parent value + Assert.equal(childState.get("a"), "child_a", "Child should have overridden value for 'a'"); + Assert.equal(childState.get("b"), "parent_b", "Child should have parent value for 'b'"); + Assert.equal(childState.get("c"), "parent_c", "Child should have parent value for 'c'"); + Assert.equal(childState.get("d"), "child_d", "Child should have its own value for 'd'"); + + // Take snapshot + const snapshot = snapshotW3cTraceState(childState); + + Assert.equal(snapshot.keys.length, 4, "Snapshot should have 4 keys"); + Assert.equal(snapshot.get("a"), "child_a", "Snapshot should have overridden value for 'a'"); + Assert.equal(snapshot.get("b"), "parent_b", "Snapshot should have parent value for 'b'"); + Assert.equal(snapshot.get("c"), "parent_c", "Snapshot should have parent value for 'c'"); + Assert.equal(snapshot.get("d"), "child_d", "Snapshot should have child value for 'd'"); + + // Verify snapshot string representation + const snapshotStr = asString(snapshot); + Assert.ok(snapshotStr.includes("a=child_a"), "Snapshot should include overridden value"); + Assert.ok(snapshotStr.includes("b=parent_b"), "Snapshot should include parent value"); + Assert.ok(snapshotStr.includes("c=parent_c"), "Snapshot should include parent value"); + Assert.ok(snapshotStr.includes("d=child_d"), "Snapshot should include child value"); + + // Modify parent's overridden key + parentState.set("a", "new_parent_a"); + + // Snapshot should still have the overridden value + Assert.equal(snapshot.get("a"), "child_a", "Snapshot should keep overridden value despite parent change"); + } + }); + + this.testCase({ + name: "snapshotW3cTraceState: snapshot with deleted parent keys", + test: () => { + const parentState = createW3cTraceState("a=parent_a,b=parent_b,c=parent_c"); + const childState = createW3cTraceState("d=child_d", parentState); + + // Delete a parent key in the child + childState.del("b"); + + // Verify child state behavior + Assert.equal(childState.get("a"), "parent_a", "Child should have parent value for 'a'"); + Assert.equal(childState.get("b"), undefined, "Child should not have deleted 'b' key"); + Assert.equal(childState.get("c"), "parent_c", "Child should have parent value for 'c'"); + Assert.equal(childState.get("d"), "child_d", "Child should have its own value for 'd'"); + + // Take snapshot + const snapshot = snapshotW3cTraceState(childState); + + Assert.equal(snapshot.keys.length, 3, "Snapshot should have 3 keys (deleted key excluded)"); + Assert.equal(snapshot.get("a"), "parent_a", "Snapshot should have parent value for 'a'"); + Assert.equal(snapshot.get("b"), undefined, "Snapshot should not have deleted 'b' key"); + Assert.equal(snapshot.get("c"), "parent_c", "Snapshot should have parent value for 'c'"); + Assert.equal(snapshot.get("d"), "child_d", "Snapshot should have child value for 'd'"); + + // Verify snapshot string representation doesn't include deleted key + const snapshotStr = asString(snapshot); + Assert.ok(!snapshotStr.includes("b="), "Snapshot should not include deleted key"); + Assert.ok(snapshotStr.includes("a=parent_a"), "Snapshot should include parent value"); + Assert.ok(snapshotStr.includes("c=parent_c"), "Snapshot should include parent value"); + Assert.ok(snapshotStr.includes("d=child_d"), "Snapshot should include child value"); + + // Re-add the deleted key in parent after snapshot + parentState.set("b", "restored_parent_b"); + + // Snapshot should still not have the key + Assert.equal(snapshot.get("b"), undefined, "Snapshot should not have restored key"); + + // Child state should also not have the key since it was deleted before child creation + Assert.equal(childState.get("b"), undefined, "Child should not have restored key"); + } + }); + } +} diff --git a/shared/AppInsightsCore/Tests/Unit/src/trace/span.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/trace/span.Tests.ts new file mode 100644 index 000000000..20c195960 --- /dev/null +++ b/shared/AppInsightsCore/Tests/Unit/src/trace/span.Tests.ts @@ -0,0 +1,1970 @@ +import { Assert, AITestClass } from "@microsoft/ai-test-framework"; +import { getDeferred, ICachedValue, isNullOrUndefined, mathMin, objDefine, perfNow, strSubstr } from "@nevware21/ts-utils"; +import { createSpan } from "../../../../src/otel/api/trace/span"; +import { createOTelApi } from "../../../../src/otel/api/OTelApi"; +import { IOTelSpanCtx } from "../../../../src/interfaces/otel/trace/IOTelSpanCtx"; +import { IOTelApi } from "../../../../src/interfaces/otel/IOTelApi"; +import { IOTelConfig } from "../../../../src/interfaces/otel/config/IOTelConfig"; +import { eOTelSpanStatusCode } from "../../../../src/enums/otel/OTelSpanStatus"; +import { IOTelAttributes } from "../../../../src/interfaces/otel/IOTelAttributes"; +import { IReadableSpan } from "../../../../src/interfaces/otel/trace/IReadableSpan"; +import { IDistributedTraceContext } from "../../../../src/interfaces/ai/IDistributedTraceContext"; +import { createDistributedTraceContext } from "../../../../src/core/TelemetryHelpers"; +import { generateW3CId } from "../../../../src/utils/CoreUtils"; +import { suppressTracing, unsuppressTracing, isTracingSuppressed, useSpan, withSpan } from "../../../../src/otel/api/trace/utils"; +import { ITraceCfg } from "../../../../src/interfaces/otel/config/IOTelTraceCfg"; +import { AppInsightsCore } from "../../../../src/core/AppInsightsCore"; +import { IConfiguration } from "../../../../src/interfaces/ai/IConfiguration"; +import { ITraceProvider, ISpanScope, ITraceHost } from "../../../../src/interfaces/ai/ITraceProvider"; +import { IOTelSpanOptions } from "../../../../src/interfaces/otel/trace/IOTelSpanOptions"; +import { eOTelSpanKind } from "../../../../src/enums/otel/OTelSpanKind"; + +export class SpanTests extends AITestClass { + + private _mockApi!: IOTelApi; + private _mockSpanContext!: IDistributedTraceContext; + private _onEndCalls!: IReadableSpan[]; + private _core!: AppInsightsCore; + + public testInitialize() { + super.testInitialize(); + this._onEndCalls = []; + + // Create mock span context + this._mockSpanContext = createDistributedTraceContext({ + traceId: "12345678901234567890123456789012", + spanId: "1234567890123456", + traceFlags: 1, + isRemote: false + }); + + // Create mock API + this._mockApi = { + cfg: { + errorHandlers: {} + } as IOTelConfig + } as IOTelApi; + } + + public testCleanup() { + super.testCleanup(); + this._onEndCalls = []; + + // Clean up AppInsightsCore instance if initialized + if (this._core && this._core.isInitialized()) { + this._core.unload(false); + } + this._core = undefined as any; + } + + /** + * Helper function to create a simple trace provider with onEnd callback + */ + private _createTestTraceProvider(host: ITraceHost, onEnd?: (span: IReadableSpan) => void): ICachedValue { + const actualOnEnd = onEnd || ((span) => this._onEndCalls.push(span)); + + return getDeferred(() => { + const provider: ITraceProvider = { + api: this._mockApi, + createSpan: (name: string, options?: IOTelSpanOptions, parent?: IDistributedTraceContext): IReadableSpan => { + // Create a new distributed trace context for this span + let newCtx: IDistributedTraceContext; + let parentCtx: IDistributedTraceContext | undefined; + + if (options && options.root) { + newCtx = createDistributedTraceContext(); + } else { + newCtx = createDistributedTraceContext(parent || host.getTraceCtx()); + if (newCtx.parentCtx) { + parentCtx = newCtx.parentCtx; + } + } + + // Always generate a new spanId + newCtx.spanId = strSubstr(generateW3CId(), 0, 16); + + // Get configuration from the core if available + let isRecording = options?.recording !== false; + if (this._core && this._core.config && this._core.config.traceCfg && this._core.config.traceCfg.suppressTracing) { + isRecording = false; + } + + // Create the span context + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: newCtx, + attributes: options?.attributes, + startTime: options?.startTime, + isRecording: isRecording, + onEnd: actualOnEnd + }; + + if (parentCtx) { + objDefine(spanCtx, "parentSpanContext", { + v: parentCtx, + w: false + }); + } + + return createSpan(spanCtx, name, options?.kind || eOTelSpanKind.INTERNAL); + }, + getProviderId: (): string => "test-provider", + isAvailable: (): boolean => true + }; + + return provider; + }); + } + + /** + * Helper function to set up AppInsightsCore with trace provider + */ + private _setupCore(config?: Partial): AppInsightsCore { + this._core = new AppInsightsCore(); + + // Create a simple test channel + const testChannel = { + identifier: "TestChannel", + priority: 1001, + initialize: () => {}, + processTelemetry: () => {}, + teardown: () => {}, + isInitialized: () => true + }; + + const coreConfig: IConfiguration = { + instrumentationKey: "test-ikey-12345", + traceCfg: { + serviceName: "test-service" + }, + ...config + }; + + // Initialize the core with the test channel + this._core.initialize(coreConfig, [testChannel]); + + // Set up the trace provider + const traceProvider = this._createTestTraceProvider(this._core); + this._core.setTraceProvider(traceProvider); + + return this._core; + } + + public registerTests() { + this.testCase({ + name: "createSpan: should create span with basic properties", + test: () => { + // Arrange + const spanName = "test-span"; + const spanKind = eOTelSpanKind.CLIENT; + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + + // Act + const span = createSpan(spanCtx, spanName, spanKind); + + // Assert + Assert.ok(span, "Span should be created"); + Assert.equal(span.name, spanName, "Span name should match"); + Assert.equal(span.kind, spanKind, "Span kind should match"); + Assert.equal(span.spanContext().traceId, this._mockSpanContext.traceId, "Trace ID should match"); + Assert.equal(span.spanContext().spanId, this._mockSpanContext.spanId, "Span ID should match"); + Assert.ok(span.isRecording(), "Span should be recording by default"); + Assert.ok(!span.ended, "Span should not be ended initially"); + } + }); + + this.testCase({ + name: "suppressTracing: span with suppressTracing config should not be recording", + test: () => { + // Arrange - create mock API with suppressTracing enabled + const mockApiWithSuppression: IOTelApi = { + cfg: { + traceCfg: { + suppressTracing: true + } as ITraceCfg, + errorHandlers: {} + } as IOTelConfig + } as IOTelApi; + + const spanCtx: IOTelSpanCtx = { + api: mockApiWithSuppression, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + + // Act + const span = createSpan(spanCtx, "suppressed-span", eOTelSpanKind.CLIENT); + + // Assert + Assert.ok(!span.isRecording(), "Span should not be recording when suppressTracing is enabled"); + Assert.ok(span, "Span should still be created"); + Assert.equal(span.name, "suppressed-span", "Span name should be set correctly"); + } + }); + + this.testCase({ + name: "suppressTracing: suppressTracing() helper should set config and return context", + test: () => { + // Arrange - create a context with traceCfg (following IConfiguration pattern) + const testContext = { + traceCfg: { + suppressTracing: false + } as ITraceCfg + }; + + // Act + const returnedContext = suppressTracing(testContext); + + // Assert + Assert.equal(returnedContext, testContext, "suppressTracing should return the same context"); + Assert.ok(testContext.traceCfg.suppressTracing, "suppressTracing config should be set to true"); + Assert.ok(isTracingSuppressed(testContext), "isTracingSuppressed should return true"); + } + }); + + this.testCase({ + name: "suppressTracing: unsuppressTracing() helper should clear config and return context", + test: () => { + // Arrange - create a context with suppressTracing enabled (following IConfiguration pattern) + const testContext = { + traceCfg: { + suppressTracing: true + } as ITraceCfg + }; + + // Act + const returnedContext = unsuppressTracing(testContext); + + // Assert + Assert.equal(returnedContext, testContext, "unsuppressTracing should return the same context"); + Assert.ok(!testContext.traceCfg.suppressTracing, "suppressTracing config should be set to false"); + Assert.ok(!isTracingSuppressed(testContext), "isTracingSuppressed should return false"); + } + }); + + this.testCase({ + name: "createSpan: should create span with options", + test: () => { + // Arrange + const spanName = "test-span-with-options"; + const spanKind = eOTelSpanKind.SERVER; + const attributes: IOTelAttributes = { + "service.name": "test-service", + "http.method": "GET", + "http.status_code": 200 + }; + + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + + // Act + const span = createSpan(spanCtx, spanName, spanKind); + + // Set attributes after creation + span.setAttributes(attributes); + + // Assert + Assert.ok(span, "Span should be created"); + Assert.equal(span.name, spanName, "Span name should match"); + Assert.equal(span.kind, spanKind, "Span kind should match"); + Assert.ok(span.isRecording(), "Span should be recording"); + Assert.ok(!span.ended, "Span should not be ended initially"); + } + }); + + this.testCase({ + name: "createSpan: should end span and call onEnd callback", + test: () => { + // Arrange + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + const span = createSpan(spanCtx, "test-span", eOTelSpanKind.CLIENT); + + // Act + span.end(); + + // Assert + Assert.ok(span.ended, "Span should be marked as ended"); + Assert.equal(this._onEndCalls.length, 1, "onEnd callback should be called once"); + Assert.equal(this._onEndCalls[0], span, "onEnd should be called with the span"); + } + }); + + this.testCase({ + name: "createSpan: should set and get attributes", + test: () => { + // Arrange + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + const span = createSpan(spanCtx, "test-span", eOTelSpanKind.CLIENT); + + // Act + span.setAttribute("http.method", "POST"); + span.setAttribute("http.status_code", 201); + span.setAttributes({ + "service.name": "test-service", + "user.authenticated": true + }); + + // Assert - Note: The exact attribute retrieval method depends on implementation + // This test verifies that setAttribute and setAttributes don't throw errors + Assert.ok(span, "Span should still be valid after setting attributes"); + Assert.ok(span.isRecording(), "Span should still be recording"); + } + }); + + this.testCase({ + name: "createSpan: should set span status", + test: () => { + // Arrange + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + const span = createSpan(spanCtx, "test-span", eOTelSpanKind.CLIENT); + + // Act + span.setStatus({ + code: eOTelSpanStatusCode.ERROR, + message: "Something went wrong" + }); + + // Assert + Assert.ok(span, "Span should still be valid after setting status"); + Assert.ok(span.isRecording(), "Span should still be recording"); + } + }); + + this.testCase({ + name: "createSpan: should handle events (not yet implemented)", + test: () => { + // Arrange + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + const span = createSpan(spanCtx, "test-span", eOTelSpanKind.CLIENT); + + // Act & Assert - Currently events are not supported, so this test just verifies span remains valid + // span.addEvent("request.started"); // Not implemented yet + // span.addEvent("response.received", { ... }); // Not implemented yet + + Assert.ok(span, "Span should still be valid"); + Assert.ok(span.isRecording(), "Span should still be recording"); + } + }); + + this.testCase({ + name: "createSpan: should record exception", + test: () => { + // Arrange + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + const span = createSpan(spanCtx, "test-span", eOTelSpanKind.CLIENT); + const testError = new Error("Test error"); + + // Act + span.recordException(testError); + span.recordException({ + name: "CustomError", + message: "Custom error message", + stack: "Error stack trace" + }); + + // Assert + Assert.ok(span, "Span should still be valid after recording exceptions"); + Assert.ok(span.isRecording(), "Span should still be recording"); + } + }); + + this.testCase({ + name: "createSpan: should update span name", + test: () => { + // Arrange + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + const span = createSpan(spanCtx, "original-name", eOTelSpanKind.CLIENT); + + // Act + span.updateName("updated-name"); + + // Assert + Assert.equal(span.name, "updated-name", "Span name should be updated"); + Assert.ok(span.isRecording(), "Span should still be recording"); + } + }); + + this.testCase({ + name: "createSpan: should handle multiple end calls gracefully", + test: () => { + // Arrange + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + const span = createSpan(spanCtx, "test-span", eOTelSpanKind.CLIENT); + + // Act + span.end(); + span.end(); // Second call should be ignored + span.end(); // Third call should be ignored + + // Assert + Assert.ok(span.ended, "Span should be marked as ended"); + Assert.equal(this._onEndCalls.length, 1, "onEnd callback should be called only once"); + } + }); + + this.testCase({ + name: "createSpan: should not record operations after span is ended", + test: () => { + // Arrange + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + const span = createSpan(spanCtx, "test-span", eOTelSpanKind.CLIENT); + span.end(); + + // Act & Assert - These operations should not throw but should be ignored + span.setAttribute("test.attr", "test-value"); + // span.addEvent("test.event"); // Not implemented yet + span.setStatus({ code: eOTelSpanStatusCode.ERROR }); + span.updateName("new-name"); + span.recordException(new Error("Test error")); + + // The span name should not change after ending + Assert.equal(span.name, "test-span", "Span name should not change after ending"); + Assert.ok(span.ended, "Span should remain ended"); + Assert.ok(!span.isRecording(), "Span should not be recording after ending"); + } + }); + + this.testCase({ + name: "createSpan: should handle invalid attribute values", + test: () => { + // Arrange + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + const span = createSpan(spanCtx, "test-span", eOTelSpanKind.CLIENT); + + // Act & Assert - These should not throw errors + span.setAttribute("", "empty-key"); + span.setAttribute("null-value", null as any); + span.setAttribute("undefined-value", undefined as any); + span.setAttribute("object-value", { nested: "object" } as any); + span.setAttribute("array-value", [1, 2, 3] as any); + + Assert.ok(span, "Span should remain valid after setting invalid attributes"); + Assert.ok(span.isRecording(), "Span should still be recording"); + } + }); + + this.testCase({ + name: "createSpan: should work with different span kinds", + test: () => { + // Test each span kind + const spanKinds = [ + eOTelSpanKind.INTERNAL, + eOTelSpanKind.SERVER, + eOTelSpanKind.CLIENT, + eOTelSpanKind.PRODUCER, + eOTelSpanKind.CONSUMER + ]; + + spanKinds.forEach((kind, index) => { + // Arrange + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + + // Act + const span = createSpan(spanCtx, `test-span-${index}`, kind); + + // Assert + Assert.ok(span, `Span should be created for kind ${kind}`); + Assert.equal(span.kind, kind, `Span kind should match ${kind}`); + Assert.ok(span.isRecording(), `Span should be recording for kind ${kind}`); + }); + } + }); + + this.testCase({ + name: "createSpan: should handle span context correctly", + test: () => { + // Arrange + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + + // Act + const span = createSpan(spanCtx, "test-span", eOTelSpanKind.CLIENT); + const spanContext = span.spanContext(); + + // Assert + Assert.ok(spanContext, "Span context should be available"); + Assert.equal(spanContext.traceId, this._mockSpanContext.traceId, "Trace ID should match"); + Assert.equal(spanContext.spanId, this._mockSpanContext.spanId, "Span ID should match"); + Assert.equal(spanContext.traceFlags, this._mockSpanContext.traceFlags, "Trace flags should match"); + } + }); + + // === AppInsightsCore Integration Tests === + + this.testCase({ + name: "AppInsightsCore Integration: should create span using core.startSpan with trace provider", + test: () => { + // Arrange + const core = this._setupCore(); + + // Act + const span = core.startSpan("integration-test-span", { + kind: eOTelSpanKind.SERVER, + attributes: { + "service.name": "test-service", + "operation.type": "web-request" + } + }); + + // Assert + Assert.ok(span, "Span should be created via core.startSpan"); + const readableSpan = span as IReadableSpan; + Assert.equal(readableSpan.name, "integration-test-span", "Span name should match"); + Assert.equal(readableSpan.kind, eOTelSpanKind.SERVER, "Span kind should match"); + Assert.ok(span?.isRecording(), "Span should be recording"); + Assert.ok(!readableSpan.ended, "Span should not be ended initially"); + } + }); + + this.testCase({ + name: "AppInsightsCore Integration: should call onEnd callback when span ends", + test: () => { + // Arrange + const core = this._setupCore(); + const span = core.startSpan("callback-test-span"); + Assert.equal(this._onEndCalls.length, 0, "No onEnd calls initially"); + + // Act + Assert.ok(span, "Span should be created"); + span?.end(); + + // Assert + const readableSpan = span as IReadableSpan; + Assert.ok(readableSpan.ended, "Span should be ended"); + Assert.equal(this._onEndCalls.length, 1, "onEnd callback should be called once"); + Assert.equal(this._onEndCalls[0].name, "callback-test-span", "onEnd should receive the correct span"); + } + }); + + this.testCase({ + name: "AppInsightsCore Integration: should inherit trace context from core", + test: () => { + // Arrange + const core = this._setupCore(); + + // Set a specific trace context on the core + const parentTraceContext = createDistributedTraceContext({ + traceId: "parent-trace-12345678901234567890123456789012", + spanId: "parent-span-1234567890123456", + traceFlags: 1 + }); + core.setTraceCtx(parentTraceContext); + + // Act + const span = core.startSpan("child-span"); + + // Assert + Assert.ok(span, "Child span should be created"); + const spanContext = span?.spanContext(); + Assert.equal(spanContext?.traceId, parentTraceContext.traceId, "Child span should inherit parent trace ID"); + Assert.notEqual(spanContext?.spanId, parentTraceContext.spanId, "Child span should have different span ID"); + } + }); + + this.testCase({ + name: "AppInsightsCore Integration: should handle configuration with suppressTracing", + test: () => { + // Arrange + const core = this._setupCore({ + traceCfg: { + suppressTracing: true, + serviceName: "suppressed-service" + } + }); + + // Act + const span = core.startSpan("suppressed-span"); + + // Assert + Assert.ok(span, "Span should still be created when suppressTracing is enabled"); + const readableSpan = span as IReadableSpan; + Assert.ok(!span?.isRecording(), "Span should not be recording when suppressTracing is enabled"); + Assert.equal(readableSpan.name, "suppressed-span", "Span name should be set correctly"); + } + }); + + this.testCase({ + name: "AppInsightsCore Integration: should support active span management", + test: () => { + // Arrange + const core = this._setupCore(); + let initialActiveSpan = core.getActiveSpan(); + Assert.ok(!isNullOrUndefined(initialActiveSpan), "Initially, activeSpan should not be null with a trace provider"); + + // Act + const span = core.startSpan("active-span-test"); + Assert.ok(span, "Span should be created"); + + // Debug: Check if trace provider exists + const traceProvider = core.getTraceProvider(); + Assert.ok(traceProvider, "Trace provider should exist"); + Assert.ok(traceProvider?.isAvailable(), "Trace provider should be available"); + + // Debug: Check trace provider before setActiveSpan + const providerActiveSpanBefore = core.getActiveSpan(); + Assert.equal(providerActiveSpanBefore, initialActiveSpan, "Trace provider should return the initially active span before setActiveSpan"); + + // Manually set as active span (this would normally be done by startActiveSpan) + const scope = core.setActiveSpan(span); + + // Assert scope object + Assert.ok(scope, "Scope should be returned"); + Assert.equal(scope.span, span, "Scope.span should equal the passed span"); + + // Debug: Check trace provider directly after setActiveSpan + const providerActiveSpanAfter = core.getActiveSpan(); + Assert.ok(providerActiveSpanAfter, "Trace provider should have active span after setActiveSpan"); + Assert.equal(providerActiveSpanAfter, span, "Trace provider active span should be the same instance"); + + // Assert + const activeSpan = core.getActiveSpan(); + Assert.ok(activeSpan, "Active span should be available"); + if (activeSpan) { + const readableActiveSpan = activeSpan as IReadableSpan; + Assert.equal(readableActiveSpan.name, "active-span-test", "Active span should be the correct span"); + Assert.equal(activeSpan, span, "Active span should be the same instance"); + } + } + }); + + this.testCase({ + name: "AppInsightsCore Integration: should create child spans with proper parent-child relationship", + test: () => { + // Arrange + const core = this._setupCore(); + + // Act + const parentSpan = core.startSpan("parent-operation", { + kind: eOTelSpanKind.SERVER, + attributes: { "operation.name": "process-request" } + }); + Assert.ok(parentSpan, "Parent span should be created"); + + // Set parent as active + const scope = core.setActiveSpan(parentSpan!); + const activeSpan = core.getActiveSpan(); + + // Assert scope and activeSpan + Assert.ok(scope, "Scope should be returned"); + Assert.equal(scope.span, parentSpan, "Scope.span should equal the parent span"); + Assert.equal(activeSpan, parentSpan, "GetGetGetGetGetGetGetGetGetActiveSpan() should return the parent span"); + + const childSpan = core.startSpan("child-operation", { + kind: eOTelSpanKind.CLIENT, + attributes: { "operation.name": "database-query" } + }); + Assert.ok(childSpan, "Child span should be created"); + + // Assert + const parentContext = parentSpan!.spanContext(); + const childContext = childSpan!.spanContext(); + + Assert.equal(childContext.traceId, parentContext.traceId, "Child should have same trace ID as parent"); + Assert.notEqual(childContext.spanId, parentContext.spanId, "Child should have different span ID from parent"); + } + }); + + this.testCase({ + name: "AppInsightsCore Integration: should handle span attributes and status correctly", + test: () => { + // Arrange + const core = this._setupCore(); + const span = core.startSpan("attribute-test-span", { + attributes: { + "initial.attribute": "initial-value" + } + }); + + // Act + Assert.ok(span, "Span should be created"); + span!.setAttribute("http.method", "POST"); + span!.setAttribute("http.status_code", 201); + span!.setAttributes({ + "service.version": "1.2.3", + "user.authenticated": true + }); + + span!.setStatus({ + code: eOTelSpanStatusCode.OK, + message: "Operation completed successfully" + }); + + // Assert + const readableSpan = span! as IReadableSpan; + Assert.ok(span!.isRecording(), "Span should still be recording"); + Assert.ok(!readableSpan.ended, "Span should not be ended"); + } + }); + + this.testCase({ + name: "AppInsightsCore Integration: should handle span lifecycle properly", + test: () => { + // Arrange + const core = this._setupCore(); + const span = core.startSpan("lifecycle-test-span"); + Assert.equal(this._onEndCalls.length, 0, "No onEnd calls initially"); + + // Act - Perform operations during span lifetime + Assert.ok(span, "Span should be created"); + span!.setAttribute("test.phase", "active"); + span!.recordException(new Error("Test exception for logging")); + span!.updateName("lifecycle-test-span-updated"); + + // End the span + span!.end(); + + // Assert + const readableSpan = span! as IReadableSpan; + Assert.ok(readableSpan.ended, "Span should be ended"); + Assert.equal(readableSpan.name, "lifecycle-test-span-updated", "Span name should be updated"); + Assert.ok(!span!.isRecording(), "Span should not be recording after ending"); + Assert.equal(this._onEndCalls.length, 1, "onEnd callback should be called once"); + + // Verify operations after end are ignored + span!.setAttribute("test.phase", "completed"); + span!.updateName("should-not-change"); + Assert.equal(readableSpan.name, "lifecycle-test-span-updated", "Span name should not change after ending"); + } + }); + + this.testCase({ + name: "AppInsightsCore Integration: should handle multiple spans and proper cleanup", + test: () => { + // Arrange + const core = this._setupCore(); + const spans: IReadableSpan[] = []; + + // Act - Create multiple spans + for (let i = 0; i < 5; i++) { + const span = core.startSpan(`batch-span-${i}`, { + kind: eOTelSpanKind.INTERNAL, + attributes: { + "span.index": i, + "batch.id": "test-batch-123" + } + }); + Assert.ok(span, `Span ${i} should be created`); + spans.push(span!); + } + + // End all spans + spans.forEach(span => span.end()); + + // Assert + Assert.equal(spans.length, 5, "Should have created 5 spans"); + Assert.equal(this._onEndCalls.length, 5, "Should have 5 onEnd callback calls"); + + spans.forEach((span, index) => { + const readableSpan = span as IReadableSpan; + Assert.ok(readableSpan.ended, `Span ${index} should be ended`); + Assert.equal(readableSpan.name, `batch-span-${index}`, `Span ${index} should have correct name`); + }); + + // Verify all spans in onEnd calls + this._onEndCalls.forEach((readableSpan, index) => { + Assert.equal(readableSpan.name, `batch-span-${index}`, `onEnd span ${index} should have correct name`); + }); + } + }); + + this.testCase({ + name: "AppInsightsCore Integration: should handle trace provider configuration changes", + test: () => { + // Arrange + const core = this._setupCore(); + const originalProvider = core.getTraceProvider(); + Assert.ok(originalProvider, "Original trace provider should be available"); + + // Act - Create new trace provider with different onEnd behavior + let alternativeOnEndCalls: IReadableSpan[] = []; + const newProvider = this._createTestTraceProvider(core, (span) => { + alternativeOnEndCalls.push(span); + }); + + core.setTraceProvider(newProvider); + const updatedProvider = core.getTraceProvider(); + + // Create spans with new provider + const span1 = core.startSpan("provider-change-test-1"); + Assert.ok(span1, "Span should be created with new provider"); + span1!.end(); + + // Assert + Assert.notEqual(updatedProvider, originalProvider, "Trace provider should be updated"); + Assert.equal(this._onEndCalls.length, 0, "Original onEnd callback should not be called"); + Assert.equal(alternativeOnEndCalls.length, 1, "New onEnd callback should be called"); + Assert.equal(alternativeOnEndCalls[0].name, "provider-change-test-1", "New callback should receive correct span"); + } + }); + + this.testCase({ + name: "AppInsightsCore Integration: should handle configuration inheritance from core config", + test: () => { + // Arrange + const core = this._setupCore({ + traceCfg: { + serviceName: "integration-test-service", + generalLimits: { + attributeCountLimit: 64, + attributeValueLengthLimit: 256 + // }, + // spanLimits: { + // attributeCountLimit: 32, + // eventCountLimit: 16, + // linkCountLimit: 8 + } + } + }); + + // Act + const span = core.startSpan("config-inheritance-test", { + attributes: { + "service.name": "integration-test-service", // Should inherit from traceCfg + "test.configured": true + } + }); + + // Assert + Assert.ok(span, "Span should be created with inherited configuration"); + const readableSpan = span! as IReadableSpan; + Assert.ok(span!.isRecording(), "Span should be recording"); + + // The span should exist and be functional - detailed config validation + // would require access to internal span configuration which is not exposed + Assert.equal(readableSpan.name, "config-inheritance-test", "Span name should be correct"); + } + }); + + // === Tracer startActiveSpan Tests === + + this.testCase({ + name: "Tracer.startActiveSpan: should set span as active during callback execution", + test: () => { + // Arrange + const core = this._setupCore(); + const otelApi = createOTelApi({ host: core }); + const tracer = otelApi.trace.getTracer("test-tracer"); + const initialActiveSpan = core.getActiveSpan(); + + let activeSpanInsideCallback: IReadableSpan | null | undefined = null; + let callbackExecuted = false; + + // Act + const result = tracer.startActiveSpan("test-operation", (span) => { + callbackExecuted = true; + + // Check if the span is set as active in the host instance + activeSpanInsideCallback = core.getActiveSpan(); + + span.setAttribute("test.key", "test-value"); + return "callback-result"; + }); + + // Assert + Assert.ok(callbackExecuted, "Callback should have been executed"); + Assert.equal(result, "callback-result", "Should return callback result"); + Assert.ok(activeSpanInsideCallback, "Active span should be set during callback"); + Assert.equal(activeSpanInsideCallback?.name, "test-operation", "Active span should be the created span"); + + // Verify active span is restored after callback + const activeSpanAfterCallback = core.getActiveSpan(); + Assert.equal(activeSpanAfterCallback, initialActiveSpan, "Active span should be restored to initial state"); + } + }); + + this.testCase({ + name: "Tracer.startActiveSpan: should automatically end span after callback completes", + test: () => { + // Arrange + const core = this._setupCore(); + const otelApi = createOTelApi({ host: core }); + const tracer = otelApi.trace.getTracer("test-tracer"); + + let spanInsideCallback: IReadableSpan | null = null; + const initialOnEndCount = this._onEndCalls.length; + + // Act + tracer.startActiveSpan("auto-end-test", (span) => { + spanInsideCallback = span; + Assert.ok(!span.ended, "Span should not be ended during callback"); + span.setAttribute("operation.type", "test"); + }); + + // Assert + Assert.ok(spanInsideCallback, "Span should have been passed to callback"); + Assert.ok(spanInsideCallback!.ended, "Span should be ended after callback completes"); + Assert.equal(this._onEndCalls.length, initialOnEndCount + 1, "onEnd should have been called"); + Assert.equal(this._onEndCalls[this._onEndCalls.length - 1].name, "auto-end-test", "onEnd should receive the correct span"); + } + }); + + this.testCase({ + name: "Tracer.startActiveSpan: should handle nested startActiveSpan calls", + test: () => { + // Arrange + const core = this._setupCore(); + const otelApi = createOTelApi({ host: core }); + const tracer = otelApi.trace.getTracer("test-tracer"); + + const executionTrace: string[] = []; + + // Act + const result = tracer.startActiveSpan("outer-operation", (outerSpan) => { + const outerActiveSpan = core.getActiveSpan(); + executionTrace.push(`outer-start: ${outerActiveSpan?.name}`); + + outerSpan.setAttribute("level", "outer"); + + // Nested startActiveSpan + const innerResult = tracer.startActiveSpan("inner-operation", (innerSpan) => { + const innerActiveSpan = core.getActiveSpan(); + executionTrace.push(`inner: ${innerActiveSpan?.name}`); + + innerSpan.setAttribute("level", "inner"); + + // Verify the inner span is now active + Assert.equal(innerActiveSpan?.name, "inner-operation", "Inner span should be active during inner callback"); + + return "inner-result"; + }); + + // After inner callback, outer should be active again + const outerActiveSpanRestored = core.getActiveSpan(); + executionTrace.push(`outer-end: ${outerActiveSpanRestored?.name}`); + + Assert.equal(outerActiveSpanRestored?.name, "outer-operation", "Outer span should be restored as active after inner callback"); + + return `outer(${innerResult})`; + }); + + // Assert + Assert.equal(result, "outer(inner-result)", "Nested startActiveSpan should work correctly"); + Assert.equal(executionTrace.length, 3, "Should have captured 3 execution points"); + Assert.equal(executionTrace[0], "outer-start: outer-operation", "Outer callback should see outer span active"); + Assert.equal(executionTrace[1], "inner: inner-operation", "Inner callback should see inner span active"); + Assert.equal(executionTrace[2], "outer-end: outer-operation", "Outer callback should see outer span restored after inner"); + } + }); + + this.testCase({ + name: "Tracer.startActiveSpan: should handle async callback with active span management", + test: () => { + // Arrange + const core = this._setupCore(); + const otelApi = createOTelApi({ host: core }); + const tracer = otelApi.trace.getTracer("test-tracer"); + const initialActiveSpan = core.getActiveSpan(); + + let activeSpanDuringAsync: IReadableSpan | null | undefined = null; + + // Act + return tracer.startActiveSpan("async-operation", async (span) => { + // Check active span at start + activeSpanDuringAsync = core.getActiveSpan(); + Assert.equal(activeSpanDuringAsync?.name, "async-operation", "Span should be active at start of async callback"); + + span.setAttribute("async.phase", "start"); + + // Simulate async operation + await new Promise(resolve => setTimeout(resolve, 100)); + + // Check active span after async operation + const activeSpanAfterAwait = core.getActiveSpan(); + Assert.equal(activeSpanAfterAwait?.name, "async-operation", "Span should still be active after await"); + + span.setAttribute("async.phase", "end"); + + return "async-result"; + }).then(result => { + // Assert + Assert.equal(result, "async-result", "Should return async callback result"); + Assert.ok(activeSpanDuringAsync, "Active span should have been set during callback"); + + // Verify active span is restored after async callback completes + const activeSpanAfterCallback = core.getActiveSpan(); + Assert.equal(activeSpanAfterCallback, initialActiveSpan, "Active span should be restored after async callback"); + }); + } + }); + + this.testCase({ + name: "Tracer.startActiveSpan: should handle exceptions and still restore active span", + test: () => { + // Arrange + const core = this._setupCore(); + const otelApi = createOTelApi({ host: core }); + const tracer = otelApi.trace.getTracer("test-tracer"); + const initialActiveSpan = core.getActiveSpan(); + + let spanInsideCallback: IReadableSpan | null = null; + let exceptionThrown = false; + + // Act + try { + tracer.startActiveSpan("error-operation", (span) => { + spanInsideCallback = span; + const activeSpan = core.getActiveSpan(); + Assert.equal(activeSpan?.name, "error-operation", "Span should be active before exception"); + + throw new Error("Test exception"); + }); + } catch (error: any) { + exceptionThrown = true; + Assert.equal(error.message, "Test exception", "Should propagate exception"); + } + + // Assert + Assert.ok(exceptionThrown, "Exception should have been thrown"); + Assert.ok(spanInsideCallback, "Span should have been created"); + Assert.ok(spanInsideCallback!.ended, "Span should be ended even after exception"); + + // Verify active span is restored even after exception + const activeSpanAfterException = core.getActiveSpan(); + Assert.equal(activeSpanAfterException, initialActiveSpan, "Active span should be restored even after exception"); + } + }); + + this.testCase({ + name: "Tracer.startActiveSpan: should work with options parameter", + test: () => { + // Arrange + const core = this._setupCore(); + const otelApi = createOTelApi({ host: core }); + const tracer = otelApi.trace.getTracer("test-tracer"); + + let activeSpanInsideCallback: IReadableSpan | null | undefined = null; + + // Act + const result = tracer.startActiveSpan("options-test", + { + kind: eOTelSpanKind.SERVER, + attributes: { + "http.method": "POST", + "http.route": "/api/test" + } + }, + (span) => { + activeSpanInsideCallback = core.getActiveSpan(); + + Assert.equal(span.kind, eOTelSpanKind.SERVER, "Span kind should match options"); + span.setAttribute("response.status", 200); + + return "options-result"; + } + ); + + // Assert + Assert.equal(result, "options-result", "Should return callback result"); + Assert.ok(activeSpanInsideCallback, "Active span should be set during callback"); + Assert.equal(activeSpanInsideCallback?.name, "options-test", "Active span should be the created span"); + } + }); + + this.testCase({ + name: "Tracer.startActiveSpan: child spans should inherit parent from active span", + test: () => { + // Arrange + const core = this._setupCore(); + const otelApi = createOTelApi({ host: core }); + const tracer = otelApi.trace.getTracer("test-tracer"); + + let parentSpanId: string | undefined; + let childSpanParentId: string | undefined; + + // Act + tracer.startActiveSpan("parent-operation", (parentSpan) => { + parentSpanId = parentSpan.spanContext().spanId; + + // Create a child span using startSpan (not startActiveSpan) + // It should automatically use the active span (parent-operation) as parent + const childSpan = core.startSpan("child-operation"); + + Assert.ok(childSpan, "Child span should be created"); + const childContext = childSpan!.spanContext(); + childSpanParentId = childSpan!.parentSpanId; + + // Verify parent-child relationship + Assert.equal(childContext.traceId, parentSpan.spanContext().traceId, "Child should have same trace ID as parent"); + + childSpan!.end(); + }); + + // Assert + Assert.ok(parentSpanId, "Parent span ID should be captured"); + Assert.equal(childSpanParentId, parentSpanId, "Child span parent ID should match parent span ID"); + } + }); + + // === withSpan Helper Tests === + + this.testCase({ + name: "withSpan: should execute function with span as active span", + test: () => { + // Arrange + const core = this._setupCore(); + let initialActiveSpan = core.getActiveSpan(); + Assert.ok(!isNullOrUndefined(initialActiveSpan), "Initially, activeSpan should not be null with a trace provider"); + const testSpan = core.startSpan("withSpan-test-active"); + Assert.ok(testSpan, "Test span should be created"); + + let capturedActiveSpan: IReadableSpan | null = null; + const testFunction = () => { + capturedActiveSpan = core.getActiveSpan(); + return "test-result"; + }; + + // Act + const result = withSpan(core, testSpan!, testFunction); + + // Assert + Assert.equal(result, "test-result", "withSpan should return function result"); + Assert.ok(capturedActiveSpan, "Function should have access to active span"); + Assert.equal(capturedActiveSpan, testSpan, "Active span should be the provided span"); + Assert.equal(core.getActiveSpan(), initialActiveSpan, "Active span should be restored after execution"); + } + }); + + this.testCase({ + name: "withSpan: should restore previous active span after execution", + test: () => { + // Arrange + const core = this._setupCore(); + const previousSpan = core.startSpan("previous-span"); + const testSpan = core.startSpan("withSpan-test-restore"); + Assert.ok(previousSpan && testSpan, "Both spans should be created"); + + // Set previous span as active + core.setActiveSpan(previousSpan!); + Assert.equal(core.getActiveSpan(), previousSpan, "Previous span should be active initially"); + + let capturedActiveSpan: IReadableSpan | null = null; + const testFunction = () => { + capturedActiveSpan = core.getActiveSpan(); + return 42; + }; + + // Act + const result = withSpan(core, testSpan!, testFunction); + + // Assert + Assert.equal(result, 42, "withSpan should return function result"); + Assert.equal(capturedActiveSpan, testSpan, "Function should have access to test span"); + Assert.equal(core.getActiveSpan(), previousSpan, "Previous active span should be restored"); + } + }); + + this.testCase({ + name: "withSpan: should handle function with arguments", + test: () => { + // Arrange + const core = this._setupCore(); + const testSpan = core.startSpan("withSpan-test-args"); + Assert.ok(testSpan, "Test span should be created"); + + let capturedArgs: any[] = []; + const testFunction = (...args: any[]) => { + capturedArgs = args; + return args.reduce((sum, val) => sum + val, 0); + }; + + // Act + const result = withSpan(core, testSpan!, testFunction, undefined, 10, 20, 30); + + // Assert + Assert.equal(result, 60, "withSpan should return correct sum"); + Assert.equal(capturedArgs.length, 3, "Function should receive all arguments"); + Assert.equal(capturedArgs[0], 10, "First argument should be correct"); + Assert.equal(capturedArgs[1], 20, "Second argument should be correct"); + Assert.equal(capturedArgs[2], 30, "Third argument should be correct"); + } + }); + + this.testCase({ + name: "withSpan: should handle function with thisArg context", + test: () => { + // Arrange + const core = this._setupCore(); + const testSpan = core.startSpan("withSpan-test-this"); + Assert.ok(testSpan, "Test span should be created"); + + const contextObject = { + value: 100, + getValue: function(multiplier: number) { + return this.value * multiplier; + } + }; + + // Act + const result = withSpan(core, testSpan!, contextObject.getValue, contextObject, 2); + + // Assert + Assert.equal(result, 200, "withSpan should execute with correct this context"); + } + }); + + this.testCase({ + name: "withSpan: should handle exceptions and still restore active span", + test: () => { + // Arrange + const core = this._setupCore(); + const previousSpan = core.startSpan("previous-span-exception"); + const testSpan = core.startSpan("withSpan-test-exception"); + Assert.ok(previousSpan && testSpan, "Both spans should be created"); + + core.setActiveSpan(previousSpan!); + + const testFunction = () => { + throw new Error("Test exception"); + }; + + // Act & Assert + let thrownError: Error | null = null; + try { + withSpan(core, testSpan!, testFunction); + } catch (error) { + thrownError = error as Error; + } + + Assert.ok(thrownError, "Exception should be thrown"); + Assert.equal(thrownError!.message, "Test exception", "Exception message should be preserved"); + Assert.equal(core.getActiveSpan(), previousSpan, "Previous active span should be restored even after exception"); + } + }); + + this.testCase({ + name: "withSpan: should work with functions returning different types", + test: () => { + // Arrange + const core = this._setupCore(); + const testSpan = core.startSpan("withSpan-test-types"); + Assert.ok(testSpan, "Test span should be created"); + + // Test string return + const stringResult = withSpan(core, testSpan!, () => "hello world"); + Assert.equal(stringResult, "hello world", "String return should work"); + + // Test number return + const numberResult = withSpan(core, testSpan!, () => 123.45); + Assert.equal(numberResult, 123.45, "Number return should work"); + + // Test boolean return + const booleanResult = withSpan(core, testSpan!, () => true); + Assert.equal(booleanResult, true, "Boolean return should work"); + + // Test object return + const objectResult = withSpan(core, testSpan!, () => ({ key: "value" })); + Assert.ok(objectResult && objectResult.key === "value", "Object return should work"); + + // Test undefined return + const undefinedResult = withSpan(core, testSpan!, () => undefined); + Assert.equal(undefinedResult, undefined, "Undefined return should work"); + + // Test null return + const nullResult = withSpan(core, testSpan!, () => null); + Assert.equal(nullResult, null, "Null return should work"); + } + }); + + this.testCase({ + name: "withSpan: should work with async-like function patterns", + test: () => { + // Arrange + const core = this._setupCore(); + let initialActiveSpan = core.getActiveSpan(); + Assert.ok(!isNullOrUndefined(initialActiveSpan), "Initially, activeSpan should not be null with a trace provider"); + const testSpan = core.startSpan("withSpan-test-async-pattern"); + Assert.ok(testSpan, "Test span should be created"); + + let spanDuringExecution: IReadableSpan | null = null; + + // Simulate async-like pattern with callback + const asyncFunction = (callback: (result: string) => void) => { + spanDuringExecution = core.getActiveSpan(); + // Simulate some async work completing synchronously for this test + callback("async-result"); + return "function-result"; + }; + + let callbackResult = ""; + const callback = (result: string) => { + callbackResult = result; + }; + + // Act + const result = withSpan(core, testSpan!, asyncFunction, undefined, callback); + + // Assert + Assert.equal(result, "function-result", "withSpan should return main function result"); + Assert.equal(callbackResult, "async-result", "Callback should be executed"); + Assert.equal(spanDuringExecution, testSpan, "Active span should be available during execution"); + Assert.equal(core.getActiveSpan(), initialActiveSpan, "Active span should be restored after completion"); + } + }); + + this.testCase({ + name: "withSpan: should work when no previous active span exists", + test: () => { + // Arrange + const core = this._setupCore(); + let initialActiveSpan = core.getActiveSpan(); + const testSpan = core.startSpan("withSpan-test-no-previous"); + Assert.ok(testSpan, "Test span should be created"); + Assert.equal(core.getActiveSpan(), initialActiveSpan, "Just starting a span should not change active span"); + + let capturedActiveSpan: IReadableSpan | null = null; + const testFunction = () => { + capturedActiveSpan = core.getActiveSpan(); + return "success"; + }; + + // Act + const result = withSpan(core, testSpan!, testFunction); + + // Assert + Assert.equal(result, "success", "Function should execute successfully"); + Assert.equal(capturedActiveSpan, testSpan, "Test span should be active during execution"); + Assert.equal(core.getActiveSpan(), initialActiveSpan, "No active span should be restored"); + } + }); + + this.testCase({ + name: "withSpan: should work with nested withSpan calls", + test: () => { + // Arrange + const core = this._setupCore(); + let initialActiveSpan = core.getActiveSpan(); + Assert.ok(!isNullOrUndefined(initialActiveSpan), "Initially, activeSpan should not be null with a trace provider"); + const outerSpan = core.startSpan("outer-span"); + const innerSpan = core.startSpan("inner-span"); + Assert.ok(outerSpan && innerSpan, "Both spans should be created"); + + const executionTrace: string[] = []; + + const innerFunction = () => { + const activeSpan = core.getActiveSpan(); + executionTrace.push(`inner: ${activeSpan ? (activeSpan as IReadableSpan).name : 'null'}`); + return "inner-result"; + }; + + const outerFunction = () => { + const activeSpanBefore = core.getActiveSpan(); + executionTrace.push(`outer-start: ${activeSpanBefore ? (activeSpanBefore as IReadableSpan).name : 'null'}`); + + const innerResult = withSpan(core, innerSpan!, innerFunction); + + const activeSpanAfter = core.getActiveSpan(); + executionTrace.push(`outer-end: ${activeSpanAfter ? (activeSpanAfter as IReadableSpan).name : 'null'}`); + + return `outer(${innerResult})`; + }; + + // Act + const result = withSpan(core, outerSpan!, outerFunction); + + // Assert + Assert.equal(result, "outer(inner-result)", "Nested withSpan should work correctly"); + Assert.equal(executionTrace.length, 3, "Should have captured 3 execution points"); + Assert.equal(executionTrace[0], "outer-start: outer-span", "Outer function should see outer span"); + Assert.equal(executionTrace[1], "inner: inner-span", "Inner function should see inner span"); + Assert.equal(executionTrace[2], "outer-end: outer-span", "Outer function should see outer span restored"); + Assert.equal(core.getActiveSpan(), initialActiveSpan, "No active span should remain after nested execution"); + } + }); + + this.testCase({ + name: "withSpan: should handle span operations within withSpan context", + test: () => { + // Arrange + const core = this._setupCore(); + let initialActiveSpan = core.getActiveSpan(); + Assert.ok(!isNullOrUndefined(initialActiveSpan), "Initially, activeSpan should not be null with a trace provider"); + const testSpan = core.startSpan("withSpan-test-operations"); + Assert.ok(testSpan, "Test span should be created"); + + const testFunction = () => { + const activeSpan = core.getActiveSpan(); + Assert.ok(activeSpan, "Should have active span in function"); + + // Perform span operations + activeSpan!.setAttribute("operation.name", "test-operation"); + activeSpan!.setAttribute("operation.step", 1); + + // Create child span + const childSpan = core.startSpan("child-operation"); + Assert.ok(childSpan, "Child span should be created"); + + childSpan!.setAttribute("child.attribute", "child-value"); + childSpan!.end(); + + return "operations-completed"; + }; + + // Act + const result = withSpan(core, testSpan!, testFunction); + + // Assert + Assert.equal(result, "operations-completed", "Function should complete successfully"); + Assert.equal(core.getActiveSpan(), initialActiveSpan, "Active span should be restored"); + + // Verify span operations were applied (span should still be valid) + const readableSpan = testSpan! as IReadableSpan; + Assert.ok(!readableSpan.ended, "Test span should not be ended"); + Assert.ok(testSpan!.isRecording(), "Test span should still be recording"); + } + }); + + this.testCase({ + name: "withSpan: should work with core that has no trace provider", + test: () => { + // Arrange + const core = new AppInsightsCore(); + + // Create a simple test channel + const testChannel = { + identifier: "TestChannel", + priority: 1001, + initialize: () => {}, + processTelemetry: () => {}, + teardown: () => {}, + isInitialized: () => true + }; + + core.initialize({ instrumentationKey: "test-key" }, [testChannel]); // Initialize with channel but no trace provider + + // Create a mock span (this would need to come from somewhere else since no provider) + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + const mockSpan = createSpan(spanCtx, "mock-span", eOTelSpanKind.CLIENT); + + let functionExecuted = false; + const testFunction = () => { + functionExecuted = true; + return "no-provider-result"; + }; + + // Act + const result = withSpan(core, mockSpan, testFunction); + + // Assert + Assert.equal(result, "no-provider-result", "Function should execute even without trace provider"); + Assert.ok(functionExecuted, "Function should have been executed"); + + // Cleanup + core.unload(false); + } + }); + + this.testCase({ + name: "withSpan: performance test - should not add significant overhead", + test: () => { + // Arrange + const core = this._setupCore(); + const testSpan = core.startSpan("withSpan-performance-test"); + Assert.ok(testSpan, "Test span should be created"); + + const iterations = 10000; + let computeResult = 0; + + const computeFunction = (base: number, multiplier: number) => { + // Simple computation to measure overhead + return base * multiplier + Math.sqrt(base); + }; + + // Measure time without withSpan + const startWithout = perfNow(); + for (let i = 0; i < iterations; i++) { + computeResult += computeFunction(i, 2); + } + const timeWithout = perfNow() - startWithout; + + // Reset result + computeResult = 0; + + // Measure time with withSpan + const startWith = perfNow(); + for (let i = 0; i < iterations; i++) { + computeResult += withSpan(core, testSpan!, computeFunction, undefined, i, 2); + } + const timeWith = perfNow() - startWith; + + // Assert reasonable performance characteristics + // withSpan should not add more than 10x overhead (very generous threshold) + const overhead = timeWith / (timeWithout || 1); + Assert.ok(overhead < 15, `withSpan overhead should be reasonable: ${overhead.toFixed(2)}x`); + + // Results should be the same + Assert.ok(computeResult > 0, "Computations should have produced results"); + } + }); + // === useSpan Helper Tests === + + this.testCase({ + name: "useSpan: should execute function with span as active span", + test: () => { + // Arrange + const core = this._setupCore(); + let initialActiveSpan = core.getActiveSpan(); + Assert.ok(!isNullOrUndefined(initialActiveSpan), "Initially, activeSpan should not be null with a trace provider"); + const testSpan = core.startSpan("useSpan-test-active"); + Assert.ok(testSpan, "Test span should be created"); + + let capturedActiveSpan: IReadableSpan | null = null; + const testFunction = () => { + capturedActiveSpan = core.getActiveSpan(); + return "test-result"; + }; + + // Act + const result = useSpan(core, testSpan!, testFunction); + + // Assert + Assert.equal(result, "test-result", "useSpan should return function result"); + Assert.ok(capturedActiveSpan, "Function should have access to active span"); + Assert.equal(capturedActiveSpan, testSpan, "Active span should be the provided span"); + Assert.equal(core.getActiveSpan(), initialActiveSpan, "Active span should be restored after execution"); + } + }); + + this.testCase({ + name: "useSpan: should restore previous active span after execution", + test: () => { + // Arrange + const core = this._setupCore(); + const previousSpan = core.startSpan("previous-span"); + const testSpan = core.startSpan("useSpan-test-restore"); + Assert.ok(previousSpan && testSpan, "Both spans should be created"); + + // Set previous span as active + core.setActiveSpan(previousSpan!); + Assert.equal(core.getActiveSpan(), previousSpan, "Previous span should be active initially"); + + let capturedActiveSpan: IReadableSpan | null = null; + const testFunction = () => { + capturedActiveSpan = core.getActiveSpan(); + return 42; + }; + + // Act + const result = useSpan(core, testSpan!, testFunction); + + // Assert + Assert.equal(result, 42, "useSpan should return function result"); + Assert.equal(capturedActiveSpan, testSpan, "Function should have access to test span"); + Assert.equal(core.getActiveSpan(), previousSpan, "Previous active span should be restored"); + } + }); + + this.testCase({ + name: "useSpan: should handle function with arguments", + test: () => { + // Arrange + const core = this._setupCore(); + const testSpan = core.startSpan("useSpan-test-args"); + Assert.ok(testSpan, "Test span should be created"); + + let capturedArgs: any[] = []; + const testFunction = (scope: ISpanScope, ...args: any[]) => { + capturedArgs = args; + return args.reduce((sum, val) => sum + val, 0); + }; + + // Act + const result = useSpan(core, testSpan!, testFunction, undefined, 10, 20, 30); + + // Assert + Assert.equal(result, 60, "useSpan should return correct sum"); + Assert.equal(capturedArgs.length, 3, "Function should receive all arguments"); + Assert.equal(capturedArgs[0], 10, "First argument should be correct"); + Assert.equal(capturedArgs[1], 20, "Second argument should be correct"); + Assert.equal(capturedArgs[2], 30, "Third argument should be correct"); + } + }); + + this.testCase({ + name: "useSpan: should handle function with thisArg context", + test: () => { + // Arrange + const core = this._setupCore(); + const testSpan = core.startSpan("useSpan-test-this"); + Assert.ok(testSpan, "Test span should be created"); + + const contextObject = { + value: 100, + getValue: function(scope: ISpanScope, multiplier: number) { + return this.value * multiplier; + } + }; + + // Act + const result = useSpan(core, testSpan!, contextObject.getValue, contextObject, 2); + + // Assert + Assert.equal(result, 200, "useSpan should execute with correct this context"); + } + }); + + this.testCase({ + name: "useSpan: should handle exceptions and still restore active span", + test: () => { + // Arrange + const core = this._setupCore(); + const previousSpan = core.startSpan("previous-span-exception"); + const testSpan = core.startSpan("useSpan-test-exception"); + Assert.ok(previousSpan && testSpan, "Both spans should be created"); + + core.setActiveSpan(previousSpan!); + + const testFunction = () => { + throw new Error("Test exception"); + }; + + // Act & Assert + let thrownError: Error | null = null; + try { + useSpan(core, testSpan!, testFunction); + } catch (error) { + thrownError = error as Error; + } + + Assert.ok(thrownError, "Exception should be thrown"); + Assert.equal(thrownError!.message, "Test exception", "Exception message should be preserved"); + Assert.equal(core.getActiveSpan(), previousSpan, "Previous active span should be restored even after exception"); + } + }); + + this.testCase({ + name: "useSpan: should work with functions returning different types", + test: () => { + // Arrange + const core = this._setupCore(); + const testSpan = core.startSpan("useSpan-test-types"); + Assert.ok(testSpan, "Test span should be created"); + + // Test string return + const stringResult = useSpan(core, testSpan!, () => "hello world"); + Assert.equal(stringResult, "hello world", "String return should work"); + + // Test number return + const numberResult = useSpan(core, testSpan!, () => 123.45); + Assert.equal(numberResult, 123.45, "Number return should work"); + + // Test boolean return + const booleanResult = useSpan(core, testSpan!, () => true); + Assert.equal(booleanResult, true, "Boolean return should work"); + + // Test object return + const objectResult = useSpan(core, testSpan!, () => ({ key: "value" })); + Assert.ok(objectResult && objectResult.key === "value", "Object return should work"); + + // Test undefined return + const undefinedResult = useSpan(core, testSpan!, () => undefined); + Assert.equal(undefinedResult, undefined, "Undefined return should work"); + + // Test null return + const nullResult = useSpan(core, testSpan!, () => null); + Assert.equal(nullResult, null, "Null return should work"); + } + }); + + this.testCase({ + name: "useSpan: should work with async-like function patterns", + test: () => { + // Arrange + const core = this._setupCore(); + let initialActiveSpan = core.getActiveSpan(); + Assert.ok(!isNullOrUndefined(initialActiveSpan), "Initially, activeSpan should not be null with a trace provider"); + const testSpan = core.startSpan("useSpan-test-async-pattern"); + Assert.ok(testSpan, "Test span should be created"); + + let spanDuringExecution: IReadableSpan | null = null; + + // Simulate async-like pattern with callback + const asyncFunction = (scope: ISpanScope, callback: (result: string) => void) => { + spanDuringExecution = scope.span; + // Simulate some async work completing synchronously for this test + callback("async-result"); + return "function-result"; + }; + + let callbackResult = ""; + const callback = (result: string) => { + callbackResult = result; + }; + + // Act + const result = useSpan(core, testSpan!, asyncFunction, undefined, callback); + + // Assert + Assert.equal(result, "function-result", "useSpan should return main function result"); + Assert.equal(callbackResult, "async-result", "Callback should be executed"); + Assert.equal(spanDuringExecution, testSpan, "Active span should be available during execution"); + Assert.equal(core.getActiveSpan(), initialActiveSpan, "Active span should be restored after completion"); + } + }); + + this.testCase({ + name: "useSpan: should work when no previous active span exists", + test: () => { + // Arrange + const core = this._setupCore(); + let initialActiveSpan = core.getActiveSpan(); + const testSpan = core.startSpan("useSpan-test-no-previous"); + Assert.ok(testSpan, "Test span should be created"); + Assert.ok(!isNullOrUndefined(initialActiveSpan), "With a traceprovider, activeSpan should not be null"); + + let capturedActiveSpan: IReadableSpan | null = null; + const testFunction = () => { + capturedActiveSpan = core.getActiveSpan(); + return "success"; + }; + + // Act + const result = useSpan(core, testSpan!, testFunction); + + // Assert + Assert.equal(result, "success", "Function should execute successfully"); + Assert.equal(capturedActiveSpan, testSpan, "Test span should be active during execution"); + Assert.equal(core.getActiveSpan(), initialActiveSpan, "No active span should be restored (was null)"); + } + }); + + this.testCase({ + name: "useSpan: should work with nested useSpan calls", + test: () => { + // Arrange + const core = this._setupCore(); + let initialActiveSpan = core.getActiveSpan(); + Assert.ok(!isNullOrUndefined(initialActiveSpan), "Initially, activeSpan should not be null with a trace provider"); + const outerSpan = core.startSpan("outer-span"); + const innerSpan = core.startSpan("inner-span"); + Assert.ok(outerSpan && innerSpan, "Both spans should be created"); + + const executionTrace: string[] = []; + + const innerFunction = (scope: ISpanScope) => { + const activeSpan = scope.span; + executionTrace.push(`inner: ${activeSpan ? (activeSpan as IReadableSpan).name : 'null'}`); + return "inner-result"; + }; + + const outerFunction = (scope: ISpanScope) => { + const activeSpanBefore = scope.span; + executionTrace.push(`outer-start: ${activeSpanBefore ? (activeSpanBefore as IReadableSpan).name : 'null'}`); + + const innerResult = useSpan(core, innerSpan!, innerFunction); + + const activeSpanAfter = core.getActiveSpan(); + executionTrace.push(`outer-end: ${activeSpanAfter ? (activeSpanAfter as IReadableSpan).name : 'null'}`); + + return `outer(${innerResult})`; + }; + + // Act + const result = useSpan(core, outerSpan!, outerFunction); + + // Assert + Assert.equal(result, "outer(inner-result)", "Nested useSpan should work correctly"); + Assert.equal(executionTrace.length, 3, "Should have captured 3 execution points"); + Assert.equal(executionTrace[0], "outer-start: outer-span", "Outer function should see outer span"); + Assert.equal(executionTrace[1], "inner: inner-span", "Inner function should see inner span"); + Assert.equal(executionTrace[2], "outer-end: outer-span", "Outer function should see outer span restored"); + Assert.equal(core.getActiveSpan?.(), initialActiveSpan, "The initial active span should be restored after nested execution"); + } + }); + + this.testCase({ + name: "useSpan: should handle span operations within useSpan context", + test: () => { + // Arrange + const core = this._setupCore(); + let initialActiveSpan = core.getActiveSpan(); + Assert.ok(!isNullOrUndefined(initialActiveSpan), "Initially, activeSpan should not be null with a trace provider"); + const testSpan = core.startSpan("useSpan-test-operations"); + Assert.ok(testSpan, "Test span should be created"); + + const testFunction = (scope: ISpanScope) => { + const activeSpan = scope.span; + Assert.ok(activeSpan, "Should have active span in function"); + + // Perform span operations + activeSpan.setAttribute("operation.name", "test-operation"); + activeSpan.setAttribute("operation.step", 1); + + // Create child span + const childSpan = core.startSpan("child-operation"); + Assert.ok(childSpan, "Child span should be created"); + + childSpan?.setAttribute("child.attribute", "child-value"); + childSpan?.end(); + + return "operations-completed"; + }; + + // Act + const result = useSpan(core, testSpan!, testFunction); + + // Assert + Assert.equal(result, "operations-completed", "Function should complete successfully"); + Assert.equal(core.getActiveSpan(), initialActiveSpan, "Active span should be restored"); + + // Verify span operations were applied (span should still be valid) + const readableSpan = testSpan! as IReadableSpan; + Assert.ok(!readableSpan.ended, "Test span should not be ended"); + Assert.ok(testSpan!.isRecording(), "Test span should still be recording"); + } + }); + + this.testCase({ + name: "useSpan: should work with core that has no trace provider", + test: () => { + // Arrange + const core = new AppInsightsCore(); + + // Create a simple test channel + const testChannel = { + identifier: "TestChannel", + priority: 1001, + initialize: () => {}, + processTelemetry: () => {}, + teardown: () => {}, + isInitialized: () => true + }; + + core.initialize({ instrumentationKey: "test-key" }, [testChannel]); // Initialize with channel but no trace provider + + // Create a mock span (this would need to come from somewhere else since no provider) + const spanCtx: IOTelSpanCtx = { + api: this._mockApi, + spanContext: this._mockSpanContext, + onEnd: (span) => this._onEndCalls.push(span) + }; + const mockSpan = createSpan(spanCtx, "mock-span", eOTelSpanKind.CLIENT); + + let functionExecuted = false; + const testFunction = () => { + functionExecuted = true; + return "no-provider-result"; + }; + + // Act + const result = useSpan(core, mockSpan, testFunction); + + // Assert + Assert.equal(result, "no-provider-result", "Function should execute even without trace provider"); + Assert.ok(functionExecuted, "Function should have been executed"); + + // Cleanup + core.unload(false); + } + }); + + this.testCase({ + name: "useSpan: performance test - should not add significant overhead", + test: () => { + // Arrange + const core = this._setupCore(); + const testSpan = core.startSpan("useSpan-performance-test"); + Assert.ok(testSpan, "Test span should be created"); + + const iterations = 10000; + let computeResult = 0; + + const computeFunction = (_scope: ISpanScope, base: number, multiplier: number) => { + // Simple computation to measure overhead + return base * multiplier + Math.sqrt(base); + }; + + let maxOverhead: number = 100; + + // Perform multiple runs to get a stable measurement + for (let lp = 0; lp < 10; lp++) { + // Measure time without useSpan + const startWithout = perfNow(); + for (let i = 0; i < iterations; i++) { + computeResult += computeFunction(null as any as ISpanScope, i, 2); + } + const timeWithout = perfNow() - startWithout; + + // Reset result + computeResult = 0; + + // Measure time with useSpan + const startWith = perfNow(); + for (let i = 0; i < iterations; i++) { + computeResult += useSpan(core, testSpan!, computeFunction, undefined, i, 2); + } + + // Results should be the same + Assert.ok(computeResult > 0, "Computations should have produced results"); + + const timeWith = perfNow() - startWith; + + const overhead = timeWith / (timeWithout || 1); + + if (lp === 0) { + maxOverhead = overhead; + } + maxOverhead = mathMin(maxOverhead, overhead); + } + + // Assert reasonable performance characteristics + // useSpan should not add more than 10x overhead (very generous threshold) + Assert.ok(maxOverhead < 10, `useSpan overhead should be reasonable: ${maxOverhead.toFixed(2)}x`); + + } + }); + } +} \ No newline at end of file diff --git a/shared/AppInsightsCore/Tests/Unit/src/trace/traceState.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/trace/traceState.Tests.ts new file mode 100644 index 000000000..86c6ad9a6 --- /dev/null +++ b/shared/AppInsightsCore/Tests/Unit/src/trace/traceState.Tests.ts @@ -0,0 +1,448 @@ +import { Assert, AITestClass } from "@microsoft/ai-test-framework"; +import { createOTelTraceState } from "../../../../src/otel/api/trace/traceState"; +import { strRepeat } from "@nevware21/ts-utils"; + +export class OTelTraceApiTests extends AITestClass { + + public testInitialize() { + super.testInitialize(); + } + + public testCleanup() { + super.testCleanup(); + } + + public registerTests() { + + this.testCase({ + name: "TraceState: serialize", + test: () => { + const traceState = createOTelTraceState("a=1,b=2,c=3"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle empty string", + test: () => { + const traceState = createOTelTraceState(""); + Assert.equal(traceState.serialize(), ""); + } + }); + + this.testCase({ + name: "TraceState: handle null", + test: () => { + const traceState = createOTelTraceState(null as any); + Assert.equal(traceState.serialize(), ""); + } + }); + + this.testCase({ + name: "TraceState: handle undefined", + test: () => { + const traceState = createOTelTraceState(undefined); + Assert.equal(traceState.serialize(), ""); + } + }); + + this.testCase({ + name: "TraceState: handle invalid input", + test: () => { + const traceState = createOTelTraceState({} as any); + Assert.equal(traceState.serialize(), ""); + } + }); + + this.testCase({ + name: "TraceState: new / updated keys are added to the front", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set("d", "4"); + Assert.equal(traceState.serialize(), "d=4,a=1,b=2,c=3"); + + traceState = traceState.set("a", "5"); + Assert.equal(traceState.serialize(), "a=5,d=4,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: must create new instances for each state", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + let traceState2 = createOTelTraceState(traceState.serialize()); + traceState2 = traceState2.set("d", "4"); + + Assert.notEqual(traceState, traceState2); + Assert.notDeepEqual(traceState, traceState2); + + Assert.equal(traceState.serialize(), "a=1,b=2,c=3", "Actual: " + traceState.serialize() + " expected a=1,b=2,c=3"); + Assert.equal(traceState2.serialize(), "d=4,a=1,b=2,c=3", "Actual: " + traceState2.serialize() + " expected d=4,a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: unset", + test: () => { + let traceState = createOTelTraceState("a=4,b=5,c=6"); + traceState = traceState.unset("b"); + Assert.equal(traceState.serialize(), "a=4,c=6"); + + traceState = traceState.unset("a"); + Assert.equal(traceState.serialize(), "c=6"); + } + }); + + this.testCase({ + name: "TraceState: get", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), "2"); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(traceState.get("d"), undefined); + } + }); + + this.testCase({ + name: "TraceState: serialize with spaces", + test: () => { + const traceState = createOTelTraceState("a=1, b=2, c=3"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: serialize with tabs", + test: () => { + const traceState = createOTelTraceState("a=1\t,b=2\t,c=3"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: serialize with newlines", + test: () => { + const traceState = createOTelTraceState("a=1\n,b=2\n,c=3"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: serialize with multiple commas", + test: () => { + const traceState = createOTelTraceState("a=1,,b=2,,c=3"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: serialize with multiple equals", + test: () => { + const traceState = createOTelTraceState("a==1,b==2,c==3"); + Assert.equal(traceState.serialize(), ""); + } + }); + + this.testCase({ + name: "TraceState: serialize with multiple spaces", + test: () => { + const traceState = createOTelTraceState("a=1 , b=2 , c=3"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: serialize with multiple tabs", + test: () => { + const traceState = createOTelTraceState("a=1\t\t,b=2\t\t,c=3"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: serialize with multiple newlines", + test: () => { + const traceState = createOTelTraceState("a=1\n\n,b=2\n\n,c=3"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: serialize with multiple commas and spaces", + test: () => { + const traceState = createOTelTraceState("a=1, ,b=2, ,c=3"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle unsetting non-existent keys", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.unset("d"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting empty key", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set("", "4"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting empty value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set("d", ""); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting empty string value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set("d", " "); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting empty key and value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set("", ""); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting null key", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set(null as any, "4"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting null value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set("d", null as any); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting null string value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set("d", "null"); + Assert.equal(traceState.serialize(), "d=null,a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting null key and value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set(null as any, null as any); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting undefined key", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set(undefined as any, "4"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + this.testCase({ + name: "TraceState: handle setting undefined value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set("d", undefined as any); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting undefined key and value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set(undefined as any, undefined as any); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting invalid key", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set({} as any, "4"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting invalid value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set("d", {} as any); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting invalid key and value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set({} as any, {} as any); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting invalid key and valid value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set({} as any, "4"); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle setting valid key and invalid value", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set("d", {} as any); + Assert.equal(traceState.serialize(), "a=1,b=2,c=3"); + } + }); + + this.testCase({ + name: "TraceState: handle dropping states when the max number of members limit is reached", + test: () => { + let traceState = createOTelTraceState("a=1,b=2,c=3"); + traceState = traceState.set("d", "4"); + traceState = traceState.set("e", "5"); + traceState = traceState.set("f", "6"); + traceState = traceState.set("g", "7"); + traceState = traceState.set("h", "8"); + traceState = traceState.set("i", "9"); + traceState = traceState.set("j", "10"); + traceState = traceState.set("k", "11"); + traceState = traceState.set("l", "12"); + traceState = traceState.set("m", "13"); + traceState = traceState.set("n", "14"); + traceState = traceState.set("o", "15"); + traceState = traceState.set("p", "16"); + traceState = traceState.set("q", "17"); + traceState = traceState.set("r", "18"); + traceState = traceState.set("s", "19"); + traceState = traceState.set("t", "20"); + traceState = traceState.set("u", "21"); + traceState = traceState.set("v", "22"); + traceState = traceState.set("w", "23"); + traceState = traceState.set("x", "24"); + traceState = traceState.set("y", "25"); + traceState = traceState.set("z", "26"); + traceState = traceState.set("aa", "27"); + traceState = traceState.set("ab", "28"); + traceState = traceState.set("ac", "29"); + traceState = traceState.set("ad", "30"); + traceState = traceState.set("ae", "31"); + traceState = traceState.set("af", "32"); + Assert.equal(traceState.serialize(), "af=32,ae=31,ad=30,ac=29,ab=28,aa=27,z=26,y=25,x=24,w=23,v=22,u=21,t=20,s=19,r=18,q=17,p=16,o=15,n=14,m=13,l=12,k=11,j=10,i=9,h=8,g=7,f=6,e=5,d=4,a=1,b=2,c=3"); + traceState = traceState.set("ag", "33"); + Assert.equal(traceState.serialize(), "ag=33,af=32,ae=31,ad=30,ac=29,ab=28,aa=27,z=26,y=25,x=24,w=23,v=22,u=21,t=20,s=19,r=18,q=17,p=16,o=15,n=14,m=13,l=12,k=11,j=10,i=9,h=8,g=7,f=6,e=5,d=4,a=1,b=2"); + } + }); + + this.testCase({ + name: "TraceState: drop states when the items are too long", + test: () => { + const traceState = createOTelTraceState("a=" + strRepeat("b", 512)); + Assert.equal(traceState.get("a"), undefined); + Assert.equal(traceState.serialize(), ""); + } + }); + + this.testCase({ + name: "TraceState: drop items that are invalid", + test: () => { + const traceState = createOTelTraceState("a=1,b,c=3"); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), undefined); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(traceState.serialize(), "a=1,c=3"); + } + }); + + this.testCase({ + name: "TraceState: drop items that are invalid with spaces", + test: () => { + const traceState = createOTelTraceState("a=1, b, c=3"); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), undefined); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(traceState.serialize(), "a=1,c=3"); + } + }); + + this.testCase({ + name: "TraceState: drop items that are invalid with tabs", + test: () => { + const traceState = createOTelTraceState("a=1\t,b\t,c=3"); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), undefined); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(traceState.serialize(), "a=1,c=3"); + } + }); + + this.testCase({ + name: "TraceState: drop items that have a single value with an '=' sign", + test: () => { + const traceState = createOTelTraceState("a=1,b=2=,c=3,d="); + Assert.equal(traceState.get("a"), "1"); + Assert.equal(traceState.get("b"), undefined); + Assert.equal(traceState.get("c"), "3"); + Assert.equal(traceState.get("d"), undefined); + Assert.equal(traceState.serialize(), "a=1,c=3"); + } + }); + + this.testCase({ + name: "TraceState: must handle valid state key ranges", + test: () => { + const traceState = createOTelTraceState("a-b=1,c/d=2,e_f=3,g*h=4"); + Assert.equal(traceState.get("a-b"), "1"); + Assert.equal(traceState.get("c/d"), "2"); + Assert.equal(traceState.get("e_f"), "3"); + Assert.equal(traceState.get("g*h"), "4"); + Assert.equal(traceState.serialize(), "a-b=1,c/d=2,e_f=3,g*h=4"); + } + }); + + this.testCase({ + name: "TraceState: handle values with embedded spaces", + test: () => { + const traceState = createOTelTraceState("a=1 b,c=2 d,e=3 f"); + Assert.equal(traceState.get("a"), "1 b"); + Assert.equal(traceState.get("c"), "2 d"); + Assert.equal(traceState.get("e"), "3 f"); + Assert.equal(traceState.serialize(), "a=1 b,c=2 d,e=3 f"); + } + }); + + } +} \ No newline at end of file diff --git a/shared/AppInsightsCore/Tests/Unit/src/trace/traceUtils.Tests.ts b/shared/AppInsightsCore/Tests/Unit/src/trace/traceUtils.Tests.ts new file mode 100644 index 000000000..4e53d2e4c --- /dev/null +++ b/shared/AppInsightsCore/Tests/Unit/src/trace/traceUtils.Tests.ts @@ -0,0 +1,1355 @@ +import { AITestClass, Assert } from "@microsoft/ai-test-framework"; +import { isSpanContextValid, wrapSpanContext, createNonRecordingSpan, isReadableSpan, useSpan, withSpan } from "../../../../src/otel/api/trace/utils"; +import { createTraceProvider } from "../../../../src/otel/api/trace/traceProvider"; +import { IDistributedTraceContext } from "../../../../src/interfaces/ai/IDistributedTraceContext"; +import { createPromise, createRejectedPromise, doAwait } from "@nevware21/ts-async"; +import { createCachedValue, isNullOrUndefined } from "@nevware21/ts-utils"; +import { IOTelApi } from "../../../../src/interfaces/otel/IOTelApi"; +import { createOTelApi } from "../../../../src/otel/api/OTelApi"; +import { eOTelSpanKind } from "../../../../src/enums/otel/OTelSpanKind"; +import { IOTelSpanCtx } from "../../../../src/interfaces/otel/trace/IOTelSpanCtx"; +import { createSpan } from "../../../../src/otel/api/trace/span"; +import { createW3cTraceState } from "../../../../src/telemetry/W3cTraceState"; +import { createDistributedTraceContext } from "../../../../src/core/TelemetryHelpers"; +import { IAppInsightsCore } from "../../../../src/interfaces/ai/IAppInsightsCore"; +import { AppInsightsCore } from "../../../../src/index"; +import { IChannelControls } from "../../../../src/interfaces/ai/IChannelControls"; + +function _createDistributedContext(traceId: string, spanId: string, traceFlags: number, traceState?: string): IDistributedTraceContext { + const theContext: IDistributedTraceContext = { + traceId: traceId, + spanId: spanId, + traceFlags: traceFlags, + getName: function (): string { + throw new Error("Function not implemented."); + }, + setName: function (pageName: string): void { + throw new Error("Function not implemented."); + }, + getTraceId: function (): string { + throw new Error("Function not implemented."); + }, + setTraceId: function (newValue: string): void { + throw new Error("Function not implemented."); + }, + getSpanId: function (): string { + throw new Error("Function not implemented."); + }, + setSpanId: function (newValue: string): void { + throw new Error("Function not implemented."); + }, + getTraceFlags: function (): number | undefined { + throw new Error("Function not implemented."); + }, + setTraceFlags: function (newValue?: number): void { + throw new Error("Function not implemented."); + }, + pageName: "", + isRemote: false, + traceState: traceState ? createW3cTraceState(traceState) : createW3cTraceState() + }; + + return theContext; +} + +export class TraceUtilsTests extends AITestClass { + private _core: IAppInsightsCore = null as any; + private _otelApi!: IOTelApi; + private _validSpanContext!: IDistributedTraceContext; + + public testInitialize() { + // Create a minimal mock channel to satisfy core initialization requirements + const mockChannel: IChannelControls = { + pause: () => {}, + resume: () => {}, + flush: () => {}, + teardown: () => {}, + processTelemetry: () => {}, + initialize: () => {}, + identifier: "mockChannel", + priority: 1001 + } as any; + + this._core = new AppInsightsCore(); + this._core.initialize({ + instrumentationKey: "00000000-0000-0000-0000-000000000000", + disableInstrumentationKeyValidation: true, + traceCfg: {}, + errorHandlers: { + attribError: (message: string, key: string, value: any) => { + console.error(message); + }, + spanError: (message: string, spanName: string) => { + console.error(message); + }, + debug: (message: string) => { + console.error(message); + }, + warn: (message: string) => { + console.error(message); + }, + error: (message: string) => { + console.error(message); + }, + notImplemented: (message: string) => { + console.error(message); + } + } + }, [mockChannel]); + + this._otelApi = createOTelApi({ host: this._core }); + + // Set up a simple trace provider for the core + this._setupTraceProvider(); + + // Valid span context with proper IDs + this._validSpanContext = _createDistributedContext("12345678901234567890123456789012", "1234567890123456", 1); + } + + private _setupTraceProvider(): void { + // Using an cached value to wrap the provider, so it is created immediately + const provider = createCachedValue(createTraceProvider( + this._core, + "test-provider", + this._otelApi + )); + + this._core.setTraceProvider(provider); + } + + public testCleanup() { + if (this._core) { + this._core.unload(false); + this._core = null as any; + } + this._otelApi = null as any; + this._validSpanContext = null as any; + } + + public registerTests() { + this.addIsSpanContextValidTests(); + this.addWrapSpanContextTests(); + this.addCreateNonRecordingSpanTests(); + this.addIsReadableSpanTests(); + this.addWithSpanTests(); + this.addUseSpanTests(); + } + + private addIsSpanContextValidTests(): void { + this.testCase({ + name: "isSpanContextValid: should return true for valid span context", + test: () => { + // Act + const result = isSpanContextValid(this._validSpanContext); + + // Assert + Assert.ok(result, "Should return true for valid span context"); + } + }); + + this.testCase({ + name: "isSpanContextValid: should return false for invalid trace ID", + test: () => { + // Arrange + const invalidContext: IDistributedTraceContext = _createDistributedContext("invalid-trace-id", "1234567890123456", 1); + + // Act + const result = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!result, "Should return false for invalid trace ID"); + } + }); + + this.testCase({ + name: "isSpanContextValid: should return false for invalid span ID", + test: () => { + // Arrange + const invalidContext: IDistributedTraceContext = _createDistributedContext("12345678901234567890123456789012", "bad", 1); + // Act + const result = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!result, "Should return false for invalid span ID"); + } + }); + + this.testCase({ + name: "isSpanContextValid: should return false for null context", + test: () => { + // Act + const result = isSpanContextValid(null as any); + + // Assert + Assert.ok(!result, "Should return false for null context"); + } + }); + + this.testCase({ + name: "isSpanContextValid: should return false for undefined context", + test: () => { + // Act + const result = isSpanContextValid(undefined as any); + + // Assert + Assert.ok(!result, "Should return false for undefined context"); + } + }); + + this.testCase({ + name: "isSpanContextValid: should return false for empty trace ID", + test: () => { + // Arrange + const invalidContext: IDistributedTraceContext = _createDistributedContext("", "1234567890123456", 1); + // Act + const result = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!result, "Should return false for empty trace ID"); + } + }); + + this.testCase({ + name: "isSpanContextValid: should return false for empty span ID", + test: () => { + // Arrange + const invalidContext: IDistributedTraceContext = _createDistributedContext("12345678901234567890123456789012", "", 1); + // Act + const result = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!result, "Should return false for empty span ID"); + } + }); + + this.testCase({ + name: "isSpanContextValid: should return false for all-zero trace ID", + test: () => { + // Arrange + const invalidContext: IDistributedTraceContext = _createDistributedContext("00000000000000000000000000000000", "1234567890123456", 1); + + // Act + const result = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!result, "Should return false for all-zero trace ID"); + } + }); + + this.testCase({ + name: "isSpanContextValid: should return false for all-zero span ID", + test: () => { + // Arrange + const invalidContext: IDistributedTraceContext = _createDistributedContext("12345678901234567890123456789012", "0000000000000000", 1); + // Act + const result = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!result, "Should return false for all-zero span ID"); + } + }); + + this.testCase({ + name: "isSpanContextValid: should return false for trace ID with wrong length", + test: () => { + // Arrange + const invalidContext: IDistributedTraceContext = _createDistributedContext("123456789012", "1234567890123456", 1); + + // Act + const result = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!result, "Should return false for trace ID with wrong length"); + } + }); + + this.testCase({ + name: "isSpanContextValid: should return false for span ID with wrong length", + test: () => { + // Arrange + const invalidContext: IDistributedTraceContext = _createDistributedContext("12345678901234567890123456789012", "1234", 1); + // Act + const result = isSpanContextValid(invalidContext); + + // Assert + Assert.ok(!result, "Should return false for span ID with wrong length"); + } + }); + } + + private addWrapSpanContextTests(): void { + this.testCase({ + name: "wrapSpanContext: should create non-recording span from valid context", + test: () => { + // Act + const wrappedSpan = wrapSpanContext(this._otelApi, this._validSpanContext); + + // Assert + Assert.ok(wrappedSpan, "Should create a span"); + Assert.ok(!wrappedSpan.isRecording(), "Wrapped span should not be recording"); + Assert.equal(wrappedSpan.spanContext().traceId, this._validSpanContext.traceId, "Trace ID should match"); + Assert.equal(wrappedSpan.spanContext().spanId, this._validSpanContext.spanId, "Span ID should match"); + } + }); + + this.testCase({ + name: "wrapSpanContext: should include span ID in wrapped span name", + test: () => { + // Act + const wrappedSpan = wrapSpanContext(this._otelApi, this._validSpanContext); + + // Assert + Assert.ok(wrappedSpan.name.includes(this._validSpanContext.spanId), + "Span name should include original span ID"); + Assert.ok(wrappedSpan.name.includes("wrapped"), + "Span name should indicate it's a wrapped span"); + } + }); + + this.testCase({ + name: "wrapSpanContext: should preserve trace flags", + test: () => { + // Arrange + const contextWithFlags: IDistributedTraceContext = _createDistributedContext("12345678901234567890123456789012", "1234567890123456", 1); + + // Act + const wrappedSpan = wrapSpanContext(this._otelApi, contextWithFlags); + + // Assert + Assert.equal(wrappedSpan.spanContext().traceFlags, contextWithFlags.traceFlags, + "Trace flags should be preserved"); + } + }); + + this.testCase({ + name: "wrapSpanContext: should handle context with tracestate", + test: () => { + // Arrange + let contextWithState: IDistributedTraceContext = createDistributedTraceContext(); + contextWithState.traceId = "12345678901234567890123456789012"; + contextWithState.spanId = "1234567890123456"; + contextWithState.traceFlags = 1; + contextWithState.traceState.set("vendor", "value"); + + // Act + const wrappedSpan = wrapSpanContext(this._otelApi, contextWithState); + + // Assert + Assert.ok(wrappedSpan, "Should create span with tracestate"); + Assert.equal(wrappedSpan.spanContext().traceState, contextWithState.traceState, + "Trace state should be preserved if present"); + } + }); + + this.testCase({ + name: "wrapSpanContext: wrapped span should not be ended", + test: () => { + // Act + const wrappedSpan = wrapSpanContext(this._otelApi, this._validSpanContext); + + // Assert + Assert.ok(!wrappedSpan.ended, "Wrapped span should not be ended initially"); + } + }); + + this.testCase({ + name: "wrapSpanContext: wrapped span should be internal kind", + test: () => { + // Act + const wrappedSpan = wrapSpanContext(this._otelApi, this._validSpanContext); + + // Assert + Assert.equal(wrappedSpan.kind, eOTelSpanKind.INTERNAL, + "Wrapped span should have INTERNAL kind"); + } + }); + } + + private addCreateNonRecordingSpanTests(): void { + this.testCase({ + name: "createNonRecordingSpan: should create non-recording span", + test: () => { + // Arrange + const spanName = "test-non-recording"; + + // Act + const span = createNonRecordingSpan(this._otelApi, spanName, this._validSpanContext); + + // Assert + Assert.ok(span, "Should create a span"); + Assert.ok(!span.isRecording(), "Span should not be recording"); + Assert.equal(span.name, spanName, "Span name should match"); + } + }); + + this.testCase({ + name: "createNonRecordingSpan: should use provided span context", + test: () => { + // Arrange + const spanName = "context-test"; + + // Act + const span = createNonRecordingSpan(this._otelApi, spanName, this._validSpanContext); + + // Assert + Assert.equal(span.spanContext().traceId, this._validSpanContext.traceId, + "Trace ID should match provided context"); + Assert.equal(span.spanContext().spanId, this._validSpanContext.spanId, + "Span ID should match provided context"); + Assert.equal(span.spanContext().traceFlags, this._validSpanContext.traceFlags, + "Trace flags should match provided context"); + } + }); + + this.testCase({ + name: "createNonRecordingSpan: should create span with INTERNAL kind", + test: () => { + // Act + const span = createNonRecordingSpan(this._otelApi, "test", this._validSpanContext); + + // Assert + Assert.equal(span.kind, eOTelSpanKind.INTERNAL, + "Non-recording span should have INTERNAL kind"); + } + }); + + this.testCase({ + name: "createRecordingSpan: recording span with onEnd should still trigger callback", + test: () => { + // Arrange + let onEndCalled = false; + + // Create a non-recording span directly with createSpan to test onEnd callback behavior + const spanCtx: IOTelSpanCtx = { + api: this._otelApi, + spanContext: this._validSpanContext, + onEnd: () => { + onEndCalled = true; + } + }; + + // Act - Create span directly with our custom context that includes onEnd callback + const span = createSpan(spanCtx, "test-recording-with-callback", eOTelSpanKind.INTERNAL); + Assert.ok(span.isRecording(), "Span should be recording"); + span.end(); + + // Assert + // onEnd callbacks are called regardless of recording state when the callback is provided + Assert.ok(onEndCalled, "onEnd callback should be called even for non-recording spans when callback is registered"); + } + }); + + this.testCase({ + name: "createNonRecordingSpan: non-recording span with onEnd should still trigger callback", + test: () => { + // Arrange + let onEndCalled = false; + + // Create a non-recording span directly with createSpan to test onEnd callback behavior + const spanCtx: IOTelSpanCtx = { + api: this._otelApi, + spanContext: this._validSpanContext, + isRecording: false, // Non-recording span + onEnd: () => { + onEndCalled = true; + } + }; + + // Act - Create span directly with our custom context that includes onEnd callback + const span = createSpan(spanCtx, "test-non-recording-with-callback", eOTelSpanKind.INTERNAL); + Assert.ok(!span.isRecording(), "Span should not be recording"); + span.end(); + + // Assert + // onEnd callbacks are called regardless of recording state when the callback is provided + // Allows for post-end processing even for non-recording spans, including tracking the number of non-recording spans ended + Assert.ok(onEndCalled, "onEnd callback should be called even for non-recording spans when callback is registered"); + } + }); + + this.testCase({ + name: "createNonRecordingSpan: utility function creates span without onEnd callback", + test: () => { + // Arrange & Act + // createNonRecordingSpan is a utility that doesn't accept an onEnd callback parameter + const span = createNonRecordingSpan(this._otelApi, "test-non-recording", this._validSpanContext); + + // Assert + Assert.ok(!span.isRecording(), "Span should not be recording"); + // This just verifies the utility function works as expected - no onEnd callback to test + + span.end(); + + // Validate that calling after end does not cause issues + Assert.ok(!span.isRecording(), "Span should not be recording"); + } + }); + + this.testCase({ + name: "createNonRecordingSpan: should accept custom span names", + test: () => { + // Arrange + const customNames = [ + "operation-1", + "http-request", + "database-query", + "cache-lookup", + "" + ]; + + // Act & Assert + customNames.forEach(name => { + const span = createNonRecordingSpan(this._otelApi, name, this._validSpanContext); + Assert.equal(span.name, name, `Span name should be '${name}'`); + Assert.ok(!span.isRecording(), "All non-recording spans should not be recording"); + }); + } + }); + + this.testCase({ + name: "createNonRecordingSpan: should preserve tracestate from context", + test: () => { + // Arrange + const contextWithState: IDistributedTraceContext = _createDistributedContext("12345678901234567890123456789012", "1234567890123456", 1, "vendor1=value1,vendor2=value2"); + + // Act + const span = createNonRecordingSpan(this._otelApi, "test", contextWithState); + + // Assert + Assert.equal(span.spanContext().traceState, contextWithState.traceState, + "Trace state should be preserved"); + } + }); + } + + private addIsReadableSpanTests(): void { + this.testCase({ + name: "isReadableSpan: should return true for valid span", + test: () => { + // Arrange - create a real span + const span = createNonRecordingSpan(this._otelApi, "test", this._validSpanContext); + + // Act + const result = isReadableSpan(span); + + // Assert + Assert.ok(result, "Should return true for a valid readable span"); + } + }); + + this.testCase({ + name: "isReadableSpan: should return false for null", + test: () => { + // Act + const result = isReadableSpan(null); + + // Assert + Assert.ok(!result, "Should return false for null"); + } + }); + + this.testCase({ + name: "isReadableSpan: should return false for undefined", + test: () => { + // Act + const result = isReadableSpan(undefined); + + // Assert + Assert.ok(!result, "Should return false for undefined"); + } + }); + + this.testCase({ + name: "isReadableSpan: should return false for empty object", + test: () => { + // Act + const result = isReadableSpan({}); + + // Assert + Assert.ok(!result, "Should return false for empty object"); + } + }); + + this.testCase({ + name: "isReadableSpan: should return false for object with only some properties", + test: () => { + // Arrange + const partialSpan = { + name: "test", + kind: eOTelSpanKind.CLIENT + }; + + // Act + const result = isReadableSpan(partialSpan); + + // Assert + Assert.ok(!result, "Should return false for object missing required properties"); + } + }); + + this.testCase({ + name: "isReadableSpan: should return false for object missing spanContext method", + test: () => { + // Arrange + const invalidSpan = { + name: "test", + kind: eOTelSpanKind.CLIENT, + duration: [0, 0], + ended: false, + startTime: [0, 0], + endTime: [0, 0], + attributes: {}, + links: [], + events: [], + status: { code: 0 }, + resource: {}, + instrumentationScope: {}, + droppedAttributesCount: 0, + isRecording: () => false, + setStatus: () => {}, + updateName: () => {}, + setAttribute: () => {}, + setAttributes: () => {}, + end: () => {}, + recordException: () => {} + // Missing spanContext method + }; + + // Act + const result = isReadableSpan(invalidSpan); + + // Assert + Assert.ok(!result, "Should return false when spanContext method is missing"); + } + }); + + this.testCase({ + name: "isReadableSpan: should return false for object with non-function methods", + test: () => { + // Arrange + const invalidSpan = { + name: "test", + kind: eOTelSpanKind.CLIENT, + spanContext: "not a function", + duration: [0, 0], + ended: false, + startTime: [0, 0], + endTime: [0, 0], + attributes: {}, + links: [], + events: [], + status: { code: 0 }, + resource: {}, + instrumentationScope: {}, + droppedAttributesCount: 0, + isRecording: () => false, + setStatus: () => {}, + updateName: () => {}, + setAttribute: () => {}, + setAttributes: () => {}, + end: () => {}, + recordException: () => {} + }; + + // Act + const result = isReadableSpan(invalidSpan); + + // Assert + Assert.ok(!result, "Should return false when required methods are not functions"); + } + }); + + this.testCase({ + name: "isReadableSpan: should return false for primitive values", + test: () => { + // Act & Assert + Assert.ok(!isReadableSpan("string"), "Should return false for string"); + Assert.ok(!isReadableSpan(123), "Should return false for number"); + Assert.ok(!isReadableSpan(true), "Should return false for boolean"); + } + }); + + this.testCase({ + name: "isReadableSpan: should return false for array", + test: () => { + // Act + const result = isReadableSpan([]); + + // Assert + Assert.ok(!result, "Should return false for array"); + } + }); + + this.testCase({ + name: "isReadableSpan: should validate all required properties exist", + test: () => { + // Arrange - create a valid span to ensure our check is comprehensive + const span = createNonRecordingSpan(this._otelApi, "validation-test", this._validSpanContext); + + // Act - verify the span has all required properties + const hasName = "name" in span; + const hasKind = "kind" in span; + const hasSpanContext = typeof span.spanContext === "function"; + const hasDuration = "duration" in span; + const hasEnded = "ended" in span; + const hasStartTime = "startTime" in span; + const hasEndTime = "endTime" in span; + const hasAttributes = "attributes" in span; + const hasLinks = "links" in span; + const hasEvents = "events" in span; + const hasStatus = "status" in span; + // const hasResource = "resource" in span; + // const hasInstrumentationScope = "instrumentationScope" in span; + const hasDroppedAttributesCount = "droppedAttributesCount" in span; + const hasIsRecording = typeof span.isRecording === "function"; + const hasSetStatus = typeof span.setStatus === "function"; + const hasUpdateName = typeof span.updateName === "function"; + const hasSetAttribute = typeof span.setAttribute === "function"; + const hasSetAttributes = typeof span.setAttributes === "function"; + const hasEnd = typeof span.end === "function"; + const hasRecordException = typeof span.recordException === "function"; + + // Assert + Assert.ok(hasName, "Should have name property"); + Assert.ok(hasKind, "Should have kind property"); + Assert.ok(hasSpanContext, "Should have spanContext method"); + Assert.ok(hasDuration, "Should have duration property"); + Assert.ok(hasEnded, "Should have ended property"); + Assert.ok(hasStartTime, "Should have startTime property"); + Assert.ok(hasEndTime, "Should have endTime property"); + Assert.ok(hasAttributes, "Should have attributes property"); + Assert.ok(hasLinks, "Should have links property"); + Assert.ok(hasEvents, "Should have events property"); + Assert.ok(hasStatus, "Should have status property"); + // Assert.ok(hasResource, "Should have resource property"); + // Assert.ok(hasInstrumentationScope, "Should have instrumentationScope property"); + Assert.ok(hasDroppedAttributesCount, "Should have droppedAttributesCount property"); + Assert.ok(hasIsRecording, "Should have isRecording method"); + Assert.ok(hasSetStatus, "Should have setStatus method"); + Assert.ok(hasUpdateName, "Should have updateName method"); + Assert.ok(hasSetAttribute, "Should have setAttribute method"); + Assert.ok(hasSetAttributes, "Should have setAttributes method"); + Assert.ok(hasEnd, "Should have end method"); + Assert.ok(hasRecordException, "Should have recordException method"); + + // Final validation + Assert.ok(isReadableSpan(span), "isReadableSpan should return true for complete span"); + } + }); + + this.testCase({ + name: "isReadableSpan: should work with recording spans", + test: () => { + // Arrange - this would be a recording span in real usage + const recordingSpan = createNonRecordingSpan(this._otelApi, "recording-test", this._validSpanContext); + + // Act + const result = isReadableSpan(recordingSpan); + + // Assert + Assert.ok(result, "Should return true for both recording and non-recording spans"); + } + }); + } + + private addWithSpanTests(): void { + this.testCase({ + name: "withSpan: should execute synchronous callback and return value", + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-sync", this._validSpanContext); + const expectedValue = "sync-result"; + + // Act + const result = withSpan(this._core, span, function() { + return expectedValue; + }); + + // Assert + Assert.equal(result, expectedValue, "Should return the callback result"); + } + }); + + this.testCase({ + name: "withSpan: should set and restore active span for synchronous callback", + test: () => { + // Arrange + const _self = this; + const span = createNonRecordingSpan(this._otelApi, "test-active-span", this._validSpanContext); + let activeSpanDuringCallback: any = null; + let coreActiveSpanDuringCallback: any = null; + const originalActiveSpan = this._core.getActiveSpan(); + + Assert.ok(!isNullOrUndefined(originalActiveSpan), "Original active span should not be null or undefined"); + + // Act + withSpan(this._core, span, function() { + activeSpanDuringCallback = _self._otelApi.trace.getActiveSpan(); + coreActiveSpanDuringCallback = _self._core.getActiveSpan(); + + Assert.equal(coreActiveSpanDuringCallback, activeSpanDuringCallback, "Core active span and OTEL active span should match during callback"); + }); + + const activeSpanAfterCallback = this._otelApi.trace.getActiveSpan(); + const coreActiveSpanAfterCallback = this._core.getActiveSpan(); + + Assert.equal(coreActiveSpanAfterCallback, activeSpanAfterCallback, "Core active span and OTEL active span should match after callback"); + + // Assert + Assert.equal(activeSpanDuringCallback, span, "Active span should be set during callback"); + Assert.equal(activeSpanAfterCallback, originalActiveSpan, "Active span should be restored after callback"); + } + }); + + this.testCase({ + name: "withSpan: should handle promise that resolves", + test: () => { + // Arrange + const _self = this; + const span = createNonRecordingSpan(_self._otelApi, "test-promise-resolve", _self._validSpanContext); + const expectedValue = "resolved-value"; + let activeSpanDuringCallback: any = null; + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + + // Act + const promise = withSpan(_self._core, span, function() { + activeSpanDuringCallback = _self._otelApi.trace.getActiveSpan(); + return createPromise((resolve) => { + setTimeout(() => resolve(expectedValue), 10); + }); + }, _self); + + const activeSpanAfterCallback = this._otelApi.trace.getActiveSpan(); + + // Assert + Assert.equal(activeSpanDuringCallback, span, "Active span should be set during callback"); + + // Assert active span immediately after callback + Assert.equal(activeSpanAfterCallback, span, "Active span should still be set after callback until promise resolves"); + + return doAwait(promise, (result) => { + Assert.equal(result, expectedValue, "Should resolve with the expected value"); + const activeSpanAfterResolve = this._otelApi.trace.getActiveSpan(); + Assert.equal(activeSpanAfterResolve, originalActiveSpan, "Active span should be restored after promise resolves"); + }); + } + }); + + this.testCase({ + name: "withSpan: should handle promise that rejects", + test: () => { + // Arrange + const _self = this; + const span = createNonRecordingSpan(_self._otelApi, "test-promise-reject", _self._validSpanContext); + const expectedError = new Error("test-error"); + let activeSpanDuringCallback: any = null; + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + + // Act + const promise = withSpan(_self._core, span, function() { + activeSpanDuringCallback = _self._otelApi.trace.getActiveSpan(); + return createRejectedPromise(expectedError); + }, _self); + + // Assert + Assert.equal(activeSpanDuringCallback, span, "Active span should be set during callback"); + + return doAwait(promise, + () => { + Assert.ok(false, "Promise should have rejected"); + }, + (error) => { + Assert.equal(error, expectedError, "Should reject with the expected error"); + const activeSpanAfterReject = this._otelApi.trace.getActiveSpan(); + Assert.equal(activeSpanAfterReject, originalActiveSpan, "Active span should be restored after promise rejects"); + } + ); + } + }); + + this.testCase({ + name: "withSpan: should handle async/await pattern with resolved promise", + useFakeTimers: true, + test: () => { + // Arrange + const _self = this; + const span = createNonRecordingSpan(_self._otelApi, "test-async-await", _self._validSpanContext); + const expectedValue = 42; + let activeSpanDuringCallback: any = null; + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + + // Act + const promise = withSpan(_self._core, span, function() { + activeSpanDuringCallback = _self._otelApi.trace.getActiveSpan(); + return createPromise((resolve) => { + setTimeout(() => resolve(expectedValue), 100); + }); + }, _self); + + // Assert + Assert.equal(activeSpanDuringCallback, span, "Active span should be set during callback"); + + // Advance timers to trigger promise resolution + this.clock.tick(100); + + return doAwait(promise, (result) => { + Assert.equal(result, expectedValue, "Should resolve with the expected value"); + const activeSpanAfterResolve = this._otelApi.trace.getActiveSpan(); + Assert.equal(activeSpanAfterResolve, originalActiveSpan, "Active span should be restored after async operation"); + }); + } + }); + + this.testCase({ + name: "withSpan: should handle async/await pattern with rejected promise", + useFakeTimers: true, + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-async-reject", this._validSpanContext); + const expectedError = new Error("async-error"); + const originalActiveSpan = this._otelApi.trace.getActiveSpan(); + + // Act + const promise = withSpan(this._core, span, function() { + return createPromise((resolve, reject) => { + setTimeout(() => reject(expectedError), 50); + }); + }); + + // Advance timers + this.clock.tick(50); + + return doAwait(promise, + () => { + Assert.ok(false, "Promise should have rejected"); + }, + (error) => { + Assert.equal(error, expectedError, "Should reject with the expected error"); + const activeSpanAfterReject = this._otelApi.trace.getActiveSpan(); + Assert.equal(activeSpanAfterReject, originalActiveSpan, "Active span should be restored after rejection"); + } + ); + } + }); + + this.testCase({ + name: "withSpan: should pass arguments to callback", + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-args", this._validSpanContext); + const arg1 = "hello"; + const arg2 = 123; + let receivedArgs: any[] = []; + + // Act + withSpan(this._core, span, function(...args: any[]) { + receivedArgs = args; + }, undefined, arg1, arg2); + + // Assert + Assert.equal(receivedArgs.length, 2, "Should receive both arguments"); + Assert.equal(receivedArgs[0], arg1, "First argument should match"); + Assert.equal(receivedArgs[1], arg2, "Second argument should match"); + } + }); + + this.testCase({ + name: "withSpan: should use provided thisArg", + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-this", this._validSpanContext); + const thisArg = { testProperty: "test-value" }; + let capturedThis: any = null; + + // Act + withSpan(this._core, span, function() { + capturedThis = this; + }, thisArg); + + // Assert + Assert.equal(capturedThis, thisArg, "Should use provided thisArg"); + Assert.equal(capturedThis.testProperty, "test-value", "thisArg properties should be accessible"); + } + }); + + this.testCase({ + name: "withSpan: should handle synchronous exception", + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-exception", this._validSpanContext); + const expectedError = new Error("sync-exception"); + const originalActiveSpan = this._otelApi.trace.getActiveSpan(); + + // Act & Assert + try { + withSpan(this._core, span, function() { + throw expectedError; + }); + Assert.ok(false, "Should have thrown an exception"); + } catch (error) { + Assert.equal(error, expectedError, "Should throw the expected error"); + const activeSpanAfterException = this._otelApi.trace.getActiveSpan(); + Assert.equal(activeSpanAfterException, originalActiveSpan, "Active span should be restored after exception"); + } + } + }); + } + + private addUseSpanTests(): void { + this.testCase({ + name: "useSpan: should execute synchronous callback and return value", + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-sync", this._validSpanContext); + const expectedValue = "sync-result"; + + // Act + const result = useSpan(this._core, span, function(scope) { + return expectedValue; + }); + + // Assert + Assert.equal(result, expectedValue, "Should return the callback result"); + } + }); + + this.testCase({ + name: "useSpan: should provide scope as first parameter", + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-scope", this._validSpanContext); + let capturedScope: any = null; + + // Act + useSpan(this._core, span, function(scope) { + capturedScope = scope; + }); + + // Assert + Assert.ok(capturedScope, "Should provide scope"); + Assert.ok(typeof capturedScope.restore === "function", "Scope should have restore method"); + } + }); + + this.testCase({ + name: "useSpan: should set and restore active span for synchronous callback", + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-active-span", this._validSpanContext); + let activeSpanDuringCallback: any = null; + const _self = this; + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + + // Act + useSpan(this._core, span, function(scope) { + activeSpanDuringCallback = _self._otelApi.trace.getActiveSpan(); + }, this); + + const activeSpanAfterCallback = this._otelApi.trace.getActiveSpan(); + + // Assert + Assert.equal(activeSpanDuringCallback, span, "Active span should be set during callback"); + Assert.equal(activeSpanAfterCallback, originalActiveSpan, "Active span should be restored after callback"); + } + }); + + this.testCase({ + name: "useSpan: should handle promise that resolves", + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-promise-resolve", this._validSpanContext); + const expectedValue = "resolved-value"; + let activeSpanDuringCallback: any = null; + let scopeFromCallback: any = null; + const _self = this; + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + + // Act + const promise = useSpan(this._core, span, function(scope) { + activeSpanDuringCallback = _self._otelApi.trace.getActiveSpan(); + scopeFromCallback = scope; + return createPromise((resolve) => { + setTimeout(() => resolve(expectedValue), 10); + }); + }, this); + + // Assert + Assert.equal(activeSpanDuringCallback, span, "Active span should be set during callback"); + Assert.ok(scopeFromCallback, "Scope should be provided to callback"); + + return doAwait(promise, (result) => { + Assert.equal(result, expectedValue, "Should resolve with the expected value"); + const activeSpanAfterResolve = this._otelApi.trace.getActiveSpan(); + Assert.equal(activeSpanAfterResolve, originalActiveSpan, "Active span should be restored after promise resolves"); + }); + } + }); + + this.testCase({ + name: "useSpan: should handle promise that rejects", + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-promise-reject", this._validSpanContext); + const expectedError = new Error("test-error"); + let activeSpanDuringCallback: any = null; + const _self = this; + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + + // Act + const promise = useSpan(this._core, span, function(scope) { + activeSpanDuringCallback = _self._otelApi.trace.getActiveSpan(); + return createRejectedPromise(expectedError); + }, this); + + // Assert + Assert.equal(activeSpanDuringCallback, span, "Active span should be set during callback"); + + return doAwait(promise, + () => { + Assert.ok(false, "Promise should have rejected"); + }, + (error) => { + Assert.equal(error, expectedError, "Should reject with the expected error"); + const activeSpanAfterReject = this._otelApi.trace.getActiveSpan(); + Assert.equal(activeSpanAfterReject, originalActiveSpan, "Active span should be restored after promise rejects"); + } + ); + } + }); + + this.testCase({ + name: "useSpan: should handle async/await pattern with resolved promise", + useFakeTimers: true, + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-async-await", this._validSpanContext); + const expectedValue = 42; + let activeSpanDuringCallback: any = null; + const _self = this; + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + + // Act + const promise = useSpan(this._core, span, function(scope) { + activeSpanDuringCallback = _self._otelApi.trace.getActiveSpan(); + return createPromise((resolve) => { + setTimeout(() => resolve(expectedValue), 100); + }); + }, this); + + // Assert + Assert.equal(activeSpanDuringCallback, span, "Active span should be set during callback"); + + // Advance timers to trigger promise resolution + this.clock.tick(100); + + return doAwait(promise, (result) => { + Assert.equal(result, expectedValue, "Should resolve with the expected value"); + const activeSpanAfterResolve = this._otelApi.trace.getActiveSpan(); + Assert.equal(activeSpanAfterResolve, originalActiveSpan, "Active span should be restored after async operation"); + }); + } + }); + + this.testCase({ + name: "useSpan: should handle async/await pattern with rejected promise", + useFakeTimers: true, + test: () => { + // Arrange + const span = createNonRecordingSpan(this._otelApi, "test-async-reject", this._validSpanContext); + const expectedError = new Error("async-error"); + const _self = this; + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + + // Act + const promise = useSpan(this._core, span, function(scope) { + return createPromise((resolve, reject) => { + setTimeout(() => reject(expectedError), 50); + }); + }); + + // Advance timers + this.clock.tick(50); + + return doAwait(promise, + () => { + Assert.ok(false, "Promise should have rejected"); + }, + (error) => { + Assert.equal(error, expectedError, "Should reject with the expected error"); + const activeSpanAfterReject = this._otelApi.trace.getActiveSpan(); + Assert.equal(activeSpanAfterReject, originalActiveSpan, "Active span should be restored after rejection"); + } + ); + } + }); + + this.testCase({ + name: "useSpan: should pass additional arguments to callback", + test: () => { + // Arrange + const _self = this; + const span = createNonRecordingSpan(_self._otelApi, "test-args", _self._validSpanContext); + const arg1 = "hello"; + const arg2 = 123; + let receivedScope: any = null; + let receivedArgs: any[] = []; + + // Act + useSpan(this._core, span, function(scope, ...args: any[]) { + receivedScope = scope; + receivedArgs = args; + }, undefined, arg1, arg2); + + // Assert + Assert.ok(receivedScope, "Should receive scope as first argument"); + Assert.equal(receivedArgs.length, 2, "Should receive additional arguments"); + Assert.equal(receivedArgs[0], arg1, "First additional argument should match"); + Assert.equal(receivedArgs[1], arg2, "Second additional argument should match"); + } + }); + + this.testCase({ + name: "useSpan: should use provided thisArg", + test: () => { + // Arrange + const _self = this; + const span = createNonRecordingSpan(_self._otelApi, "test-this", _self._validSpanContext); + const thisArg = { testProperty: "test-value" }; + let capturedThis: any = null; + + // Act + useSpan(this._core, span, function(scope) { + capturedThis = this; + }, thisArg); + + // Assert + Assert.equal(capturedThis, thisArg, "Should use provided thisArg"); + Assert.equal(capturedThis.testProperty, "test-value", "thisArg properties should be accessible"); + } + }); + + this.testCase({ + name: "useSpan: should use scope as thisArg when not provided", + test: () => { + // Arrange + const _self = this; + const span = createNonRecordingSpan(_self._otelApi, "test-default-this", _self._validSpanContext); + let capturedThis: any = null; + let capturedScope: any = null; + + // Act + useSpan(_self._core, span, function(scope) { + capturedThis = this; + capturedScope = scope; + }); + + // Assert + Assert.equal(capturedThis, capturedScope, "Should use scope as thisArg when not provided"); + } + }); + + this.testCase({ + name: "useSpan: should handle synchronous exception", + test: () => { + // Arrange + const _self = this; + const span = createNonRecordingSpan(_self._otelApi, "test-exception", _self._validSpanContext); + const expectedError = new Error("sync-exception"); + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + + // Act & Assert + try { + useSpan(_self._core, span, function(scope) { + throw expectedError; + }); + Assert.ok(false, "Should have thrown an exception"); + } catch (error) { + Assert.equal(error, expectedError, "Should throw the expected error"); + const activeSpanAfterException = this._otelApi.trace.getActiveSpan(); + Assert.equal(activeSpanAfterException, originalActiveSpan, "Active span should be restored after exception"); + } + } + }); + + this.testCase({ + name: "useSpan: scope.restore should be callable manually", + test: () => { + // Arrange + const _self = this; + const span = createNonRecordingSpan(_self._otelApi, "test-manual-restore", _self._validSpanContext); + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + let scopeRestored = false; + + // Act + useSpan(_self._core, span, function(scope) { + Assert.equal(_self._otelApi.trace.getActiveSpan(), span, "Active span should be set"); + // Manually restore (though the framework will restore again) + scope.restore(); + scopeRestored = true; + }, _self); + // Assert + Assert.ok(scopeRestored, "Scope restore should have been called"); + const activeSpanAfterCallback = this._otelApi.trace.getActiveSpan(); + Assert.equal(activeSpanAfterCallback, originalActiveSpan, "Active span should be restored"); + } + }); + + this.testCase({ + name: "useSpan: should handle promise rejection with ts-async doAwait", + test: () => { + // Arrange + const _self = this; + const span = createNonRecordingSpan(_self._otelApi, "test-doawait-reject", _self._validSpanContext); + const expectedError = new Error("doAwait-error"); + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + + // Act + const promise = useSpan(_self._core, span, function(scope) { + return createPromise((resolve, reject) => { + setTimeout(() => reject(expectedError), 5); + }); + }); + + // Use doAwait pattern to handle rejection + return doAwait(promise, + (value) => { + Assert.ok(false, "Should not resolve"); + }, + (reason) => { + Assert.equal(reason, expectedError, "Should reject with expected error"); + Assert.equal(this._otelApi.trace.getActiveSpan(), originalActiveSpan, "Active span should be restored after rejection"); + } + ); + } + }); + + this.testCase({ + name: "useSpan: should handle complex promise chain", + test: () => { + // Arrange + const _self = this; + const span = createNonRecordingSpan(_self._otelApi, "test-chain", _self._validSpanContext); + const originalActiveSpan = _self._otelApi.trace.getActiveSpan(); + + // Act + const promise = useSpan(_self._core, span, function(scope) { + return createPromise((resolve) => { + setTimeout(() => resolve(1), 5); + }); + }); + + // Chain multiple operations + return doAwait(promise, (value) => { + Assert.equal(value, 1, "First promise should resolve with 1"); + + return doAwait(createPromise((resolve) => { + setTimeout(() => resolve(value + 1), 5); + }), (value2) => { + Assert.equal(value2, 2, "Second promise should resolve with 2"); + Assert.equal(this._otelApi.trace.getActiveSpan(), originalActiveSpan, "Active span should be restored"); + }); + }); + } + }); + } +} diff --git a/shared/AppInsightsCore/Tests/tsconfig.json b/shared/AppInsightsCore/Tests/tsconfig.json index a43b959cf..481ad4275 100644 --- a/shared/AppInsightsCore/Tests/tsconfig.json +++ b/shared/AppInsightsCore/Tests/tsconfig.json @@ -5,7 +5,7 @@ "noImplicitAny": false, "module": "amd", "moduleResolution": "node", - "target": "es6", + "target": "es2017", "importHelpers": true, "noEmitHelpers": true, "skipLibCheck": true, diff --git a/shared/AppInsightsCore/api-extractor.json b/shared/AppInsightsCore/api-extractor.json index 217a20851..1209227df 100644 --- a/shared/AppInsightsCore/api-extractor.json +++ b/shared/AppInsightsCore/api-extractor.json @@ -45,7 +45,7 @@ * * SUPPORTED TOKENS: , , */ - "mainEntryPointFilePath": "/build/types/applicationinsights-core-js.d.ts", + "mainEntryPointFilePath": "/build/types/index.d.ts", /** * A list of NPM package names whose exports should be treated as part of this package. diff --git a/shared/AppInsightsCore/package.json b/shared/AppInsightsCore/package.json index 859b9714d..2fa5d145a 100644 --- a/shared/AppInsightsCore/package.json +++ b/shared/AppInsightsCore/package.json @@ -13,8 +13,8 @@ "browser performance monitoring", "web analytics" ], - "main": "dist/es5/applicationinsights-core-js.js", - "module": "dist-es5/applicationinsights-core-js.js", + "main": "dist/es5/index.js", + "module": "dist-es5/index.js", "types": "types/applicationinsights-core-js.d.ts", "scripts": { "clean": "git clean -xdf", diff --git a/shared/AppInsightsCore/rollup.config.js b/shared/AppInsightsCore/rollup.config.js index 7ff1b6e4e..c4d0270c6 100644 --- a/shared/AppInsightsCore/rollup.config.js +++ b/shared/AppInsightsCore/rollup.config.js @@ -2,34 +2,35 @@ import { createUnVersionedConfig } from "../../rollup.base.config"; import { updateDistEsmFiles } from "../../tools/updateDistEsm/updateDistEsm"; const version = require("./package.json").version; -const entryPointName = "applicationinsights-core-js"; +const entryPointName = "index"; +const browserOutputName = "applicationinsights-core-js"; // MUST NOT CHANGE - CDN compatibility const banner = [ - "/*!", - ` * Application Insights JavaScript SDK - Core, ${version}`, - " * Copyright (c) Microsoft and contributors. All rights reserved.", - " */" + "/*!", + ` * Application Insights JavaScript SDK - Core, ${version}`, + " * Copyright (c) Microsoft and contributors. All rights reserved.", + " */" ].join("\n"); const replaceValues = { - "// Copyright (c) Microsoft Corporation. All rights reserved.": "", - "// Licensed under the MIT License.": "" + "// Copyright (c) Microsoft Corporation. All rights reserved.": "", + "// Licensed under the MIT License.": "" }; updateDistEsmFiles(replaceValues, banner, true, true, "dist-es5"); -export default createUnVersionedConfig(banner, - { - namespace: "Microsoft.ApplicationInsights", - version: version, - node: { - entryPoint: entryPointName, - outputName: entryPointName +export default createUnVersionedConfig(banner, + { + namespace: "Microsoft.ApplicationInsights", + version: version, + node: { + entryPoint: entryPointName, + outputName: entryPointName + }, + browser: { + entryPoint: entryPointName, + outputName: browserOutputName // Browser keeps original name for CDN + } }, - browser: { - entryPoint: entryPointName, - outputName: entryPointName - }, - }, - [ "applicationinsights-core-js" ], - false + ["index", "applicationinsights-core-js"], + false ); \ No newline at end of file diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IDistributedTraceContext.ts b/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IDistributedTraceContext.ts deleted file mode 100644 index 1655ce517..000000000 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IDistributedTraceContext.ts +++ /dev/null @@ -1,54 +0,0 @@ -// Copyright (c) Microsoft Corporation. All rights reserved. -// Licensed under the MIT License. - -export interface IDistributedTraceContext { - - /** - * Returns the current name of the page - */ - getName(): string; - - /** - * Sets the current name of the page - * @param pageName - The name of the page - */ - setName(pageName: string): void; - - /** - * Returns the unique identifier for a trace. All requests / spans from the same trace share the same traceId. - * Must be read from incoming headers or generated according to the W3C TraceContext specification, - * in a hex representation of 16-byte array. A.k.a. trace-id, TraceID or Distributed TraceID - */ - getTraceId(): string; - - /** - * Set the unique identifier for a trace. All requests / spans from the same trace share the same traceId. - * Must be conform to the W3C TraceContext specification, in a hex representation of 16-byte array. - * A.k.a. trace-id, TraceID or Distributed TraceID https://www.w3.org/TR/trace-context/#trace-id - */ - setTraceId(newValue: string): void; - - /** - * Self-generated 8-bytes identifier of the incoming request. Must be a hex representation of 8-byte array. - * Also know as the parentId, used to link requests together - */ - getSpanId(): string; - - /** - * Self-generated 8-bytes identifier of the incoming request. Must be a hex representation of 8-byte array. - * Also know as the parentId, used to link requests together - * https://www.w3.org/TR/trace-context/#parent-id - */ - setSpanId(newValue: string): void; - - /** - * An integer representation of the W3C TraceContext trace-flags. - */ - getTraceFlags(): number | undefined; - - /** - * https://www.w3.org/TR/trace-context/#trace-flags - * @param newValue - An integer representation of the W3C TraceContext trace-flags. - */ - setTraceFlags(newValue?: number): void; -} diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/Constants.ts b/shared/AppInsightsCore/src/JavaScriptSDK/Constants.ts deleted file mode 100644 index 38e9922eb..000000000 --- a/shared/AppInsightsCore/src/JavaScriptSDK/Constants.ts +++ /dev/null @@ -1,14 +0,0 @@ -// Copyright (c) Microsoft Corporation. All rights reserved. -// Licensed under the MIT License. - -export const ChannelControllerPriority = 500; - -export const DisabledPropertyName: string = "Microsoft_ApplicationInsights_BypassAjaxInstrumentation"; -// export const SampleRate = "sampleRate"; -// export const ProcessLegacy = "ProcessLegacy"; -// export const HttpMethod = "http.method"; -// export const DEFAULT_BREEZE_ENDPOINT = "https://dc.services.visualstudio.com"; -// export const DEFAULT_BREEZE_PATH = "/v2/track"; -// export const strNotSpecified = "not_specified"; -// export const strIkey = "iKey"; - diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/TelemetryHelpers.ts b/shared/AppInsightsCore/src/JavaScriptSDK/TelemetryHelpers.ts deleted file mode 100644 index 5e4c87281..000000000 --- a/shared/AppInsightsCore/src/JavaScriptSDK/TelemetryHelpers.ts +++ /dev/null @@ -1,183 +0,0 @@ -// Copyright (c) Microsoft Corporation. All rights reserved. -// Licensed under the MIT License. - -import { arrForEach, isFunction } from "@nevware21/ts-utils"; -import { IAppInsightsCore } from "../JavaScriptSDK.Interfaces/IAppInsightsCore"; -import { IDistributedTraceContext } from "../JavaScriptSDK.Interfaces/IDistributedTraceContext"; -import { IProcessTelemetryContext, IProcessTelemetryUnloadContext } from "../JavaScriptSDK.Interfaces/IProcessTelemetryContext"; -import { IPlugin, ITelemetryPlugin } from "../JavaScriptSDK.Interfaces/ITelemetryPlugin"; -import { ITelemetryPluginChain } from "../JavaScriptSDK.Interfaces/ITelemetryPluginChain"; -import { ITelemetryUnloadState } from "../JavaScriptSDK.Interfaces/ITelemetryUnloadState"; -import { ITraceParent } from "../JavaScriptSDK.Interfaces/ITraceParent"; -import { IUnloadableComponent } from "../JavaScriptSDK.Interfaces/IUnloadableComponent"; -import { createElmNodeData } from "./DataCacheHelper"; -import { STR_CORE, STR_PRIORITY, STR_PROCESS_TELEMETRY } from "./InternalConstants"; -import { isValidSpanId, isValidTraceId } from "./W3cTraceParent"; - -export interface IPluginState { - core?: IAppInsightsCore; - isInitialized?: boolean; - teardown?: boolean; - disabled?: boolean; -} - -const pluginStateData = createElmNodeData("plugin"); - -export function _getPluginState(plugin: IPlugin): IPluginState { - return pluginStateData.get(plugin, "state", {}, true) -} - -/** - * Initialize the queue of plugins - * @param plugins - The array of plugins to initialize and setting of the next plugin - * @param config - The current config for the instance - * @param core - THe current core instance - * @param extensions - The extensions - */ -export function initializePlugins(processContext: IProcessTelemetryContext, extensions: IPlugin[]) { - - // Set the next plugin and identified the uninitialized plugins - let initPlugins: ITelemetryPlugin[] = []; - let lastPlugin: ITelemetryPlugin = null; - let proxy: ITelemetryPluginChain = processContext.getNext(); - let pluginState: IPluginState; - - while (proxy) { - let thePlugin = proxy.getPlugin(); - if (thePlugin) { - if (lastPlugin && lastPlugin.setNextPlugin && thePlugin.processTelemetry) { - // Set this plugin as the next for the previous one - lastPlugin.setNextPlugin(thePlugin); - } - - pluginState = _getPluginState(thePlugin); - - let isInitialized = !!pluginState.isInitialized; - if (thePlugin.isInitialized) { - isInitialized = thePlugin.isInitialized(); - } - - if (!isInitialized) { - initPlugins.push(thePlugin); - } - - lastPlugin = thePlugin; - proxy = proxy.getNext(); - } - } - - // Now initialize the plugins - arrForEach(initPlugins, thePlugin => { - let core = processContext.core(); - - thePlugin.initialize( - processContext.getCfg(), - core, - extensions, - processContext.getNext()); - - pluginState = _getPluginState(thePlugin); - - // Only add the core to the state if the plugin didn't set it (doesn't extend from BaseTelemetryPlugin) - if (!thePlugin[STR_CORE] && !pluginState[STR_CORE]) { - pluginState[STR_CORE] = core; - } - - pluginState.isInitialized = true; - delete pluginState.teardown; - }); -} - -export function sortPlugins(plugins:T[]) { - // Sort by priority - return plugins.sort((extA, extB) => { - let result = 0; - if (extB) { - let bHasProcess = extB[STR_PROCESS_TELEMETRY]; - if (extA[STR_PROCESS_TELEMETRY]) { - result = bHasProcess ? extA[STR_PRIORITY] - extB[STR_PRIORITY] : 1; - } else if (bHasProcess) { - result = -1; - } - } else { - result = extA ? 1 : -1; - } - - return result; - }); - // sort complete -} - -/** - * Teardown / Unload helper to perform teardown/unloading operations for the provided components synchronously or asynchronously, this will call any - * _doTeardown() or _doUnload() functions on the provided components to allow them to finish removal. - * @param components - The components you want to unload - * @param unloadCtx - This is the context that should be used during unloading. - * @param unloadState - The details / state of the unload process, it holds details like whether it should be unloaded synchronously or asynchronously and the reason for the unload. - * @param asyncCallback - An optional callback that the plugin must call if it returns true to inform the caller that it has completed any async unload/teardown operations. - * @returns boolean - true if the plugin has or will call asyncCallback, this allows the plugin to perform any asynchronous operations. - */ -export function unloadComponents(components: any | IUnloadableComponent[], unloadCtx?: IProcessTelemetryUnloadContext, unloadState?: ITelemetryUnloadState, asyncCallback?: () => void): void | boolean { - let idx = 0; - - function _doUnload(): void | boolean { - while (idx < components.length) { - let component = components[idx++]; - if (component) { - let func = component._doUnload || component._doTeardown; - if (isFunction(func)) { - if (func.call(component, unloadCtx, unloadState, _doUnload) === true) { - return true; - } - } - } - } - } - - return _doUnload(); -} - - -/** - * Creates a IDistributedTraceContext which optionally also "sets" the value on a parent - * @param parentCtx - An optional parent distributed trace instance - * @returns A new IDistributedTraceContext instance that uses an internal temporary object - */ -export function createDistributedTraceContext(parentCtx?: IDistributedTraceContext): IDistributedTraceContext { - let trace: ITraceParent = {} as ITraceParent; - - return { - getName: (): string => { - return (trace as any).name; - }, - setName: (newValue: string): void => { - parentCtx && parentCtx.setName(newValue); - (trace as any).name = newValue; - }, - getTraceId: (): string => { - return trace.traceId; - }, - setTraceId: (newValue: string): void => { - parentCtx && parentCtx.setTraceId(newValue); - if (isValidTraceId(newValue)) { - trace.traceId = newValue - } - }, - getSpanId: (): string => { - return trace.spanId; - }, - setSpanId: (newValue: string): void => { - parentCtx && parentCtx.setSpanId(newValue); - if (isValidSpanId(newValue)) { - trace.spanId = newValue - } - }, - getTraceFlags: (): number => { - return trace.traceFlags; - }, - setTraceFlags: (newTraceFlags?: number): void => { - parentCtx && parentCtx.setTraceFlags(newTraceFlags); - trace.traceFlags = newTraceFlags - } - }; -} diff --git a/shared/AppInsightsCore/src/applicationinsights-core-js.ts b/shared/AppInsightsCore/src/applicationinsights-core-js.ts deleted file mode 100644 index 12a4be475..000000000 --- a/shared/AppInsightsCore/src/applicationinsights-core-js.ts +++ /dev/null @@ -1,112 +0,0 @@ -// Copyright (c) Microsoft Corporation. All rights reserved. -// Licensed under the MIT License. -export { IConfiguration } from "./JavaScriptSDK.Interfaces/IConfiguration"; -export { IChannelControls, MinChannelPriorty, IInternalOfflineSupport } from "./JavaScriptSDK.Interfaces/IChannelControls"; -export { IChannelControlsHost } from "./JavaScriptSDK.Interfaces/IChannelControlsHost"; -export { ITelemetryPlugin, IPlugin } from "./JavaScriptSDK.Interfaces/ITelemetryPlugin"; -export { IExceptionConfig } from "./JavaScriptSDK.Interfaces/IExceptionConfig"; -export { IAppInsightsCore, ILoadedPlugin } from "./JavaScriptSDK.Interfaces/IAppInsightsCore"; -export { ITelemetryItem, ICustomProperties, Tags } from "./JavaScriptSDK.Interfaces/ITelemetryItem"; -export { IBaseProcessingContext, IProcessTelemetryContext, IProcessTelemetryUnloadContext, IProcessTelemetryUpdateContext } from "./JavaScriptSDK.Interfaces/IProcessTelemetryContext"; -export { INotificationListener } from "./JavaScriptSDK.Interfaces/INotificationListener"; -export { ITelemetryPluginChain } from "./JavaScriptSDK.Interfaces/ITelemetryPluginChain"; -export { IDiagnosticLogger } from "./JavaScriptSDK.Interfaces/IDiagnosticLogger"; -export { InstrumentorHooksCallback, IInstrumentHooksCallbacks, IInstrumentHooks, IInstrumentHook, IInstrumentCallDetails } from "./JavaScriptSDK.Interfaces/IInstrumentHooks"; -export { IUnloadableComponent } from "./JavaScriptSDK.Interfaces/IUnloadableComponent"; -export { IPayloadData, SendPOSTFunction, IXHROverride, OnCompleteCallback } from "./JavaScriptSDK.Interfaces/IXHROverride" -export { IUnloadHook, ILegacyUnloadHook } from "./JavaScriptSDK.Interfaces/IUnloadHook"; -export { eEventsDiscardedReason, EventsDiscardedReason, eBatchDiscardedReason, BatchDiscardedReason } from "./JavaScriptSDK.Enums/EventsDiscardedReason"; -export { SendRequestReason, TransportType } from "./JavaScriptSDK.Enums/SendRequestReason"; -//export { StatsType, eStatsType } from "./JavaScriptSDK.Enums/StatsType"; -export { TelemetryUpdateReason } from "./JavaScriptSDK.Enums/TelemetryUpdateReason"; -export { TelemetryUnloadReason } from "./JavaScriptSDK.Enums/TelemetryUnloadReason"; -export { eActiveStatus, ActiveStatus } from "./JavaScriptSDK.Enums/InitActiveStatusEnum" -export { throwAggregationError } from "./JavaScriptSDK/AggregationError"; -export { AppInsightsCore } from "./JavaScriptSDK/AppInsightsCore"; -export { BaseTelemetryPlugin } from "./JavaScriptSDK/BaseTelemetryPlugin"; -export { randomValue, random32, mwcRandomSeed, mwcRandom32, newId } from "./JavaScriptSDK/RandomHelper"; -export { Undefined, newGuid, generateW3CId } from "./JavaScriptSDK/CoreUtils"; -export { runTargetUnload, doUnloadAll } from "./JavaScriptSDK/AsyncUtils"; -export { - normalizeJsName, toISOString, getExceptionName, strContains, setValue, getSetValue, - proxyAssign, proxyFunctions, proxyFunctionAs, createClassFromInterface, optimizeObject, - isNotUndefined, isNotNullOrUndefined, objExtend, isFeatureEnabled, getResponseText, formatErrorMessageXdr, formatErrorMessageXhr, prependTransports, - openXhr, _appendHeader, _getAllResponseHeaders, convertAllHeadersToMap -} from "./JavaScriptSDK/HelperFuncs"; -export { parseResponse } from "./JavaScriptSDK/ResponseHelpers"; -export { IXDomainRequest, IBackendResponse } from "./JavaScriptSDK.Interfaces/IXDomainRequest"; -export { _ISenderOnComplete, _ISendPostMgrConfig, _ITimeoutOverrideWrapper, _IInternalXhrOverride } from "./JavaScriptSDK.Interfaces/ISenderPostManager"; -export { SenderPostManager } from "./JavaScriptSDK/SenderPostManager"; -//export { IStatsBeat, IStatsBeatConfig, IStatsBeatKeyMap as IStatsBeatEndpoints, IStatsBeatState} from "./JavaScriptSDK.Interfaces/IStatsBeat"; -//export { IStatsEventData } from "./JavaScriptSDK.Interfaces/IStatsEventData"; -//export { IStatsMgr, IStatsMgrConfig } from "./JavaScriptSDK.Interfaces/IStatsMgr"; -//export { createStatsMgr } from "./JavaScriptSDK/StatsBeat"; -export { - isArray, isTypeof, isUndefined, isNullOrUndefined, objHasOwnProperty as hasOwnProperty, isObject, isFunction, - strEndsWith, strStartsWith, isDate, isError, isString, isNumber, isBoolean, arrForEach, arrIndexOf, - arrReduce, arrMap, strTrim, objKeys, objDefineAccessors, throwError, isSymbol, - isNotTruthy, isTruthy, objFreeze, objSeal, objToString, objDeepFreeze as deepFreeze, - getInst as getGlobalInst, hasWindow, getWindow, hasDocument, getDocument, hasNavigator, getNavigator, hasHistory, - getHistory, dumpObj, asString, objForEachKey, getPerformance, utcNow as dateNow, perfNow -} from "@nevware21/ts-utils"; -export { EnumValue, createEnumStyle, createValueMap } from "./JavaScriptSDK.Enums/EnumHelperFuncs"; -export { - attachEvent, detachEvent, addEventHandler, addEventListeners, addPageUnloadEventListener, addPageHideEventListener, addPageShowEventListener, - removeEventHandler, removeEventListeners, removePageUnloadEventListener, removePageHideEventListener, removePageShowEventListener, eventOn, eventOff, - mergeEvtNamespace, _IRegisteredEvents, __getRegisteredEvents -} from "./JavaScriptSDK/EventHelpers"; - -export { - getCrypto, getMsCrypto, getLocation, hasJSON, getJSON, - isReactNative, getConsole, isIE, getIEVersion, isSafari, - setEnableEnvMocks, isBeaconsSupported, isFetchSupported, useXDomainRequest, isXhrSupported, - findMetaTag, findNamedServerTiming, sendCustomEvent, dispatchEvent, createCustomDomEvent, fieldRedaction -} from "./JavaScriptSDK/EnvUtils"; -export { - getGlobal, - strShimPrototype as strPrototype, - strShimFunction as strFunction, - strShimUndefined as strUndefined, - strShimObject as strObject -} from "@microsoft/applicationinsights-shims"; -export { NotificationManager } from "./JavaScriptSDK/NotificationManager"; -export { INotificationManager } from "./JavaScriptSDK.Interfaces/INotificationManager"; -export { IPerfEvent } from "./JavaScriptSDK.Interfaces/IPerfEvent"; -export { IPerfManager, IPerfManagerProvider } from "./JavaScriptSDK.Interfaces/IPerfManager"; -export { PerfEvent, PerfManager, doPerf, getGblPerfMgr, setGblPerfMgr } from "./JavaScriptSDK/PerfManager"; -export { IFeatureOptInDetails, IFeatureOptIn } from "./JavaScriptSDK.Interfaces/IFeatureOptIn"; -export { FeatureOptInMode, CdnFeatureMode } from "./JavaScriptSDK.Enums/FeatureOptInEnums" -export { safeGetLogger, DiagnosticLogger, _InternalLogMessage, _throwInternal, _warnToConsole, _logInternalMessage } from "./JavaScriptSDK/DiagnosticLogger"; -export { - ProcessTelemetryContext, createProcessTelemetryContext - // Explicitly NOT exporting createProcessTelemetryUnloadContext() and createProcessTelemetryUpdateContext() as these should only be created internally -} from "./JavaScriptSDK/ProcessTelemetryContext"; -export { initializePlugins, sortPlugins, unloadComponents } from "./JavaScriptSDK/TelemetryHelpers"; -export { _eInternalMessageId, _InternalMessageId, LoggingSeverity, eLoggingSeverity } from "./JavaScriptSDK.Enums/LoggingEnums"; -export { InstrumentProto, InstrumentProtos, InstrumentFunc, InstrumentFuncs, InstrumentEvent } from "./JavaScriptSDK/InstrumentHooks"; -export { ICookieMgr, ICookieMgrConfig } from "./JavaScriptSDK.Interfaces/ICookieMgr"; -export { - createCookieMgr, safeGetCookieMgr, uaDisallowsSameSiteNone, areCookiesSupported -} from "./JavaScriptSDK/CookieMgr"; -export { IDbgExtension } from "./JavaScriptSDK.Interfaces/IDbgExtension"; -export { getDebugListener, getDebugExt } from "./JavaScriptSDK/DbgExtensionUtils" -export { TelemetryInitializerFunction, ITelemetryInitializerHandler, ITelemetryInitializerContainer } from "./JavaScriptSDK.Interfaces/ITelemetryInitializers"; -export { createUniqueNamespace } from "./JavaScriptSDK/DataCacheHelper"; -export { UnloadHandler, IUnloadHandlerContainer, createUnloadHandlerContainer } from "./JavaScriptSDK/UnloadHandlerContainer"; -export { IUnloadHookContainer, createUnloadHookContainer, _testHookMaxUnloadHooksCb } from "./JavaScriptSDK/UnloadHookContainer"; -export { ITelemetryUpdateState } from "./JavaScriptSDK.Interfaces/ITelemetryUpdateState"; -export { ITelemetryUnloadState } from "./JavaScriptSDK.Interfaces/ITelemetryUnloadState"; -export { IDistributedTraceContext } from "./JavaScriptSDK.Interfaces/IDistributedTraceContext"; -export { ITraceParent } from "./JavaScriptSDK.Interfaces/ITraceParent"; -export { - createTraceParent, parseTraceParent, isValidTraceId, isValidSpanId, isValidTraceParent, isSampledFlag, formatTraceParent, findW3cTraceParent, findAllScripts -} from "./JavaScriptSDK/W3cTraceParent"; - -// Dynamic Config definitions -export { IConfigCheckFn, IConfigDefaultCheck, IConfigDefaults, IConfigSetFn } from "./Config/IConfigDefaults"; -export { IDynamicConfigHandler } from "./Config/IDynamicConfigHandler"; -export { IDynamicPropertyHandler } from "./Config/IDynamicPropertyHandler"; -export { IWatchDetails, IWatcherHandler, WatcherFunction } from "./Config/IDynamicWatcher"; -export { createDynamicConfig, onConfigChange } from "./Config/DynamicConfig"; -export { getDynamicConfigHandler, blockDynamicConversion, forceDynamicConversion } from "./Config/DynamicSupport"; -export { cfgDfValidate, cfgDfMerge, cfgDfBoolean, cfgDfFunc, cfgDfString, cfgDfSet, cfgDfBlockPropValue } from "./Config/ConfigDefaultHelpers"; \ No newline at end of file diff --git a/shared/AppInsightsCore/src/Config/ConfigDefaultHelpers.ts b/shared/AppInsightsCore/src/config/ConfigDefaultHelpers.ts similarity index 93% rename from shared/AppInsightsCore/src/Config/ConfigDefaultHelpers.ts rename to shared/AppInsightsCore/src/config/ConfigDefaultHelpers.ts index 8c7eedeca..5ef7fa72d 100644 --- a/shared/AppInsightsCore/src/Config/ConfigDefaultHelpers.ts +++ b/shared/AppInsightsCore/src/config/ConfigDefaultHelpers.ts @@ -2,9 +2,9 @@ // Licensed under the MIT License. import { asString, isBoolean, isFunction, isNullOrUndefined, isString } from "@nevware21/ts-utils"; -import { STR_EMPTY } from "../JavaScriptSDK/InternalConstants"; -import { IConfiguration } from "../applicationinsights-core-js"; -import { IConfigCheckFn, IConfigDefaultCheck, IConfigDefaults, IConfigSetFn } from "./IConfigDefaults"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { IConfigCheckFn, IConfigDefaultCheck, IConfigDefaults, IConfigSetFn } from "../interfaces/config/IConfigDefaults"; /** * @internal @@ -31,7 +31,7 @@ function _stringToBoolOrDefault(theValue: any, defaultValue: boolean, theConf * @param defaultValue - The default value to apply it not provided or it's not valid * @returns a new IConfigDefaultCheck structure */ -export function cfgDfMerge(defaultValue: V | IConfigDefaults): IConfigDefaultCheck { +export function cfgDfMerge(defaultValue: (V | undefined) | IConfigDefaults): IConfigDefaultCheck { return { mrg: true, v: defaultValue diff --git a/shared/AppInsightsCore/src/Config/ConfigDefaults.ts b/shared/AppInsightsCore/src/config/ConfigDefaults.ts similarity index 98% rename from shared/AppInsightsCore/src/Config/ConfigDefaults.ts rename to shared/AppInsightsCore/src/config/ConfigDefaults.ts index be8e6621d..1586917b0 100644 --- a/shared/AppInsightsCore/src/Config/ConfigDefaults.ts +++ b/shared/AppInsightsCore/src/config/ConfigDefaults.ts @@ -4,8 +4,8 @@ import { asString, isArray, isDefined, isNullOrUndefined, isObject, isPlainObject, isUndefined, objForEachKey, objHasOwn } from "@nevware21/ts-utils"; -import { IConfigCheckFn, IConfigDefaultCheck, IConfigDefaults, IConfigSetFn } from "./IConfigDefaults"; -import { IDynamicConfigHandler } from "./IDynamicConfigHandler"; +import { IConfigCheckFn, IConfigDefaultCheck, IConfigDefaults, IConfigSetFn } from "../interfaces/config/IConfigDefaults"; +import { IDynamicConfigHandler } from "../interfaces/config/IDynamicConfigHandler"; function _isConfigDefaults(value: any): value is IConfigDefaultCheck { return (value && isObject(value) && !isArray(value) && (value.isVal || value.fb || objHasOwn(value, "v") || objHasOwn(value, "mrg") || objHasOwn(value, "ref") || value.set)); diff --git a/shared/AppInsightsCore/src/Config/DynamicConfig.ts b/shared/AppInsightsCore/src/config/DynamicConfig.ts similarity index 91% rename from shared/AppInsightsCore/src/Config/DynamicConfig.ts rename to shared/AppInsightsCore/src/config/DynamicConfig.ts index eb7a15c41..ad1fcbce5 100644 --- a/shared/AppInsightsCore/src/Config/DynamicConfig.ts +++ b/shared/AppInsightsCore/src/config/DynamicConfig.ts @@ -2,21 +2,21 @@ // Licensed under the MIT License. import { dumpObj, isUndefined, objDefine, objForEachKey } from "@nevware21/ts-utils"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; -import { createUniqueNamespace } from "../JavaScriptSDK/DataCacheHelper"; -import { STR_NOT_DYNAMIC_ERROR } from "../JavaScriptSDK/InternalConstants"; +import { STR_EMPTY, STR_NOT_DYNAMIC_ERROR } from "../constants/InternalConstants"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { IConfigDefaults } from "../interfaces/config/IConfigDefaults"; +import { IDynamicConfigHandler, _IInternalDynamicConfigHandler } from "../interfaces/config/IDynamicConfigHandler"; +import { IWatcherHandler, WatcherFunction } from "../interfaces/config/IDynamicWatcher"; +import { _IDynamicConfigHandlerState } from "../interfaces/config/_IDynamicConfigHandlerState"; +import { createUniqueNamespace } from "../utils/DataCacheHelper"; import { _applyDefaultValue } from "./ConfigDefaults"; import { _eSetDynamicPropertyFlags, _makeDynamicObject, _setDynamicProperty, _setDynamicPropertyState, _throwDynamicError } from "./DynamicProperty"; import { _createState } from "./DynamicState"; import { CFG_HANDLER_LINK, _cfgDeepCopy, getDynamicConfigHandler, throwInvalidAccess } from "./DynamicSupport"; -import { IConfigDefaults } from "./IConfigDefaults"; -import { IDynamicConfigHandler, _IInternalDynamicConfigHandler } from "./IDynamicConfigHandler"; -import { IWatcherHandler, WatcherFunction } from "./IDynamicWatcher"; -import { _IDynamicConfigHandlerState } from "./_IDynamicConfigHandlerState"; /** * Identifies a function which will be re-called whenever any of it's accessed configuration values @@ -36,7 +36,7 @@ function _createAndUseHandler(state: _IDynamicConfigHandlerState, configHa } }; - objDefine(handler, "toJSON", { v: () => "WatcherHandler" + (handler.fn ? "" : "[X]") }); + objDefine(handler, "toJSON", { v: () => "WatcherHandler" + (handler.fn ? STR_EMPTY : "[X]") }); state.use(handler, configHandler); diff --git a/shared/AppInsightsCore/src/Config/DynamicProperty.ts b/shared/AppInsightsCore/src/config/DynamicProperty.ts similarity index 96% rename from shared/AppInsightsCore/src/Config/DynamicProperty.ts rename to shared/AppInsightsCore/src/config/DynamicProperty.ts index 826a79029..2aee68895 100644 --- a/shared/AppInsightsCore/src/Config/DynamicProperty.ts +++ b/shared/AppInsightsCore/src/config/DynamicProperty.ts @@ -4,12 +4,12 @@ import { arrForEach, arrIndexOf, dumpObj, isArray, objDefine, objDefineProp, objForEachKey, objGetOwnPropertyDescriptor } from "@nevware21/ts-utils"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; -import { UNDEFINED_VALUE } from "../JavaScriptSDK/InternalConstants"; +import { UNDEFINED_VALUE } from "../constants/InternalConstants"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { IWatcherHandler, _IDynamicDetail } from "../interfaces/config/IDynamicWatcher"; +import { _IDynamicConfigHandlerState, _IDynamicGetter } from "../interfaces/config/_IDynamicConfigHandlerState"; import { CFG_HANDLER_LINK, _canMakeDynamic, blockDynamicConversion, throwInvalidAccess } from "./DynamicSupport"; -import { IWatcherHandler, _IDynamicDetail } from "./IDynamicWatcher"; -import { _IDynamicConfigHandlerState, _IDynamicGetter } from "./_IDynamicConfigHandlerState"; export const enum _eSetDynamicPropertyFlags { inPlace = 0, diff --git a/shared/AppInsightsCore/src/Config/DynamicState.ts b/shared/AppInsightsCore/src/config/DynamicState.ts similarity index 94% rename from shared/AppInsightsCore/src/Config/DynamicState.ts rename to shared/AppInsightsCore/src/config/DynamicState.ts index 60851283b..1109d7505 100644 --- a/shared/AppInsightsCore/src/Config/DynamicState.ts +++ b/shared/AppInsightsCore/src/config/DynamicState.ts @@ -2,11 +2,11 @@ // Licensed under the MIT License. import { ITimerHandler, arrForEach, arrIndexOf, dumpObj, newSymbol, scheduleTimeout } from "@nevware21/ts-utils"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { throwAggregationError } from "../JavaScriptSDK/AggregationError"; -import { _IInternalDynamicConfigHandler } from "./IDynamicConfigHandler"; -import { IWatcherHandler, WatcherFunction, _IDynamicDetail } from "./IDynamicWatcher"; -import { _IDynamicConfigHandlerState } from "./_IDynamicConfigHandlerState"; +import { throwAggregationError } from "../core/AggregationError"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { _IInternalDynamicConfigHandler } from "../interfaces/config/IDynamicConfigHandler"; +import { IWatcherHandler, WatcherFunction, _IDynamicDetail } from "../interfaces/config/IDynamicWatcher"; +import { _IDynamicConfigHandlerState } from "../interfaces/config/_IDynamicConfigHandlerState"; const symPrefix = "[[ai_"; const symPostfix = "]]"; diff --git a/shared/AppInsightsCore/src/Config/DynamicSupport.ts b/shared/AppInsightsCore/src/config/DynamicSupport.ts similarity index 94% rename from shared/AppInsightsCore/src/Config/DynamicSupport.ts rename to shared/AppInsightsCore/src/config/DynamicSupport.ts index c7e8e2193..5a7cad09d 100644 --- a/shared/AppInsightsCore/src/Config/DynamicSupport.ts +++ b/shared/AppInsightsCore/src/config/DynamicSupport.ts @@ -2,9 +2,9 @@ // Licensed under the MIT License. import { isArray, isPlainObject, objForEachKey, symbolFor, throwTypeError } from "@nevware21/ts-utils"; -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { IDynamicConfigHandler } from "./IDynamicConfigHandler"; -import { _IDynamicConfigHandlerState } from "./_IDynamicConfigHandlerState"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { IDynamicConfigHandler } from "../interfaces/config/IDynamicConfigHandler"; +import { _IDynamicConfigHandlerState } from "../interfaces/config/_IDynamicConfigHandlerState"; // Using Symbol.for so that if the same symbol was already created it would be returned // To handle multiple instances using potentially different versions we are not using @@ -55,7 +55,8 @@ export function _cfgDeepCopy(source: T): T { /** * @internal * Get the dynamic config handler if the value is already dynamic - * @returns + * @param value - The value to check for dynamic config handler + * @returns The dynamic config handler if present, null otherwise */ export function getDynamicConfigHandler(value: V | IDynamicConfigHandler): IDynamicConfigHandler | null { if (value) { diff --git a/shared/AppInsightsCommon/src/Constants.ts b/shared/AppInsightsCore/src/constants/Constants.ts similarity index 95% rename from shared/AppInsightsCommon/src/Constants.ts rename to shared/AppInsightsCore/src/constants/Constants.ts index fefde7f7b..f78f707bc 100644 --- a/shared/AppInsightsCommon/src/Constants.ts +++ b/shared/AppInsightsCore/src/constants/Constants.ts @@ -9,6 +9,9 @@ * @ignore */ export const DisabledPropertyName: string = "Microsoft_ApplicationInsights_BypassAjaxInstrumentation"; + +export const ChannelControllerPriority = 500; + export const SampleRate = "sampleRate"; export const ProcessLegacy = "ProcessLegacy"; export const HttpMethod = "http.method"; @@ -16,3 +19,4 @@ export const DEFAULT_BREEZE_ENDPOINT = "https://dc.services.visualstudio.com"; export const DEFAULT_BREEZE_PATH = "/v2/track"; export const strNotSpecified = "not_specified"; export const strIkey = "iKey"; + diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/InternalConstants.ts b/shared/AppInsightsCore/src/constants/InternalConstants.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK/InternalConstants.ts rename to shared/AppInsightsCore/src/constants/InternalConstants.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/AggregationError.ts b/shared/AppInsightsCore/src/core/AggregationError.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK/AggregationError.ts rename to shared/AppInsightsCore/src/core/AggregationError.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/AppInsightsCore.ts b/shared/AppInsightsCore/src/core/AppInsightsCore.ts similarity index 79% rename from shared/AppInsightsCore/src/JavaScriptSDK/AppInsightsCore.ts rename to shared/AppInsightsCore/src/core/AppInsightsCore.ts index 1650c97e2..171574e46 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/AppInsightsCore.ts +++ b/shared/AppInsightsCore/src/core/AppInsightsCore.ts @@ -4,65 +4,78 @@ import dynamicProto from "@microsoft/dynamicproto-js"; import { IPromise, createPromise, createSyncAllSettledPromise, doAwaitResponse } from "@nevware21/ts-async"; import { - ITimerHandler, arrAppend, arrForEach, arrIndexOf, createTimeout, deepExtend, hasDocument, isFunction, isNullOrUndefined, isPlainObject, - isPromiseLike, objDeepFreeze, objDefine, objForEachKey, objFreeze, objHasOwn, scheduleTimeout, throwError + ICachedValue, ITimerHandler, arrAppend, arrForEach, arrIndexOf, createTimeout, deepExtend, hasDocument, isFunction, isNullOrUndefined, + isPlainObject, isPromiseLike, objAssign, objDeepFreeze, objDefine, objForEachKey, objFreeze, objHasOwn, scheduleTimeout, throwError } from "@nevware21/ts-utils"; -import { createDynamicConfig, onConfigChange } from "../Config/DynamicConfig"; -import { IConfigDefaults } from "../Config/IConfigDefaults"; -import { IDynamicConfigHandler, _IInternalDynamicConfigHandler } from "../Config/IDynamicConfigHandler"; -import { IWatchDetails, WatcherFunction } from "../Config/IDynamicWatcher"; -import { eEventsDiscardedReason } from "../JavaScriptSDK.Enums/EventsDiscardedReason"; -import { ActiveStatus, eActiveStatus } from "../JavaScriptSDK.Enums/InitActiveStatusEnum"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { SendRequestReason } from "../JavaScriptSDK.Enums/SendRequestReason"; -import { TelemetryUnloadReason } from "../JavaScriptSDK.Enums/TelemetryUnloadReason"; -import { TelemetryUpdateReason } from "../JavaScriptSDK.Enums/TelemetryUpdateReason"; -import { IAppInsightsCore, ILoadedPlugin } from "../JavaScriptSDK.Interfaces/IAppInsightsCore"; -import { IChannelControls } from "../JavaScriptSDK.Interfaces/IChannelControls"; -import { IChannelControlsHost } from "../JavaScriptSDK.Interfaces/IChannelControlsHost"; -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { ICookieMgr } from "../JavaScriptSDK.Interfaces/ICookieMgr"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; -import { IDistributedTraceContext } from "../JavaScriptSDK.Interfaces/IDistributedTraceContext"; -import { INotificationListener } from "../JavaScriptSDK.Interfaces/INotificationListener"; -import { INotificationManager } from "../JavaScriptSDK.Interfaces/INotificationManager"; -import { IPerfManager } from "../JavaScriptSDK.Interfaces/IPerfManager"; -import { IProcessTelemetryContext, IProcessTelemetryUpdateContext } from "../JavaScriptSDK.Interfaces/IProcessTelemetryContext"; -import { ITelemetryInitializerHandler, TelemetryInitializerFunction } from "../JavaScriptSDK.Interfaces/ITelemetryInitializers"; -import { ITelemetryItem } from "../JavaScriptSDK.Interfaces/ITelemetryItem"; -import { IPlugin, ITelemetryPlugin } from "../JavaScriptSDK.Interfaces/ITelemetryPlugin"; -import { ITelemetryPluginChain } from "../JavaScriptSDK.Interfaces/ITelemetryPluginChain"; -import { ITelemetryUnloadState } from "../JavaScriptSDK.Interfaces/ITelemetryUnloadState"; -import { ITelemetryUpdateState } from "../JavaScriptSDK.Interfaces/ITelemetryUpdateState"; -import { ILegacyUnloadHook, IUnloadHook } from "../JavaScriptSDK.Interfaces/IUnloadHook"; +import { cfgDfMerge } from "../config/ConfigDefaultHelpers"; +import { createDynamicConfig, onConfigChange } from "../config/DynamicConfig"; +import { ChannelControllerPriority } from "../constants/Constants"; +import { + STR_CHANNELS, STR_CREATE_PERF_MGR, STR_DISABLED, STR_EMPTY, STR_EXTENSIONS, STR_EXTENSION_CONFIG, UNDEFINED_VALUE +} from "../constants/InternalConstants"; +import { DiagnosticLogger, _InternalLogMessage, _throwInternal, _warnToConsole } from "../diagnostics/DiagnosticLogger"; +import { eEventsDiscardedReason } from "../enums/ai/EventsDiscardedReason"; +import { ActiveStatus, eActiveStatus } from "../enums/ai/InitActiveStatusEnum"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { SendRequestReason } from "../enums/ai/SendRequestReason"; +import { TelemetryUnloadReason } from "../enums/ai/TelemetryUnloadReason"; +import { TelemetryUpdateReason } from "../enums/ai/TelemetryUpdateReason"; +import { eTraceHeadersMode } from "../enums/ai/TraceHeadersMode"; +import { IAppInsightsCore, ILoadedPlugin } from "../interfaces/ai/IAppInsightsCore"; +import { IChannelControls } from "../interfaces/ai/IChannelControls"; +import { IChannelControlsHost } from "../interfaces/ai/IChannelControlsHost"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { ICookieMgr } from "../interfaces/ai/ICookieMgr"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { IDistributedTraceContext } from "../interfaces/ai/IDistributedTraceContext"; +import { INotificationListener } from "../interfaces/ai/INotificationListener"; +import { INotificationManager } from "../interfaces/ai/INotificationManager"; +import { IPerfManager } from "../interfaces/ai/IPerfManager"; +import { IProcessTelemetryContext, IProcessTelemetryUpdateContext } from "../interfaces/ai/IProcessTelemetryContext"; +import { ITelemetryInitializerHandler, TelemetryInitializerFunction } from "../interfaces/ai/ITelemetryInitializers"; +import { ITelemetryItem } from "../interfaces/ai/ITelemetryItem"; +import { IPlugin, ITelemetryPlugin } from "../interfaces/ai/ITelemetryPlugin"; +import { ITelemetryPluginChain } from "../interfaces/ai/ITelemetryPluginChain"; +import { ITelemetryUnloadState } from "../interfaces/ai/ITelemetryUnloadState"; +import { ITelemetryUpdateState } from "../interfaces/ai/ITelemetryUpdateState"; +import { ISpanScope, ITraceProvider } from "../interfaces/ai/ITraceProvider"; +import { ILegacyUnloadHook, IUnloadHook } from "../interfaces/ai/IUnloadHook"; +import { IConfigDefaults } from "../interfaces/config/IConfigDefaults"; +import { IDynamicConfigHandler, _IInternalDynamicConfigHandler } from "../interfaces/config/IDynamicConfigHandler"; +import { IWatchDetails, WatcherFunction } from "../interfaces/config/IDynamicWatcher"; +import { ITraceCfg } from "../interfaces/otel/config/IOTelTraceCfg"; +import { IOTelSpanContext } from "../interfaces/otel/trace/IOTelSpanContext"; +import { IOTelSpanOptions } from "../interfaces/otel/trace/IOTelSpanOptions"; +import { IReadableSpan } from "../interfaces/otel/trace/IReadableSpan"; +import { _noopVoid } from "../internal/noopHelpers"; +import { findW3cTraceState } from "../telemetry/W3cTraceState"; +import { createUniqueNamespace } from "../utils/DataCacheHelper"; +import { getSetValue, isNotNullOrUndefined, proxyFunctionAs, proxyFunctions, toISOString } from "../utils/HelperFuncs"; +import { findW3cTraceParent } from "../utils/TraceParent"; import { doUnloadAll, runTargetUnload } from "./AsyncUtils"; -import { ChannelControllerPriority } from "./Constants"; import { createCookieMgr } from "./CookieMgr"; -import { createUniqueNamespace } from "./DataCacheHelper"; import { getDebugListener } from "./DbgExtensionUtils"; -import { DiagnosticLogger, _InternalLogMessage, _throwInternal, _warnToConsole } from "./DiagnosticLogger"; -import { getSetValue, isNotNullOrUndefined, proxyFunctionAs, proxyFunctions, toISOString } from "./HelperFuncs"; -import { - STR_CHANNELS, STR_CREATE_PERF_MGR, STR_DISABLED, STR_EMPTY, STR_EXTENSIONS, STR_EXTENSION_CONFIG, UNDEFINED_VALUE -} from "./InternalConstants"; import { NotificationManager } from "./NotificationManager"; import { PerfManager, doPerf, getGblPerfMgr } from "./PerfManager"; import { createProcessTelemetryContext, createProcessTelemetryUnloadContext, createProcessTelemetryUpdateContext, createTelemetryProxyChain } from "./ProcessTelemetryContext"; -import { _getPluginState, createDistributedTraceContext, initializePlugins, sortPlugins } from "./TelemetryHelpers"; +import { + _getPluginState, createDistributedTraceContext, initializePlugins, isDistributedTraceContext, sortPlugins +} from "./TelemetryHelpers"; import { TelemetryInitializerPlugin } from "./TelemetryInitializerPlugin"; import { IUnloadHandlerContainer, UnloadHandler, createUnloadHandlerContainer } from "./UnloadHandlerContainer"; import { IUnloadHookContainer, createUnloadHookContainer } from "./UnloadHookContainer"; -// import { IStatsBeat, IStatsBeatConfig, IStatsBeatState } from "../JavaScriptSDK.Interfaces/IStatsBeat"; -// import { IStatsMgr } from "../JavaScriptSDK.Interfaces/IStatsMgr"; +// import { IStatsBeat, IStatsBeatConfig, IStatsBeatState } from "../interfaces/ai/IStatsBeat"; +// import { IStatsMgr } from "../interfaces/ai/IStatsMgr"; const strValidationError = "Plugins must provide initialize method"; const strNotificationManager = "_notificationManager"; const strSdkUnloadingError = "SDK is still unloading..."; const strSdkNotInitialized = "SDK is not initialized"; const maxInitQueueSize = 100; const maxInitTimeout = 50000; +const maxAttributeCount = 128; // const strPluginUnloadFailed = "Failed to unload plugin"; // /** @@ -90,23 +103,76 @@ const maxInitTimeout = 50000; const defaultConfig: IConfigDefaults = objDeepFreeze({ cookieCfg: {}, [STR_EXTENSIONS]: { rdOnly: true, ref: true, v: [] }, - [STR_CHANNELS]: { rdOnly: true, ref: true, v:[] }, + [STR_CHANNELS]: { rdOnly: true, ref: true, v: [] }, [STR_EXTENSION_CONFIG]: { ref: true, v: {} }, [STR_CREATE_PERF_MGR]: UNDEFINED_VALUE, loggingLevelConsole: eLoggingSeverity.DISABLED, - diagnosticLogInterval: UNDEFINED_VALUE + diagnosticLogInterval: UNDEFINED_VALUE, + traceHdrMode: eTraceHeadersMode.All, + traceCfg: cfgDfMerge({ + generalLimits: cfgDfMerge({ + attributeValueLengthLimit: undefined, + attributeCountLimit: maxAttributeCount + }), + // spanLimits: cfgDfMerge({ + // attributeValueLengthLimit: undefined, + // attributeCountLimit: maxAttributeCount, + // linkCountLimit: maxAttributeCount, + // eventCountLimit: maxAttributeCount, + // attributePerEventCountLimit: maxAttributeCount, + // attributePerLinkCountLimit: maxAttributeCount + // }), + // idGenerator: null, + serviceName: null, + suppressTracing: false + }) // _sdk: { rdOnly: true, ref: true, v: defaultSdkConfig } }); +function _getDefaultConfig(core: IAppInsightsCore): IConfigDefaults { + let handlers = { + // Dynamic Default Error Handlers + errorHandlers: cfgDfMerge({ + attribError: (message: string, key: string, value: any) => { + core.logger.throwInternal(eLoggingSeverity.WARNING, _eInternalMessageId.AttributeError, message, { + attribName: key, + value: value + }); + }, + spanError: (message: string, spanName: string) => { + core.logger.throwInternal(eLoggingSeverity.WARNING, _eInternalMessageId.SpanError, message, { + spanName: spanName + }); + }, + debug: (message: string) => { + core.logger.debugToConsole(message); + }, + warn: (message: string) => { + core.logger.warnToConsole(message) + }, + error: (message: string) => { + core.logger.throwInternal(eLoggingSeverity.CRITICAL, _eInternalMessageId.TraceError, message); + }, + notImplemented: (message: string) => { + core.logger.throwInternal(eLoggingSeverity.CRITICAL, _eInternalMessageId.NotImplementedError, message); + } + }) + }; + + return objDeepFreeze(objAssign({}, defaultConfig as any, handlers)); +} + /** * Helper to create the default performance manager * @param core - The AppInsightsCore instance * @param notificationMgr - The notification manager */ +/*#__NO_SIDE_EFFECTS__*/ function _createPerfManager (core: IAppInsightsCore, notificationMgr: INotificationManager) { return new PerfManager(notificationMgr); } +/*#__NO_SIDE_EFFECTS__*/ function _validateExtensions(logger: IDiagnosticLogger, channelPriority: number, allExtensions: IPlugin[]): { core: IPlugin[], channels: IChannelControls[] } { // Concat all available extensions let coreExtensions: ITelemetryPlugin[] = []; @@ -114,7 +180,7 @@ function _validateExtensions(logger: IDiagnosticLogger, channelPriority: number, // Check if any two extensions have the same priority, then warn to console // And extract the local extensions from the - let extPriorities = {}; + let extPriorities: any = {}; // Extension validation arrForEach(allExtensions, (ext: ITelemetryPlugin) => { @@ -150,6 +216,7 @@ function _validateExtensions(logger: IDiagnosticLogger, channelPriority: number, }; } +/*#__NO_SIDE_EFFECTS__*/ function _isPluginPresent(thePlugin: IPlugin, plugins: IPlugin[]) { let exists = false; @@ -173,7 +240,7 @@ function _deepMergeConfig(details: IWatchDetails, target: any, n _deepMergeConfig(details, target[key], value, merge); } } - + if (merge && isPlainObject(value) && isPlainObject(target[key])) { // The target is an object and it has a value _deepMergeConfig(details, target[key], value, merge); @@ -185,8 +252,9 @@ function _deepMergeConfig(details: IWatchDetails, target: any, n } } -function _findWatcher(listeners: { rm: () => void, w: WatcherFunction}[], newWatcher: WatcherFunction) { - let theListener: { rm: () => void, w: WatcherFunction} = null; +/*#__NO_SIDE_EFFECTS__*/ +function _findWatcher(listeners: { rm: () => void, w: WatcherFunction }[], newWatcher: WatcherFunction) { + let theListener: { rm: () => void, w: WatcherFunction } = null; let idx: number = -1; arrForEach(listeners, (listener, lp) => { if (listener.w === newWatcher) { @@ -199,7 +267,7 @@ function _findWatcher(listeners: { rm: () => void, w: WatcherFunction void, w: WatcherFunction}[], newWatcher: WatcherFunction) { +function _addDelayedCfgListener(listeners: { rm: () => void, w: WatcherFunction }[], newWatcher: WatcherFunction) { let theListener = _findWatcher(listeners, newWatcher).l; if (!theListener) { @@ -218,7 +286,7 @@ function _addDelayedCfgListener(listeners: { rm: () => void, w: WatcherFunction< return theListener; } -function _registerDelayedCfgListener(config: IConfiguration, listeners: { rm: () => void, w: WatcherFunction}[], logger: IDiagnosticLogger) { +function _registerDelayedCfgListener(config: IConfiguration, listeners: { rm: () => void, w: WatcherFunction }[], logger: IDiagnosticLogger) { arrForEach(listeners, (listener) => { let unloadHdl = onConfigChange(config, listener.w, logger); delete listener.w; // Clear the listener reference so it will get garbage collected. @@ -251,6 +319,7 @@ function _initDebugListener(configHandler: IDynamicConfigHandler } // Moved this outside of the closure to reduce the retained memory footprint +/*#__NO_SIDE_EFFECTS__*/ function _createUnloadHook(unloadHook: IUnloadHook): IUnloadHook { return objDefine({ rm: () => { @@ -259,6 +328,25 @@ function _createUnloadHook(unloadHook: IUnloadHook): IUnloadHook { }, "toJSON", { v: () => "aicore::onCfgChange<" + JSON.stringify(unloadHook) + ">" }); } +/*#__NO_SIDE_EFFECTS__*/ +function _getParentTraceCtx(mode: eTraceHeadersMode): IDistributedTraceContext | null { + let spanContext: IDistributedTraceContext | null = null; + const parentTrace = (mode & eTraceHeadersMode.TraceParent) ? findW3cTraceParent() : null; + const parentTraceState = (mode & eTraceHeadersMode.TraceState) ? findW3cTraceState() : null; + + if (parentTrace || parentTraceState) { + spanContext = createDistributedTraceContext({ + traceId: parentTrace ? parentTrace.traceId : null, + spanId: parentTrace ? parentTrace.spanId : null, + traceFlags: parentTrace ? parentTrace.traceFlags : UNDEFINED_VALUE, + isRemote: true, // Mark as remote since it's from an external source + traceState: parentTraceState + }); + } + + return spanContext; +} + /** * @group Classes * @group Entrypoint @@ -305,14 +393,18 @@ export class AppInsightsCore im let _channels: IChannelControls[] | null; let _isUnloading: boolean; let _telemetryInitializerPlugin: TelemetryInitializerPlugin; + let _serverOTelCtx: IDistributedTraceContext | null; + let _serverTraceHdrMode: eTraceHeadersMode; let _internalLogsEventName: string | null; let _evtNamespace: string; let _unloadHandlers: IUnloadHandlerContainer; let _hookContainer: IUnloadHookContainer; let _debugListener: INotificationListener | null; let _traceCtx: IDistributedTraceContext | null; + let _traceProvider: ICachedValue | null; + let _activeSpan: IReadableSpan | null; let _instrumentationKey: string | null; - let _cfgListeners: { rm: () => void, w: WatcherFunction}[]; + let _cfgListeners: { rm: () => void, w: WatcherFunction }[]; let _extensions: IPlugin[]; let _pluginVersionStringArr: string[]; let _pluginVersionString: string; @@ -321,7 +413,7 @@ export class AppInsightsCore im let _initInMemoMaxSize: number; // max event count limit during wait for init promises to be resolved let _isStatusSet: boolean; // track if active status is set in case of init timeout and init promises setting the status twice let _initTimer: ITimerHandler; - + /** * Internal log poller */ @@ -335,7 +427,7 @@ export class AppInsightsCore im _initDefaults(); // Special internal method to allow the unit tests and DebugPlugin to hook embedded objects - _self["_getDbgPlgTargets"] = () => { + (_self as any)["_getDbgPlgTargets"] = () => { return [_extensions, _eventQueue]; }; @@ -355,13 +447,13 @@ export class AppInsightsCore im if (_isUnloading) { throwError(strSdkUnloadingError); } - + // Make sure core is only initialized once if (_self.isInitialized()) { throwError("Core cannot be initialized more than once"); } - _configHandler = createDynamicConfig(config, defaultConfig as any, logger || _self.logger, false); + _configHandler = createDynamicConfig(config, _getDefaultConfig(_self), logger || _self.logger, false); // Re-assigning the local config property so we don't have any references to the passed value and it can be garbage collected config = _configHandler.cfg; @@ -381,6 +473,12 @@ export class AppInsightsCore im objForEachKey(extCfg, (key) => { details.ref(extCfg, key); }); + + if (rootCfg.traceHdrMode !== _serverTraceHdrMode) { + // Create a new trace context if it doesn't exist and the mode is not None + _serverOTelCtx = _getParentTraceCtx(rootCfg.traceHdrMode); + _serverTraceHdrMode = rootCfg.traceHdrMode; + } })); _notificationManager = notificationManager; @@ -475,7 +573,7 @@ export class AppInsightsCore im if (!_notificationManager) { _notificationManager = new NotificationManager(_configHandler.cfg); // For backward compatibility only - _self[strNotificationManager] = _notificationManager; + (_self as any)[strNotificationManager] = _notificationManager; } return _notificationManager; @@ -958,8 +1056,9 @@ export class AppInsightsCore im _self.flush = _flushChannels; _self.getTraceCtx = (createNew?: boolean): IDistributedTraceContext | null => { - if (!_traceCtx) { - _traceCtx = createDistributedTraceContext(); + + if ((!_traceCtx && createNew !== false) || createNew === true) { + _traceCtx = createDistributedTraceContext(_serverOTelCtx); } return _traceCtx; @@ -969,6 +1068,132 @@ export class AppInsightsCore im _traceCtx = traceCtx || null; }; + _self.startSpan = (name: string, options?: IOTelSpanOptions, parent?: IDistributedTraceContext): IReadableSpan | null => { + if (!_traceProvider || !_traceProvider.v || !_traceProvider.v.isAvailable()) { + // No trace provider available or provider is not ready + return null; + } + + return _traceProvider.v.createSpan(name, options, parent || _self.getTraceCtx()); + }; + + /** + * Return the current active span + * @param createNew - Optional flag to create a non-recording span if no active span exists, defaults to true + */ + _self.getActiveSpan = (createNew?: boolean): IReadableSpan | null => { + // Special case for when there is no active span but there is a trace provider + if (createNew !== false && _traceProvider && !_activeSpan && _traceProvider.v) { + // Now that we have a trace provider, ensure that we return an active span (non-recording) + _activeSpan = _traceProvider.v.createSpan(_traceProvider.v.getProviderId(), { + recording: false, + root: true + }); + } + + return _activeSpan; + }; + + /** + * Set the current Active Span + * @param span - The span to set as the active span + */ + _self.setActiveSpan = (span: IReadableSpan): ISpanScope => { + let theSpanContext: IDistributedTraceContext | null = null; + let currentCtx: IDistributedTraceContext = _self.getTraceCtx(); + let currentSpan: IReadableSpan | null = _activeSpan; + let scope: ISpanScope; + + if (span) { + let otelSpanContext: IDistributedTraceContext | IOTelSpanContext = null; + if (span.spanContext) { + // May be a valid IDistributedTraceContext or an OpenTelemetry SpanContext + otelSpanContext = span.spanContext(); + } else if ((span as any).context) { + // Legacy OpenTelemetry API support (Note: The returned context won't be a valid IDistributedTraceContext) + otelSpanContext = (span as any).context(); + } + + if (otelSpanContext) { + if (isDistributedTraceContext(otelSpanContext)) { + theSpanContext = otelSpanContext; + } else { + // Support Spans from other libraries that may not be using the IDistributedTraceContext + // If the spanContext is not a valid IDistributedTraceContext then we need to create a new one + // and optionally set the parentSpanContext if it exists + + // Create a new context using the current trace context as the parent + theSpanContext = createDistributedTraceContext(currentCtx); + + let parentContext: any = span.parentSpanContext; + if (!parentContext) { + if (span.parentSpanId) { + parentContext = { + traceId: (otelSpanContext as any).traceId, + spanId: span.parentSpanId + }; + } + } + + // Was there a defined parent context and is it different from the current basic context + if (parentContext && parentContext.traceId !== theSpanContext.traceId && + parentContext.spanId !== theSpanContext.spanId && + parentContext.traceFlags !== theSpanContext.traceFlags) { + + // Assign the parent details to this new context + theSpanContext.traceId = parentContext.traceId; + theSpanContext.spanId = parentContext.spanId; + theSpanContext.traceFlags = parentContext.traceFlags; + + // Now create a new "Child" context which is extending the parent context + theSpanContext = createDistributedTraceContext(theSpanContext); + } + + theSpanContext.traceId = (otelSpanContext as any).traceId; + theSpanContext.spanId = (otelSpanContext as any).spanId; + theSpanContext.traceFlags = (otelSpanContext as any).traceFlags; + } + } + } + + scope = { + host: _self, + span: span, + prvSpan: currentSpan, + restore: () => { + // Restore the current span and trace context + if (currentSpan) { + _self.setActiveSpan(currentSpan); + } else { + _activeSpan = null; + _self.setTraceCtx(currentCtx); + } + + // Clear the restore function, so that multiple calls to restore do not have any effect + scope.restore = _noopVoid; + } + }; + + // Change the active span to the new span + _activeSpan = span; + + // Set the current trace context for the core SDK + // This is REQUIRED for the SDK to correctly associate telemetry with the current span context + if (theSpanContext) { + _self.setTraceCtx(theSpanContext); + } + + return scope; + }; + + _self.setTraceProvider = (traceProvider: ICachedValue): void => { + _traceProvider = traceProvider; + }; + + _self.getTraceProvider = (): ITraceProvider | null => { + return _traceProvider ? _traceProvider.v : null; + }; + _self.addUnloadHook = _addUnloadHook; // Create the addUnloadCb @@ -1083,6 +1308,8 @@ export class AppInsightsCore im arrAppend(cfgExtensions, _extensions); _telemetryInitializerPlugin = new TelemetryInitializerPlugin(); + _serverOTelCtx = null; + _serverTraceHdrMode = eTraceHeadersMode.None; _eventQueue = []; runTargetUnload(_notificationManager, false); _notificationManager = null; @@ -1101,6 +1328,7 @@ export class AppInsightsCore im _evtNamespace = createUniqueNamespace("AIBaseCore", true); _unloadHandlers = createUnloadHandlerContainer(); _traceCtx = null; + _traceProvider = null; _instrumentationKey = null; _hookContainer = createUnloadHookContainer(); _cfgListeners = []; @@ -1620,15 +1848,24 @@ export class AppInsightsCore im * @param callBack - if specified, notify caller when send is complete, the channel should return true to indicate to the caller that it will be called. * If the caller doesn't return true the caller should assume that it may never be called. * @param sendReason - specify the reason that you are calling "flush" defaults to ManualFlush (1) if not specified - * @returns - true if the callback will be return after the flush is complete otherwise the caller should assume that any provided callback will never be called + * @returns true if the callback will be return after the flush is complete otherwise the caller should assume that any provided callback will never be called */ public flush(isAsync?: boolean, callBack?: (flushComplete?: boolean) => void, sendReason?: SendRequestReason): void { // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging } /** - * Gets the current distributed trace context for this instance if available - * @param createNew - Optional flag to create a new instance if one doesn't currently exist, defaults to true + * Gets the current distributed trace context for this instance if available, you can optional + * create a new instance if one does not currently exist or return null if one does not currently exist. + * When a server context is available it will be used as the parent context for the any new instance created + * (when createNew is true or no instance currently exists), calling this function will not + * change the current distributed trace context, it will only return the current context + * or create a new instance if one does not currently exist. + * @param createNew - Optional flag to create a new instance if one doesn't currently exist, defaults to + * undefined which will only create a new instance if one does not currently exist. + * If set to `false` then it will return null if no distributed trace context is available. + * If set to `true` then a new instance will be created even if one already exists. + * @param createNew - Optional flag to create a new instance if one doesn't currently exist, defaults to */ public getTraceCtx(createNew?: boolean): IDistributedTraceContext | null { // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging @@ -1642,6 +1879,71 @@ export class AppInsightsCore im // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging } + /** + * Start a new span with the given name and optional parent context. + * The span will become the active span for its duration unless a different + * span is explicitly set as active. + * + * @param name - The name of the span + * @param options - Options for creating the span (kind, attributes, startTime) + * @param parent - Optional parent context. If not provided, uses the current active trace context + * @returns A new span instance, or null if no trace provider is available + * @since 3.4.0 + */ + public startSpan(name: string, options?: IOTelSpanOptions, parent?: IDistributedTraceContext): IReadableSpan | null { + // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging + return null; + } + + /** + * Return the current active span, if no trace provider is available null will be returned + * but when a trace provider is available a span instance will always be returned, even if + * there is no active span (in which case a non-recording span will be returned). + * @param createNew - Optional flag to create a non-recording span if no active span exists, defaults to true. + * When false, returns the existing active span or null without creating a non-recording span. + * @returns The current active span or null if no trace provider is available or if createNew is false and no active span exists + * @since 3.4.0 + */ + public getActiveSpan(createNew?: boolean): IReadableSpan | null { + // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging + return null; + } + + /** + * Set the current Active Span + * @param span - The span to set as the active span + * @returns An ISpanScope instance that provides the current scope, the span will always be the span passed in + * even when no trace provider is available + * @since 3.4.0 + */ + public setActiveSpan(span: IReadableSpan): ISpanScope { + // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging + return null; + } + + /** + * Set the trace provider for creating spans. + * This allows different SKUs to provide their own span implementations. + * + * @param provider - The trace provider to use for span creation, it is passed as a cached value so that it may + * be implemented via a lazy / deferred initializer. + * @since 3.4.0 + */ + public setTraceProvider(provider: ICachedValue): void { + // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging + } + + /** + * Get the current trace provider. + * + * @returns The current trace provider, or null if none is set + * @since 3.4.0 + */ + public getTraceProvider(): ITraceProvider | null { + // @DynamicProtoStub -- DO NOT add any code as this will be removed during packaging + return null; + } + /** * Add this hook so that it is automatically removed during unloading * @param hooks - The single hook or an array of IInstrumentHook objects diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/AsyncUtils.ts b/shared/AppInsightsCore/src/core/AsyncUtils.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK/AsyncUtils.ts rename to shared/AppInsightsCore/src/core/AsyncUtils.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/BaseTelemetryPlugin.ts b/shared/AppInsightsCore/src/core/BaseTelemetryPlugin.ts similarity index 92% rename from shared/AppInsightsCore/src/JavaScriptSDK/BaseTelemetryPlugin.ts rename to shared/AppInsightsCore/src/core/BaseTelemetryPlugin.ts index f13197054..73ae606ae 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/BaseTelemetryPlugin.ts +++ b/shared/AppInsightsCore/src/core/BaseTelemetryPlugin.ts @@ -4,25 +4,25 @@ import dynamicProto from "@microsoft/dynamicproto-js"; import { isFunction, objDefine } from "@nevware21/ts-utils"; -import { createDynamicConfig } from "../Config/DynamicConfig"; -import { IConfigDefaults } from "../Config/IConfigDefaults"; -import { TelemetryUnloadReason } from "../JavaScriptSDK.Enums/TelemetryUnloadReason"; -import { TelemetryUpdateReason } from "../JavaScriptSDK.Enums/TelemetryUpdateReason"; -import { IAppInsightsCore } from "../JavaScriptSDK.Interfaces/IAppInsightsCore"; -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; +import { createDynamicConfig } from "../config/DynamicConfig"; +import { STR_EXTENSION_CONFIG } from "../constants/InternalConstants"; +import { safeGetLogger } from "../diagnostics/DiagnosticLogger"; +import { TelemetryUnloadReason } from "../enums/ai/TelemetryUnloadReason"; +import { TelemetryUpdateReason } from "../enums/ai/TelemetryUpdateReason"; +import { IAppInsightsCore } from "../interfaces/ai/IAppInsightsCore"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; import { IProcessTelemetryContext, IProcessTelemetryUnloadContext, IProcessTelemetryUpdateContext -} from "../JavaScriptSDK.Interfaces/IProcessTelemetryContext"; -import { ITelemetryItem } from "../JavaScriptSDK.Interfaces/ITelemetryItem"; -import { IPlugin, ITelemetryPlugin } from "../JavaScriptSDK.Interfaces/ITelemetryPlugin"; -import { ITelemetryPluginChain } from "../JavaScriptSDK.Interfaces/ITelemetryPluginChain"; -import { ITelemetryUnloadState } from "../JavaScriptSDK.Interfaces/ITelemetryUnloadState"; -import { ITelemetryUpdateState } from "../JavaScriptSDK.Interfaces/ITelemetryUpdateState"; -import { ILegacyUnloadHook, IUnloadHook } from "../JavaScriptSDK.Interfaces/IUnloadHook"; -import { safeGetLogger } from "./DiagnosticLogger"; -import { isNotNullOrUndefined, proxyFunctionAs } from "./HelperFuncs"; -import { STR_EXTENSION_CONFIG } from "./InternalConstants"; +} from "../interfaces/ai/IProcessTelemetryContext"; +import { ITelemetryItem } from "../interfaces/ai/ITelemetryItem"; +import { IPlugin, ITelemetryPlugin } from "../interfaces/ai/ITelemetryPlugin"; +import { ITelemetryPluginChain } from "../interfaces/ai/ITelemetryPluginChain"; +import { ITelemetryUnloadState } from "../interfaces/ai/ITelemetryUnloadState"; +import { ITelemetryUpdateState } from "../interfaces/ai/ITelemetryUpdateState"; +import { ILegacyUnloadHook, IUnloadHook } from "../interfaces/ai/IUnloadHook"; +import { IConfigDefaults } from "../interfaces/config/IConfigDefaults"; +import { isNotNullOrUndefined, proxyFunctionAs } from "../utils/HelperFuncs"; import { createProcessTelemetryContext, createProcessTelemetryUnloadContext, createProcessTelemetryUpdateContext } from "./ProcessTelemetryContext"; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/CookieMgr.ts b/shared/AppInsightsCore/src/core/CookieMgr.ts similarity index 95% rename from shared/AppInsightsCore/src/JavaScriptSDK/CookieMgr.ts rename to shared/AppInsightsCore/src/core/CookieMgr.ts index b5f057723..1184a3c10 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/CookieMgr.ts +++ b/shared/AppInsightsCore/src/core/CookieMgr.ts @@ -2,22 +2,22 @@ // Licensed under the MIT License. import { IPromise } from "@nevware21/ts-async"; import { - ILazyValue, arrForEach, arrIndexOf, dumpObj, getDocument, getLazy, getNavigator, isArray, isFunction, isNullOrUndefined, isString, - isTruthy, isUndefined, objForEachKey, strEndsWith, strIndexOf, strLeft, strSubstring, strTrim, utcNow + ILazyValue, arrForEach, arrIndexOf, dumpObj, getDocument, getLazy, isArray, isFunction, isNullOrUndefined, isString, isTruthy, + isUndefined, objForEachKey, strEndsWith, strIndexOf, strLeft, strSubstring, strTrim, utcNow } from "@nevware21/ts-utils"; -import { cfgDfMerge } from "../Config/ConfigDefaultHelpers"; -import { createDynamicConfig, onConfigChange } from "../Config/DynamicConfig"; -import { IConfigDefaults } from "../Config/IConfigDefaults"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { IAppInsightsCore } from "../JavaScriptSDK.Interfaces/IAppInsightsCore"; -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { ICookieMgr, ICookieMgrConfig } from "../JavaScriptSDK.Interfaces/ICookieMgr"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; -import { IUnloadHook } from "../JavaScriptSDK.Interfaces/IUnloadHook"; -import { _throwInternal } from "./DiagnosticLogger"; -import { getLocation, isIE } from "./EnvUtils"; -import { getExceptionName, isNotNullOrUndefined, setValue, strContains } from "./HelperFuncs"; -import { STR_DOMAIN, STR_EMPTY, STR_PATH, UNDEFINED_VALUE } from "./InternalConstants"; +import { cfgDfMerge } from "../config/ConfigDefaultHelpers"; +import { createDynamicConfig, onConfigChange } from "../config/DynamicConfig"; +import { STR_DOMAIN, STR_EMPTY, STR_PATH, UNDEFINED_VALUE } from "../constants/InternalConstants"; +import { _throwInternal } from "../diagnostics/DiagnosticLogger"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { IAppInsightsCore } from "../interfaces/ai/IAppInsightsCore"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { ICookieMgr, ICookieMgrConfig } from "../interfaces/ai/ICookieMgr"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { IUnloadHook } from "../interfaces/ai/IUnloadHook"; +import { IConfigDefaults } from "../interfaces/config/IConfigDefaults"; +import { getLocation, getUserAgentString, isIE } from "../utils/EnvUtils"; +import { getExceptionName, isNotNullOrUndefined, setValue, strContains } from "../utils/HelperFuncs"; const strToGMTString = "toGMTString"; const strToUTCString = "toUTCString"; @@ -154,8 +154,8 @@ export function safeGetCookieMgr(core: IAppInsightsCore, config?: IConfiguration cookieMgr = core.getCookieMgr(); } else if (config) { let cookieCfg = config.cookieCfg; - if (cookieCfg && cookieCfg[strConfigCookieMgr]) { - cookieMgr = cookieCfg[strConfigCookieMgr]; + if (cookieCfg && (cookieCfg as any)[strConfigCookieMgr]) { + cookieMgr = (cookieCfg as any)[strConfigCookieMgr]; } else { cookieMgr = createCookieMgr(config); } @@ -242,7 +242,7 @@ export function createCookieMgr(rootConfig?: IConfiguration, logger?: IDiagnosti // Only set same site if not also secure if (_allowUaSameSite === null) { - _allowUaSameSite = !uaDisallowsSameSiteNone((getNavigator() || {} as Navigator).userAgent); + _allowUaSameSite = !uaDisallowsSameSiteNone(getUserAgentString()); } if (_allowUaSameSite) { diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/DbgExtensionUtils.ts b/shared/AppInsightsCore/src/core/DbgExtensionUtils.ts similarity index 84% rename from shared/AppInsightsCore/src/JavaScriptSDK/DbgExtensionUtils.ts rename to shared/AppInsightsCore/src/core/DbgExtensionUtils.ts index e900b32ea..e3de89323 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/DbgExtensionUtils.ts +++ b/shared/AppInsightsCore/src/core/DbgExtensionUtils.ts @@ -2,10 +2,10 @@ // Licensed under the MIT License. import { getInst } from "@nevware21/ts-utils"; -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { IDbgExtension } from "../JavaScriptSDK.Interfaces/IDbgExtension"; -import { INotificationListener } from "../JavaScriptSDK.Interfaces/INotificationListener"; -import { STR_EVENTS_DISCARDED, STR_EVENTS_SEND_REQUEST, STR_EVENTS_SENT, STR_PERF_EVENT } from "./InternalConstants"; +import { STR_EVENTS_DISCARDED, STR_EVENTS_SEND_REQUEST, STR_EVENTS_SENT, STR_PERF_EVENT } from "../constants/InternalConstants"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { IDbgExtension } from "../interfaces/ai/IDbgExtension"; +import { INotificationListener } from "../interfaces/ai/INotificationListener"; const listenerFuncs = [ STR_EVENTS_SENT, STR_EVENTS_DISCARDED, STR_EVENTS_SEND_REQUEST, STR_PERF_EVENT ]; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/InstrumentHooks.ts b/shared/AppInsightsCore/src/core/InstrumentHooks.ts similarity index 99% rename from shared/AppInsightsCore/src/JavaScriptSDK/InstrumentHooks.ts rename to shared/AppInsightsCore/src/core/InstrumentHooks.ts index ee050b64f..a14fcbde7 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/InstrumentHooks.ts +++ b/shared/AppInsightsCore/src/core/InstrumentHooks.ts @@ -5,8 +5,8 @@ import { strShimFunction, strShimPrototype } from "@microsoft/applicationinsight import { getInst, objHasOwnProperty } from "@nevware21/ts-utils"; import { IInstrumentCallDetails, IInstrumentHook, IInstrumentHooks, IInstrumentHooksCallbacks, InstrumentorHooksCallback -} from "../JavaScriptSDK.Interfaces/IInstrumentHooks"; -import { _getObjProto } from "./HelperFuncs"; +} from "../interfaces/ai/IInstrumentHooks"; +import { _getObjProto } from "../utils/HelperFuncs"; const aiInstrumentHooks = "_aiHooks"; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/NotificationManager.ts b/shared/AppInsightsCore/src/core/NotificationManager.ts similarity index 95% rename from shared/AppInsightsCore/src/JavaScriptSDK/NotificationManager.ts rename to shared/AppInsightsCore/src/core/NotificationManager.ts index dba9e53ce..d2a2834a5 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/NotificationManager.ts +++ b/shared/AppInsightsCore/src/core/NotificationManager.ts @@ -3,16 +3,17 @@ import dynamicProto from "@microsoft/dynamicproto-js"; import { IPromise, createAllPromise, createPromise, doAwaitResponse } from "@nevware21/ts-async"; import { ITimerHandler, arrForEach, arrIndexOf, objDefine, safe, scheduleTimeout } from "@nevware21/ts-utils"; -import { createDynamicConfig } from "../Config/DynamicConfig"; -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { INotificationListener } from "../JavaScriptSDK.Interfaces/INotificationListener"; -import { INotificationManager } from "../JavaScriptSDK.Interfaces/INotificationManager"; -import { IPerfEvent } from "../JavaScriptSDK.Interfaces/IPerfEvent"; -import { ITelemetryItem } from "../JavaScriptSDK.Interfaces/ITelemetryItem"; -import { IPayloadData, IUnloadHook } from "../applicationinsights-core-js"; +import { createDynamicConfig } from "../config/DynamicConfig"; import { STR_EVENTS_DISCARDED, STR_EVENTS_SEND_REQUEST, STR_EVENTS_SENT, STR_OFFLINE_DROP, STR_OFFLINE_SENT, STR_OFFLINE_STORE, STR_PERF_EVENT -} from "./InternalConstants"; +} from "../constants/InternalConstants"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { INotificationListener } from "../interfaces/ai/INotificationListener"; +import { INotificationManager } from "../interfaces/ai/INotificationManager"; +import { IPerfEvent } from "../interfaces/ai/IPerfEvent"; +import { ITelemetryItem } from "../interfaces/ai/ITelemetryItem"; +import { IUnloadHook } from "../interfaces/ai/IUnloadHook"; +import { IPayloadData } from "../interfaces/ai/IXHROverride"; const defaultValues = { perfEvtsSendAll: false diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/PerfManager.ts b/shared/AppInsightsCore/src/core/PerfManager.ts similarity index 96% rename from shared/AppInsightsCore/src/JavaScriptSDK/PerfManager.ts rename to shared/AppInsightsCore/src/core/PerfManager.ts index f04e8f18c..a934dd86f 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/PerfManager.ts +++ b/shared/AppInsightsCore/src/core/PerfManager.ts @@ -2,10 +2,11 @@ // Licensed under the MIT License. import dynamicProto from "@microsoft/dynamicproto-js"; import { isArray, isFunction, objDefine, utcNow } from "@nevware21/ts-utils"; -import { INotificationManager } from "../JavaScriptSDK.Interfaces/INotificationManager"; -import { IPerfEvent } from "../JavaScriptSDK.Interfaces/IPerfEvent"; -import { IPerfManager, IPerfManagerProvider } from "../JavaScriptSDK.Interfaces/IPerfManager"; -import { STR_GET_PERF_MGR } from "./InternalConstants"; +import { STR_GET_PERF_MGR } from "../constants/InternalConstants"; +import { INotificationManager } from "../interfaces/ai/INotificationManager"; +import { IPerfEvent } from "../interfaces/ai/IPerfEvent"; +import { IPerfManager, IPerfManagerProvider } from "../interfaces/ai/IPerfManager"; +import { _noopVoid } from "../internal/noopHelpers"; const strExecutionContextKey = "ctx"; const strParentContextKey = "ParentContextKey"; @@ -130,7 +131,7 @@ export class PerfEvent implements IPerfEvent { _self.time = utcNow() - _self.start; _self.exTime = _self.time - childTime; - _self.complete = () => {}; + _self.complete = _noopVoid; }; } } diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/ProcessTelemetryContext.ts b/shared/AppInsightsCore/src/core/ProcessTelemetryContext.ts similarity index 78% rename from shared/AppInsightsCore/src/JavaScriptSDK/ProcessTelemetryContext.ts rename to shared/AppInsightsCore/src/core/ProcessTelemetryContext.ts index d5b10999f..2a1b8f38e 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/ProcessTelemetryContext.ts +++ b/shared/AppInsightsCore/src/core/ProcessTelemetryContext.ts @@ -2,28 +2,25 @@ // Licensed under the MIT License. "use strict"; -import { - arrForEach, dumpObj, isArray, isFunction, isNullOrUndefined, isUndefined, objForEachKey, objFreeze, objKeys -} from "@nevware21/ts-utils"; -import { _applyDefaultValue } from "../Config/ConfigDefaults"; -import { createDynamicConfig } from "../Config/DynamicConfig"; -import { IConfigDefaults } from "../Config/IConfigDefaults"; -import { IDynamicConfigHandler } from "../Config/IDynamicConfigHandler"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { IAppInsightsCore } from "../JavaScriptSDK.Interfaces/IAppInsightsCore"; -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; +import { arrForEach, dumpObj, isArray, isFunction, isNullOrUndefined, isUndefined, objForEachKey, objFreeze } from "@nevware21/ts-utils"; +import { _applyDefaultValue } from "../config/ConfigDefaults"; +import { createDynamicConfig } from "../config/DynamicConfig"; +import { STR_CORE, STR_DISABLED, STR_EMPTY } from "../constants/InternalConstants"; +import { _throwInternal, safeGetLogger } from "../diagnostics/DiagnosticLogger"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { IAppInsightsCore } from "../interfaces/ai/IAppInsightsCore"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; import { IBaseProcessingContext, IProcessTelemetryContext, IProcessTelemetryUnloadContext, IProcessTelemetryUpdateContext -} from "../JavaScriptSDK.Interfaces/IProcessTelemetryContext"; -import { ITelemetryItem } from "../JavaScriptSDK.Interfaces/ITelemetryItem"; -import { IPlugin, ITelemetryPlugin } from "../JavaScriptSDK.Interfaces/ITelemetryPlugin"; -import { ITelemetryPluginChain } from "../JavaScriptSDK.Interfaces/ITelemetryPluginChain"; -import { ITelemetryUnloadState } from "../JavaScriptSDK.Interfaces/ITelemetryUnloadState"; -import { ITelemetryUpdateState } from "../JavaScriptSDK.Interfaces/ITelemetryUpdateState"; -import { _throwInternal, safeGetLogger } from "./DiagnosticLogger"; -import { proxyFunctions } from "./HelperFuncs"; -import { STR_CORE, STR_DISABLED, STR_EMPTY, STR_EXTENSION_CONFIG } from "./InternalConstants"; +} from "../interfaces/ai/IProcessTelemetryContext"; +import { ITelemetryItem } from "../interfaces/ai/ITelemetryItem"; +import { IPlugin, ITelemetryPlugin } from "../interfaces/ai/ITelemetryPlugin"; +import { ITelemetryPluginChain } from "../interfaces/ai/ITelemetryPluginChain"; +import { ITelemetryUnloadState } from "../interfaces/ai/ITelemetryUnloadState"; +import { ITelemetryUpdateState } from "../interfaces/ai/ITelemetryUpdateState"; +import { IConfigDefaults } from "../interfaces/config/IConfigDefaults"; +import { IDynamicConfigHandler } from "../interfaces/config/IDynamicConfigHandler"; +import { _noopVoid } from "../internal/noopHelpers"; import { doPerf } from "./PerfManager"; import { _getPluginState } from "./TelemetryHelpers"; @@ -33,6 +30,13 @@ const strGetTelCtx = "_getTelCtx"; let _chainId = 0; +/** + * Identifies the type for the `extensionConfig` property + * This is a key/value pair where the key is the name of the extension and the value is the configuration + * for that extension. + */ +type ExtensionConfig = { [key: string]: any }; + interface OnCompleteCallback { func: () => void; self: any; // This for the function @@ -155,37 +159,35 @@ function _createInternalContext(telemetryChain function _getExtCfg(identifier: string, createIfMissing: boolean) { let idCfg: T = null; - let cfg = dynamicHandler.cfg; + let extCfg: ExtensionConfig = _getCfg(dynamicHandler.cfg, "extensionConfig", createIfMissing); + + if (extCfg) { + idCfg = _getCfg(extCfg, identifier, createIfMissing); + } + + return idCfg; + } + + function _getCfg(cfg: Cfg, identifier: string, createIfMissing: boolean) { + let idCfg: T = null; if (cfg && identifier) { - let extCfg = cfg.extensionConfig; - if (!extCfg && createIfMissing) { - extCfg = {}; + idCfg = cfg[identifier] as T; + if (!idCfg && createIfMissing) { + idCfg = {} as T; } // Always set the value so that the property always exists - cfg[STR_EXTENSION_CONFIG] = extCfg; // Note: it is valid for the "value" to be undefined + (cfg as any)[identifier] = idCfg; // Note: it is valid for the "value" to be undefined // Calling `ref()` has a side effect of causing the referenced property to become dynamic (if not already) - extCfg = dynamicHandler.ref(cfg, STR_EXTENSION_CONFIG); - if (extCfg) { - idCfg = extCfg[identifier]; - if (!idCfg && createIfMissing) { - idCfg = {} as T; - } - - // Always set the value so that the property always exists - extCfg[identifier] = idCfg; // Note: it is valid for the "value" to be undefined - - // Calling `ref()` has a side effect of causing the referenced property to become dynamic (if not already) - idCfg = dynamicHandler.ref(extCfg, identifier); - } + idCfg = dynamicHandler.ref(cfg, identifier); } return idCfg; } - function _resolveExtCfg(identifier: string, defaultValues: IConfigDefaults): T { - let newConfig: T = _getExtCfg(identifier, true); + function _resolveExtCfg(identifier: string, defaultValues: IConfigDefaults, rootOnly?: boolean): T { + let newConfig: T = rootOnly ? _getCfg(dynamicHandler.cfg, identifier, true) : _getExtCfg(identifier, true); if (defaultValues) { // Enumerate over the defaultValues and if not already populated attempt to @@ -551,7 +553,7 @@ export function createTelemetryPluginProxy(plugin: ITelemetryPlugin, config: ICo return hasRun; } - if (!_processChain(unloadCtx, _callTeardown, "unload", () => {}, unloadState.isAsync)) { + if (!_processChain(unloadCtx, _callTeardown, "unload", _noopVoid, unloadState.isAsync)) { // Only called if we hasRun was not true unloadCtx.processNext(unloadState); } @@ -578,7 +580,7 @@ export function createTelemetryPluginProxy(plugin: ITelemetryPlugin, config: ICo return hasRun; } - if (!_processChain(updateCtx, _callUpdate, "update", () => {}, false)) { + if (!_processChain(updateCtx, _callUpdate, "update", _noopVoid, false)) { // Only called if we hasRun was not true updateCtx.processNext(updateState); } @@ -586,89 +588,3 @@ export function createTelemetryPluginProxy(plugin: ITelemetryPlugin, config: ICo return objFreeze(proxyChain); } - -/** - * This class will be removed! - * @deprecated use createProcessTelemetryContext() instead - */ -export class ProcessTelemetryContext implements IProcessTelemetryContext { - /** - * Gets the current core config instance - */ - public getCfg: () => IConfiguration; - - public getExtCfg: (identifier:string, defaultValue?: IConfigDefaults) => T; - - public getConfig: (identifier:string, field: string, defaultValue?: number | string | boolean | string[] | RegExp[] | Function) => number | string | boolean | string[] | RegExp[] | Function; - - /** - * Returns the IAppInsightsCore instance for the current request - */ - public core: () => IAppInsightsCore; - - /** - * Returns the current IDiagnosticsLogger for the current request - */ - public diagLog: () => IDiagnosticLogger; - - /** - * Helper to allow inherited classes to check and possibly shortcut executing code only - * required if there is a nextPlugin - */ - public hasNext: () => boolean; - - /** - * Returns the next configured plugin proxy - */ - public getNext: () => ITelemetryPluginChain; - - /** - * Helper to set the next plugin proxy - */ - public setNext: (nextCtx:ITelemetryPluginChain) => void; - - /** - * Call back for telemetry processing before it it is sent - * @param env - This is the current event being reported - * @param itemCtx - This is the context for the current request, ITelemetryPlugin instances - * can optionally use this to access the current core instance or define / pass additional information - * to later plugins (vs appending items to the telemetry item) - * @returns boolean (true) if there is no more plugins to process otherwise false or undefined (void) - */ - public processNext: (env: ITelemetryItem) => boolean | void; - - /** - * Synchronously iterate over the context chain running the callback for each plugin, once - * every plugin has been executed via the callback, any associated onComplete will be called. - * @param callback - The function call for each plugin in the context chain - */ - public iterate: (callback: (plugin: T) => void) => void; - - /** - * Create a new context using the core and config from the current instance - * @param plugins - The execution order to process the plugins, if null or not supplied - * then the current execution order will be copied. - * @param startAt - The plugin to start processing from, if missing from the execution - * order then the next plugin will be NOT set. - */ - public createNew: (plugins?:IPlugin[]|ITelemetryPluginChain, startAt?:IPlugin) => IProcessTelemetryContext; - - /** - * Set the function to call when the current chain has executed all processNext or unloadNext items. - */ - public onComplete: (onComplete: () => void) => void; - - /** - * Creates a new Telemetry Item context with the current config, core and plugin execution chain - * @param plugins - The plugin instances that will be executed - * @param config - The current config - * @param core - The current core instance - */ - constructor(pluginChain: ITelemetryPluginChain, config: IConfiguration, core: IAppInsightsCore, startAt?:IPlugin) { - let _self = this; - - let context = createProcessTelemetryContext(pluginChain, config, core, startAt); - // Proxy all functions of the context to this object - proxyFunctions(_self, context, objKeys(context) as any); - } -} diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/ResponseHelpers.ts b/shared/AppInsightsCore/src/core/ResponseHelpers.ts similarity index 68% rename from shared/AppInsightsCore/src/JavaScriptSDK/ResponseHelpers.ts rename to shared/AppInsightsCore/src/core/ResponseHelpers.ts index 46190deda..a81cce235 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/ResponseHelpers.ts +++ b/shared/AppInsightsCore/src/core/ResponseHelpers.ts @@ -2,11 +2,12 @@ // Licensed under the MIT License. import { dumpObj } from "@nevware21/ts-utils"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; -import { IBackendResponse } from "../JavaScriptSDK.Interfaces/IXDomainRequest"; -import { _throwInternal } from "./DiagnosticLogger"; -import { getJSON } from "./EnvUtils"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { _throwInternal } from "../diagnostics/DiagnosticLogger"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { IBackendResponse } from "../interfaces/ai/IXDomainRequest"; +import { getJSON } from "../utils/EnvUtils"; /** * Parses the response from the backend. @@ -14,7 +15,7 @@ import { getJSON } from "./EnvUtils"; */ export function parseResponse(response: any, diagLog?: IDiagnosticLogger): IBackendResponse { try { - if (response && response !== "") { + if (response && response !== STR_EMPTY) { const result = getJSON().parse(response); if (result && result.itemsReceived && result.itemsReceived >= result.itemsAccepted && diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/SenderPostManager.ts b/shared/AppInsightsCore/src/core/SenderPostManager.ts similarity index 96% rename from shared/AppInsightsCore/src/JavaScriptSDK/SenderPostManager.ts rename to shared/AppInsightsCore/src/core/SenderPostManager.ts index 1540eeb3c..43977e2ff 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/SenderPostManager.ts +++ b/shared/AppInsightsCore/src/core/SenderPostManager.ts @@ -4,22 +4,23 @@ import dynamicProto from "@microsoft/dynamicproto-js"; import { AwaitResponse, IPromise, createPromise, doAwaitResponse } from "@nevware21/ts-async"; import { arrForEach, dumpObj, getInst, getNavigator, getWindow, isFunction, isString, objKeys } from "@nevware21/ts-utils"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { SendRequestReason, TransportType } from "../JavaScriptSDK.Enums/SendRequestReason"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; -import { IProcessTelemetryUnloadContext } from "../JavaScriptSDK.Interfaces/IProcessTelemetryContext"; +import { DisabledPropertyName } from "../constants/Constants"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { _throwInternal, _warnToConsole } from "../diagnostics/DiagnosticLogger"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { SendRequestReason, TransportType } from "../enums/ai/SendRequestReason"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { IProcessTelemetryUnloadContext } from "../interfaces/ai/IProcessTelemetryContext"; import { _IInternalXhrOverride, _ISendPostMgrConfig, _ISenderOnComplete, _ITimeoutOverrideWrapper -} from "../JavaScriptSDK.Interfaces/ISenderPostManager"; -import { ITelemetryUnloadState } from "../JavaScriptSDK.Interfaces/ITelemetryUnloadState"; -import { IXDomainRequest } from "../JavaScriptSDK.Interfaces/IXDomainRequest"; -import { IPayloadData, IXHROverride, OnCompleteCallback, SendPOSTFunction } from "../JavaScriptSDK.Interfaces/IXHROverride"; -import { DisabledPropertyName } from "./Constants"; -import { _throwInternal, _warnToConsole } from "./DiagnosticLogger"; -import { getLocation, isBeaconsSupported, isFetchSupported, isXhrSupported, useXDomainRequest } from "./EnvUtils"; -import { _getAllResponseHeaders, formatErrorMessageXdr, formatErrorMessageXhr, getResponseText, openXhr } from "./HelperFuncs"; - -const STR_EMPTY = ""; +} from "../interfaces/ai/ISenderPostManager"; +import { ITelemetryUnloadState } from "../interfaces/ai/ITelemetryUnloadState"; +import { IXDomainRequest } from "../interfaces/ai/IXDomainRequest"; +import { IPayloadData, IXHROverride, OnCompleteCallback, SendPOSTFunction } from "../interfaces/ai/IXHROverride"; +import { _noopVoid } from "../internal/noopHelpers"; +import { getLocation, isBeaconsSupported, isFetchSupported, isXhrSupported, useXDomainRequest } from "../utils/EnvUtils"; +import { _getAllResponseHeaders, formatErrorMessageXdr, formatErrorMessageXhr, getResponseText, openXhr } from "../utils/HelperFuncs"; + const STR_NO_RESPONSE_BODY = "NoResponseBody"; const _noResponseQs = "&" + STR_NO_RESPONSE_BODY + "=true"; const STR_POST_METHOD = "POST"; @@ -34,6 +35,7 @@ declare var XDomainRequest: { * Manager SendPost functions * SendPostManger * @internal for internal use only + * @since 3.0.0 */ export class SenderPostManager { @@ -470,7 +472,7 @@ export class SenderPostManager { _syncFetchPayload += batchLength; if (_isOneDs) { - if (payload["_sendReason"] === SendRequestReason.Unload) { + if ((payload as any)["_sendReason"] === SendRequestReason.Unload) { // As a sync request (during unload), it is unlikely that we will get a chance to process the response so // just like beacon send assume that the events have been accepted and processed ignoreResponse = true; @@ -487,7 +489,7 @@ export class SenderPostManager { const request = new Request(endPointUrl, init); try { // Also try and tag the request (just in case the value in init is not copied over) - request[DisabledPropertyName] = true; + (request as any)[DisabledPropertyName] = true; } catch(e) { // If the environment has locked down the XMLHttpRequest (preventExtensions and/or freeze), this would // cause the request to fail and we no telemetry would be sent @@ -650,11 +652,11 @@ export class SenderPostManager { }; - xdr.onprogress = () => { }; + xdr.onprogress = _noopVoid; // XDomainRequest requires the same protocol as the hosting page. // If the protocol doesn't match, we can't send the telemetry :(. - const hostingProtocol = _window && _window.location && _window.location.protocol || ""; + const hostingProtocol = _window && _window.location && _window.location.protocol || STR_EMPTY; let endpoint = payload.urlString; if (!endpoint) { _onNoPayloadUrl(oncomplete); @@ -670,7 +672,7 @@ export class SenderPostManager { return; } - const endpointUrl = _isOneDs? endpoint : endpoint.replace(/^(https?:)/, ""); + const endpointUrl = _isOneDs? endpoint : endpoint.replace(/^(https?:)/, STR_EMPTY); xdr.open(STR_POST_METHOD, endpointUrl); if (payload.timeout) { xdr.timeout = payload.timeout; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/StatsBeat.ts b/shared/AppInsightsCore/src/core/StatsBeat.ts similarity index 94% rename from shared/AppInsightsCore/src/JavaScriptSDK/StatsBeat.ts rename to shared/AppInsightsCore/src/core/StatsBeat.ts index 1834903dc..ef0578302 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/StatsBeat.ts +++ b/shared/AppInsightsCore/src/core/StatsBeat.ts @@ -4,19 +4,19 @@ import { ITimerHandler, arrForEach, isNumber, makeGlobRegex, objDefineProps, scheduleTimeout, strIndexOf, strLower, utcNow } from "@nevware21/ts-utils"; -import { onConfigChange } from "../Config/DynamicConfig"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { eStatsType } from "../JavaScriptSDK.Enums/StatsType"; -import { IAppInsightsCore } from "../JavaScriptSDK.Interfaces/IAppInsightsCore"; -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { INetworkStatsbeat } from "../JavaScriptSDK.Interfaces/INetworkStatsbeat"; -import { IStatsBeat, IStatsBeatConfig, IStatsBeatState, IStatsEndpointConfig } from "../JavaScriptSDK.Interfaces/IStatsBeat"; -import { IStatsMgr, IStatsMgrConfig } from "../JavaScriptSDK.Interfaces/IStatsMgr"; -import { ITelemetryItem } from "../JavaScriptSDK.Interfaces/ITelemetryItem"; -import { IPayloadData } from "../JavaScriptSDK.Interfaces/IXHROverride"; -import { _throwInternal, safeGetLogger } from "./DiagnosticLogger"; -import { isFeatureEnabled } from "./HelperFuncs"; -import { STR_EMPTY } from "./InternalConstants"; +import { onConfigChange } from "../config/DynamicConfig"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { _throwInternal, safeGetLogger } from "../diagnostics/DiagnosticLogger"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { eStatsType } from "../enums/ai/StatsType"; +import { IAppInsightsCore } from "../interfaces/ai/IAppInsightsCore"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { INetworkStatsbeat } from "../interfaces/ai/INetworkStatsbeat"; +import { IStatsBeat, IStatsBeatConfig, IStatsBeatState, IStatsEndpointConfig } from "../interfaces/ai/IStatsBeat"; +import { IStatsMgr, IStatsMgrConfig } from "../interfaces/ai/IStatsMgr"; +import { ITelemetryItem } from "../interfaces/ai/ITelemetryItem"; +import { IPayloadData } from "../interfaces/ai/IXHROverride"; +import { isFeatureEnabled } from "../utils/HelperFuncs"; const STATS_COLLECTION_SHORT_INTERVAL: number = 900000; // 15 minutes const STATS_MIN_INTERVAL_SECONDS = 60; // 1 minute diff --git a/shared/AppInsightsCore/src/core/TelemetryHelpers.ts b/shared/AppInsightsCore/src/core/TelemetryHelpers.ts new file mode 100644 index 000000000..f1ce02ccf --- /dev/null +++ b/shared/AppInsightsCore/src/core/TelemetryHelpers.ts @@ -0,0 +1,374 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { arrForEach, isFunction, objDefineProps } from "@nevware21/ts-utils"; +import { STR_CORE, STR_EMPTY, STR_PRIORITY, STR_PROCESS_TELEMETRY, UNDEFINED_VALUE } from "../constants/InternalConstants"; +import { IAppInsightsCore } from "../interfaces/ai/IAppInsightsCore"; +import { IDistributedTraceContext, IDistributedTraceInit } from "../interfaces/ai/IDistributedTraceContext"; +import { IProcessTelemetryContext, IProcessTelemetryUnloadContext } from "../interfaces/ai/IProcessTelemetryContext"; +import { IPlugin, ITelemetryPlugin } from "../interfaces/ai/ITelemetryPlugin"; +import { ITelemetryPluginChain } from "../interfaces/ai/ITelemetryPluginChain"; +import { ITelemetryUnloadState } from "../interfaces/ai/ITelemetryUnloadState"; +import { IUnloadableComponent } from "../interfaces/ai/IUnloadableComponent"; +import { IW3cTraceState } from "../interfaces/ai/IW3cTraceState"; +import { IOTelSpanContext } from "../interfaces/otel/trace/IOTelSpanContext"; +import { createOTelSpanContext } from "../otel/api/trace/spanContext"; +import { isOTelTraceState } from "../otel/api/trace/traceState"; +import { createW3cTraceState } from "../telemetry/W3cTraceState"; +import { generateW3CId } from "../utils/CoreUtils"; +import { createElmNodeData } from "../utils/DataCacheHelper"; +import { getLocation } from "../utils/EnvUtils"; +import { setProtoTypeName } from "../utils/HelperFuncs"; +import { isValidSpanId, isValidTraceId } from "../utils/TraceParent"; + +export interface IPluginState { + core?: IAppInsightsCore; + isInitialized?: boolean; + teardown?: boolean; + disabled?: boolean; +} + +const pluginStateData = createElmNodeData("plugin"); + +export function _getPluginState(plugin: IPlugin): IPluginState { + return pluginStateData.get(plugin, "state", {}, true) +} + +/** + * Initialize the queue of plugins + * @param plugins - The array of plugins to initialize and setting of the next plugin + * @param config - The current config for the instance + * @param core - THe current core instance + * @param extensions - The extensions + */ +export function initializePlugins(processContext: IProcessTelemetryContext, extensions: IPlugin[]) { + + // Set the next plugin and identified the uninitialized plugins + let initPlugins: ITelemetryPlugin[] = []; + let lastPlugin: ITelemetryPlugin = null; + let proxy: ITelemetryPluginChain = processContext.getNext(); + let pluginState: IPluginState; + + while (proxy) { + let thePlugin = proxy.getPlugin(); + if (thePlugin) { + if (lastPlugin && lastPlugin.setNextPlugin && thePlugin.processTelemetry) { + // Set this plugin as the next for the previous one + lastPlugin.setNextPlugin(thePlugin); + } + + pluginState = _getPluginState(thePlugin); + + let isInitialized = !!pluginState.isInitialized; + if (thePlugin.isInitialized) { + isInitialized = thePlugin.isInitialized(); + } + + if (!isInitialized) { + initPlugins.push(thePlugin); + } + + lastPlugin = thePlugin; + proxy = proxy.getNext(); + } + } + + // Now initialize the plugins + arrForEach(initPlugins, thePlugin => { + let core = processContext.core(); + + thePlugin.initialize( + processContext.getCfg(), + core, + extensions, + processContext.getNext()); + + pluginState = _getPluginState(thePlugin); + + // Only add the core to the state if the plugin didn't set it (doesn't extend from BaseTelemetryPlugin) + if (!thePlugin[STR_CORE] && !pluginState[STR_CORE]) { + pluginState[STR_CORE] = core; + } + + pluginState.isInitialized = true; + delete pluginState.teardown; + }); +} + +export function sortPlugins(plugins: T[]) { + // Sort by priority + return plugins.sort((extA: any, extB: any) => { + let result = 0; + if (extB) { + let bHasProcess = extB[STR_PROCESS_TELEMETRY]; + if (extA[STR_PROCESS_TELEMETRY]) { + result = bHasProcess ? extA[STR_PRIORITY] - extB[STR_PRIORITY] : 1; + } else if (bHasProcess) { + result = -1; + } + } else { + result = extA ? 1 : -1; + } + + return result; + }); + // sort complete +} + +/** + * Teardown / Unload helper to perform teardown/unloading operations for the provided components synchronously or asynchronously, this will call any + * _doTeardown() or _doUnload() functions on the provided components to allow them to finish removal. + * @param components - The components you want to unload + * @param unloadCtx - This is the context that should be used during unloading. + * @param unloadState - The details / state of the unload process, it holds details like whether it should be unloaded synchronously or asynchronously and the reason for the unload. + * @param asyncCallback - An optional callback that the plugin must call if it returns true to inform the caller that it has completed any async unload/teardown operations. + * @returns boolean - true if the plugin has or will call asyncCallback, this allows the plugin to perform any asynchronous operations. + */ +export function unloadComponents(components: any | IUnloadableComponent[], unloadCtx?: IProcessTelemetryUnloadContext, unloadState?: ITelemetryUnloadState, asyncCallback?: () => void): void | boolean { + let idx = 0; + + function _doUnload(): void | boolean { + while (idx < components.length) { + let component = components[idx++]; + if (component) { + let func = component._doUnload || component._doTeardown; + if (isFunction(func)) { + if (func.call(component, unloadCtx, unloadState, _doUnload) === true) { + return true; + } + } + } + } + } + + return _doUnload(); +} + +/*#__NO_SIDE_EFFECTS__*/ +export function isDistributedTraceContext(obj: any): obj is IDistributedTraceContext { + return obj && + isFunction(obj.getName) && + isFunction(obj.getTraceId) && + isFunction(obj.getSpanId) && + isFunction(obj.getTraceFlags) && + isFunction(obj.setName) && + isFunction(obj.setTraceId) && + isFunction(obj.setSpanId) && + isFunction(obj.setTraceFlags); +} + +/** + * Creates an IDistributedTraceContext instance that ensures a valid traceId is always available. + * The traceId will be inherited from the parent context if valid, otherwise a new random W3C trace ID is generated. + * + * @param parent - An optional parent {@link IDistributedTraceContext} or {@link IOTelSpanContext} to inherit + * trace context values from. If provided, the traceId and spanId will be copied from the parent if they are valid. + * When the parent is an {@link IDistributedTraceContext}, it will be set as the parentCtx property to maintain + * hierarchical relationships and enable parent context updates. + * When the parent is an {@link IOTelSpanContext}, the parentCtx will be null because OpenTelemetry span contexts + * are read-only data sources that don't support the same hierarchical management methods as IDistributedTraceContext. + * The core instance will create a wrapped IDistributedTraceContext instance from the IOTelSpanContext data + * to enable Application Insights distributed tracing functionality while maintaining OpenTelemetry compatibility. + * + * @returns A new IDistributedTraceContext instance with the following behavior: + * - **traceId**: Always present - either inherited from parent (if valid) or newly generated W3C trace ID + * - **spanId**: Inherited from parent if valid, otherwise empty string + * - **traceFlags**: Inherited from parent if available, otherwise undefined + * - **pageName**: Inherited from parent context or derived from current location, defaults to "_unknown_" + * - **traceState**: Lazily created W3C trace state, inheriting from parent if available + * + * @remarks + * This function ensures consistent distributed tracing by guaranteeing that every context has a valid traceId, + * which is essential for the refactored W3C trace state implementation. The spanId may be empty until a + * specific span is created, which is normal behavior for trace contexts. + * + * The distinction between IDistributedTraceContext and IOTelSpanContext parents is important: + * - IDistributedTraceContext parents enable bidirectional updates and hierarchical management + * - IOTelSpanContext parents are used only for initial data extraction and OpenTelemetry compatibility + */ +/*#__NO_SIDE_EFFECTS__*/ +export function createDistributedTraceContext(parent?: IDistributedTraceContext | IOTelSpanContext | IDistributedTraceInit | undefined | null): IDistributedTraceContext { + let parentCtx: IDistributedTraceContext = null; + let initCtx: IDistributedTraceInit | IOTelSpanContext = null; + let traceId = (parent && isValidTraceId(parent.traceId)) ? parent.traceId : generateW3CId(); + let spanId = (parent && isValidSpanId(parent.spanId)) ? parent.spanId : STR_EMPTY; + let traceFlags = parent ? parent.traceFlags : UNDEFINED_VALUE; + let isRemote = parent ? parent.isRemote : false; + let pageName = STR_EMPTY; + let traceState: IW3cTraceState = null; + + if (parent) { + if (isDistributedTraceContext(parent)) { + parentCtx = parent; + pageName = parentCtx.getName(); + } else { + initCtx = parent; + } + } + + if (!pageName) { + pageName = "_unknown_"; + // If we have a location, use that as the page name + let location = getLocation(); + if (location && location.pathname) { + pageName = location.pathname + (location.hash || ""); + } + } + + function _getName(): string { + return pageName; + } + + function _setPageNameFn(updateParent: boolean) { + return function (newValue: string): void { + if (updateParent) { + parentCtx && parentCtx.setName(newValue); + } + + pageName = newValue; + }; + } + + function _getTraceId(): string { + return traceId; + } + + function _setTraceIdFn(updateParent: boolean) { + return function (newValue: string): void { + if (updateParent) { + parentCtx && parentCtx.setTraceId(newValue); + } + + if (isValidTraceId(newValue)) { + traceId = newValue + } + }; + } + + function _getSpanId(): string { + return spanId; + } + + function _setSpanIdFn(updateParent: boolean) { + return function (newValue: string): void { + if (updateParent) { + parentCtx && parentCtx.setSpanId(newValue); + } + + if (isValidSpanId(newValue)) { + spanId = newValue + } + }; + } + + function _getTraceFlags(): number { + return traceFlags; + } + + function _setTraceFlagsFn(updateParent: boolean) { + return function (newTraceFlags?: number): void { + if (updateParent) { + parentCtx && parentCtx.setTraceFlags(newTraceFlags); + } + + traceFlags = newTraceFlags; + }; + } + + function _getTraceState(): IW3cTraceState { + if (!traceState) { + if (!parentCtx) { + // The passed in parent was not an IDistributedTraceContext + if (initCtx) { + if (isOTelTraceState(initCtx.traceState)) { + // This looks like an IOTelSpanContext, so we have to just use the passed traceState as-is as it doesn't support + // the W3cTraceState heirarchy methods + traceState = createW3cTraceState(initCtx.traceState.serialize() || STR_EMPTY, parentCtx ? parentCtx.traceState : undefined); + } else { + // This looks like an IDistributedTraceInit, so we can create a new W3cTraceState + traceState = createW3cTraceState(STR_EMPTY, initCtx.traceState || (parentCtx ? parentCtx.traceState : undefined)); + } + } + } + + if (!traceState) { + traceState = createW3cTraceState(STR_EMPTY, parentCtx ? parentCtx.traceState : undefined); + } + } + + return traceState; + } + + let otelSpanCtx: IOTelSpanContext = null; + let traceCtx: IDistributedTraceContext = setProtoTypeName({ + getName: _getName, + setName: _setPageNameFn(true), + getTraceId: _getTraceId, + setTraceId: _setTraceIdFn(true), + getSpanId: _getSpanId, + setSpanId: _setSpanIdFn(true), + getTraceFlags: _getTraceFlags, + setTraceFlags: _setTraceFlagsFn(true), + traceId, + spanId, + traceFlags, + traceState, + isRemote, + pageName, + getOTelSpanContext: () => { + if (!otelSpanCtx) { + // Lazily create the OTel Span Context + otelSpanCtx = createOTelSpanContext(traceCtx); + } + + return otelSpanCtx; + } + }, "DistributedTraceContext"); + + return objDefineProps(traceCtx, { + pageName: { + g: _getName, + s: _setPageNameFn(false) + }, + traceId: { + g: _getTraceId, + s: _setTraceIdFn(false) + }, + spanId: { + g: _getSpanId, + s: _setSpanIdFn(false) + }, + traceFlags: { + g: _getTraceFlags, + s: _setTraceFlagsFn(false) + }, + isRemote: { + v: isRemote, + w: false + + }, + traceState: { + g: _getTraceState + }, + parentCtx: { + g: () => parentCtx + }, + _parent: { + g: () => { + let result: any; + if (parentCtx) { + result = { + t: "traceCtx", + v: parentCtx + }; + } else if(initCtx) { + result = { + t: "initCtx", + v: initCtx + }; + } + return result; + } + } + }); +} diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/TelemetryInitializerPlugin.ts b/shared/AppInsightsCore/src/core/TelemetryInitializerPlugin.ts similarity index 87% rename from shared/AppInsightsCore/src/JavaScriptSDK/TelemetryInitializerPlugin.ts rename to shared/AppInsightsCore/src/core/TelemetryInitializerPlugin.ts index 5001682f4..783cfe6f4 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/TelemetryInitializerPlugin.ts +++ b/shared/AppInsightsCore/src/core/TelemetryInitializerPlugin.ts @@ -1,18 +1,18 @@ -// // Copyright (c) Microsoft Corporation. All rights reserved. -// // Licensed under the MIT License. +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. import dynamicProto from "@microsoft/dynamicproto-js"; import { arrAppend, arrForEach, dumpObj } from "@nevware21/ts-utils"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; -import { IProcessTelemetryContext } from "../JavaScriptSDK.Interfaces/IProcessTelemetryContext"; +import { _throwInternal } from "../diagnostics/DiagnosticLogger"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { IProcessTelemetryContext } from "../interfaces/ai/IProcessTelemetryContext"; import { ITelemetryInitializerContainer, ITelemetryInitializerHandler, TelemetryInitializerFunction -} from "../JavaScriptSDK.Interfaces/ITelemetryInitializers"; -import { ITelemetryItem } from "../JavaScriptSDK.Interfaces/ITelemetryItem"; +} from "../interfaces/ai/ITelemetryInitializers"; +import { ITelemetryItem } from "../interfaces/ai/ITelemetryItem"; +import { getExceptionName } from "../utils/HelperFuncs"; import { BaseTelemetryPlugin } from "./BaseTelemetryPlugin"; -import { _throwInternal } from "./DiagnosticLogger"; -import { getExceptionName } from "./HelperFuncs"; interface _IInternalTelemetryInitializerHandler { id: number; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/UnloadHandlerContainer.ts b/shared/AppInsightsCore/src/core/UnloadHandlerContainer.ts similarity index 79% rename from shared/AppInsightsCore/src/JavaScriptSDK/UnloadHandlerContainer.ts rename to shared/AppInsightsCore/src/core/UnloadHandlerContainer.ts index 565b73d54..d272ea41e 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/UnloadHandlerContainer.ts +++ b/shared/AppInsightsCore/src/core/UnloadHandlerContainer.ts @@ -2,10 +2,10 @@ // Licensed under the MIT License. import { arrForEach, dumpObj } from "@nevware21/ts-utils"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { IProcessTelemetryUnloadContext } from "../JavaScriptSDK.Interfaces/IProcessTelemetryContext"; -import { ITelemetryUnloadState } from "../JavaScriptSDK.Interfaces/ITelemetryUnloadState"; -import { _throwInternal } from "./DiagnosticLogger"; +import { _throwInternal } from "../diagnostics/DiagnosticLogger"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { IProcessTelemetryUnloadContext } from "../interfaces/ai/IProcessTelemetryContext"; +import { ITelemetryUnloadState } from "../interfaces/ai/ITelemetryUnloadState"; export type UnloadHandler = (itemCtx: IProcessTelemetryUnloadContext, unloadState: ITelemetryUnloadState) => void; @@ -14,6 +14,7 @@ export interface IUnloadHandlerContainer { run: (itemCtx: IProcessTelemetryUnloadContext, unloadState: ITelemetryUnloadState) => void } +/*#__NO_SIDE_EFFECTS__*/ export function createUnloadHandlerContainer(): IUnloadHandlerContainer { let handlers: UnloadHandler[] = []; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/UnloadHookContainer.ts b/shared/AppInsightsCore/src/core/UnloadHookContainer.ts similarity index 90% rename from shared/AppInsightsCore/src/JavaScriptSDK/UnloadHookContainer.ts rename to shared/AppInsightsCore/src/core/UnloadHookContainer.ts index 32895ed0f..a433c3f84 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/UnloadHookContainer.ts +++ b/shared/AppInsightsCore/src/core/UnloadHookContainer.ts @@ -2,10 +2,10 @@ // Licensed under the MIT License. import { arrAppend, arrForEach, dumpObj } from "@nevware21/ts-utils"; -import { _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; -import { ILegacyUnloadHook, IUnloadHook } from "../JavaScriptSDK.Interfaces/IUnloadHook"; -import { _throwInternal } from "./DiagnosticLogger"; +import { _throwInternal } from "../diagnostics/DiagnosticLogger"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { ILegacyUnloadHook, IUnloadHook } from "../interfaces/ai/IUnloadHook"; let _maxHooks: number | undefined; let _hookAddMonitor: (state: string, hooks: Array) => void | undefined; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/DiagnosticLogger.ts b/shared/AppInsightsCore/src/diagnostics/DiagnosticLogger.ts similarity index 93% rename from shared/AppInsightsCore/src/JavaScriptSDK/DiagnosticLogger.ts rename to shared/AppInsightsCore/src/diagnostics/DiagnosticLogger.ts index 9fa60e83e..0c604c49a 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/DiagnosticLogger.ts +++ b/shared/AppInsightsCore/src/diagnostics/DiagnosticLogger.ts @@ -3,16 +3,17 @@ "use strict" import dynamicProto from "@microsoft/dynamicproto-js"; import { IPromise } from "@nevware21/ts-async"; -import { dumpObj, isFunction, isUndefined } from "@nevware21/ts-utils"; -import { createDynamicConfig, onConfigChange } from "../Config/DynamicConfig"; -import { LoggingSeverity, _InternalMessageId, _eInternalMessageId, eLoggingSeverity } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { IAppInsightsCore } from "../JavaScriptSDK.Interfaces/IAppInsightsCore"; -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; -import { IConfigDefaults, IUnloadHook } from "../applicationinsights-core-js"; -import { getDebugExt } from "./DbgExtensionUtils"; -import { getConsole, getJSON, hasJSON } from "./EnvUtils"; -import { STR_EMPTY } from "./InternalConstants"; +import { dumpObj, isFunction, isUndefined, objDefine } from "@nevware21/ts-utils"; +import { createDynamicConfig, onConfigChange } from "../config/DynamicConfig"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { getDebugExt } from "../core/DbgExtensionUtils"; +import { LoggingSeverity, _InternalMessageId, _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { IAppInsightsCore } from "../interfaces/ai/IAppInsightsCore"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { IUnloadHook } from "../interfaces/ai/IUnloadHook"; +import { IConfigDefaults } from "../interfaces/config/IConfigDefaults"; +import { getConsole, getJSON, hasJSON } from "../utils/EnvUtils"; const STR_WARN_TO_CONSOLE = "warnToConsole"; @@ -36,7 +37,7 @@ const defaultValues: IConfigDefaults = { loggingLevelTelemetry: 1, maxMessageLimit: 25, enableDebug: false -} +}; const _logFuncs: { [key in eLoggingSeverity]: keyof IDiagnosticLogger} = { [eLoggingSeverity.DISABLED]: null, @@ -94,6 +95,7 @@ export class _InternalLogMessage{ } } +/*#__NO_SIDE_EFFECTS__*/ export function safeGetLogger(core: IAppInsightsCore, config?: IConfiguration): IDiagnosticLogger { return (core || {} as any).logger || new DiagnosticLogger(config); } @@ -101,6 +103,8 @@ export function safeGetLogger(core: IAppInsightsCore, config?: IConfiguration): export class DiagnosticLogger implements IDiagnosticLogger { public identifier = "DiagnosticLogger"; + public readonly dbgMode: boolean; + /** * The internal logging queue */ @@ -193,6 +197,10 @@ export class DiagnosticLogger implements IDiagnosticLogger { _unloadHandler = null; }; + objDefine(_self, "dbgMode", { + g: () => _enableDebug + }); + function _logInternalMessage(severity: LoggingSeverity, message: _InternalLogMessage): void { if (_areInternalMessagesThrottled()) { return; diff --git a/shared/AppInsightsCommon/src/ThrottleMgr.ts b/shared/AppInsightsCore/src/diagnostics/ThrottleMgr.ts similarity index 94% rename from shared/AppInsightsCommon/src/ThrottleMgr.ts rename to shared/AppInsightsCore/src/diagnostics/ThrottleMgr.ts index 83e8ccada..4f669fd4b 100644 --- a/shared/AppInsightsCommon/src/ThrottleMgr.ts +++ b/shared/AppInsightsCore/src/diagnostics/ThrottleMgr.ts @@ -1,11 +1,17 @@ -import { - IAppInsightsCore, IConfiguration, IDiagnosticLogger, _eInternalMessageId, _throwInternal, arrIndexOf, eLoggingSeverity, - isNotNullOrUndefined, isNullOrUndefined, onConfigChange, randomValue, safeGetLogger, strTrim -} from "@microsoft/applicationinsights-core-js"; -import { arrForEach, mathFloor, mathMin, objForEachKey } from "@nevware21/ts-utils"; -import { IThrottleInterval, IThrottleLocalStorageObj, IThrottleMgrConfig, IThrottleResult } from "./Interfaces/IThrottleMgr"; -import { utlCanUseLocalStorage, utlGetLocalStorage, utlSetLocalStorage } from "./StorageHelperFuncs"; -import { IConfig } from "./applicationinsights-common"; +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { arrForEach, arrIndexOf, isNullOrUndefined, mathFloor, mathMin, objForEachKey, strTrim } from "@nevware21/ts-utils"; +import { onConfigChange } from "../config/DynamicConfig"; +import { _throwInternal, safeGetLogger } from "../diagnostics/DiagnosticLogger"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { IAppInsightsCore } from "../interfaces/ai/IAppInsightsCore"; +import { IConfig } from "../interfaces/ai/IConfig"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { IThrottleInterval, IThrottleLocalStorageObj, IThrottleMgrConfig, IThrottleResult } from "../interfaces/ai/IThrottleMgr"; +import { randomValue } from "../utils/RandomHelper"; +import { utlCanUseLocalStorage, utlGetLocalStorage, utlSetLocalStorage } from "../utils/StorageHelperFuncs"; const THROTTLE_STORAGE_PREFIX = "appInsightsThrottle"; @@ -197,7 +203,7 @@ export class ThrottleMgr { _queue = {}; _config = {}; _setCfgByKey(_eInternalMessageId.DefaultThrottleMsgKey); - _namePrefix = isNotNullOrUndefined(namePrefix)? namePrefix : ""; + _namePrefix = !isNullOrUndefined(namePrefix)? namePrefix : ""; core.addUnloadHook(onConfigChange(core.config, (details) => { let coreConfig = details.cfg; @@ -260,7 +266,7 @@ export class ThrottleMgr { } function _canThrottle(config: IThrottleMgrConfig, canUseLocalStorage: boolean, localStorageObj: IThrottleLocalStorageObj) { - if (config && !config.disabled && canUseLocalStorage && isNotNullOrUndefined(localStorageObj)) { + if (config && !config.disabled && canUseLocalStorage && !isNullOrUndefined(localStorageObj)) { let curDate = _getThrottleDate(); let date = localStorageObj.date; let interval = config.interval; @@ -284,7 +290,7 @@ export class ThrottleMgr { } function _getLocalStorageName(msgKey: _eInternalMessageId | number, prefix?: string) { - let fix = isNotNullOrUndefined(prefix)? prefix : ""; + let fix = !isNullOrUndefined(prefix)? prefix : ""; if (msgKey) { return THROTTLE_STORAGE_PREFIX + fix + "-" + msgKey; } diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/EnumHelperFuncs.ts b/shared/AppInsightsCore/src/enums/EnumHelperFuncs.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Enums/EnumHelperFuncs.ts rename to shared/AppInsightsCore/src/enums/EnumHelperFuncs.ts diff --git a/shared/AppInsightsCore/src/enums/W3CTraceFlags.ts b/shared/AppInsightsCore/src/enums/W3CTraceFlags.ts new file mode 100644 index 000000000..b9d329b6e --- /dev/null +++ b/shared/AppInsightsCore/src/enums/W3CTraceFlags.ts @@ -0,0 +1,18 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +/** + * The TelemetryUpdateReason enumeration contains a set of bit-wise values that specify the reason for update request. + */ +export const enum eW3CTraceFlags { + /** + * No sampling decision has been made. + */ + None = 0, + + /** + * Represents that the trace has been sampled. + * @remarks This value is used to indicate that the trace has been sampled. + */ + Sampled = 1 +} diff --git a/shared/AppInsightsCore/src/enums/ai/DependencyTypes.ts b/shared/AppInsightsCore/src/enums/ai/DependencyTypes.ts new file mode 100644 index 000000000..0b943cef4 --- /dev/null +++ b/shared/AppInsightsCore/src/enums/ai/DependencyTypes.ts @@ -0,0 +1,50 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { createEnumStyle } from "../EnumHelperFuncs"; + +export const enum eDependencyTypes { + InProc = "InProc", + QueueMessage = "Queue Message", + Sql = "SQL", + Http = "Http", + Grpc = "GRPC", + Wcf = "WCF Service", +} + +/** + * The EventsDiscardedReason enumeration contains a set of values that specify the reason for discarding an event. + */ +export const DependencyTypes = (/* @__PURE__ */ createEnumStyle({ + /** + * InProc + */ + InProc: eDependencyTypes.InProc, + + /** + * Quene Message + */ + QueueMessage: eDependencyTypes.QueueMessage, + + /** + * Sql + */ + Sql: eDependencyTypes.Sql, + + /** + * Http + */ + Http: eDependencyTypes.Http, + + /** + * Grpc + */ + Grpc: eDependencyTypes.Grpc, + + /** + * Wcf + */ + Wcf: eDependencyTypes.Wcf +})); + +export type DependencyTypes = string | eDependencyTypes; \ No newline at end of file diff --git a/shared/AppInsightsCore/src/enums/ai/Enums.ts b/shared/AppInsightsCore/src/enums/ai/Enums.ts new file mode 100644 index 000000000..6ac091dbe --- /dev/null +++ b/shared/AppInsightsCore/src/enums/ai/Enums.ts @@ -0,0 +1,149 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { createEnumStyle } from "../EnumHelperFuncs"; + +/** + * Type of storage to differentiate between local storage and session storage + */ +export const enum eStorageType { + LocalStorage, + SessionStorage +} + +export const StorageType = (/* @__PURE__ */ createEnumStyle({ + LocalStorage: eStorageType.LocalStorage, + SessionStorage: eStorageType.SessionStorage +})); +export type StorageType = eStorageType | typeof StorageType; + +/** + * Enum is used in aiDataContract to describe how fields are serialized. + * For instance: (Fieldtype.Required | FieldType.Array) will mark the field as required and indicate it's an array + */ +export const enum FieldType { Default = 0, Required = 1, Array = 2, Hidden = 4 } + +export const enum eDistributedTracingModes { + /** + * Send only the legacy Application Insights correlation headers + * + * Headers Sent: + * - `Request-Id` (Legacy Application Insights header for older Server side SDKs) + * + * Config Decimal Value: `0` (Zero) + */ + AI = 0x00, + + /** + * (Default) Send both W3C Trace parent header and back-compatibility Application Insights headers + * - `Request-Id` + * - [`traceparent`](https://www.w3.org/TR/trace-context/#traceparent-header) + * + * Config Decimal Value: `1` (One) + */ + AI_AND_W3C = 0x01, + + /** + * Send Only the W3C Trace parent header + * + * Headers Sent: + * - [`traceparent`](https://www.w3.org/TR/trace-context/#traceparent-header) + * + * Config Decimal Value: `2` (Two) + */ + W3C = 0x02, + + /** + * @internal + * Bitwise mask used to separate the base distributed tracing mode from the additional optional + * tracing modes. + * @since 3.4.0 + */ + _BaseMask = 0x0F, // Mask to get the base distributed tracing mode + + /** + * @internal + * Enabling this bit will send the W3C Trace State header, it is not intended to be used directly + * or on its own. The code may assume that if this bit is set, then the W3C Trace Context headers + * will also be included. + * + * Config Decimal Value: `16` (Sixteen in decimal) + * @since 3.4.0 + */ + _W3CTraceState = 0x10, // Bit mask to enable sending the W3C Trace State headers + + /** + * Send all of the W3C Trace Context headers and the W3C Trace State headers and back-compatibility + * Application Insights headers. + * + * Currently sent headers: + * - `Request-Id` (Legacy Application Insights header for older Server side SDKs) + * - [`traceparent`](https://www.w3.org/TR/trace-context/#traceparent-header) + * - [`tracestate`](https://www.w3.org/TR/trace-context/#tracestate-header) + * + * NOTE!: Additional headers may be added as part of a future update should the W3C Trace Context specification be updated + * to include additional headers. + * + * Config Decimal Value: `17` (Seventeen in decimal) + * @since 3.4.0 + */ + AI_AND_W3C_TRACE = AI_AND_W3C | _W3CTraceState, + + /** + * Send all of the W3C Trace Context headers and the W3C Trace State headers. + * + * Currently sent headers: + * - [`traceparent`](https://www.w3.org/TR/trace-context/#traceparent-header) + * - [`tracestate`](https://www.w3.org/TR/trace-context/#tracestate-header) + * + * NOTE!: Additional headers may be added as part of a future update should the W3C Trace Context specification be updated + * to include additional headers. + * + * Config Decimal Value: `18` (Eighteen in decimal) + * @since 3.4.0 + */ + W3C_TRACE = W3C | _W3CTraceState +} + +export const DistributedTracingModes = (/* @__PURE__ */ createEnumStyle({ + AI: eDistributedTracingModes.AI, + AI_AND_W3C: eDistributedTracingModes.AI_AND_W3C, + W3C: eDistributedTracingModes.W3C, + AI_AND_W3C_TRACE: eDistributedTracingModes.AI_AND_W3C_TRACE, + W3C_TRACE: eDistributedTracingModes.W3C_TRACE, + + // Internal mask values + _BaseMask: eDistributedTracingModes._BaseMask, + _W3CTraceState: eDistributedTracingModes._W3CTraceState +})); +export type DistributedTracingModes = number | eDistributedTracingModes; + +/** + * The EventPersistence contains a set of values that specify the event's persistence. + */ +export const enum EventPersistenceValue { + /** + * Normal persistence. + */ + Normal = 1, + /** + * Critical persistence. + */ + Critical = 2 +} + +/** + * The EventPersistence contains a set of values that specify the event's persistence. + */ +export const EventPersistence = (/* @__PURE__ */ createEnumStyle({ + /** + * Normal persistence. + */ + Normal: EventPersistenceValue.Normal, + + /** + * Critical persistence. + */ + Critical: EventPersistenceValue.Critical +})); +export type EventPersistence = number | EventPersistenceValue; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/EventsDiscardedReason.ts b/shared/AppInsightsCore/src/enums/ai/EventsDiscardedReason.ts similarity index 97% rename from shared/AppInsightsCore/src/JavaScriptSDK.Enums/EventsDiscardedReason.ts rename to shared/AppInsightsCore/src/enums/ai/EventsDiscardedReason.ts index 5a7775307..a83463ab2 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/EventsDiscardedReason.ts +++ b/shared/AppInsightsCore/src/enums/ai/EventsDiscardedReason.ts @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { createEnumStyle } from "../JavaScriptSDK.Enums/EnumHelperFuncs"; +import { createEnumStyle } from "../EnumHelperFuncs"; /** * The eEventsDiscardedReason enumeration contains a set of values that specify the reason for discarding an event. diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/FeatureOptInEnums.ts b/shared/AppInsightsCore/src/enums/ai/FeatureOptInEnums.ts similarity index 86% rename from shared/AppInsightsCore/src/JavaScriptSDK.Enums/FeatureOptInEnums.ts rename to shared/AppInsightsCore/src/enums/ai/FeatureOptInEnums.ts index a0e1756bd..bec50ef13 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/FeatureOptInEnums.ts +++ b/shared/AppInsightsCore/src/enums/ai/FeatureOptInEnums.ts @@ -1,3 +1,6 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + export const enum FeatureOptInMode { /** * not set, completely depends on cdn cfg diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/InitActiveStatusEnum.ts b/shared/AppInsightsCore/src/enums/ai/InitActiveStatusEnum.ts similarity index 84% rename from shared/AppInsightsCore/src/JavaScriptSDK.Enums/InitActiveStatusEnum.ts rename to shared/AppInsightsCore/src/enums/ai/InitActiveStatusEnum.ts index 4b3789867..d2725aca0 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/InitActiveStatusEnum.ts +++ b/shared/AppInsightsCore/src/enums/ai/InitActiveStatusEnum.ts @@ -1,4 +1,7 @@ -import { createEnumStyle } from "./EnumHelperFuncs"; +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { createEnumStyle } from "../EnumHelperFuncs"; export const enum eActiveStatus { diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/LoggingEnums.ts b/shared/AppInsightsCore/src/enums/ai/LoggingEnums.ts similarity index 95% rename from shared/AppInsightsCore/src/JavaScriptSDK.Enums/LoggingEnums.ts rename to shared/AppInsightsCore/src/enums/ai/LoggingEnums.ts index 2ff7d885c..82f50a817 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/LoggingEnums.ts +++ b/shared/AppInsightsCore/src/enums/ai/LoggingEnums.ts @@ -2,7 +2,7 @@ // Licensed under the MIT License. // @skip-file-minify -import { createEnumStyle } from "../JavaScriptSDK.Enums/EnumHelperFuncs"; +import { createEnumStyle } from "../EnumHelperFuncs"; export const enum eLoggingSeverity { /** @@ -130,6 +130,11 @@ export const enum _eInternalMessageId { InitPromiseException = 112, StatsBeatManagerException = 113, StatsBeatException = 114, + AttributeError = 115, + SpanError = 116, + TraceError = 117, + NotImplementedError = 118, + VersionMismatch = 119 } export type _InternalMessageId = number | _eInternalMessageId; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/SendRequestReason.ts b/shared/AppInsightsCore/src/enums/ai/SendRequestReason.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Enums/SendRequestReason.ts rename to shared/AppInsightsCore/src/enums/ai/SendRequestReason.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/StatsType.ts b/shared/AppInsightsCore/src/enums/ai/StatsType.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Enums/StatsType.ts rename to shared/AppInsightsCore/src/enums/ai/StatsType.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/TelemetryUnloadReason.ts b/shared/AppInsightsCore/src/enums/ai/TelemetryUnloadReason.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Enums/TelemetryUnloadReason.ts rename to shared/AppInsightsCore/src/enums/ai/TelemetryUnloadReason.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Enums/TelemetryUpdateReason.ts b/shared/AppInsightsCore/src/enums/ai/TelemetryUpdateReason.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Enums/TelemetryUpdateReason.ts rename to shared/AppInsightsCore/src/enums/ai/TelemetryUpdateReason.ts diff --git a/shared/AppInsightsCore/src/enums/ai/TraceHeadersMode.ts b/shared/AppInsightsCore/src/enums/ai/TraceHeadersMode.ts new file mode 100644 index 000000000..047399022 --- /dev/null +++ b/shared/AppInsightsCore/src/enums/ai/TraceHeadersMode.ts @@ -0,0 +1,29 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +/** + * Controls how the SDK should look for trace headers (traceparent/tracestate) from the initial page load + * The values are bitwise OR'd together to allow for multiple values to be set at once. + * @since 3.4.0 + */ +export const enum eTraceHeadersMode { + /** + * Don't look for any trace headers + */ + None = 0x00, + + /** + * Look for traceparent header/meta tag + */ + TraceParent = 0x01, + + /** + * Look for tracestate header/meta tag + */ + TraceState = 0x02, + + /** + * Look for both traceparent and tracestate headers/meta tags + */ + All = 0x03 +} diff --git a/shared/AppInsightsCore/src/enums/otel/OTelSamplingDecision.ts b/shared/AppInsightsCore/src/enums/otel/OTelSamplingDecision.ts new file mode 100644 index 000000000..a400b76b3 --- /dev/null +++ b/shared/AppInsightsCore/src/enums/otel/OTelSamplingDecision.ts @@ -0,0 +1,33 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { createEnumStyle } from "../EnumHelperFuncs"; + +/** + * A sampling decision that determines how a {@link IOTelSpan} will be recorded + * and collected. + */ +export const enum eOTelSamplingDecision { + /** + * `Span.isRecording() === false`, span will not be recorded and all events + * and attributes will be dropped. + */ + NOT_RECORD = 0, + /** + * `Span.isRecording() === true`, but `Sampled` flag in {@link eW3CTraceFlags} + * MUST NOT be set. + */ + RECORD = 1, + /** + * `Span.isRecording() === true` AND `Sampled` flag in {@link eW3CTraceFlags} + * MUST be set. + */ + RECORD_AND_SAMPLED = 2, +} + +export const OTelSamplingDecision = (/* @__PURE__ */createEnumStyle({ + NOT_RECORD: eOTelSamplingDecision.NOT_RECORD, + RECORD: eOTelSamplingDecision.RECORD, + RECORD_AND_SAMPLED: eOTelSamplingDecision.RECORD_AND_SAMPLED +})); +export type OTelSamplingDecision = number | eOTelSamplingDecision; diff --git a/shared/AppInsightsCore/src/enums/otel/OTelSpanKind.ts b/shared/AppInsightsCore/src/enums/otel/OTelSpanKind.ts new file mode 100644 index 000000000..d76042aa0 --- /dev/null +++ b/shared/AppInsightsCore/src/enums/otel/OTelSpanKind.ts @@ -0,0 +1,53 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { createEnumStyle } from "../EnumHelperFuncs"; + +/** + * The defined set of Span Kinds as defined by the OpenTelemetry. + */ +export const enum eOTelSpanKind { + /** Default value. Indicates that the span is used internally. */ + INTERNAL = 0, + + /** + * Indicates that the span covers server-side handling of an RPC or other + * remote request. + */ + SERVER = 1, + + /** + * Indicates that the span covers the client-side wrapper around an RPC or + * other remote request. + */ + CLIENT = 2, + + /** + * Indicates that the span describes producer sending a message to a + * broker. Unlike client and server, there is no direct critical path latency + * relationship between producer and consumer spans. + */ + PRODUCER = 3, + + /** + * Indicates that the span describes consumer receiving a message from a + * broker. Unlike client and server, there is no direct critical path latency + * relationship between producer and consumer spans. + */ + CONSUMER = 4, +} + +/** + * Creates an enum style object for the OTelSpanKind enum, providing the enum + * values as properties of the object as both string and number types. + * This allows for easy access to the enum values in a more readable format. + */ +export const OTelSpanKind = (/* @__PURE__ */createEnumStyle({ + INTERNAL: eOTelSpanKind.INTERNAL, + SERVER: eOTelSpanKind.SERVER, + CLIENT: eOTelSpanKind.CLIENT, + PRODUCER: eOTelSpanKind.PRODUCER, + CONSUMER: eOTelSpanKind.CONSUMER +})); + +export type OTelSpanKind = number | eOTelSpanKind; diff --git a/shared/AppInsightsCore/src/enums/otel/OTelSpanStatus.ts b/shared/AppInsightsCore/src/enums/otel/OTelSpanStatus.ts new file mode 100644 index 000000000..ae9c1c2fd --- /dev/null +++ b/shared/AppInsightsCore/src/enums/otel/OTelSpanStatus.ts @@ -0,0 +1,34 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { createEnumStyle } from "../EnumHelperFuncs"; + +/** + * An enumeration of status codes, matching the OpenTelemetry specification. + * + * @since 3.4.0 + */ +export const enum eOTelSpanStatusCode { + /** + * The default status. + */ + UNSET = 0, + /** + * The operation has been validated by an Application developer or + * Operator to have completed successfully. + */ + OK = 1, + /** + * The operation contains an error. + */ + ERROR = 2, +} + + +export const OTelSpanStatusCode = (/* @__PURE__ */createEnumStyle({ + UNSET: eOTelSpanStatusCode.UNSET, + OK: eOTelSpanStatusCode.OK, + ERROR: eOTelSpanStatusCode.ERROR +})); + +export type OTelSpanStatusCode = number | eOTelSpanStatusCode; diff --git a/shared/AppInsightsCore/src/enums/otel/eAttributeChangeOp.ts b/shared/AppInsightsCore/src/enums/otel/eAttributeChangeOp.ts new file mode 100644 index 000000000..efe0a33b9 --- /dev/null +++ b/shared/AppInsightsCore/src/enums/otel/eAttributeChangeOp.ts @@ -0,0 +1,46 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { createEnumStyle } from "../EnumHelperFuncs"; + +/** + * Const enum for attribute change operation types + */ +export const enum eAttributeChangeOp { + /** + * Clear operation - clearing all attributes + */ + Clear = 0, + + /** + * Set operation - setting an attribute value (generic) + */ + Set = 1, + + /** + * Add operation - adding a new attribute that didn't exist before + */ + Add = 2, + + /** + * Delete operation - deleting an attribute + */ + Delete = 3, + + /** + * Dropped attributes - attributes that were dropped due to size limits + */ + DroppedAttributes = 4 +} + +/** + * Runtime enum style object for attribute change operation types + */ +export const AttributeChangeOp = createEnumStyle({ + Clear: eAttributeChangeOp.Clear, + Set: eAttributeChangeOp.Set, + Add: eAttributeChangeOp.Add, + Delete: eAttributeChangeOp.Delete +}); + +export type AttributeChangeOp = number | eAttributeChangeOp; \ No newline at end of file diff --git a/shared/AppInsightsCore/src/index.ts b/shared/AppInsightsCore/src/index.ts new file mode 100644 index 000000000..cb5abe5d9 --- /dev/null +++ b/shared/AppInsightsCore/src/index.ts @@ -0,0 +1,370 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +export { IConfiguration } from "./interfaces/ai/IConfiguration"; +export { IChannelControls, MinChannelPriorty, IInternalOfflineSupport } from "./interfaces/ai/IChannelControls"; +export { IChannelControlsHost } from "./interfaces/ai/IChannelControlsHost"; +export { ITelemetryPlugin, IPlugin } from "./interfaces/ai/ITelemetryPlugin"; +export { IExceptionConfig } from "./interfaces/ai/IExceptionConfig"; +export { IAppInsightsCore, ILoadedPlugin } from "./interfaces/ai/IAppInsightsCore"; +export { ITelemetryItem, ICustomProperties, Tags } from "./interfaces/ai/ITelemetryItem"; +export { IBaseProcessingContext, IProcessTelemetryContext, IProcessTelemetryUnloadContext, IProcessTelemetryUpdateContext } from "./interfaces/ai/IProcessTelemetryContext"; +export { INotificationListener } from "./interfaces/ai/INotificationListener"; +export { ITelemetryPluginChain } from "./interfaces/ai/ITelemetryPluginChain"; +export { IDiagnosticLogger } from "./interfaces/ai/IDiagnosticLogger"; +export { InstrumentorHooksCallback, IInstrumentHooksCallbacks, IInstrumentHooks, IInstrumentHook, IInstrumentCallDetails } from "./interfaces/ai/IInstrumentHooks"; +export { IUnloadableComponent } from "./interfaces/ai/IUnloadableComponent"; +export { IPayloadData, SendPOSTFunction, IXHROverride, OnCompleteCallback } from "./interfaces/ai/IXHROverride"; +export { IUnloadHook, ILegacyUnloadHook } from "./interfaces/ai/IUnloadHook"; +export { eEventsDiscardedReason, EventsDiscardedReason, eBatchDiscardedReason, BatchDiscardedReason } from "./enums/ai/EventsDiscardedReason"; +export { eDependencyTypes, DependencyTypes } from "./enums/ai/DependencyTypes"; +export { SendRequestReason, TransportType } from "./enums/ai/SendRequestReason"; +//export { StatsType, eStatsType } from "./enums/ai/StatsType"; +export { TelemetryUpdateReason } from "./enums/ai/TelemetryUpdateReason"; +export { TelemetryUnloadReason } from "./enums/ai/TelemetryUnloadReason"; +export { eActiveStatus, ActiveStatus } from "./enums/ai/InitActiveStatusEnum"; +export { throwAggregationError } from "./core/AggregationError"; +export { AppInsightsCore } from "./core/AppInsightsCore"; +export { BaseTelemetryPlugin } from "./core/BaseTelemetryPlugin"; +export { randomValue, random32, mwcRandomSeed, mwcRandom32, newId } from "./utils/RandomHelper"; +export { Undefined, newGuid, generateW3CId } from "./utils/CoreUtils"; +export { runTargetUnload, doUnloadAll } from "./core/AsyncUtils"; +export { + normalizeJsName, toISOString, getExceptionName, strContains, setValue, getSetValue, + proxyAssign, proxyFunctions, proxyFunctionAs, createClassFromInterface, optimizeObject, + isNotUndefined, isNotNullOrUndefined, objExtend, isFeatureEnabled, getResponseText, formatErrorMessageXdr, formatErrorMessageXhr, prependTransports, + openXhr, _appendHeader, _getAllResponseHeaders, setObjStringTag, setProtoTypeName, isTimeSpan +} from "./utils/HelperFuncs"; +export { parseResponse } from "./core/ResponseHelpers"; +export { IXDomainRequest, IBackendResponse } from "./interfaces/ai/IXDomainRequest"; +export { _ISenderOnComplete, _ISendPostMgrConfig, _ITimeoutOverrideWrapper, _IInternalXhrOverride } from "./interfaces/ai/ISenderPostManager"; +export { SenderPostManager } from "./core/SenderPostManager"; +//export { IStatsBeat, IStatsBeatConfig, IStatsBeatKeyMap as IStatsBeatEndpoints, IStatsBeatState} from "./interfaces/ai/IStatsBeat"; +//export { IStatsEventData } from "./interfaces/ai/IStatsEventData"; +//export { IStatsMgr, IStatsMgrConfig } from "./interfaces/ai/IStatsMgr"; +//export { createStatsMgr } from "./core/StatsBeat"; +export { + isArray, isTypeof, isUndefined, isNullOrUndefined, isStrictUndefined, objHasOwnProperty as hasOwnProperty, isObject, isFunction, + strEndsWith, strStartsWith, isDate, isError, isString, isNumber, isBoolean, arrForEach, arrIndexOf, + arrReduce, arrMap, strTrim, objKeys, objCreate, objDefine, objDefineProp, objDefineAccessors, throwError, isSymbol, + isNotTruthy, isTruthy, objFreeze, objSeal, objToString, objDeepFreeze as deepFreeze, + getInst as getGlobalInst, hasWindow, getWindow, hasDocument, getDocument, hasNavigator, getNavigator, hasHistory, + getHistory, dumpObj, asString, objForEachKey, getPerformance, utcNow as dateNow, perfNow, + ObjDefinePropDescriptor +} from "@nevware21/ts-utils"; +export { EnumValue, createEnumStyle, createValueMap } from "./enums/EnumHelperFuncs"; +export { + attachEvent, detachEvent, addEventHandler, addEventListeners, addPageUnloadEventListener, addPageHideEventListener, addPageShowEventListener, + removeEventHandler, removeEventListeners, removePageUnloadEventListener, removePageHideEventListener, removePageShowEventListener, eventOn, eventOff, + mergeEvtNamespace, _IRegisteredEvents, __getRegisteredEvents +} from "./internal/EventHelpers"; + +export { + getCrypto, getMsCrypto, getLocation, hasJSON, getJSON, + isReactNative, getConsole, isIE, getIEVersion, isSafari, + setEnableEnvMocks, isBeaconsSupported, isFetchSupported, useXDomainRequest, isXhrSupported, + findMetaTag, findNamedServerTiming, sendCustomEvent, dispatchEvent, createCustomDomEvent, fieldRedaction +} from "./utils/EnvUtils"; +export { + getGlobal, + strShimPrototype as strPrototype, + strShimFunction as strFunction, + strShimUndefined as strUndefined, + strShimObject as strObject +} from "@microsoft/applicationinsights-shims"; +export { NotificationManager } from "./core/NotificationManager"; +export { INotificationManager } from "./interfaces/ai/INotificationManager"; +export { IPerfEvent } from "./interfaces/ai/IPerfEvent"; +export { IPerfManager, IPerfManagerProvider } from "./interfaces/ai/IPerfManager"; +export { PerfEvent, PerfManager, doPerf, getGblPerfMgr, setGblPerfMgr } from "./core/PerfManager"; +export { IFeatureOptInDetails, IFeatureOptIn } from "./interfaces/ai/IFeatureOptIn"; +export { FeatureOptInMode, CdnFeatureMode } from "./enums/ai/FeatureOptInEnums"; +export { safeGetLogger, DiagnosticLogger, _InternalLogMessage, _throwInternal, _warnToConsole, _logInternalMessage } from "./diagnostics/DiagnosticLogger"; +export { + createProcessTelemetryContext + // Explicitly NOT exporting createProcessTelemetryUnloadContext() and createProcessTelemetryUpdateContext() as these should only be created internally +} from "./core/ProcessTelemetryContext"; +export { initializePlugins, sortPlugins, unloadComponents, createDistributedTraceContext, isDistributedTraceContext } from "./core/TelemetryHelpers"; +export { _eInternalMessageId, _InternalMessageId, LoggingSeverity, eLoggingSeverity } from "./enums/ai/LoggingEnums"; +export { InstrumentProto, InstrumentProtos, InstrumentFunc, InstrumentFuncs, InstrumentEvent } from "./core/InstrumentHooks"; +export { ICookieMgr, ICookieMgrConfig } from "./interfaces/ai/ICookieMgr"; +export { + createCookieMgr, safeGetCookieMgr, uaDisallowsSameSiteNone, areCookiesSupported +} from "./core/CookieMgr"; +export { IDbgExtension } from "./interfaces/ai/IDbgExtension"; +export { getDebugListener, getDebugExt } from "./core/DbgExtensionUtils"; +export { TelemetryInitializerFunction, ITelemetryInitializerHandler, ITelemetryInitializerContainer } from "./interfaces/ai/ITelemetryInitializers"; +export { createUniqueNamespace } from "./utils/DataCacheHelper"; +export { UnloadHandler, IUnloadHandlerContainer, createUnloadHandlerContainer } from "./core/UnloadHandlerContainer"; +export { IUnloadHookContainer, createUnloadHookContainer, _testHookMaxUnloadHooksCb } from "./core/UnloadHookContainer"; +export { ITelemetryUpdateState } from "./interfaces/ai/ITelemetryUpdateState"; +export { ITelemetryUnloadState } from "./interfaces/ai/ITelemetryUnloadState"; +export { IDistributedTraceContext, IDistributedTraceInit } from "./interfaces/ai/IDistributedTraceContext"; +export { ITraceParent } from "./interfaces/ai/ITraceParent"; +export { + createTraceParent, parseTraceParent, isValidTraceId, isValidSpanId, isValidTraceParent, isSampledFlag, formatTraceParent, findW3cTraceParent, + findAllScripts, INVALID_TRACE_ID, INVALID_SPAN_ID, scriptsInfo +} from "./utils/TraceParent"; + +// Dynamic Config definitions +export { IConfigCheckFn, IConfigDefaultCheck, IConfigDefaults, IConfigSetFn } from "./interfaces/config/IConfigDefaults"; +export { IDynamicConfigHandler } from "./interfaces/config/IDynamicConfigHandler"; +export { IDynamicPropertyHandler } from "./interfaces/config/IDynamicPropertyHandler"; +export { IWatchDetails, IWatcherHandler, WatcherFunction } from "./interfaces/config/IDynamicWatcher"; +export { createDynamicConfig, onConfigChange } from "./config/DynamicConfig"; +export { getDynamicConfigHandler, blockDynamicConversion, forceDynamicConversion } from "./config/DynamicSupport"; +export { cfgDfValidate, cfgDfMerge, cfgDfBoolean, cfgDfFunc, cfgDfString, cfgDfSet, cfgDfBlockPropValue } from "./config/ConfigDefaultHelpers"; + +// W3c TraceState support +export { eW3CTraceFlags } from "./enums/W3CTraceFlags"; +export { IW3cTraceState } from "./interfaces/ai/IW3cTraceState"; +export { createW3cTraceState, findW3cTraceState, isW3cTraceState, snapshotW3cTraceState } from "./telemetry/W3cTraceState"; + +// ========================================================================== +// OpenTelemetry exports +// ========================================================================== + +// OpenTelemetry Trace support +export { IOTelTraceState } from "./interfaces/otel/trace/IOTelTraceState"; +export { IOTelSpan } from "./interfaces/otel/trace/IOTelSpan"; +export { IOTelTracer } from "./interfaces/otel/trace/IOTelTracer"; +export { IOTelTracerProvider } from "./interfaces/otel/trace/IOTelTracerProvider"; +export { IOTelTracerOptions } from "./interfaces/otel/trace/IOTelTracerOptions"; +export { ITraceProvider, ITraceHost, ISpanScope } from "./interfaces/ai/ITraceProvider"; +export { IOTelSpanOptions } from "./interfaces/otel/trace/IOTelSpanOptions"; +export { createOTelTraceState, isOTelTraceState } from "./otel/api/trace/traceState"; +export { createSpan } from "./otel/api/trace/span"; +export { createTraceProvider } from "./otel/api/trace/traceProvider"; +export { + createNonRecordingSpan, isSpanContextValid, + wrapSpanContext, isReadableSpan, isTracingSuppressed, suppressTracing, unsuppressTracing, useSpan, withSpan, startActiveSpan +} from "./otel/api/trace/utils"; + +export { + AzureMonitorSampleRate, ApplicationInsightsCustomEventName, MicrosoftClientIp, ApplicationInsightsMessageName, + ApplicationInsightsExceptionName, ApplicationInsightsPageViewName, ApplicationInsightsAvailabilityName, + ApplicationInsightsEventName, ApplicationInsightsBaseType, ApplicationInsightsMessageBaseType, + ApplicationInsightsExceptionBaseType, ApplicationInsightsPageViewBaseType, ApplicationInsightsAvailabilityBaseType, + ApplicationInsightsEventBaseType, ATTR_ENDUSER_ID, ATTR_ENDUSER_PSEUDO_ID, ATTR_HTTP_ROUTE, SEMATTRS_NET_PEER_IP, + SEMATTRS_NET_PEER_NAME, SEMATTRS_NET_HOST_IP, SEMATTRS_PEER_SERVICE, SEMATTRS_HTTP_USER_AGENT, SEMATTRS_HTTP_METHOD, + SEMATTRS_HTTP_URL, SEMATTRS_HTTP_STATUS_CODE, SEMATTRS_HTTP_ROUTE, SEMATTRS_HTTP_HOST, SEMATTRS_DB_SYSTEM, + SEMATTRS_DB_STATEMENT, SEMATTRS_DB_OPERATION, SEMATTRS_DB_NAME, SEMATTRS_RPC_SYSTEM, SEMATTRS_RPC_GRPC_STATUS_CODE, + SEMATTRS_EXCEPTION_TYPE, SEMATTRS_EXCEPTION_MESSAGE, SEMATTRS_EXCEPTION_STACKTRACE, SEMATTRS_HTTP_SCHEME, + SEMATTRS_HTTP_TARGET, SEMATTRS_HTTP_FLAVOR, SEMATTRS_NET_TRANSPORT, SEMATTRS_NET_HOST_NAME, SEMATTRS_NET_HOST_PORT, + SEMATTRS_NET_PEER_PORT, SEMATTRS_HTTP_CLIENT_IP, SEMATTRS_ENDUSER_ID, ATTR_CLIENT_ADDRESS, ATTR_CLIENT_PORT, + ATTR_SERVER_ADDRESS, ATTR_SERVER_PORT, ATTR_URL_FULL, ATTR_URL_PATH, ATTR_URL_QUERY, ATTR_URL_SCHEME, + ATTR_ERROR_TYPE, ATTR_NETWORK_LOCAL_ADDRESS, ATTR_NETWORK_LOCAL_PORT, ATTR_NETWORK_PROTOCOL_NAME, + ATTR_NETWORK_PEER_ADDRESS, ATTR_NETWORK_PEER_PORT, ATTR_NETWORK_PROTOCOL_VERSION, ATTR_NETWORK_TRANSPORT, + ATTR_USER_AGENT_ORIGINAL, ATTR_HTTP_REQUEST_METHOD, ATTR_HTTP_RESPONSE_STATUS_CODE, ATTR_EXCEPTION_TYPE, + ATTR_EXCEPTION_MESSAGE, ATTR_EXCEPTION_STACKTRACE, EXP_ATTR_ENDUSER_ID, EXP_ATTR_ENDUSER_PSEUDO_ID, + EXP_ATTR_SYNTHETIC_TYPE, DBSYSTEMVALUES_MONGODB, DBSYSTEMVALUES_COSMOSDB, DBSYSTEMVALUES_MYSQL, + DBSYSTEMVALUES_POSTGRESQL, DBSYSTEMVALUES_REDIS, DBSYSTEMVALUES_DB2, DBSYSTEMVALUES_DERBY, DBSYSTEMVALUES_MARIADB, + DBSYSTEMVALUES_MSSQL, DBSYSTEMVALUES_ORACLE, DBSYSTEMVALUES_SQLITE, DBSYSTEMVALUES_OTHER_SQL, DBSYSTEMVALUES_HSQLDB, + DBSYSTEMVALUES_H2 +} from "./otel/attribute/SemanticConventions" + +// OpenTelemetry Core API Interfaces +export { IOTelApi } from "./interfaces/otel/IOTelApi"; +export { IOTelApiCtx } from "./interfaces/otel/IOTelApiCtx"; +export { IOTelAttributes, OTelAttributeValue, ExtendedOTelAttributeValue } from "./interfaces/otel/IOTelAttributes"; +export { OTelException, IOTelExceptionWithCode, IOTelExceptionWithMessage, IOTelExceptionWithName } from "./interfaces/IException"; +export { IOTelHrTime, OTelTimeInput } from "./interfaces/IOTelHrTime"; +export { createOTelApi } from "./otel/api/OTelApi"; + +// OpenTelemetry Trace Interfaces +export { ITraceApi } from "./interfaces/otel/trace/IOTelTraceApi"; +export { IOTelSpanCtx } from "./interfaces/otel/trace/IOTelSpanCtx"; +export { IOTelSpanStatus } from "./interfaces/otel/trace/IOTelSpanStatus"; +export { IReadableSpan } from "./interfaces/otel/trace/IReadableSpan"; +export { IOTelSpanContext } from "./interfaces/otel/trace/IOTelSpanContext"; + +// OpenTelemetry Configuration Interfaces +export { IOTelConfig } from "./interfaces/otel/config/IOTelConfig"; +export { IOTelAttributeLimits } from "./interfaces/otel/config/IOTelAttributeLimits"; +export { IOTelErrorHandlers } from "./interfaces/otel/config/IOTelErrorHandlers"; +export { ITraceCfg } from "./interfaces/otel/config/IOTelTraceCfg"; + +// OpenTelemetry Attribute Support +export { IAttributeContainer, IAttributeChangeInfo } from "./interfaces/otel/attribute/IAttributeContainer"; +export { eAttributeChangeOp, AttributeChangeOp } from "./enums/otel/eAttributeChangeOp"; +export { createAttributeContainer, addAttributes, isAttributeContainer, createAttributeSnapshot } from "./otel/attribute/attributeContainer"; +export { eAttributeFilter, AttributeFilter } from "./interfaces/otel/attribute/IAttributeContainer"; + +// OpenTelemetry Enums +export { eOTelSpanKind, OTelSpanKind } from "./enums/otel/OTelSpanKind"; +export { eOTelSpanStatusCode, OTelSpanStatusCode } from "./enums/otel/OTelSpanStatus"; + +// OpenTelemetry Helper Utilities +export { + hrTime, hrTimeToTimeStamp, hrTimeDuration, hrTimeToMilliseconds, timeInputToHrTime, millisToHrTime, hrTimeToNanoseconds, + addHrTimes, hrTimeToMicroseconds, zeroHrTime, nanosToHrTime, isTimeInput, isTimeInputHrTime +} from "./internal/timeHelpers"; +export { isAttributeValue, isAttributeKey, sanitizeAttributes } from "./internal/attributeHelpers"; +export { + getSyntheticType, isSyntheticSource, serializeAttribute, getUrl, getPeerIp, getHttpMethod, getHttpUrl, getHttpHost, getHttpScheme, + getHttpTarget, getNetPeerName, getNetPeerPort, getUserAgent, getLocationIp, getHttpStatusCode, getHttpClientIp, + getDependencyTarget, isSqlDB +} from "./internal/commonUtils"; + +// OpenTelemetry Error Handlers +export { + handleAttribError, handleSpanError, handleDebug, handleWarn, handleError, handleNotImplemented +} from "./internal/handleErrors"; + +// OpenTelemetry Error Classes +export { OpenTelemetryError, OpenTelemetryErrorConstructor, getOpenTelemetryError, throwOTelError } from "./otel/api/errors/OTelError"; +export { OTelInvalidAttributeError, throwOTelInvalidAttributeError } from "./otel/api/errors/OTelInvalidAttributeError"; +export { OTelSpanError, throwOTelSpanError } from "./otel/api/errors/OTelSpanError"; + +// ======================================== +// Application Insights Common Exports +// ======================================== + +// Utility functions +export { + correlationIdSetPrefix, correlationIdGetPrefix, correlationIdCanIncludeCorrelationHeader, + correlationIdGetCorrelationContext, correlationIdGetCorrelationContextValue, + dateTimeUtilsNow, dateTimeUtilsDuration, isInternalApplicationInsightsEndpoint, + createDistributedTraceContextFromTrace +} from "./utils/Util"; + +export { ThrottleMgr } from "./diagnostics/ThrottleMgr"; +export { parseConnectionString, ConnectionStringParser } from "./telemetry/ConnectionStringParser"; +export { ConnectionString } from "./interfaces/ai/ConnectionString"; +export { FieldType } from "./enums/ai/Enums"; +export { IRequestHeaders, RequestHeaders, eRequestHeaders } from "./telemetry/RequestResponseHeaders"; +export { + DisabledPropertyName, ProcessLegacy, SampleRate, HttpMethod, + DEFAULT_BREEZE_ENDPOINT, DEFAULT_BREEZE_PATH, strNotSpecified, + ChannelControllerPriority +} from "./constants/Constants"; + +// Contracts +export { IData as AIData } from "./interfaces/ai/contracts/IData"; +export { IBase as AIBase } from "./interfaces/ai/contracts/IBase"; +export { IDomain } from "./interfaces/ai/contracts/IDomain"; +export { ISerializable } from "./interfaces/ai/telemetry/ISerializable"; +export { IEnvelope } from "./interfaces/ai/telemetry/IEnvelope"; +export { IStackFrame } from "./interfaces/ai/contracts/IStackFrame"; +export { IExceptionDetails } from "./interfaces/ai/contracts/IExceptionDetails"; +export { IExceptionData } from "./interfaces/ai/contracts/IExceptionData"; +export { IEventData } from "./interfaces/ai/contracts/IEventData"; +export { IMessageData } from "./interfaces/ai/contracts/IMessageData"; +export { IMetricData } from "./interfaces/ai/contracts/IMetricData"; +export { IDataPoint } from "./interfaces/ai/contracts/IDataPoint"; +export { DataPointType } from "./interfaces/ai/contracts/DataPointType"; +export { IPageViewPerfData } from "./interfaces/ai/contracts/IPageViewPerfData"; + +// Telemetry classes +export { Envelope } from "./telemetry/ai/Common/Envelope"; +export { Event } from "./telemetry/ai/Event"; +export { Exception } from "./telemetry/ai/Exception"; +export { Metric } from "./telemetry/ai/Metric"; +export { PageView } from "./telemetry/ai/PageView"; +export { IPageViewData } from "./interfaces/ai/contracts/IPageViewData"; +export { RemoteDependencyData } from "./telemetry/ai/RemoteDependencyData"; +export { IRemoteDependencyData } from "./interfaces/ai/contracts/IRemoteDependencyData"; +export { Trace } from "./telemetry/ai/Trace"; +export { PageViewPerformance } from "./telemetry/ai/PageViewPerformance"; +export { Data } from "./telemetry/ai/Common/Data"; +export { DataPoint } from "./telemetry/ai/Common/DataPoint"; + +// Telemetry interfaces +export { IEventTelemetry } from "./interfaces/ai/IEventTelemetry"; +export { ITraceTelemetry } from "./interfaces/ai/ITraceTelemetry"; +export { IMetricTelemetry } from "./interfaces/ai/IMetricTelemetry"; +export { IDependencyTelemetry } from "./interfaces/ai/IDependencyTelemetry"; +export { IExceptionTelemetry, IAutoExceptionTelemetry, IExceptionInternal } from "./interfaces/ai/IExceptionTelemetry"; +export { IPageViewTelemetry, IPageViewTelemetryInternal } from "./interfaces/ai/IPageViewTelemetry"; +export { IPageViewPerformanceTelemetry, IPageViewPerformanceTelemetryInternal } from "./interfaces/ai/IPageViewPerformanceTelemetry"; +export { IRequestTelemetry } from "./interfaces/ai/IRequestTelemetry"; + +// Severity level +export { eSeverityLevel, SeverityLevel } from "./interfaces/ai/contracts/SeverityLevel"; + +// Configuration +export { IConfig, ConfigurationManager } from "./interfaces/ai/IConfig"; +export { IStorageBuffer } from "./interfaces/ai/IStorageBuffer"; +export { ICorrelationConfig } from "./interfaces/ai/ICorrelationConfig"; + +// Context tags and keys +export { IContextTagKeys, ContextTagKeys } from "./interfaces/ai/contracts/ContextTagKeys"; +export { CtxTagKeys, Extensions } from "./interfaces/ai/PartAExtensions"; + +// Data types and envelope types +export { + EventDataType, ExceptionDataType, MetricDataType, PageViewDataType, + PageViewPerformanceDataType, RemoteDependencyDataType, RequestDataType, TraceDataType +} from "./telemetry/ai/DataTypes"; + +export { + EventEnvelopeType, ExceptionEnvelopeType, MetricEnvelopeType, PageViewEnvelopeType, + PageViewPerformanceEnvelopeType, RemoteDependencyEnvelopeType, RequestEnvelopeType, TraceEnvelopeType +} from "./telemetry/ai/EnvelopeTypes"; + +// Data sanitization +export { + DataSanitizerValues, dataSanitizeKeyAndAddUniqueness, dataSanitizeKey, dataSanitizeString, + dataSanitizeUrl, dataSanitizeMessage, dataSanitizeException, dataSanitizeProperties, + dataSanitizeMeasurements, dataSanitizeId, dataSanitizeInput, dsPadNumber +} from "./telemetry/ai/Common/DataSanitizer"; + +// Telemetry item creator +export { TelemetryItemCreator, createTelemetryItem } from "./telemetry/TelemetryItemCreator"; + +// Application Insights interfaces +export { IAppInsights } from "./interfaces/ai/IAppInsights"; +export { ITelemetryContext } from "./interfaces/ai/ITelemetryContext"; +export { IPropertiesPlugin } from "./interfaces/ai/IPropertiesPlugin"; +export { IRequestContext } from "./interfaces/ai/IRequestContext"; + +// Context interfaces +export { IWeb } from "./interfaces/ai/context/IWeb"; +export { ISession } from "./interfaces/ai/context/ISession"; +export { ISessionManager } from "./interfaces/ai/context/ISessionManager"; +export { IApplication } from "./interfaces/ai/context/IApplication"; +export { IDevice } from "./interfaces/ai/context/IDevice"; +export { IInternal } from "./interfaces/ai/context/IInternal"; +export { ILocation } from "./interfaces/ai/context/ILocation"; +export { ISample } from "./interfaces/ai/context/ISample"; +export { IOperatingSystem } from "./interfaces/ai/context/IOperatingSystem"; +export { IUser, IUserContext } from "./interfaces/ai/context/IUser"; +export { ITelemetryTrace } from "./interfaces/ai/context/ITelemetryTrace"; + +// Enums +export { eDistributedTracingModes, DistributedTracingModes, EventPersistence } from "./enums/ai/Enums"; +export { eTraceHeadersMode } from "./enums/ai/TraceHeadersMode"; + +// Helper functions +export { stringToBoolOrDefault, msToTimeSpan, getExtensionByName, isCrossOriginError } from "./utils/HelperFuncs"; +export { createDomEvent } from "./utils/DomHelperFuncs"; + +// Storage helpers +export { + utlDisableStorage, utlEnableStorage, utlCanUseLocalStorage, utlGetLocalStorage, + utlSetLocalStorage, utlRemoveStorage, utlCanUseSessionStorage, utlGetSessionStorageKeys, + utlGetSessionStorage, utlSetSessionStorage, utlRemoveSessionStorage, utlSetStoragePrefix +} from "./utils/StorageHelperFuncs"; + +// URL helpers +export { + urlParseUrl, urlGetAbsoluteUrl, urlGetPathName, urlGetCompleteUrl, + urlParseHost, urlParseFullHost +} from "./utils/UrlHelperFuncs"; + +// Throttle manager interfaces +export { + IThrottleLimit, IThrottleInterval, IThrottleMgrConfig, + IThrottleLocalStorageObj, IThrottleResult +} from "./interfaces/ai/IThrottleMgr"; + +// Offline support +export { + IOfflineListener, createOfflineListener, IOfflineState, + eOfflineValue, OfflineCallback +} from "./utils/Offline"; + +// Plugin identifiers +export const PropertiesPluginIdentifier = "AppInsightsPropertiesPlugin"; +export const BreezeChannelIdentifier = "AppInsightsChannelPlugin"; +export const AnalyticsPluginIdentifier = "ApplicationInsightsAnalytics"; diff --git a/shared/AppInsightsCore/src/interfaces/IException.ts b/shared/AppInsightsCore/src/interfaces/IException.ts new file mode 100644 index 000000000..7636f0caf --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/IException.ts @@ -0,0 +1,37 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +export interface IOTelExceptionWithCode { + code: string | number; + name?: string; + message?: string; + stack?: string; +} + +export interface IOTelExceptionWithMessage { + code?: string | number; + message: string; + name?: string; + stack?: string; +} + +export interface IOTelExceptionWithName { + code?: string | number; + message?: string; + name: string; + stack?: string; +} + +/** + * Defines Exception. + * + * string or an object with one of (message or name or code) and optional stack + * + * @since 3.4.0 + */ +export type OTelException = + | IOTelExceptionWithCode + | IOTelExceptionWithMessage + | IOTelExceptionWithName + | string; + \ No newline at end of file diff --git a/shared/AppInsightsCore/src/interfaces/IOTelHrTime.ts b/shared/AppInsightsCore/src/interfaces/IOTelHrTime.ts new file mode 100644 index 000000000..b2b329ff8 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/IOTelHrTime.ts @@ -0,0 +1,81 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +/** + * High-resolution time represented as a tuple of [seconds, nanoseconds]. + * This is the base type for all OpenTelemetry high-resolution time values. + * + * @remarks + * The first element represents seconds since Unix epoch, and the second element + * represents nanoseconds (0-999,999,999) within that second. + * + * @example + * ```typescript + * const hrTime: OTelHrTimeBase = [1609459200, 500000000]; // 2021-01-01 00:00:00.5 UTC + * ``` + * + * @since 3.4.0 + */ +export type OTelHrTimeBase = [number, number]; + +/** + * Enhanced high-resolution time interface that extends the base tuple with additional properties. + * Provides a more structured way to work with high-resolution timestamps. + * + * @example + * ```typescript + * const hrTime: IOTelHrTime = { + * 0: 1609459200, // seconds since Unix epoch + * 1: 500000000, // nanoseconds (0-999,999,999) + * }; + * ``` + * + * @since 3.4.0 + */ +export interface IOTelHrTime extends OTelHrTimeBase { + /** + * Seconds since Unix epoch (January 1, 1970 00:00:00 UTC). + * Must be a non-negative integer. + */ + 0: number; + + /** + * Nanoseconds within the second specified by index 0. + * Must be in the range [0, 999999999]. + */ + 1: number; + + /** + * Optional total nanoseconds since Unix epoch. + * When provided, this should be equivalent to (this[0] * 1e9) + this[1]. + * + * @remarks + * This field may be used for more efficient time calculations or when + * working with systems that natively use nanosecond timestamps. + */ + // unixNano?: number; +} + +/** + * Union type representing all valid time input formats accepted by OpenTelemetry APIs. + * + * @remarks + * - `IOTelHrTime`: High-resolution time with nanosecond precision + * - `number`: Milliseconds since Unix epoch (JavaScript Date.now() format) + * - `Date`: JavaScript Date object + * + * @example + * ```typescript + * // All of these are valid time inputs: + * const hrTime: OTelTimeInput = [1609459200, 500000000]; + * const msTime: OTelTimeInput = Date.now(); + * const dateTime: OTelTimeInput = new Date(); + * + * span.addEvent("event", {}, hrTime); + * span.addEvent("event", {}, msTime); + * span.addEvent("event", {}, dateTime); + * ``` + * + * @since 3.4.0 + */ +export type OTelTimeInput = IOTelHrTime | number | Date; diff --git a/shared/AppInsightsCommon/src/Interfaces/ConnectionString.ts b/shared/AppInsightsCore/src/interfaces/ai/ConnectionString.ts similarity index 67% rename from shared/AppInsightsCommon/src/Interfaces/ConnectionString.ts rename to shared/AppInsightsCore/src/interfaces/ai/ConnectionString.ts index a878724a4..c5b781800 100644 --- a/shared/AppInsightsCommon/src/Interfaces/ConnectionString.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/ConnectionString.ts @@ -1,3 +1,6 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + export type ConnectionStringKey = "authorization" | "instrumentationkey" | "ingestionendpoint" | "location" | "endpointsuffix"; export type ConnectionString = { [key in ConnectionStringKey]?: string }; \ No newline at end of file diff --git a/shared/AppInsightsCommon/src/Interfaces/IAppInsights.ts b/shared/AppInsightsCore/src/interfaces/ai/IAppInsights.ts similarity index 94% rename from shared/AppInsightsCommon/src/Interfaces/IAppInsights.ts rename to shared/AppInsightsCore/src/interfaces/ai/IAppInsights.ts index 711dafe09..149f49f56 100644 --- a/shared/AppInsightsCommon/src/Interfaces/IAppInsights.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IAppInsights.ts @@ -1,12 +1,13 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { ICookieMgr, ITelemetryItem } from "@microsoft/applicationinsights-core-js"; +import { ICookieMgr } from "./ICookieMgr"; import { IEventTelemetry } from "./IEventTelemetry"; import { IAutoExceptionTelemetry, IExceptionTelemetry } from "./IExceptionTelemetry"; import { IMetricTelemetry } from "./IMetricTelemetry"; import { IPageViewPerformanceTelemetry } from "./IPageViewPerformanceTelemetry"; import { IPageViewTelemetry } from "./IPageViewTelemetry"; +import { ITelemetryItem } from "./ITelemetryItem"; import { ITraceTelemetry } from "./ITraceTelemetry"; export interface IAppInsights { diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IAppInsightsCore.ts b/shared/AppInsightsCore/src/interfaces/ai/IAppInsightsCore.ts similarity index 92% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IAppInsightsCore.ts rename to shared/AppInsightsCore/src/interfaces/ai/IAppInsightsCore.ts index 39859e849..198b6ee0e 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IAppInsightsCore.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IAppInsightsCore.ts @@ -2,16 +2,15 @@ // Licensed under the MIT License. import { IPromise } from "@nevware21/ts-async"; -import { ITimerHandler } from "@nevware21/ts-utils"; -import { WatcherFunction } from "../Config/IDynamicWatcher"; -import { eActiveStatus } from "../JavaScriptSDK.Enums/InitActiveStatusEnum"; -import { SendRequestReason } from "../JavaScriptSDK.Enums/SendRequestReason"; -import { UnloadHandler } from "../JavaScriptSDK/UnloadHandlerContainer"; +import { ICachedValue, ITimerHandler } from "@nevware21/ts-utils"; +import { UnloadHandler } from "../../core/UnloadHandlerContainer"; +import { eActiveStatus } from "../../enums/ai/InitActiveStatusEnum"; +import { SendRequestReason } from "../../enums/ai/SendRequestReason"; +import { WatcherFunction } from "../config/IDynamicWatcher"; import { IChannelControls } from "./IChannelControls"; import { IConfiguration } from "./IConfiguration"; import { ICookieMgr } from "./ICookieMgr"; import { IDiagnosticLogger } from "./IDiagnosticLogger"; -import { IDistributedTraceContext } from "./IDistributedTraceContext"; import { INotificationListener } from "./INotificationListener"; import { INotificationManager } from "./INotificationManager"; import { IPerfManagerProvider } from "./IPerfManager"; @@ -20,6 +19,7 @@ import { ITelemetryInitializerHandler, TelemetryInitializerFunction } from "./IT import { ITelemetryItem } from "./ITelemetryItem"; import { IPlugin, ITelemetryPlugin } from "./ITelemetryPlugin"; import { ITelemetryUnloadState } from "./ITelemetryUnloadState"; +import { ITraceHost, ITraceProvider } from "./ITraceProvider"; import { ILegacyUnloadHook, IUnloadHook } from "./IUnloadHook"; // import { IStatsBeat, IStatsBeatState } from "./IStatsBeat"; @@ -45,12 +45,7 @@ export interface ILoadedPlugin { remove: (isAsync?: boolean, removeCb?: (removed?: boolean) => void) => void; } -export interface IAppInsightsCore extends IPerfManagerProvider { - - /* - * Config object used to initialize AppInsights - */ - readonly config: CfgType; +export interface IAppInsightsCore extends IPerfManagerProvider, ITraceHost { /** * The current logger instance for this instance. @@ -221,15 +216,13 @@ export interface IAppInsightsCore void, sendReason?: SendRequestReason, cbTimeout?: number): boolean | void; /** - * Gets the current distributed trace context for this instance if available - * @param createNew - Optional flag to create a new instance if one doesn't currently exist, defaults to true - */ - getTraceCtx(createNew?: boolean): IDistributedTraceContext | null; - - /** - * Sets the current distributed trace context for this instance if available + * Set the trace provider for creating spans. + * This allows different SKUs to provide their own span implementations. + * + * @param provider - The trace provider to use for span creation + * @since 3.4.0 */ - setTraceCtx(newTraceCtx: IDistributedTraceContext | null | undefined): void; + setTraceProvider(provider: ICachedValue): void; /** * Watches and tracks changes for accesses to the current config, and if the accessed config changes the diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IChannelControls.ts b/shared/AppInsightsCore/src/interfaces/ai/IChannelControls.ts similarity index 98% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IChannelControls.ts rename to shared/AppInsightsCore/src/interfaces/ai/IChannelControls.ts index 038874ba5..1c8b940b4 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IChannelControls.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IChannelControls.ts @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. import { IPromise } from "@nevware21/ts-async"; -import { SendRequestReason } from "../JavaScriptSDK.Enums/SendRequestReason"; +import { SendRequestReason } from "../../enums/ai/SendRequestReason"; import { IProcessTelemetryUnloadContext } from "./IProcessTelemetryContext"; import { ITelemetryItem } from "./ITelemetryItem"; import { ITelemetryPlugin } from "./ITelemetryPlugin"; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IChannelControlsHost.ts b/shared/AppInsightsCore/src/interfaces/ai/IChannelControlsHost.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IChannelControlsHost.ts rename to shared/AppInsightsCore/src/interfaces/ai/IChannelControlsHost.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/IConfig.ts b/shared/AppInsightsCore/src/interfaces/ai/IConfig.ts similarity index 98% rename from shared/AppInsightsCommon/src/Interfaces/IConfig.ts rename to shared/AppInsightsCore/src/interfaces/ai/IConfig.ts index a30f31ba3..3e643f147 100644 --- a/shared/AppInsightsCommon/src/Interfaces/IConfig.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IConfig.ts @@ -1,9 +1,11 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IConfiguration, ICustomProperties, isNullOrUndefined } from "@microsoft/applicationinsights-core-js"; -import { DistributedTracingModes } from "../Enums"; +import { isNullOrUndefined } from "@nevware21/ts-utils"; +import { DistributedTracingModes } from "../../enums/ai/Enums"; +import { IConfiguration } from "./IConfiguration"; import { IRequestContext } from "./IRequestContext"; import { IStorageBuffer } from "./IStorageBuffer"; +import { ICustomProperties } from "./ITelemetryItem"; import { IThrottleMgrConfig } from "./IThrottleMgr"; /** diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IConfiguration.ts b/shared/AppInsightsCore/src/interfaces/ai/IConfiguration.ts similarity index 95% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IConfiguration.ts rename to shared/AppInsightsCore/src/interfaces/ai/IConfiguration.ts index bca694a56..c3db75b74 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IConfiguration.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IConfiguration.ts @@ -1,6 +1,8 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. import { IPromise } from "@nevware21/ts-async"; +import { eTraceHeadersMode } from "../../enums/ai/TraceHeadersMode"; +import { IOTelConfig } from "../otel/config/IOTelConfig"; import { IAppInsightsCore } from "./IAppInsightsCore"; import { IChannelControls } from "./IChannelControls"; import { ICookieMgrConfig } from "./ICookieMgr"; @@ -10,13 +12,10 @@ import { INotificationManager } from "./INotificationManager"; import { IPerfManager } from "./IPerfManager"; import { ITelemetryPlugin } from "./ITelemetryPlugin"; -//import { IStatsBeatConfig } from "./IStatsBeat"; -"use strict"; - /** * Configuration provided to SDK core */ -export interface IConfiguration { +export interface IConfiguration extends IOTelConfig { /** * Instrumentation key of resource. Either this or connectionString must be specified. */ @@ -252,6 +251,13 @@ export interface IConfiguration { // * @internal // */ //_sdk?: IInternalSdkConfiguration; + + /** + * [Optional] Controls if the SDK should look for the `traceparent` and/or `tracestate` values from + * the service timing headers or meta tags from the initial page load. + * @defaultValue eTraceHeadersMode.All + */ + traceHdrMode?: eTraceHeadersMode; } ///** diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ICookieMgr.ts b/shared/AppInsightsCore/src/interfaces/ai/ICookieMgr.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ICookieMgr.ts rename to shared/AppInsightsCore/src/interfaces/ai/ICookieMgr.ts diff --git a/shared/AppInsightsCore/src/interfaces/ai/ICorrelationConfig.ts b/shared/AppInsightsCore/src/interfaces/ai/ICorrelationConfig.ts new file mode 100644 index 000000000..4d6d2b337 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/ai/ICorrelationConfig.ts @@ -0,0 +1,192 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { DistributedTracingModes } from "../../enums/ai/Enums"; +import { IRequestContext } from "./IRequestContext"; +import { ICustomProperties } from "./ITelemetryItem"; + +export interface ICorrelationConfig { + enableCorsCorrelation: boolean; + + /** + * [Optional] Domains to be excluded from correlation headers. + * To override or discard the default, add an array with all domains to be excluded or + * an empty array to the configuration. + * + * @example + * ```ts + * import { ApplicationInsights } from '@microsoft/applicationinsights-web'; + * const appInsights = new ApplicationInsights({ + * config: { + * connectionString: 'InstrumentationKey=YOUR_INSTRUMENTATION_KEY_GOES_HERE', + * extensionConfig: { + * AjaxDependencyPlugin: { + * // Both arrays of strings are used to match the request URL against the + * // current host and the request URL to determine if correlation headers + * // The strings are converted to RegExp objects by translating + * // - `.` to `\\.` (to match a literal dot) + * // - `*` to `.*` (to match any character) + * // - `\` to `\\` (to match a literal slash) + * // All other characters are ignored and passed to the RegExp constructor + * correlationHeaderExcludedDomains: ["test", "*.azure.com", "ignore.microsoft.com"], + * correlationHeaderDomains: ["azure.com", "prefix.bing.com", "*.microsoft.com", "example.com"] + * } + * } + * }); + * appInsights.loadAppInsights(); + * appInsights.trackPageView(); // Manually call trackPageView to establish the current user/session/pageview + * ``` + */ + correlationHeaderExcludedDomains: string[]; + + /** + * [Optional] Domains to be included in correlation headers. + * To override or discard the default, add an array with all domains to be included or + * an empty array to the configuration. + * + * @example + * ```ts + * import { ApplicationInsights } from '@microsoft/applicationinsights-web'; + * const appInsights = new ApplicationInsights({ + * config: { + * connectionString: 'InstrumentationKey=YOUR_INSTRUMENTATION_KEY_GOES_HERE', + * extensionConfig: { + * AjaxDependencyPlugin: { + * // Values MUST be RegExp objects + * correlationHeaderExcludePatterns: [/*\.azure.com/, /prefix.bing.com/, /.*\.microsoft.com/, /example.com/] + * } + * } + * }); + * appInsights.loadAppInsights(); + * appInsights.trackPageView(); // Manually call trackPageView to establish the current user/session/pageview + * ``` + */ + correlationHeaderExcludePatterns?: RegExp[]; + disableCorrelationHeaders: boolean; + + /** + * The distributed tracing mode to use for this configuration. + * Defaults to AI_AND_W3C. + * This is used to determine which headers are sent with requests and how the + * telemetry is correlated across services. + * @default AI_AND_W3C + * @see {@link eDistributedTracingModes} + */ + distributedTracingMode: DistributedTracingModes; + + maxAjaxCallsPerView: number; + disableAjaxTracking: boolean; + disableFetchTracking: boolean; + appId?: string; + enableRequestHeaderTracking?: boolean; + enableResponseHeaderTracking?: boolean; + enableAjaxErrorStatusText?: boolean; + + /** + * Flag to enable looking up and including additional browser window.performance timings + * in the reported ajax (XHR and fetch) reported metrics. + * Defaults to false. + */ + enableAjaxPerfTracking?: boolean; + + /** + * The maximum number of times to look for the window.performance timings (if available), this + * is required as not all browsers populate the window.performance before reporting the + * end of the XHR request and for fetch requests this is added after its complete + * Defaults to 3 + */ + maxAjaxPerfLookupAttempts?: number; + + /** + * The amount of time to wait before re-attempting to find the windows.performance timings + * for an ajax request, time is in milliseconds and is passed directly to setTimeout() + * Defaults to 25. + */ + ajaxPerfLookupDelay?: number; + + /** + * [Optional] Domains to be excluded from correlation headers. + * To override or discard the default, add an array with all domains to be excluded or + * an empty array to the configuration. + * + * @example + * ```ts + * import { ApplicationInsights } from '@microsoft/applicationinsights-web'; + * const appInsights = new ApplicationInsights({ + * config: { + * connectionString: 'InstrumentationKey=YOUR_INSTRUMENTATION_KEY_GOES_HERE', + * extensionConfig: { + * AjaxDependencyPlugin: { + * // Both arrays of strings are used to match the request URL against the + * // current host and the request URL to determine if correlation headers + * // The strings are converted to RegExp objects by translating + * // - `.` to `\\.` (to match a literal dot) + * // - `*` to `.*` (to match any character) + * // - `\` to `\\` (to match a literal slash) + * // All other characters are ignored and passed to the RegExp constructor + * correlationHeaderExcludedDomains: ["test", "*.azure.com", "ignore.microsoft.com"], + * correlationHeaderDomains: ["azure.com", "prefix.bing.com", "*.microsoft.com", "example.com"] + * } + * } + * }); + * appInsights.loadAppInsights(); + * appInsights.trackPageView(); // Manually call trackPageView to establish the current user/session/pageview + * ``` + */ + correlationHeaderDomains?: string[]; + + /** + * [Optional] Response and request headers to be excluded from AJAX & Fetch tracking data. + * To override or discard the default, add an array with all headers to be excluded or + * an empty array to the configuration. + * + * For example: `["Authorization", "X-API-Key", "WWW-Authenticate"]` + * + * @example + * ```js + * import { ApplicationInsights } from '@microsoft/applicationinsights-web'; + * import { AjaxPlugin } from '@microsoft/applicationinsights-dependencies-js'; + * + * const dependencyPlugin = new AjaxPlugin(); + * const appInsights = new ApplicationInsights({ + * config: { + * connectionString: 'InstrumentationKey=YOUR_INSTRUMENTATION_KEY_GOES_HERE', + * extensions: [dependencyPlugin], + * extensionConfig: { + * AjaxDependencyPlugin: { + * ignoreHeaders: [ + * "Authorization", + * "X-API-Key", + * "WWW-Authenticate" + * ] + * } + * } + * } + * }); + * appInsights.loadAppInsights(); + * appInsights.trackPageView(); // Manually call trackPageView to establish the current user/session/pageview + * ``` + */ + ignoreHeaders?: string[]; + + /** + * Provide a way to exclude specific route from automatic tracking for XMLHttpRequest or Fetch request. + * For an ajax / fetch request that the request url matches with the regex patterns, auto tracking is turned off. + * Default is undefined. + */ + excludeRequestFromAutoTrackingPatterns?: string[] | RegExp[]; + + /** + * Provide a way to enrich dependencies logs with context at the beginning of api call. + * Default is undefined. + */ + addRequestContext?: (requestContext?: IRequestContext) => ICustomProperties; + + /** + * [Optional] Flag to indicate whether the internal looking endpoints should be automatically + * added to the `excludeRequestFromAutoTrackingPatterns` collection. (defaults to true). + * This flag exists as the provided regex is generic and may unexpectedly match a domain that + * should not be excluded. + */ + addIntEndpoints?: boolean; +} diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IDbgExtension.ts b/shared/AppInsightsCore/src/interfaces/ai/IDbgExtension.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IDbgExtension.ts rename to shared/AppInsightsCore/src/interfaces/ai/IDbgExtension.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/IDependencyTelemetry.ts b/shared/AppInsightsCore/src/interfaces/ai/IDependencyTelemetry.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/IDependencyTelemetry.ts rename to shared/AppInsightsCore/src/interfaces/ai/IDependencyTelemetry.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IDiagnosticLogger.ts b/shared/AppInsightsCore/src/interfaces/ai/IDiagnosticLogger.ts similarity index 85% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IDiagnosticLogger.ts rename to shared/AppInsightsCore/src/interfaces/ai/IDiagnosticLogger.ts index cb4b03f17..54ce5c17d 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IDiagnosticLogger.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IDiagnosticLogger.ts @@ -1,10 +1,14 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. import { IPromise } from "@nevware21/ts-async"; -import { LoggingSeverity, _InternalMessageId } from "../JavaScriptSDK.Enums/LoggingEnums"; -import { _InternalLogMessage } from "../JavaScriptSDK/DiagnosticLogger"; +import { LoggingSeverity, _InternalMessageId } from "../../enums/ai/LoggingEnums"; import { ITelemetryUpdateState } from "./ITelemetryUpdateState"; +export interface IInternalLogMessage { + message: string; + messageId: _InternalMessageId; +} + export interface IDiagnosticLogger { /** @@ -17,7 +21,7 @@ export interface IDiagnosticLogger { /** * The internal logging queue */ - queue: _InternalLogMessage[]; + queue: IInternalLogMessage[]; /** * This method will throw exceptions in debug mode or attempt to log the error as a console warning. @@ -30,7 +34,7 @@ export interface IDiagnosticLogger { * This will write a debug message to the console if possible * @param message - The debug message */ - debugToConsole? (message: string): void + debugToConsole?(message: string): void /** * This will write a warning to the console if possible @@ -56,7 +60,7 @@ export interface IDiagnosticLogger { * @param severity - The severity of the log message * @param message - The message to log. */ - logInternalMessage?(severity: LoggingSeverity, message: _InternalLogMessage): void; + logInternalMessage?(severity: LoggingSeverity, message: IInternalLogMessage): void; /** * Optional Callback hook to allow the diagnostic logger to update it's configuration @@ -73,4 +77,9 @@ export interface IDiagnosticLogger { * / Promise to allow any listeners to wait for the operation to complete. */ unload?(isAsync?: boolean): void | IPromise; + + /** + * A flag that indicates whether this logger is in debug (throw real exceptions) mode + */ + readonly dbgMode?: boolean; } diff --git a/shared/AppInsightsCore/src/interfaces/ai/IDistributedTraceContext.ts b/shared/AppInsightsCore/src/interfaces/ai/IDistributedTraceContext.ts new file mode 100644 index 000000000..a19d0be4e --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/ai/IDistributedTraceContext.ts @@ -0,0 +1,334 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IOTelTraceState } from "../otel/trace/IOTelTraceState"; +import { IW3cTraceState } from "./IW3cTraceState"; + +/** + * An object that can be used to populate a new {@link IDistributedTraceContext} instance, + * the included {@link IW3cTraceState} or {@link IOTelTraceState} is used as the parent of the + * created instances traceState + */ +export interface IDistributedTraceInit { + /** + * The unique identifier for the trace that this span belongs to. + * + * The trace ID is a globally unique identifier that connects all spans within a single + * distributed trace. It consists of 16 randomly generated bytes encoded as 32 lowercase + * hexadecimal characters, providing 128 bits of entropy to ensure worldwide uniqueness + * with practically sufficient probability. + * + * @remarks + * - Must be exactly 32 lowercase hexadecimal characters + * - Represents 128 bits (16 bytes) of random data + * - Shared by all spans within the same trace + * - Used for trace correlation across distributed systems + * - Should never be all zeros (invalid trace ID) + * + * @example + * ```typescript + * // Example trace ID format + * const traceId = "4bf92f3577b34da6a3ce929d0e0e4736"; + * + * // All spans in the same trace share this ID + * console.log(parentSpan.spanContext().traceId === childSpan.spanContext().traceId); // true + * ``` + */ + traceId: string; + + /** + * The unique identifier for this specific span within the trace. + * + * The span ID uniquely identifies this span within the trace and is used to establish + * parent-child relationships between spans. It consists of 8 randomly generated bytes + * encoded as 16 lowercase hexadecimal characters, providing 64 bits of entropy to + * ensure global uniqueness with practically sufficient probability. + * + * @remarks + * - Must be exactly 16 lowercase hexadecimal characters + * - Represents 64 bits (8 bytes) of random data + * - Unique within the trace (different spans have different span IDs) + * - Used as parent ID when creating child spans + * - Should never be all zeros (invalid span ID) + * + * @example + * ```typescript + * // Example span ID format + * const spanId = "00f067aa0ba902b7"; + * + * // Each span has a unique ID within the trace + * const parentId = parentSpan.spanContext().spanId; // "00f067aa0ba902b7" + * const childId = childSpan.spanContext().spanId; // "b9c7c989f97918e1" + * + * // Child span uses parent's span ID as its parent ID + * console.log(childSpan.parentSpanId === parentId); // true + * ``` + */ + spanId: string; + + /** + * Indicates whether this span context was propagated from a remote parent span. + * + * This flag distinguishes between spans created locally within the same process + * and spans that represent operations in remote services. Remote spans are typically + * created when trace context is received via HTTP headers, message queues, or other + * inter-process communication mechanisms. + * + * @defaultValue false - spans are considered local unless explicitly marked as remote + * + * @remarks + * - True only when span context was received from another process/service + * - Helps distinguish local vs. distributed trace segments + * - Used by tracing systems for visualization and analysis + * - Local child spans of remote parents are NOT considered remote themselves + * + * @example + * ```typescript + * // HTTP service receiving trace context + * const incomingSpanContext = extractSpanContextFromHeaders(request.headers); + * console.log(incomingSpanContext.isRemote); // true + * + * // Child span created locally + * const localChild = tracer.startSpan('local-operation', { + * parent: incomingSpanContext + * }); + * console.log(localChild.spanContext().isRemote); // false + * ``` + */ + isRemote?: boolean; + + /** + * Trace flags that control trace behavior and indicate sampling decisions. + * + * The trace flags are represented as a single byte (8-bit bitmap) that carries + * trace-level information. The least significant bit (0x01) indicates whether + * the trace is sampled. When this bit is set, it documents that the caller + * may have recorded trace data. Additional bits are reserved for future use + * and should be ignored when not understood. + * + * @remarks + * - Represented as a number (0-255) corresponding to 8 bits + * - Bit 0 (0x01): Sampled flag - indicates trace may contain recorded data + * - Bits 1-7: Reserved for future use, should be preserved during propagation + * - Used by sampling algorithms to make consistent decisions across services + * - See {@link eW3CTraceFlags} for standard flag values + * + * @example + * ```typescript + * // Check if trace is sampled + * const isSampled = (spanContext.traceFlags & 0x01) === 1; + * + * // Common flag values + * const UNSAMPLED = 0x00; // 00000000 - not sampled + * const SAMPLED = 0x01; // 00000001 - sampled + * + * // Preserving unknown flags during propagation + * const preservedFlags = spanContext.traceFlags | 0x01; // Set sampled bit while preserving others + * + * // W3C traceparent header format includes these flags + * const traceparent = `00-${traceId}-${spanId}-${traceFlags.toString(16).padStart(2, '0')}`; + * ``` + */ + traceFlags?: number; + + /** + * Vendor-specific trace state information for cross-system trace correlation. + * + * The trace state carries tracing-system-specific context in a standardized format + * defined by the W3C Trace Context specification. It allows multiple tracing systems + * to participate in the same trace by providing a mechanism for each system to add + * its own metadata without interfering with others. + * + * The trace state is formatted as a comma-separated list of key-value pairs, where + * each pair represents one tracing system's contribution. Keys should be unique + * within the trace state and follow specific naming conventions. + * + * @remarks + * - Maximum of 32 list members allowed + * - Each member format: `key=value` separated by commas + * - Keys should be namespaced to avoid conflicts (e.g., `vendor@system=value`) + * - Total size should not exceed 512 characters for practical header limits + * - Spaces around list members are ignored + * - Preserves vendor-specific information during trace propagation + * + * @see {@link https://www.w3.org/TR/trace-context/#tracestate-field | W3C Trace Context Specification} + * + * @example + * ```typescript + * // Single tracing system + * const singleVendor = { + * get: (key: string) => key === 'rojo' ? '00f067aa0ba902b7' : undefined, + * set: (key: string, value: string) => { ... }, + * unset: (key: string) => { ... }, + * serialize: () => 'rojo=00f067aa0ba902b7' + * }; + * + * // Multiple tracing systems + * const multiVendor = { + * serialize: () => 'rojo=00f067aa0ba902b7,congo=t61rcWkgMzE,vendor@system=custom-value' + * }; + * + * // Accessing trace state + * const rojoValue = spanContext.traceState?.get('rojo'); + * const serialized = spanContext.traceState?.serialize(); + * + * // HTTP header format (When the traceState is an IOTelTraceState) + * headers['tracestate'] = spanContext.traceState?.serialize() || ''; + * + * // HTTP header format (When the traceState is an IW3cTraceState) + * headers['tracestate'] = spanContext.traceState?.hdrs()[0] || ''; + * ``` + */ + traceState?: IW3cTraceState | IOTelTraceState; +} + +export interface IDistributedTraceContext extends IDistributedTraceInit { + + /** + * Returns the current name of the page + */ + getName(): string; + + /** + * Sets the current name of the page, also updates the name for any parent context. + * This is used to identify the page in the telemetry data. + * @remarks This function updates the current and ALL parent contexts with the new name, + * to just update the name of the current context, use the `pageName` property. + * @param pageName - The name of the page + * @deprecated Use the `pageName` property to avoid the side effect of changing the page name of all + * parent contexts. + */ + setName(pageName: string): void; + + /** + * Returns the unique identifier for a trace. All requests / spans from the same trace share the same traceId. + * Must be read from incoming headers or generated according to the W3C TraceContext specification, + * in a hex representation of 16-byte array. A.k.a. trace-id, TraceID or Distributed TraceID + */ + getTraceId(): string; + + /** + * Set the unique identifier for a trace. All requests / spans from the same trace share the same traceId. + * Must be conform to the W3C TraceContext specification, in a hex representation of 16-byte array. + * A.k.a. trace-id, TraceID or Distributed TraceID https://www.w3.org/TR/trace-context/#trace-id + * + * @remarks Sets the traceId for the current context AND all parent contexts, if you want to set the traceId + * for the current context only, use the `traceId` property. + * @param newValue - The traceId to set + * @deprecated Use the `traceId` property to avoid the side effect of changing the traceId of all + * parent contexts. + */ + setTraceId(newValue: string): void; + + /** + * Self-generated 8-bytes identifier of the incoming request. Must be a hex representation of 8-byte array. + * Also know as the parentId, used to link requests together + */ + getSpanId(): string; + + /** + * Self-generated 8-bytes identifier of the incoming request. Must be a hex representation of 8-byte array. + * Also know as the parentId, used to link requests together + * https://www.w3.org/TR/trace-context/#parent-id + * + * @remarks Sets the spanId for the current context AND all parent contexts, if you want to set the spanId for + * the current context only, use the `spanId` property. + * @param newValue - The spanId to set + * @deprecated Use the `spanId` property to avoid the side effect of changing the spanId of all + * parent contexts. + */ + setSpanId(newValue: string): void; + + /** + * An integer representation of the W3C TraceContext trace-flags. + */ + getTraceFlags(): number | undefined; + + /** + * https://www.w3.org/TR/trace-context/#trace-flags + * @remarks Sets the trace flags for the current context and ALL parent contexts, if you want to set the trace + * flags for the current context only, use the `traceFlags` property. + * @param newValue - An integer representation of the W3C TraceContext trace-flags. + * @deprecated Use the `traceFlags` property to avoid the side effect of changing the traceFlags of all + * parent contexts. + */ + setTraceFlags(newValue?: number): void; + + /** + * Returns the current name of the page + * @remarks This function updates the current context only, to update the name of the current and ALL parent contexts, + * use the `setName` method. + * @default undefined + * @since 3.4.0 + */ + pageName: string; + + /** + * The current ID of the trace that this span belongs to. It is worldwide unique + * with practically sufficient probability by being made as 16 randomly + * generated bytes, encoded as a 32 lowercase hex characters corresponding to + * 128 bits. + * @remarks It is NOT recommended that you dynamically change this value after creation and it is actively + * being used as this may affect anyone accessing this context (as a parent for instance). You should logically + * treat this as readonly after creation. + * @remarks If you update this value, it will only update for the current context, not the parent context, + * if you need to update the current and ALL parent contexts, use the `setTraceId` method which + * provides the previous behavior. + * @since 3.4.0 + */ + traceId: string; + + /** + * The ID of the Span. It is globally unique with practically sufficient + * probability by being made as 8 randomly generated bytes, encoded as a 16 + * lowercase hex characters corresponding to 64 bits. + * If you update this value, it will only update for the current context, not the parent context. + * @remarks If you update this value, it will only update for the current context, not the parent context, + * if you need to update the current and ALL parent contexts, use the `setSpanId` method. + * @since 3.4.0 + */ + spanId: string; + + /** + * Returns true if the current context was initialized (propagated) from a remote parent. + * @since 3.4.0 + * @default false + * @returns True if the context was propagated from a remote parent + */ + readonly isRemote: boolean; + + /** + * Trace flags to propagate. + * + * It is represented as 1 byte (bitmap). Bit to represent whether trace is + * sampled or not. When set, the least significant bit documents that the + * caller may have recorded trace data. A caller who does not record trace + * data out-of-band leaves this flag unset. + * + * see {@link eW3CTraceFlags} for valid flag values. + * + * @remarks If you update this value, it will only update for the current context, not the parent context, + * if you need to update the current and ALL parent contexts, use the `setTraceFlags` method. + * @since 3.4.0 + */ + traceFlags?: number; + + /** + * Returns the current trace state which will be used to propgate context across different services. + * Updating (adding / removing keys) of the trace state will modify the current context.IOTelTraceState + * @remarks Unlike the OpenTelemetry {@link IOTelTraceState}, this value is a mutable object, so you can + * modify it directly you do not need to reassign the new value to this property. + * @since 3.4.0 + */ + readonly traceState: IW3cTraceState; + + /** + * Provides access to the parent context of the current context. + * @remarks This is a read-only property, you cannot modify the parent context directly, you can only + * modify the current context. If you need to modify the parent context, you need to do it through the + * current context using the `setTraceId`, `setSpanId`, `setTraceFlags` and `setName` methods. + * @default null + * @since 3.4.0 + */ + readonly parentCtx?: IDistributedTraceContext | null; +} diff --git a/shared/AppInsightsCommon/src/Interfaces/IEventTelemetry.ts b/shared/AppInsightsCore/src/interfaces/ai/IEventTelemetry.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/IEventTelemetry.ts rename to shared/AppInsightsCore/src/interfaces/ai/IEventTelemetry.ts index f072867f1..6cbfd5923 100644 --- a/shared/AppInsightsCommon/src/Interfaces/IEventTelemetry.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IEventTelemetry.ts @@ -1,8 +1,8 @@ -import { IPartC } from "./IPartC"; - // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. +import { IPartC } from "./IPartC"; + export interface IEventTelemetry extends IPartC { /** * @description An event name string diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IExceptionConfig.ts b/shared/AppInsightsCore/src/interfaces/ai/IExceptionConfig.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IExceptionConfig.ts rename to shared/AppInsightsCore/src/interfaces/ai/IExceptionConfig.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/IExceptionTelemetry.ts b/shared/AppInsightsCore/src/interfaces/ai/IExceptionTelemetry.ts similarity index 97% rename from shared/AppInsightsCommon/src/Interfaces/IExceptionTelemetry.ts rename to shared/AppInsightsCore/src/interfaces/ai/IExceptionTelemetry.ts index ff60cc640..c9e845a76 100644 --- a/shared/AppInsightsCommon/src/Interfaces/IExceptionTelemetry.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IExceptionTelemetry.ts @@ -1,8 +1,8 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { SeverityLevel } from "./Contracts/SeverityLevel"; import { IPartC } from "./IPartC"; +import { SeverityLevel } from "./contracts/SeverityLevel"; /** * @export diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IFeatureOptIn.ts b/shared/AppInsightsCore/src/interfaces/ai/IFeatureOptIn.ts similarity index 85% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IFeatureOptIn.ts rename to shared/AppInsightsCore/src/interfaces/ai/IFeatureOptIn.ts index 0503a051a..f05fd9173 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IFeatureOptIn.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IFeatureOptIn.ts @@ -1,4 +1,7 @@ -import { FeatureOptInMode } from "../JavaScriptSDK.Enums/FeatureOptInEnums"; +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { FeatureOptInMode } from "../../enums/ai/FeatureOptInEnums"; export interface IFeatureOptInDetails { /** @@ -10,14 +13,14 @@ export interface IFeatureOptInDetails { * Identifies configuration override values when given feature is enabled * NOTE: should use flat string for fields, for example, if you want to set value for extensionConfig.Ananlytics.disableAjaxTrackig in configurations, * you should use "extensionConfig.Ananlytics.disableAjaxTrackig" as field name: \{["extensionConfig.Analytics.disableAjaxTrackig"]:1\} - * Default: undefined + * @default undefined */ onCfg?: {[field: string]: any}; /** * Identifies configuration override values when given feature is disabled * NOTE: should use flat string for fields, for example, if you want to set value for extensionConfig.Ananlytics.disableAjaxTrackig in configurations, * you should use "extensionConfig.Ananlytics.disableAjaxTrackig" as field name: \{["extensionConfig.Analytics.disableAjaxTrackig"]:1\} - * Default: undefined + * @default undefined */ offCfg?: {[field: string]: any}; /** diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IInstrumentHooks.ts b/shared/AppInsightsCore/src/interfaces/ai/IInstrumentHooks.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IInstrumentHooks.ts rename to shared/AppInsightsCore/src/interfaces/ai/IInstrumentHooks.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/IMetricTelemetry.ts b/shared/AppInsightsCore/src/interfaces/ai/IMetricTelemetry.ts similarity index 91% rename from shared/AppInsightsCommon/src/Interfaces/IMetricTelemetry.ts rename to shared/AppInsightsCore/src/interfaces/ai/IMetricTelemetry.ts index 0c391b723..0146009fb 100644 --- a/shared/AppInsightsCommon/src/Interfaces/IMetricTelemetry.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IMetricTelemetry.ts @@ -1,3 +1,6 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + import { IPartC } from "./IPartC"; // Copyright (c) Microsoft Corporation. All rights reserved. diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/INetworkStatsbeat.ts b/shared/AppInsightsCore/src/interfaces/ai/INetworkStatsbeat.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/INetworkStatsbeat.ts rename to shared/AppInsightsCore/src/interfaces/ai/INetworkStatsbeat.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/INotificationListener.ts b/shared/AppInsightsCore/src/interfaces/ai/INotificationListener.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/INotificationListener.ts rename to shared/AppInsightsCore/src/interfaces/ai/INotificationListener.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/INotificationManager.ts b/shared/AppInsightsCore/src/interfaces/ai/INotificationManager.ts similarity index 95% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/INotificationManager.ts rename to shared/AppInsightsCore/src/interfaces/ai/INotificationManager.ts index 54149d50a..4457d643a 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/INotificationManager.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/INotificationManager.ts @@ -1,9 +1,9 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. import { IPromise } from "@nevware21/ts-async"; -import { INotificationListener } from "../JavaScriptSDK.Interfaces/INotificationListener"; -import { ITelemetryItem } from "../JavaScriptSDK.Interfaces/ITelemetryItem"; +import { INotificationListener } from "./INotificationListener"; import { IPerfEvent } from "./IPerfEvent"; +import { ITelemetryItem } from "./ITelemetryItem"; import { IPayloadData } from "./IXHROverride"; /** diff --git a/shared/AppInsightsCommon/src/Interfaces/IPageViewPerformanceTelemetry.ts b/shared/AppInsightsCore/src/interfaces/ai/IPageViewPerformanceTelemetry.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/IPageViewPerformanceTelemetry.ts rename to shared/AppInsightsCore/src/interfaces/ai/IPageViewPerformanceTelemetry.ts index 044b3d908..c5ba738b4 100644 --- a/shared/AppInsightsCommon/src/Interfaces/IPageViewPerformanceTelemetry.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IPageViewPerformanceTelemetry.ts @@ -1,8 +1,8 @@ -import { IPartC } from "./IPartC"; - // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. +import { IPartC } from "./IPartC"; + export interface IPageViewPerformanceTelemetry extends IPartC { /** * name String - The name of the page. Defaults to the document title. diff --git a/shared/AppInsightsCommon/src/Interfaces/IPageViewTelemetry.ts b/shared/AppInsightsCore/src/interfaces/ai/IPageViewTelemetry.ts similarity index 94% rename from shared/AppInsightsCommon/src/Interfaces/IPageViewTelemetry.ts rename to shared/AppInsightsCore/src/interfaces/ai/IPageViewTelemetry.ts index 4a43926b4..aff08109f 100644 --- a/shared/AppInsightsCommon/src/Interfaces/IPageViewTelemetry.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IPageViewTelemetry.ts @@ -1,3 +1,6 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + import { IPartC } from "./IPartC"; // Copyright (c) Microsoft Corporation. All rights reserved. diff --git a/shared/AppInsightsCommon/src/Interfaces/IPartC.ts b/shared/AppInsightsCore/src/interfaces/ai/IPartC.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/IPartC.ts rename to shared/AppInsightsCore/src/interfaces/ai/IPartC.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IPerfEvent.ts b/shared/AppInsightsCore/src/interfaces/ai/IPerfEvent.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IPerfEvent.ts rename to shared/AppInsightsCore/src/interfaces/ai/IPerfEvent.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IPerfManager.ts b/shared/AppInsightsCore/src/interfaces/ai/IPerfManager.ts similarity index 95% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IPerfManager.ts rename to shared/AppInsightsCore/src/interfaces/ai/IPerfManager.ts index a8d34ed31..d23a30581 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IPerfManager.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IPerfManager.ts @@ -1,6 +1,6 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IPerfEvent } from "./../JavaScriptSDK.Interfaces/IPerfEvent"; +import { IPerfEvent } from "./IPerfEvent"; /** * Identifies an interface to a host that can provide an IPerfManager implementation diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IProcessTelemetryContext.ts b/shared/AppInsightsCore/src/interfaces/ai/IProcessTelemetryContext.ts similarity index 94% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IProcessTelemetryContext.ts rename to shared/AppInsightsCore/src/interfaces/ai/IProcessTelemetryContext.ts index 891afde5e..a58cd0671 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IProcessTelemetryContext.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IProcessTelemetryContext.ts @@ -2,7 +2,7 @@ // Licensed under the MIT License. "use strict"; -import { IConfigDefaults } from "../Config/IConfigDefaults"; +import { IConfigDefaults } from "../config/IConfigDefaults"; import { IAppInsightsCore } from "./IAppInsightsCore"; import { IConfiguration } from "./IConfiguration"; import { IDiagnosticLogger } from "./IDiagnosticLogger"; @@ -35,9 +35,12 @@ export interface IBaseProcessingContext { getCfg: () => IConfiguration; /** - * Gets the named extension config + * Gets the named extension configuration + * @param identifier - The named extension identifier + * @param defaultValue - The default value(s) to return if no defined config exists + * @param rootOnly - If true, only the look for the configuration in the top level and not in the "extensionConfig" */ - getExtCfg: (identifier: string, defaultValue?: IConfigDefaults) => T; + getExtCfg: (identifier: string, defaultValue?: IConfigDefaults, rootOnly?: boolean) => T; /** * Gets the named config from either the named identifier extension or core config if neither exist then the diff --git a/shared/AppInsightsCommon/src/Interfaces/IPropertiesPlugin.ts b/shared/AppInsightsCore/src/interfaces/ai/IPropertiesPlugin.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/IPropertiesPlugin.ts rename to shared/AppInsightsCore/src/interfaces/ai/IPropertiesPlugin.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/IRequestContext.ts b/shared/AppInsightsCore/src/interfaces/ai/IRequestContext.ts similarity index 64% rename from shared/AppInsightsCommon/src/Interfaces/IRequestContext.ts rename to shared/AppInsightsCore/src/interfaces/ai/IRequestContext.ts index 1f20f7751..14823ef1d 100644 --- a/shared/AppInsightsCommon/src/Interfaces/IRequestContext.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IRequestContext.ts @@ -1,3 +1,6 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + export interface IRequestContext { status?: number; xhr?: XMLHttpRequest; diff --git a/shared/AppInsightsCore/src/interfaces/ai/IRequestTelemetry.ts b/shared/AppInsightsCore/src/interfaces/ai/IRequestTelemetry.ts new file mode 100644 index 000000000..f277b1c7d --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/ai/IRequestTelemetry.ts @@ -0,0 +1,44 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IPartC } from "./IPartC"; + +/** + * This defines the contract for request telemetry items that are passed to Application Insights API (track) + */ +export interface IRequestTelemetry extends IPartC { + /** + * Identifier of a request call instance. Used for correlation between request and other telemetry items. + */ + id: string; + + /** + * Name of the request. Represents code path taken to process request. Low cardinality value to allow better grouping of requests. For HTTP requests it represents the HTTP method and URL path template like 'GET /values/\{id\}'. + */ + name?: string; + + /** + * Request duration in milliseconds. + */ + duration: number; + + /** + * Indication of successful or unsuccessful call. + */ + success: boolean; + + /** + * Result of a request execution. HTTP status code for HTTP requests. + */ + responseCode: number; + + /** + * Source of the request. Examples are the instrumentation key of the caller or the ip address of the caller. + */ + source?: string; + + /** + * Request URL with all query string parameters. + */ + url?: string; +} \ No newline at end of file diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ISenderPostManager.ts b/shared/AppInsightsCore/src/interfaces/ai/ISenderPostManager.ts similarity index 97% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ISenderPostManager.ts rename to shared/AppInsightsCore/src/interfaces/ai/ISenderPostManager.ts index c98011931..38d4135c6 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ISenderPostManager.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/ISenderPostManager.ts @@ -1,5 +1,8 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + import { ITimerHandler } from "@nevware21/ts-utils"; -import { TransportType } from "../JavaScriptSDK.Enums/SendRequestReason"; +import { TransportType } from "../../enums/ai/SendRequestReason"; import { IXDomainRequest } from "./IXDomainRequest"; import { IPayloadData, IXHROverride, OnCompleteCallback } from "./IXHROverride"; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IStatsBeat.ts b/shared/AppInsightsCore/src/interfaces/ai/IStatsBeat.ts similarity index 98% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IStatsBeat.ts rename to shared/AppInsightsCore/src/interfaces/ai/IStatsBeat.ts index ee91549bb..09468efab 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IStatsBeat.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IStatsBeat.ts @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { StatsType } from "../JavaScriptSDK.Enums/StatsType"; +import { StatsType } from "../../enums/ai/StatsType"; import { IPayloadData } from "./IXHROverride"; /** diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IStatsEventData.ts b/shared/AppInsightsCore/src/interfaces/ai/IStatsEventData.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IStatsEventData.ts rename to shared/AppInsightsCore/src/interfaces/ai/IStatsEventData.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IStatsMgr.ts b/shared/AppInsightsCore/src/interfaces/ai/IStatsMgr.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IStatsMgr.ts rename to shared/AppInsightsCore/src/interfaces/ai/IStatsMgr.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/IStorageBuffer.ts b/shared/AppInsightsCore/src/interfaces/ai/IStorageBuffer.ts similarity index 77% rename from shared/AppInsightsCommon/src/Interfaces/IStorageBuffer.ts rename to shared/AppInsightsCore/src/interfaces/ai/IStorageBuffer.ts index c65cb1e85..ce7e80a3c 100644 --- a/shared/AppInsightsCommon/src/Interfaces/IStorageBuffer.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IStorageBuffer.ts @@ -1,4 +1,7 @@ -import { IDiagnosticLogger } from "@microsoft/applicationinsights-core-js"; +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IDiagnosticLogger } from "./IDiagnosticLogger"; /** * Identifies a simple interface to allow you to override the storage mechanism used diff --git a/shared/AppInsightsCommon/src/Interfaces/ITelemetryContext.ts b/shared/AppInsightsCore/src/interfaces/ai/ITelemetryContext.ts similarity index 68% rename from shared/AppInsightsCommon/src/Interfaces/ITelemetryContext.ts rename to shared/AppInsightsCore/src/interfaces/ai/ITelemetryContext.ts index a4a991f18..3b2c65cb6 100644 --- a/shared/AppInsightsCommon/src/Interfaces/ITelemetryContext.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/ITelemetryContext.ts @@ -1,16 +1,16 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IApplication } from "./Context/IApplication"; -import { IDevice } from "./Context/IDevice"; -import { IInternal } from "./Context/IInternal"; -import { ILocation } from "./Context/ILocation"; -import { IOperatingSystem } from "./Context/IOperatingSystem"; -import { ISession } from "./Context/ISession"; -import { ISessionManager } from "./Context/ISessionManager"; -import { ITelemetryTrace } from "./Context/ITelemetryTrace"; -import { IUserContext } from "./Context/IUser"; -import { IWeb } from "./Context/IWeb"; +import { IApplication } from "./context/IApplication"; +import { IDevice } from "./context/IDevice"; +import { IInternal } from "./context/IInternal"; +import { ILocation } from "./context/ILocation"; +import { IOperatingSystem } from "./context/IOperatingSystem"; +import { ISession } from "./context/ISession"; +import { ISessionManager } from "./context/ISessionManager"; +import { ITelemetryTrace } from "./context/ITelemetryTrace"; +import { IUserContext } from "./context/IUser"; +import { IWeb } from "./context/IWeb"; export interface ITelemetryContext { /** @@ -35,6 +35,8 @@ export interface ITelemetryContext { /** * The object describing a operation tracked by this object. + * @deprecated Use the core getTraceCtx method instead to get / set the current trace context, this is required to + * support distributed tracing and allows the core to manage the trace context. */ readonly telemetryTrace: ITelemetryTrace; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryInitializers.ts b/shared/AppInsightsCore/src/interfaces/ai/ITelemetryInitializers.ts similarity index 88% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryInitializers.ts rename to shared/AppInsightsCore/src/interfaces/ai/ITelemetryInitializers.ts index cdcad6b81..21d7787f2 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryInitializers.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/ITelemetryInitializers.ts @@ -1,3 +1,6 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + import { ITelemetryItem } from "./ITelemetryItem"; import { ILegacyUnloadHook } from "./IUnloadHook"; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryItem.ts b/shared/AppInsightsCore/src/interfaces/ai/ITelemetryItem.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryItem.ts rename to shared/AppInsightsCore/src/interfaces/ai/ITelemetryItem.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryPlugin.ts b/shared/AppInsightsCore/src/interfaces/ai/ITelemetryPlugin.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryPlugin.ts rename to shared/AppInsightsCore/src/interfaces/ai/ITelemetryPlugin.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryPluginChain.ts b/shared/AppInsightsCore/src/interfaces/ai/ITelemetryPluginChain.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryPluginChain.ts rename to shared/AppInsightsCore/src/interfaces/ai/ITelemetryPluginChain.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryUnloadState.ts b/shared/AppInsightsCore/src/interfaces/ai/ITelemetryUnloadState.ts similarity index 72% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryUnloadState.ts rename to shared/AppInsightsCore/src/interfaces/ai/ITelemetryUnloadState.ts index cf3640591..4ae741734 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryUnloadState.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/ITelemetryUnloadState.ts @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { TelemetryUnloadReason } from "../JavaScriptSDK.Enums/TelemetryUnloadReason"; +import { TelemetryUnloadReason } from "../../enums/ai/TelemetryUnloadReason"; export interface ITelemetryUnloadState { reason: TelemetryUnloadReason; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryUpdateState.ts b/shared/AppInsightsCore/src/interfaces/ai/ITelemetryUpdateState.ts similarity index 93% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryUpdateState.ts rename to shared/AppInsightsCore/src/interfaces/ai/ITelemetryUpdateState.ts index dc621366a..1cab0e568 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITelemetryUpdateState.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/ITelemetryUpdateState.ts @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { TelemetryUpdateReason } from "../JavaScriptSDK.Enums/TelemetryUpdateReason"; +import { TelemetryUpdateReason } from "../../enums/ai/TelemetryUpdateReason"; import { IConfiguration } from "./IConfiguration"; import { IPlugin } from "./ITelemetryPlugin"; diff --git a/shared/AppInsightsCommon/src/Interfaces/IThrottleMgr.ts b/shared/AppInsightsCore/src/interfaces/ai/IThrottleMgr.ts similarity index 96% rename from shared/AppInsightsCommon/src/Interfaces/IThrottleMgr.ts rename to shared/AppInsightsCore/src/interfaces/ai/IThrottleMgr.ts index 9fe41a91a..12a836235 100644 --- a/shared/AppInsightsCommon/src/Interfaces/IThrottleMgr.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IThrottleMgr.ts @@ -1,3 +1,5 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. /** * Identifies limit number/percentage of items sent per time diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITraceParent.ts b/shared/AppInsightsCore/src/interfaces/ai/ITraceParent.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/ITraceParent.ts rename to shared/AppInsightsCore/src/interfaces/ai/ITraceParent.ts diff --git a/shared/AppInsightsCore/src/interfaces/ai/ITraceProvider.ts b/shared/AppInsightsCore/src/interfaces/ai/ITraceProvider.ts new file mode 100644 index 000000000..80ccb630d --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/ai/ITraceProvider.ts @@ -0,0 +1,157 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IOTelApi } from "../otel/IOTelApi"; +import { IOTelSpanOptions } from "../otel/trace/IOTelSpanOptions"; +import { IReadableSpan } from "../otel/trace/IReadableSpan"; +import { IConfiguration } from "./IConfiguration"; +import { IDistributedTraceContext } from "./IDistributedTraceContext"; + +/** + * A trace provider interface that enables different SKUs to provide their own + * span implementations while being managed by the core SDK. + * + * This follows the OpenTelemetry TraceProvider pattern, allowing the core to + * delegate span creation to the appropriate implementation based on the SDK variant. + * + * @since 3.4.0 + */ +export interface ITraceProvider { + /** + * The OpenTelemetry API instance associated with this trace provider. + * This provides access to the tracer provider and other OpenTelemetry functionality. + * @since 3.4.0 + */ + readonly api: IOTelApi; + + /** + * Creates a new span with the given name and options. + * + * @param name - The name of the span + * @param options - Options for creating the span (kind, attributes, startTime) + * @param parent - Optional parent context. If not provided, uses the current active trace context + * @returns A new span instance specific to this provider's implementation + * @since 3.4.0 + */ + createSpan(name: string, options?: IOTelSpanOptions, parent?: IDistributedTraceContext): IReadableSpan; + + /** + * Gets the provider identifier for debugging and logging purposes. + * @returns A string identifying this trace provider implementation + * @since 3.4.0 + */ + getProviderId(): string; + + /** + * Determines if this provider is available and ready to create spans. + * @returns true if the provider can create spans, false otherwise + * @since 3.4.0 + */ + isAvailable(): boolean; +} + +/** + * Interface for OpenTelemetry trace operations. + * This interface provides span creation, context management, and trace provider operations + * that are common across different SDK implementations (Core, AISKU, etc.). + * + * @since 3.4.0 + */ +export interface ITraceHost { + + /* + * Config object that was used to initialize AppInsights / ITraceHost + */ + readonly config: CfgType; + + /** + * Gets the current distributed trace active context for this instance + * @param createNew - Optional flag to create a new instance if one doesn't currently exist, defaults to true. By default this + * will use any located parent as defined by the {@link IConfiguration.traceHdrMode} configuration for each new instance created. + */ + getTraceCtx(createNew?: boolean): IDistributedTraceContext | null; + + /** + * Sets the current distributed trace context for this instance if available + */ + setTraceCtx(newTraceCtx: IDistributedTraceContext | null | undefined): void; + + /** + * Start a new span with the given name and optional parent context. + * + * Note: This method only creates and returns the span. It does not automatically + * set the span as the active trace context. Context management should be handled + * separately using setTraceCtx() if needed. + * + * @param name - The name of the span + * @param options - Options for creating the span (kind, attributes, startTime) + * @param parent - Optional parent context. If not provided, uses the current active trace context + * @returns A new span instance, or null if no trace provider is available + * @since 3.4.0 + * + * @see {@link IReadableSpan} - Interface for individual spans + * @see {@link IOTelSpanOptions} - Configuration options for span creation + */ + startSpan(name: string, options?: IOTelSpanOptions, parent?: IDistributedTraceContext): IReadableSpan | null; + + /** + * Return the current active span, if no trace provider is available null will be returned + * but when a trace provider is available a span instance will always be returned, even if + * there is no active span (in which case a non-recording span will be returned). + * @param createNew - Optional flag to create a non-recording span if no active span exists, defaults to true. + * When false, returns the existing active span or null without creating a non-recording span, which can improve + * performance when only checking if an active span exists. + * @returns The current active span or null if no trace provider is available or if createNew is false and no active span exists + * @since 3.4.0 + */ + getActiveSpan(createNew?: boolean): IReadableSpan | null; + + /** + * Set the current Active Span, if no trace provider is available the span will be not be set as the active span. + * @param span - The span to set as the active span + * @returns An ISpanScope instance that provides the current scope, the span will always be the span passed in + * even when no trace provider is available + * @since 3.4.0 + */ + setActiveSpan(span: IReadableSpan): ISpanScope + + /** + * Get the current trace provider. + * + * @returns The current trace provider, or null if none is set + * @since 3.4.0 + */ + getTraceProvider(): ITraceProvider | null; +} + +/** + * Represents the execution scope for a span, combining the trace instance and the active span. + * This interface is used as the context for executing functions within a span's scope. + * + * @since 3.4.0 + */ +export interface ISpanScope { + /** + * The trace host (core or AISKU instance). + * @since 3.4.0 + */ + readonly host: T; + + /** + * The active span for this execution scope. + * @since 3.4.0 + */ + readonly span: IReadableSpan; + + /** + * The previously active span before this scope was created, if any. + * @since 3.4.0 + */ + readonly prvSpan?: IReadableSpan; + + /** + * Restores the previous active span in the trace instance. + * @since 3.4.0 + */ + restore(): void; +} diff --git a/shared/AppInsightsCommon/src/Interfaces/ITraceTelemetry.ts b/shared/AppInsightsCore/src/interfaces/ai/ITraceTelemetry.ts similarity index 89% rename from shared/AppInsightsCommon/src/Interfaces/ITraceTelemetry.ts rename to shared/AppInsightsCore/src/interfaces/ai/ITraceTelemetry.ts index 2c188d8e1..d4e2e38ef 100644 --- a/shared/AppInsightsCommon/src/Interfaces/ITraceTelemetry.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/ITraceTelemetry.ts @@ -1,8 +1,8 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { SeverityLevel } from "./Contracts/SeverityLevel"; import { IPartC } from "./IPartC"; +import { SeverityLevel } from "./contracts/SeverityLevel"; export interface ITraceTelemetry extends IPartC { /** diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IUnloadHook.ts b/shared/AppInsightsCore/src/interfaces/ai/IUnloadHook.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IUnloadHook.ts rename to shared/AppInsightsCore/src/interfaces/ai/IUnloadHook.ts diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IUnloadableComponent.ts b/shared/AppInsightsCore/src/interfaces/ai/IUnloadableComponent.ts similarity index 100% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IUnloadableComponent.ts rename to shared/AppInsightsCore/src/interfaces/ai/IUnloadableComponent.ts diff --git a/shared/AppInsightsCore/src/interfaces/ai/IW3cTraceState.ts b/shared/AppInsightsCore/src/interfaces/ai/IW3cTraceState.ts new file mode 100644 index 000000000..3ff00582d --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/ai/IW3cTraceState.ts @@ -0,0 +1,78 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +/** + * Represents a mutable [W3C trace state list](https://www.w3.org/TR/trace-context/#tracestate-header), this is a + * list of key/value pairs that are used to pass trace state information between different tracing systems. The + * list is ordered and the order is important as it determines the processing order. + * + * Importantly instances of this type are mutable, change made to an instance via {@link IW3cTraceState.set} or + * {@link IW3cTraceState.del} will be reflected on the instance and any child instances that use it as a parent. + * However, any parent instance associated with an instance will not be modified by operations on that particular + * instance. + * + * @since 3.4.0 + */ +export interface IW3cTraceState { + /** + * Returns a readonly array of the current keys associated with the trace state, keys are returned in the + * required processing order and if this instance has a parent the keys from the parent will be included + * unless they have been removed (deleted) from the child instance. + * Once created any modifications to the parent will also be reflected in the child, this is different from + * the OpenTelemetry implementation which creates a new instance for each call. + * @returns A readonly array of the current keys associated with the trace state + */ + readonly keys: string[]; + + /** + * Check if the trace state list is empty, meaning it has no keys or values. + * This exists to allow for quick checks without needing to create a new array of keys. + * @since 3.4.0 + * @returns true if the trace state list is empty, false otherwise + */ + readonly isEmpty: boolean; + + /** + * Get the value for the specified key that is associated with this instance, either directly or from the parent. + * @param key - The key to lookup + * @returns The value for the key, or undefined if not found + */ + get(key: string): string | undefined; + + /** + * Set the value for the specified key for this instance, returning its new location within the list. + * - 0 is the front of the list + * - -1 not set because the key/value pair is invalid + * If the key already exists it will be removed from its current location and added to the front of the list. And + * if the key was in the parent this will override the value inherited from the parent, more importantly it will + * not modify the parent value. + * @param key - The key to set + * @param value - The value to set + * @returns 0 if successful, -1 if not + */ + set(key: string, value: string): number; + + /** + * Delete the specified key from this instance, if the key was in the parent it will be removed (hidden) from + * this instance but will still be available directly from the parent. + * @param key - The key to delete + */ + del(key: string): void; + + /** + * Format the trace state list into a strings where each string can be used as a header value. + * This will return an empty array if the trace state list is empty. + * @param maxHeaders - The maximum number of entries to include in the output, once the limit is reached no more entries will be included + * @param maxKeys - The maximum number of keys to include in the output, once the limit is reached no more keys will be included + * @param maxLen - The maximum length of each header value, once the limit is reached a new header value will be created + * @returns An array of strings that can be used for the header values, if the trace state list is empty an empty array will be returned + */ + hdrs(maxHeaders?: number, maxKeys?: number, maxLen?: number): string[]; + + /** + * Create a new instance of IW3cTraceState which is a child of this instance, meaning it will inherit the keys + * and values from this instance but any changes made to the child will not affect this instance. + * @returns A new instance of IW3cTraceState which is a child of this instance + */ + child(): IW3cTraceState; +} diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IXDomainRequest.ts b/shared/AppInsightsCore/src/interfaces/ai/IXDomainRequest.ts similarity index 92% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IXDomainRequest.ts rename to shared/AppInsightsCore/src/interfaces/ai/IXDomainRequest.ts index 6d58d3548..b14c9cafa 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IXDomainRequest.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IXDomainRequest.ts @@ -1,3 +1,6 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + export interface IBackendResponse { /** * Number of items received by the backend diff --git a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IXHROverride.ts b/shared/AppInsightsCore/src/interfaces/ai/IXHROverride.ts similarity index 82% rename from shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IXHROverride.ts rename to shared/AppInsightsCore/src/interfaces/ai/IXHROverride.ts index 4d34a9d49..b56547709 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK.Interfaces/IXHROverride.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/IXHROverride.ts @@ -1,5 +1,8 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + import { IPromise } from "@nevware21/ts-async"; -import { SendRequestReason } from "../applicationinsights-core-js"; +import { SendRequestReason } from "../../enums/ai/SendRequestReason"; /** IPayloadData describes interface of payload sent via POST channel */ export interface IPayloadData { @@ -20,9 +23,7 @@ export interface IPayloadData { */ export type SendPOSTFunction = (payload: IPayloadData, oncomplete: OnCompleteCallback, sync?: boolean) => void | IPromise; -export type OnCompleteCallback = (status: number, headers: { [headerName: string]: string; }, response?: string) => void; - - +export type OnCompleteCallback = (status: number, headers: { [headerName: string]: string; }, response?: string, payload?: IPayloadData) => void; /** * The IXHROverride interface overrides the way HTTP requests are sent. @@ -30,6 +31,3 @@ export type OnCompleteCallback = (status: number, headers: { [headerName: string export interface IXHROverride { sendPOST: SendPOSTFunction; } - - - diff --git a/shared/AppInsightsCore/src/interfaces/ai/PartAExtensions.ts b/shared/AppInsightsCore/src/interfaces/ai/PartAExtensions.ts new file mode 100644 index 000000000..7905a6501 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/ai/PartAExtensions.ts @@ -0,0 +1,17 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { ContextTagKeys } from "./contracts/ContextTagKeys"; + +export const Extensions = { + UserExt: "user", + DeviceExt: "device", + TraceExt: "trace", + WebExt: "web", + AppExt: "app", + OSExt: "os", + SessionExt: "ses", + SDKExt: "sdk" +}; + +export let CtxTagKeys = (/* #__PURE__ */ new ContextTagKeys()); \ No newline at end of file diff --git a/shared/AppInsightsCommon/src/Interfaces/Context/IApplication.ts b/shared/AppInsightsCore/src/interfaces/ai/context/IApplication.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Context/IApplication.ts rename to shared/AppInsightsCore/src/interfaces/ai/context/IApplication.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Context/IDevice.ts b/shared/AppInsightsCore/src/interfaces/ai/context/IDevice.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Context/IDevice.ts rename to shared/AppInsightsCore/src/interfaces/ai/context/IDevice.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Context/IInternal.ts b/shared/AppInsightsCore/src/interfaces/ai/context/IInternal.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Context/IInternal.ts rename to shared/AppInsightsCore/src/interfaces/ai/context/IInternal.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Context/ILocation.ts b/shared/AppInsightsCore/src/interfaces/ai/context/ILocation.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Context/ILocation.ts rename to shared/AppInsightsCore/src/interfaces/ai/context/ILocation.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Context/IOperatingSystem.ts b/shared/AppInsightsCore/src/interfaces/ai/context/IOperatingSystem.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Context/IOperatingSystem.ts rename to shared/AppInsightsCore/src/interfaces/ai/context/IOperatingSystem.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Context/ISample.ts b/shared/AppInsightsCore/src/interfaces/ai/context/ISample.ts similarity index 76% rename from shared/AppInsightsCommon/src/Interfaces/Context/ISample.ts rename to shared/AppInsightsCore/src/interfaces/ai/context/ISample.ts index cc46225ef..8da501828 100644 --- a/shared/AppInsightsCommon/src/Interfaces/Context/ISample.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/context/ISample.ts @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { ITelemetryItem } from "@microsoft/applicationinsights-core-js"; +import { ITelemetryItem } from "../ITelemetryItem"; export interface ISample { /** diff --git a/shared/AppInsightsCommon/src/Interfaces/Context/ISession.ts b/shared/AppInsightsCore/src/interfaces/ai/context/ISession.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Context/ISession.ts rename to shared/AppInsightsCore/src/interfaces/ai/context/ISession.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Context/ISessionManager.ts b/shared/AppInsightsCore/src/interfaces/ai/context/ISessionManager.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Context/ISessionManager.ts rename to shared/AppInsightsCore/src/interfaces/ai/context/ISessionManager.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Context/ITelemetryTrace.ts b/shared/AppInsightsCore/src/interfaces/ai/context/ITelemetryTrace.ts similarity index 64% rename from shared/AppInsightsCommon/src/Interfaces/Context/ITelemetryTrace.ts rename to shared/AppInsightsCore/src/interfaces/ai/context/ITelemetryTrace.ts index 4b12ebf00..6df218204 100644 --- a/shared/AppInsightsCommon/src/Interfaces/Context/ITelemetryTrace.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/context/ITelemetryTrace.ts @@ -1,6 +1,11 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. +/** + * Interface for telemetry trace context. + * @deprecated Use the core getTraceCtx method instead to get / set the current trace context, this is required to + * support distributed tracing and allows the core to manage the trace context. + */ export interface ITelemetryTrace { /** * Trace id @@ -12,12 +17,6 @@ export interface ITelemetryTrace { */ parentID?: string; - /** - * @deprecated Never Used - * Trace state - */ - traceState?: ITraceState; - /** * An integer representation of the W3C TraceContext trace-flags. https://www.w3.org/TR/trace-context/#trace-flags */ @@ -28,6 +27,3 @@ export interface ITelemetryTrace { */ name?: string; } - -export interface ITraceState { -} \ No newline at end of file diff --git a/shared/AppInsightsCommon/src/Interfaces/Context/IUser.ts b/shared/AppInsightsCore/src/interfaces/ai/context/IUser.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Context/IUser.ts rename to shared/AppInsightsCore/src/interfaces/ai/context/IUser.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Context/IWeb.ts b/shared/AppInsightsCore/src/interfaces/ai/context/IWeb.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Context/IWeb.ts rename to shared/AppInsightsCore/src/interfaces/ai/context/IWeb.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/AvailabilityData.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/AvailabilityData.ts similarity index 92% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/AvailabilityData.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/AvailabilityData.ts index abf701029..bd1165c5b 100644 --- a/shared/AppInsightsCommon/src/Interfaces/Contracts/AvailabilityData.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/contracts/AvailabilityData.ts @@ -1,5 +1,5 @@ -// // Copyright (c) Microsoft Corporation. All rights reserved. -// // Licensed under the MIT License. +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. // // THIS FILE WAS AUTOGENERATED // import { Domain } from "./Domain"; diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/ContextTagKeys.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/ContextTagKeys.ts similarity index 99% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/ContextTagKeys.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/ContextTagKeys.ts index c02da207c..4576d2d54 100644 --- a/shared/AppInsightsCommon/src/Interfaces/Contracts/ContextTagKeys.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/contracts/ContextTagKeys.ts @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { createClassFromInterface } from "@microsoft/applicationinsights-core-js"; +import { createClassFromInterface } from "../../../utils/HelperFuncs"; function _aiNameFunc(baseName: string) { let aiName = "ai." + baseName + "."; diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/DataPointType.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/DataPointType.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/DataPointType.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/DataPointType.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/DependencyKind.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/DependencyKind.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/DependencyKind.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/DependencyKind.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/DependencySourceType.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/DependencySourceType.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/DependencySourceType.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/DependencySourceType.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IBase.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IBase.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IBase.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IBase.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IData.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IData.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IData.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IData.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IDataPoint.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IDataPoint.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IDataPoint.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IDataPoint.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IDomain.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IDomain.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IDomain.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IDomain.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IEventData.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IEventData.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IEventData.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IEventData.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IExceptionData.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IExceptionData.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IExceptionData.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IExceptionData.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IExceptionDetails.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IExceptionDetails.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IExceptionDetails.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IExceptionDetails.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IMessageData.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IMessageData.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IMessageData.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IMessageData.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IMetricData.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IMetricData.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IMetricData.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IMetricData.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IPageViewData.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IPageViewData.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IPageViewData.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IPageViewData.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IPageViewPerfData.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IPageViewPerfData.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IPageViewPerfData.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IPageViewPerfData.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IRemoteDependencyData.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IRemoteDependencyData.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IRemoteDependencyData.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IRemoteDependencyData.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/IStackFrame.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/IStackFrame.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/IStackFrame.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/IStackFrame.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Contracts/SeverityLevel.ts b/shared/AppInsightsCore/src/interfaces/ai/contracts/SeverityLevel.ts similarity index 90% rename from shared/AppInsightsCommon/src/Interfaces/Contracts/SeverityLevel.ts rename to shared/AppInsightsCore/src/interfaces/ai/contracts/SeverityLevel.ts index fba12fca0..f9b2dc795 100644 --- a/shared/AppInsightsCommon/src/Interfaces/Contracts/SeverityLevel.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/contracts/SeverityLevel.ts @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { createEnumStyle } from "@microsoft/applicationinsights-core-js"; +import { createEnumStyle } from "../../../enums/EnumHelperFuncs"; /** * Defines the level of severity for the event. diff --git a/shared/AppInsightsCommon/src/Interfaces/Telemetry/IEnvelope.ts b/shared/AppInsightsCore/src/interfaces/ai/telemetry/IEnvelope.ts similarity index 100% rename from shared/AppInsightsCommon/src/Interfaces/Telemetry/IEnvelope.ts rename to shared/AppInsightsCore/src/interfaces/ai/telemetry/IEnvelope.ts diff --git a/shared/AppInsightsCommon/src/Interfaces/Telemetry/ISerializable.ts b/shared/AppInsightsCore/src/interfaces/ai/telemetry/ISerializable.ts similarity index 73% rename from shared/AppInsightsCommon/src/Interfaces/Telemetry/ISerializable.ts rename to shared/AppInsightsCore/src/interfaces/ai/telemetry/ISerializable.ts index 71db06e0c..a203e5ebb 100644 --- a/shared/AppInsightsCommon/src/Interfaces/Telemetry/ISerializable.ts +++ b/shared/AppInsightsCore/src/interfaces/ai/telemetry/ISerializable.ts @@ -1,11 +1,13 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. +import { FieldType } from "../../../enums/ai/Enums"; + export interface ISerializable { /** * The set of fields for a serializable object. * This defines the serialization order and a value of true/false * for each field defines whether the field is required or not. */ - aiDataContract: any; + aiDataContract: { [key: string]: FieldType | (() => FieldType) }; } \ No newline at end of file diff --git a/shared/AppInsightsCore/src/Config/IConfigDefaults.ts b/shared/AppInsightsCore/src/interfaces/config/IConfigDefaults.ts similarity index 97% rename from shared/AppInsightsCore/src/Config/IConfigDefaults.ts rename to shared/AppInsightsCore/src/interfaces/config/IConfigDefaults.ts index be5eba12f..d9a9b5240 100644 --- a/shared/AppInsightsCore/src/Config/IConfigDefaults.ts +++ b/shared/AppInsightsCore/src/interfaces/config/IConfigDefaults.ts @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; +import { IConfiguration } from "../ai/IConfiguration"; /** * The type to identify whether the default value should be applied in preference to the provided value. diff --git a/shared/AppInsightsCore/src/Config/IDynamicConfigHandler.ts b/shared/AppInsightsCore/src/interfaces/config/IDynamicConfigHandler.ts similarity index 96% rename from shared/AppInsightsCore/src/Config/IDynamicConfigHandler.ts rename to shared/AppInsightsCore/src/interfaces/config/IDynamicConfigHandler.ts index 4c98979fa..878fdbb01 100644 --- a/shared/AppInsightsCore/src/Config/IDynamicConfigHandler.ts +++ b/shared/AppInsightsCore/src/interfaces/config/IDynamicConfigHandler.ts @@ -1,8 +1,8 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { IDiagnosticLogger } from "../JavaScriptSDK.Interfaces/IDiagnosticLogger"; +import { IConfiguration } from "../ai/IConfiguration"; +import { IDiagnosticLogger } from "../ai/IDiagnosticLogger"; import { IConfigDefaults } from "./IConfigDefaults"; import { IWatcherHandler, WatcherFunction } from "./IDynamicWatcher"; @@ -109,4 +109,4 @@ export interface _IInternalDynamicConfigHandler extends IDyn * @param allowUpdate - An optional flag to enable updating config properties marked as readonly */ _block: (configHandler: WatcherFunction, allowUpdate?: boolean) => void; -} \ No newline at end of file +} diff --git a/shared/AppInsightsCore/src/Config/IDynamicPropertyHandler.ts b/shared/AppInsightsCore/src/interfaces/config/IDynamicPropertyHandler.ts similarity index 85% rename from shared/AppInsightsCore/src/Config/IDynamicPropertyHandler.ts rename to shared/AppInsightsCore/src/interfaces/config/IDynamicPropertyHandler.ts index 9195c41c2..1aa61d2ae 100644 --- a/shared/AppInsightsCore/src/Config/IDynamicPropertyHandler.ts +++ b/shared/AppInsightsCore/src/interfaces/config/IDynamicPropertyHandler.ts @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; +import { IConfiguration } from "../ai/IConfiguration"; import { IWatcherHandler } from "./IDynamicWatcher"; export interface IDynamicPropertyHandler { @@ -14,4 +14,4 @@ export interface IDynamicPropertyHandler { * The current collection is watcher handlers which should be called if the value changes */ h: IWatcherHandler[]; -} \ No newline at end of file +} diff --git a/shared/AppInsightsCore/src/Config/IDynamicWatcher.ts b/shared/AppInsightsCore/src/interfaces/config/IDynamicWatcher.ts similarity index 94% rename from shared/AppInsightsCore/src/Config/IDynamicWatcher.ts rename to shared/AppInsightsCore/src/interfaces/config/IDynamicWatcher.ts index 88c458467..2392c3e42 100644 --- a/shared/AppInsightsCore/src/Config/IDynamicWatcher.ts +++ b/shared/AppInsightsCore/src/interfaces/config/IDynamicWatcher.ts @@ -1,8 +1,8 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { IUnloadHook } from "../JavaScriptSDK.Interfaces/IUnloadHook"; +import { IConfiguration } from "../ai/IConfiguration"; +import { IUnloadHook } from "../ai/IUnloadHook"; import { IConfigDefaults } from "./IConfigDefaults"; import { IDynamicPropertyHandler } from "./IDynamicPropertyHandler"; diff --git a/shared/AppInsightsCore/src/Config/_IDynamicConfigHandlerState.ts b/shared/AppInsightsCore/src/interfaces/config/_IDynamicConfigHandlerState.ts similarity index 95% rename from shared/AppInsightsCore/src/Config/_IDynamicConfigHandlerState.ts rename to shared/AppInsightsCore/src/interfaces/config/_IDynamicConfigHandlerState.ts index 4c2f6dbe5..77b96cdb2 100644 --- a/shared/AppInsightsCore/src/Config/_IDynamicConfigHandlerState.ts +++ b/shared/AppInsightsCore/src/interfaces/config/_IDynamicConfigHandlerState.ts @@ -1,3 +1,6 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + import { IDynamicConfigHandler } from "./IDynamicConfigHandler"; import { IDynamicPropertyHandler } from "./IDynamicPropertyHandler"; import { IWatcherHandler, WatcherFunction, _IDynamicDetail } from "./IDynamicWatcher"; diff --git a/shared/AppInsightsCore/src/interfaces/otel/IOTelApi.ts b/shared/AppInsightsCore/src/interfaces/otel/IOTelApi.ts new file mode 100644 index 000000000..2ec0f7cc7 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/IOTelApi.ts @@ -0,0 +1,58 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { ITraceHost } from "../ai/ITraceProvider"; +import { IOTelConfig } from "./config/IOTelConfig"; +import { ITraceApi } from "./trace/IOTelTraceApi"; +import { IOTelTracerProvider } from "./trace/IOTelTracerProvider"; + +/** + * The main OpenTelemetry API interface that provides access to all OpenTelemetry functionality. + * This interface extends the IOTelTracerProvider and serves as the entry point for OpenTelemetry operations. + * + * @example + * ```typescript + * // Get a tracer from the API instance + * const tracer = otelApi.getTracer("my-component"); + * + * // Create a span + * const span = tracer.startSpan("operation"); + * + * // Access context manager + * const currentContext = otelApi.context.active(); + * + * // Access trace API + * const activeSpan = otelApi.trace.getActiveSpan(); + * ``` + * + * @since 3.4.0 + */ +export interface IOTelApi extends IOTelTracerProvider { + /** + * The configuration object that contains all OpenTelemetry-specific settings. + * This includes tracing configuration, error handlers, and other OpenTelemetry options. + * + * @remarks + * Changes to this configuration after initialization may not take effect until + * the next telemetry operation, depending on the implementation. + */ + cfg: IOTelConfig; + + /** + * The current {@link ITraceHost} instance for this IOTelApi instance, this is effectively + * the OpenTelemetry ContextAPI instance without the static methods. + * @returns The ContextManager instance + */ + host: ITraceHost; + + /** + * The current {@link ITraceApi} instance for this IOTelApi instance, this is + * effectively the OpenTelemetry TraceAPI instance without the static methods. + * @returns The current {@link ITraceApi} instance + */ + trace: ITraceApi; + + // propagation?: IOTelPropagationApi; + + // metrics?: IOTelMetricsApi; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/IOTelApiCtx.ts b/shared/AppInsightsCore/src/interfaces/otel/IOTelApiCtx.ts new file mode 100644 index 000000000..03775557b --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/IOTelApiCtx.ts @@ -0,0 +1,18 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { ITraceHost } from "../ai/ITraceProvider"; + +/** + * The context for the current IOTelApi instance linking it to the core SDK instance, + * including access to the core dynamic configuration. + * + * Note: Passing the core instance within a context object to allow future expansion + * without breaking changes or modifying signatures. Also allows easier mocking for tests. + */ +export interface IOTelApiCtx { + /** + * The host instance associated with this OTel API instance + */ + host: ITraceHost; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/IOTelAttributes.ts b/shared/AppInsightsCore/src/interfaces/otel/IOTelAttributes.ts new file mode 100644 index 000000000..163451cb9 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/IOTelAttributes.ts @@ -0,0 +1,36 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +/** + * Attribute values may be any non-nullish primitive value except an object. + * + * null or undefined attribute values are invalid and will result in undefined behavior. + * + * @since 3.4.0 + */ +export type OTelAttributeValue = + | string + | number + | boolean + | Array + | Array + | Array; + +/** + * Defines extended attribute values which may contain nested attributes. + * + * @since 3.4.0 + */ +export type ExtendedOTelAttributeValue = OTelAttributeValue | IOTelAttributes; + +/** + * Attributes is a map from string to attribute values. + * + * Note: only the own enumerable keys are counted as valid attribute keys. + * + * @since 3.4.0 + */ +export interface IOTelAttributes { + [key: string]: OTelAttributeValue | undefined; +} + diff --git a/shared/AppInsightsCore/src/interfaces/otel/attribute/IAttributeContainer.ts b/shared/AppInsightsCore/src/interfaces/otel/attribute/IAttributeContainer.ts new file mode 100644 index 000000000..06a17d3e4 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/attribute/IAttributeContainer.ts @@ -0,0 +1,191 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { eAttributeChangeOp } from "../../../enums/otel/eAttributeChangeOp"; +import { IUnloadHook } from "../../ai/IUnloadHook"; +import { IOTelAttributes, OTelAttributeValue } from "../IOTelAttributes"; + +/** + * Identifies the source of an attribute value in iterator operations + * @since 3.4.0 + */ +export const enum eAttributeFilter { + /** + * The attribute exists local to the current container instance + */ + Local = 0, + + /** + * The attribute does not exist locally and is inherited from a parent container or attributes object + */ + Inherited = 1, + + /** + * The attribute exists or has been deleted locally (only) to the current container instance + */ + LocalOrDeleted = 2 +} + +export type AttributeFilter = number | eAttributeFilter; + +/** + * Information about what changed in an attribute container + */ +export interface IAttributeChangeInfo { + /** + * The Id of the container that is initiated the change (not the immediate sender -- which is always the parent) + * As children only receive listener notifications from their parent in reaction to both changes + * they make and any changes they receive from their parent + */ + frm: string; + + /** + * Operation type that occurred + */ + op: eAttributeChangeOp; + + /** + * The key that was changed (only present for 'set' operations) + */ + k?: string; + + /** + * The old value (only present for 'set' operations when replacing existing value) + */ + prev?: V; + + /** + * The new value (only present for 'set' operations) + */ + val?: V; +} + +/** + * Interface for an attribute container + * @since 3.4.0 + */ +export interface IAttributeContainer { + /** + * Unique identifier for the attribute container + */ + readonly id: string; + + /** + * The number of attributes that have been set + * @returns The number of attributes that have been set + */ + readonly size: number; + + /** + * The number of attributes that were dropped due to the attribute limit being reached + * @returns The number of attributes that were dropped due to the attribute limit being reached + */ + readonly droppedAttributes: number; + + /** + * Return a snapshot of the current attributes, including inherited ones. + * This value is read-only and reflects the state of the attributes at the time of access, + * and the returned instance will not change if any attributes are modified later, you will need + * to access the attributes property again to get the latest state. + * + * Note: As this causes a snapshot to be taken, it is an expensive operation as it enumerates all + * attributes, so you SHOULD use this property sparingly. + * @returns A read-only snapshot of the current attributes + */ + readonly attributes: IOTelAttributes; + + /** + * Clear all existing attributes from the container, this will also remove any inherited attributes + * from this instance only (it will not change the inherited attributes / container(s)) + */ + clear: () => void; + + /** + * Get the value of an attribute by key + * @param key - The attribute key to retrieve + * @param source - Optional filter to only check attributes from a specific source (Local or Inherited) + * @returns The attribute value if found, undefined otherwise + */ + get: (key: string, source?: eAttributeFilter) => V | undefined; + + /** + * Check if an attribute exists by key + * @param key - The attribute key to check + * @param source - Optional filter to only check attributes from a specific source (Local or Inherited) + * @returns True if the attribute exists, false otherwise + */ + has: (key: string, source?: eAttributeFilter) => boolean; + + /** + * Set the value of an attribute by key on this instance. + * @param key - The attribute key to set + * @param value - The value to assign to the named attribute + * @returns true if the value was successfully set / replaced + */ + set: (key: string, value: V) => boolean; + + /** + * Delete an existing attribute, if the key doesn't exist this will return false. If the key does + * exist then it will be removed from this instance and any inherited value will be hidden (even if + * the inherited value changes) + * @param key - The attribute key to delete + * @returns True if the attribute was deleted, false if it didn't exist (which includes if it has already been deleted) + */ + del: (key: string) => boolean; + + /** + * The keys() method returns a new iterator object that contains the existing keys for each element + * in this attribute container. It will return all locally set keys first and then the inherited keys. + * When a key exists in both the local and inherited attributes, only the local key will be returned. + * If the key has been deleted locally, it will not be included in the iterator. + * @returns An iterator over the keys of the attribute container + */ + keys: () => Iterator; + + /** + * The entries() method of returns a new iterator object that contains the [key, value, source?] tuples for + * each attribute, it returns all existing attributes of this instance including all inherited ones. If the + * same key exists in both the local and inherited attributes, only the first (non-deleted) tuple will be returned. + * If the key has been deleted, it will not be included in the iterator. + * + * The source value of the tuple identifies the origin of the attribute (Local or Inherited). + * @returns An iterator over the entries of the attribute container + */ + entries: () => Iterator<[string, V, eAttributeFilter]>; + + /** + * The forEach() method of executes a provided function once per each key/value pair in this attribute container, + * it will process all local attributes first, then the inherited attributes. If the same key exists in both the + * local and inherited attributes, only the first (non-deleted) key/value pair will be processed. + * If a key has been deleted, it will not be included in the set of processed key/value pairs. + * @param callback - The function to execute for each key/value pair + * @param thisArg - Optional value to use as `this` when executing `callback` + */ + forEach: (callback: (key: string, value: V, source?: eAttributeFilter) => void, thisArg?: any) => void; + + /** + * The values() method returns a new iterator instance that contains the values for each element in this + * attribute container. It will return all locally set values first and then the inherited values. If the + * same key is present in both the local or inherited attributes only the first (non-deleted) value will be + * returned. If a key has been deleted, it will not be included in the iterator. + * @returns An iterator over the values of the attribute container + */ + values: () => Iterator; + + /** + * Register a callback listener for any attribute changes, this will include local and inherited changes. + * @param callback - Function to be called when attributes change, receives change information + * @returns IUnloadHook instance with rm() function to remove this listener, once called it will never be invoked again + */ + listen: (callback: (changeInfo: IAttributeChangeInfo) => void) => IUnloadHook; + + /** + * Create a child attribute container that inherits from this one, optionally taking a snapshot + * so that any future changes to the parent container do not affect the child container. + * The child will use all of the configuration from the parent container. + * @param name - Optional name for the child container + * @param snapshot - If true, the child container will be a snapshot of the current state + * @returns A new attribute container instance + */ + child: (name?: string, snapshot?: boolean) => IAttributeContainer +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/config/IOTelAttributeLimits.ts b/shared/AppInsightsCore/src/interfaces/otel/config/IOTelAttributeLimits.ts new file mode 100644 index 000000000..f99515228 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/config/IOTelAttributeLimits.ts @@ -0,0 +1,72 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +/** + * Configuration interface for OpenTelemetry attribute limits. + * These limits help control the size and number of attributes to prevent + * excessive memory usage and ensure consistent performance. + * + * @example + * ```typescript + * const limits: IOTelAttributeLimits = { + * attributeCountLimit: 128, // Maximum 128 attributes + * attributeValueLengthLimit: 4096 // Maximum 4KB per attribute value + * }; + * ``` + * + * @remarks + * When limits are exceeded: + * - Additional attributes beyond `attributeCountLimit` are dropped + * - Attribute values longer than `attributeValueLengthLimit` are truncated + * - The behavior may vary based on the specific implementation + * + * @since 3.4.0 + */ +export interface IOTelAttributeLimits { + /** + * Maximum allowed length for attribute values in characters. + * + * @remarks + * - Values longer than this limit will be truncated + * - Applies to string attribute values only + * - Numeric and boolean values are not affected by this limit + * - Array values have this limit applied to each individual element + * + * @defaultValue 4096 + * + * @example + * ```typescript + * // If attributeValueLengthLimit is 100: + * span.setAttribute("description", "a".repeat(200)); // Will be truncated to 100 characters + * span.setAttribute("count", 12345); // Not affected (number) + * span.setAttribute("enabled", true); // Not affected (boolean) + * ``` + */ + attributeValueLengthLimit?: number; + + /** + * Maximum number of attributes allowed per telemetry item. + * + * @remarks + * - Attributes added beyond this limit will be dropped + * - The order of attributes matters; earlier attributes take precedence + * - This limit applies to the total count of attributes, regardless of their type + * - Inherited or default attributes count toward this limit + * + * @defaultValue 128 + * + * @example + * ```typescript + * // If attributeCountLimit is 5: + * span.setAttributes({ + * "attr1": "value1", // Kept + * "attr2": "value2", // Kept + * "attr3": "value3", // Kept + * "attr4": "value4", // Kept + * "attr5": "value5", // Kept + * "attr6": "value6" // Dropped (exceeds limit) + * }); + * ``` + */ + attributeCountLimit?: number; +} \ No newline at end of file diff --git a/shared/AppInsightsCore/src/interfaces/otel/config/IOTelConfig.ts b/shared/AppInsightsCore/src/interfaces/otel/config/IOTelConfig.ts new file mode 100644 index 000000000..befeeb957 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/config/IOTelConfig.ts @@ -0,0 +1,55 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IOTelErrorHandlers } from "./IOTelErrorHandlers"; +import { ITraceCfg } from "./IOTelTraceCfg"; + +/** + * OpenTelemetry configuration interface + * Provides configuration specific to the OpenTelemetry extensions + */ +export interface IOTelConfig { + /** + * Configuration interface for OpenTelemetry tracing functionality. + * This interface contains all the settings that control how traces are created, + * processed, and managed within the OpenTelemetry system. + * + * @example + * ```typescript + * const traceCfg: ITraceCfg = { + * serviceName: "my-service", + * generalLimits: { + * attributeCountLimit: 128, + * attributeValueLengthLimit: 4096 + * }, + * spanLimits: { + * attributeCountLimit: 128, + * linkCountLimit: 128, + * eventCountLimit: 128 + * } + * }; + * ``` + * + * @since 3.4.0 + */ + traceCfg?: ITraceCfg; + + /** + * Error handlers for OpenTelemetry operations. + * This interface allows you to specify custom error handling logic for various + * OpenTelemetry components, enabling better control over how errors are managed + * within the OpenTelemetry system. + * + * @see {@link IOTelErrorHandlers} + * + * @example + * ```typescript + * const errorHandlers: IOTelErrorHandlers = { + * attribError: (message, key, value) => { + * console.warn(`Attribute error for ${key}:`, message); + * } + * }; + * ``` + */ + errorHandlers?: IOTelErrorHandlers; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/config/IOTelErrorHandlers.ts b/shared/AppInsightsCore/src/interfaces/otel/config/IOTelErrorHandlers.ts new file mode 100644 index 000000000..697c11e3b --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/config/IOTelErrorHandlers.ts @@ -0,0 +1,196 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +/** + * Configuration interface for OpenTelemetry error handling callbacks. + * Provides hooks to customize how different types of errors and diagnostic + * messages are handled within the OpenTelemetry system. + * + * @example + * ```typescript + * const errorHandlers: IOTelErrorHandlers = { + * attribError: (message, key, value) => { + * console.warn(`Attribute error for ${key}:`, message); + * }, + * spanError: (message, spanName) => { + * logger.error(`Span ${spanName} error:`, message); + * }, + * warn: (message) => { + * logger.warn(message); + * }, + * error: (message) => { + * logger.error(message); + * } + * }; + * ``` + * + * @remarks + * If handlers are not provided, default behavior will be used: + * - `attribError`: Throws an `OTelInvalidAttributeError` + * - `spanError`: Logs to console or calls warn handler + * - `debug`: Logs to console.log + * - `warn`: Logs to console.warn + * - `error`: Logs to console.error + * - `notImplemented`: Logs to console.error + * + * @since 3.4.0 + */ +export interface IOTelErrorHandlers { + /** + * Handles attribute-related errors, such as invalid attribute values or keys. + * Called when an attribute operation fails validation or processing. + * + * @param message - Descriptive error message explaining what went wrong + * @param key - The attribute key that caused the error + * @param value - The attribute value that caused the error (may be of any type) + * + * @remarks + * Common scenarios that trigger this handler: + * - Invalid attribute key format + * - Attribute value exceeds length limits + * - Unsupported attribute value type + * - Attribute count exceeds limits + * + * @default Throws an `OTelInvalidAttributeError` + * + * @example + * ```typescript + * attribError: (message, key, value) => { + * metrics.increment('otel.attribute.errors', { key, type: typeof value }); + * logger.warn(`Attribute ${key} rejected: ${message}`); + * } + * ``` + */ + attribError?: (message: string, key: string, value: any) => void; + + /** + * Handles span-related errors that occur during span operations. + * Called when a span operation fails or encounters an unexpected condition. + * + * @param message - Descriptive error message explaining the span error + * @param spanName - The name of the span that encountered the error + * + * @remarks + * Common scenarios that trigger this handler: + * - Span operation called on an ended span + * - Invalid span configuration + * - Span processor errors + * - Context propagation failures + * + * @default Logs to console or calls the warn handler + * + * @example + * ```typescript + * spanError: (message, spanName) => { + * metrics.increment('otel.span.errors', { span_name: spanName }); + * logger.error(`Span operation failed for "${spanName}": ${message}`); + * } + * ``` + */ + spanError?: (message: string, spanName: string) => void; + + /** + * Handles debug-level diagnostic messages. + * Used for detailed troubleshooting information that is typically + * only relevant during development or when diagnosing issues. + * + * @param message - Debug message to be handled + * + * @remarks + * Debug messages are typically: + * - Verbose operational details + * - Internal state information + * - Performance metrics + * - Development-time diagnostics + * + * @default Logs to console.log + * + * @example + * ```typescript + * debug: (message) => { + * if (process.env.NODE_ENV === 'development') { + * console.debug('[OTel Debug]', message); + * } + * } + * ``` + */ + debug?: (message: string) => void; + + /** + * Handles warning-level messages for non-fatal issues. + * Used for conditions that are unusual but don't prevent continued operation. + * + * @param message - Warning message to be handled + * + * @remarks + * Warning scenarios include: + * - Configuration issues that fall back to defaults + * - Performance degradation + * - Deprecated API usage + * - Resource limit approaches + * + * @default Logs to console.warn + * + * @example + * ```typescript + * warn: (message) => { + * logger.warn('[OTel Warning]', message); + * metrics.increment('otel.warnings'); + * } + * ``` + */ + warn?: (message: string) => void; + + /** + * Handles general error conditions that may affect functionality. + * Used for significant errors that should be investigated but may not be fatal. + * + * @param message - Error message to be handled + * + * @remarks + * Error scenarios include: + * - Failed network requests + * - Configuration validation failures + * - Resource allocation failures + * - Unexpected runtime conditions + * + * @default Logs to console.error + * + * @example + * ```typescript + * error: (message) => { + * logger.error('[OTel Error]', message); + * errorReporting.captureException(new Error(message)); + * } + * ``` + */ + error?: (message: string) => void; + + /** + * Handles errors related to unimplemented functionality. + * Called when a method or feature is not yet implemented or is intentionally + * disabled in the current configuration. + * + * @param message - Message describing the unimplemented functionality + * + * @remarks + * Common scenarios: + * - Placeholder methods that haven't been implemented + * - Features disabled in the current build + * - Platform-specific functionality not available + * - Optional features not included in the configuration + * + * @default Logs to console.error + * + * @example + * ```typescript + * notImplemented: (message) => { + * logger.warn(`[OTel Not Implemented] ${message}`); + * if (process.env.NODE_ENV === 'development') { + * console.trace('Not implemented method called'); + * } + * } + * ``` + */ + notImplemented?: (message: string) => void; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/config/IOTelSpanLimits.ts b/shared/AppInsightsCore/src/interfaces/otel/config/IOTelSpanLimits.ts new file mode 100644 index 000000000..0a94d23bf --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/config/IOTelSpanLimits.ts @@ -0,0 +1,131 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +// import { IOTelAttributeLimits } from "./IOTelAttributeLimits"; + +// /** +// * Configuration interface for OpenTelemetry span-specific limits. +// * Extends the general attribute limits with additional constraints specific to spans, +// * including limits on events, links, and their associated attributes. +// * +// * @example +// * ```typescript +// * const spanLimits: IOTelSpanLimits = { +// * // Inherited from IOTelAttributeLimits +// * attributeCountLimit: 128, +// * attributeValueLengthLimit: 4096, +// * +// * // Span-specific limits +// * linkCountLimit: 128, +// * eventCountLimit: 128, +// * attributePerEventCountLimit: 32, +// * attributePerLinkCountLimit: 32 +// * }; +// * ``` +// * +// * @remarks +// * These limits help prevent spans from consuming excessive memory and ensure +// * consistent performance even when dealing with complex traces that have many +// * events, links, or attributes. +// * +// * @since 3.4.0 +// */ +// export interface IOTelSpanLimits extends IOTelAttributeLimits { + +// /** +// * Maximum number of links allowed per span. +// * +// * @remarks +// * - Links added beyond this limit will be dropped +// * - Links are typically added at span creation time +// * - Each link represents a causal relationship with another span +// * - Links added after creation may be subject to additional restrictions +// * +// * @defaultValue 128 +// * +// * @example +// * ```typescript +// * const span = tracer.startSpan("operation", { +// * links: [ +// * { context: relatedSpanContext1 }, +// * { context: relatedSpanContext2 }, +// * // ... up to linkCountLimit links +// * ] +// * }); +// * ``` +// */ +// linkCountLimit?: number; + +// /** +// * Maximum number of events allowed per span. +// * +// * @remarks +// * - Events added beyond this limit will be dropped +// * - Events are typically used to mark significant points during span execution +// * - Each event can have its own set of attributes (limited by attributePerEventCountLimit) +// * - Events are ordered chronologically within the span +// * +// * @defaultValue 128 +// * +// * @example +// * ```typescript +// * // If eventCountLimit is 3: +// * span.addEvent("started"); // Kept +// * span.addEvent("processing"); // Kept +// * span.addEvent("validation"); // Kept +// * span.addEvent("completed"); // Dropped (exceeds limit) +// * ``` +// */ +// eventCountLimit?: number; + +// /** +// * Maximum number of attributes allowed per span event. +// * +// * @remarks +// * - This limit applies to each individual event within a span +// * - Attributes added to events beyond this limit will be dropped +// * - This is separate from the span's own attribute limits +// * - Event attributes are useful for providing context about what happened at that point in time +// * +// * @defaultValue 32 +// * +// * @example +// * ```typescript +// * // If attributePerEventCountLimit is 2: +// * span.addEvent("user_action", { +// * "action": "click", // Kept +// * "element": "button", // Kept +// * "timestamp": "12345" // Dropped (exceeds per-event limit) +// * }); +// * ``` +// */ +// attributePerEventCountLimit?: number; + +// /** +// * Maximum number of attributes allowed per span link. +// * +// * @remarks +// * - This limit applies to each individual link within a span +// * - Attributes added to links beyond this limit will be dropped +// * - This is separate from the span's own attribute limits +// * - Link attributes provide additional context about the relationship between spans +// * +// * @defaultValue 32 +// * +// * @example +// * ```typescript +// * // If attributePerLinkCountLimit is 2: +// * const span = tracer.startSpan("operation", { +// * links: [{ +// * context: relatedSpanContext, +// * attributes: { +// * "relationship": "follows", // Kept +// * "correlation_id": "abc123", // Kept +// * "priority": "high" // Dropped (exceeds per-link limit) +// * } +// * }] +// * }); +// * ``` +// */ +// attributePerLinkCountLimit?: number; +// } diff --git a/shared/AppInsightsCore/src/interfaces/otel/config/IOTelTraceCfg.ts b/shared/AppInsightsCore/src/interfaces/otel/config/IOTelTraceCfg.ts new file mode 100644 index 000000000..c66ad552e --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/config/IOTelTraceCfg.ts @@ -0,0 +1,85 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IOTelAttributeLimits } from "./IOTelAttributeLimits"; + +/** + * Configuration interface for OpenTelemetry compatible tracing functionality. + * This interface contains all the settings that control how traces are created, + * processed, and managed within the OpenTelemetry system. + * + * @example + * ```typescript + * const traceCfg: ITraceCfg = { + * serviceName: "my-service", + * generalLimits: { + * attributeCountLimit: 128, + * attributeValueLengthLimit: 4096 + * }, + * spanLimits: { + * attributeCountLimit: 128, + * linkCountLimit: 128, + * eventCountLimit: 128 + * } + * }; + * ``` + * + * @since 3.4.0 + */ +export interface ITraceCfg { + /** + * Global attribute limits that apply to all telemetry items. + * These limits help prevent excessive memory usage and ensure consistent + * behavior across different telemetry types. + * + * @remarks + * These limits are inherited by more specific configurations unless overridden. + * For example, spans will use these limits unless `spanLimits` specifies different values. + */ + generalLimits?: IOTelAttributeLimits; + + // /** + // * Specific limits that apply only to spans. + // * These limits override the general limits for span-specific properties. + // * + // * @remarks + // * Includes limits for attributes, events, links, and their associated attributes. + // * This allows for fine-tuned control over span size and complexity. + // */ + // spanLimits?: IOTelSpanLimits; + + // idGenerator?: IOTelIdGenerator; + + // logRecordProcessors?: LogRecordProcessor[]; + // metricReader: IMetricReader; + // views: ViewOptions[]; + // instrumentations: (Instrumentation | Instrumentation[])[]; + // resource: Resource; + // resourceDetectors: Array; + + /** + * The name of the service generating telemetry data. + * This name will be included in all telemetry items as a resource attribute. + * + * @remarks + * The service name is crucial for identifying and filtering telemetry data + * in observability systems. It should be consistent across all instances + * of the same service. + * + * @example + * ```typescript + * serviceName: "user-authentication-service" + * ``` + */ + serviceName?: string; + + // spanProcessors?: SpanProcessor[]; + // traceExporter: SpanExporter; + + /** + * A flag that indicates whether the tracing (creating of a "trace" event) should be suppressed + * when a {@link IOTelSpan} ends and the span {@link IOTelSpan#isRecording | isRecording} is true. + * This value is also inherited by spans when they are created. + */ + suppressTracing?: boolean; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpan.ts b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpan.ts new file mode 100644 index 000000000..0f3bd6196 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpan.ts @@ -0,0 +1,488 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { OTelException } from "../../IException"; +import { OTelTimeInput } from "../../IOTelHrTime"; +import { IDistributedTraceContext } from "../../ai/IDistributedTraceContext"; +import { IOTelAttributes, OTelAttributeValue } from "../IOTelAttributes"; +import { IAttributeContainer } from "../attribute/IAttributeContainer"; +import { IOTelSpanStatus } from "./IOTelSpanStatus"; + +/** + * Provides an OpenTelemetry compatible interface for spans conforming to the OpenTelemetry API specification (v1.9.0). + * + * A span represents an operation within a trace and is the fundamental unit of work in distributed tracing. + * Spans can be thought of as a grouping mechanism for a set of operations that are executed as part of + * a single logical unit of work, providing timing information and contextual data about the operation. + * + * Spans form a tree structure within a trace, with a single root span that may have zero or more child spans, + * which in turn may have their own children. This hierarchical structure allows for detailed analysis of + * complex, multi-step operations across distributed systems. + * + * @since 3.4.0 + * + * @remarks + * - All spans created by this library implement the ISpan interface and extend the IReadableSpan interface + * - Spans should be ended by calling `end()` when the operation completes + * - Once ended, spans should generally not be used for further operations + * - Spans automatically track timing information from creation to end + * + * @example + * ```typescript + * // Basic span usage + * const span = tracer.startSpan('user-authentication'); + * span.setAttribute('user.id', '12345'); + * span.setAttribute('auth.method', 'oauth2'); + * + * try { + * const result = await authenticateUser(); + * span.setStatus({ code: SpanStatusCode.OK }); + * span.setAttribute('auth.success', true); + * } catch (error) { + * span.recordException(error); + * span.setStatus({ + * code: SpanStatusCode.ERROR, + * message: 'Authentication failed' + * }); + * } finally { + * span.end(); + * } + * ``` + */ +export interface IOTelSpan { + /** + * Returns the span context object associated with this span. + * + * The span context is an immutable, serializable identifier that uniquely identifies + * this span within a trace. It contains the trace ID, span ID, and trace flags that + * can be used to create new child spans or propagate trace context across process boundaries. + * + * The returned span context remains valid even after the span has ended, making it + * useful for asynchronous operations and cross-service communication. + * + * @returns The immutable span context associated with this span + * + * @remarks + * - The span context is the primary mechanism for trace propagation + * - Context can be serialized and transmitted across network boundaries + * - Contains trace ID (unique to the entire trace) and span ID (unique to this span) + * + * @example + * ```typescript + * const span = tracer.startSpan('parent-operation'); + * const spanContext = span.spanContext(); + * + * // Use context to create child spans in other parts of the system + * const childSpan = tracer.startSpan('child-operation', { + * parent: spanContext + * }); + * + * // Context can be serialized for cross-service propagation + * const traceId = spanContext.traceId; + * const spanId = spanContext.spanId; + * ``` + */ + spanContext(): IDistributedTraceContext; + + /** + * Sets a single attribute on the span with the specified key and value. + * + * Attributes provide contextual information about the operation represented by the span. + * They are key-value pairs that help with filtering, grouping, and understanding spans + * in trace analysis tools. Attributes should represent meaningful properties of the operation. + * + * @param key - The attribute key, should be descriptive and follow naming conventions + * @param value - The attribute value; null or undefined values are invalid and result in undefined behavior + * + * @returns This span instance for method chaining + * + * @remarks + * - Attribute keys should follow semantic conventions when available + * - Common attributes include service.name, http.method, db.statement, etc. + * - Setting null or undefined values is invalid and may cause unexpected behavior + * - Attributes set after span creation don't affect sampling decisions + * + * @example + * ```typescript + * const span = tracer.startSpan('http-request'); + * + * // Set individual attributes with descriptive keys + * span.setAttribute('http.method', 'POST') + * .setAttribute('http.url', 'https://api.example.com/users') + * .setAttribute('http.status_code', 201) + * .setAttribute('user.id', '12345') + * .setAttribute('request.size', 1024); + * ``` + */ + setAttribute(key: string, value: OTelAttributeValue): this; + + /** + * Sets multiple attributes on the span at once using an attributes object. + * + * This method allows efficient batch setting of multiple attributes in a single call. + * All attributes in the provided object will be added to the span, supplementing any + * existing attributes (duplicate keys will be overwritten). + * + * @param attributes - An object containing key-value pairs to set as span attributes + * + * @returns This span instance for method chaining + * + * @remarks + * - Null or undefined attribute values are invalid and will result in undefined behavior + * - More efficient than multiple `setAttribute` calls for bulk operations + * - Existing attributes with the same keys will be overwritten + * + * @example + * ```typescript + * const span = tracer.startSpan('database-query'); + * + * // Set multiple attributes efficiently + * span.setAttributes({ + * 'db.system': 'postgresql', + * 'db.name': 'user_database', + * 'db.table': 'users', + * 'db.operation': 'SELECT', + * 'db.rows_affected': 5, + * 'query.duration_ms': 15.7 + * }); + * ``` + */ + setAttributes(attributes: IOTelAttributes): this; + + /** + * The {@link IAttributeContainer | attribute container} associated with this span, providing + * advanced attribute management capabilities. Rather than using the {@link IReadableSpan#attributes} + * directly which returns a readonly {@link IOTelAttributes} map that is a snapshot of the attributes at + * the time of access, the attribute container offers methods to get, set, delete, and iterate over attributes + * with fine-grained control. + * It is recommended that you only access the {@link IReadableSpan#attributes} property sparingly due to the + * performance cost of taking a snapshot of all attributes. + */ + readonly attribContainer: IAttributeContainer; + + // /** + // * Adds an event to the span with optional attributes and timestamp. + // * + // * **Note: This method is currently not implemented and events will be dropped.** + // * + // * Events represent significant points in time during the span's execution. + // * They provide additional context about what happened during the operation, + // * such as cache hits/misses, validation steps, or other notable occurrences. + // * + // * @param name - The name of the event, should be descriptive of what occurred + // * @param attributesOrStartTime - Event attributes object, or start time if third parameter is undefined + // * @param startTime - Optional start time of the event; if not provided, current time is used + // * + // * @returns This span instance for method chaining + // * + // * @remarks + // * - **Current implementation drops events - not yet supported** + // * - Events are timestamped occurrences within a span's lifecycle + // * - Useful for marking significant points like cache hits, retries, or validation steps + // * - Should not be used for high-frequency events due to performance impact + // * + // * @example + // * ```typescript + // * const span = tracer.startSpan('user-registration'); + // * + // * // Add events to mark significant points + // * span.addEvent('validation.started') + // * .addEvent('validation.completed', { + // * 'validation.result': 'success', + // * 'validation.duration_ms': 23 + // * }) + // * .addEvent('database.save.started') + // * .addEvent('database.save.completed', { + // * 'db.rows_affected': 1 + // * }); + // * ``` + // */ + // addEvent(name: string, attributesOrStartTime?: IOTelAttributes | OTelTimeInput, startTime?: OTelTimeInput): this; + + // /** + // * Adds a single link to the span connecting it to another span. + // * + // * **Note: This method is currently not implemented and links will be dropped.** + // * + // * Links establish relationships between spans that are not in a typical parent-child + // * relationship. They are useful for connecting spans across different traces or + // * for representing batch operations where multiple spans are related but not nested. + // * + // * @param link - The link object containing span context and optional attributes + // * + // * @returns This span instance for method chaining + // * + // * @remarks + // * - **Current implementation drops links - not yet supported** + // * - Links added after span creation do not affect sampling decisions + // * - Prefer adding links during span creation when possible + // * - Useful for batch operations, fan-out scenarios, or cross-trace relationships + // * + // * @example + // * ```typescript + // * const span = tracer.startSpan('batch-processor'); + // * + // * // Link to related spans from a batch operation + // * span.addLink({ + // * context: relatedSpan.spanContext(), + // * attributes: { + // * 'link.type': 'batch_item', + // * 'batch.index': 1 + // * } + // * }); + // * ``` + // */ + // addLink(link: IOTelLink): this; + + // /** + // * Adds multiple links to the span in a single operation. + // * + // * **Note: This method is currently not implemented and links will be dropped.** + // * + // * This is an efficient way to establish multiple relationships between this span + // * and other spans. Particularly useful for batch operations, fan-out scenarios, + // * or when a single operation needs to reference multiple related operations. + // * + // * @param links - An array of link objects to add to the span + // * + // * @returns This span instance for method chaining + // * + // * @remarks + // * - **Current implementation drops links - not yet supported** + // * - More efficient than multiple `addLink` calls for bulk operations + // * - Links added after span creation do not affect sampling decisions + // * - Consider span creation time linking for sampling-sensitive scenarios + // * + // * @example + // * ```typescript + // * const span = tracer.startSpan('aggregate-results'); + // * + // * // Link to multiple related spans from parallel operations + // * span.addLinks([ + // * { + // * context: span1.spanContext(), + // * attributes: { 'operation.type': 'data_fetch', 'source': 'database' } + // * }, + // * { + // * context: span2.spanContext(), + // * attributes: { 'operation.type': 'data_fetch', 'source': 'cache' } + // * }, + // * { + // * context: span3.spanContext(), + // * attributes: { 'operation.type': 'data_transform', 'stage': 'preprocessing' } + // * } + // * ]); + // * ``` + // */ + // addLinks(links: IOTelLink[]): this; + + /** + * Sets the status of the span to indicate the success or failure of the operation. + * + * The span status provides a standardized way to indicate whether the operation + * completed successfully, encountered an error, or is in an unknown state. + * This status is used by observability tools to provide meaningful insights + * about system health and operation outcomes. + * + * @param status - The status object containing code and optional message + * + * @returns This span instance for method chaining + * + * @remarks + * - Default status is UNSET until explicitly set + * - Setting status overrides any previous status values + * - ERROR status should be accompanied by a descriptive message when possible + * - Status should reflect the final outcome of the operation + * + * @example + * ```typescript + * const span = tracer.startSpan('payment-processing'); + * + * try { + * const result = await processPayment(paymentData); + * + * // Indicate successful completion + * span.setStatus({ + * code: SpanStatusCode.OK + * }); + * + * } catch (error) { + * // Indicate operation failed + * span.setStatus({ + * code: SpanStatusCode.ERROR, + * message: 'Payment processing failed: ' + error.message + * }); + * + * span.recordException(error); + * } + * ``` + */ + setStatus(status: IOTelSpanStatus): this; + + /** + * Updates the name of the span, overriding the name provided during creation. + * + * Span names should be descriptive and represent the operation being performed. + * Updating the name can be useful when the operation's scope becomes clearer + * during execution, or when implementing generic spans that need specific naming + * based on runtime conditions. + * + * @param name - The new name for the span, should be descriptive of the operation + * + * @returns This span instance for method chaining + * + * @remarks + * - Name updates may affect sampling behavior depending on implementation + * - Choose names that are meaningful but not too specific to avoid cardinality issues + * - Follow naming conventions consistent with your observability strategy + * - Consider the impact on existing traces and dashboards when changing names + * + * @example + * ```typescript + * const span = tracer.startSpan('generic-operation'); + * + * // Update name based on runtime determination + * if (operationType === 'user-registration') { + * span.updateName('user-registration'); + * span.setAttribute('operation.type', 'registration'); + * } else if (operationType === 'user-login') { + * span.updateName('user-authentication'); + * span.setAttribute('operation.type', 'authentication'); + * } + * ``` + */ + updateName(name: string): this; + + /** + * Marks the end of the span's execution and records the end timestamp. + * + * This method finalizes the span and makes it available for export to tracing systems. + * Once ended, the span should not be used for further operations. The span's duration + * is calculated from its start time to the end time provided or current time. + * + * @param endTime - Optional end time; if not provided, current time is used + * + * @remarks + * - This method does NOT return `this` to discourage chaining after span completion + * - Ending a span has no effect on child spans, which may continue running + * - Child spans can be ended independently after their parent has ended + * - The span becomes eligible for export once ended + * - Calling end() multiple times has no additional effect + * + * @example + * ```typescript + * const span = tracer.startSpan('file-processing'); + * + * try { + * // Perform the operation + * const result = await processFile(filePath); + * + * // Record success + * span.setStatus({ code: SpanStatusCode.OK }); + * span.setAttribute('file.size', result.size); + * + * } catch (error) { + * span.recordException(error); + * span.setStatus({ + * code: SpanStatusCode.ERROR, + * message: error.message + * }); + * } finally { + * // Always end the span + * span.end(); + * // Don't use span after this point + * } + * + * // Custom end time example + * const customEndTime = Date.now() * 1000000; // nanoseconds + * span.end(customEndTime); + * ``` + */ + end(endTime?: OTelTimeInput): void; + + /** + * Returns whether this span is actively recording information. + * + * A recording span accepts and stores attributes, events, status, and other span data. + * Non-recording spans (typically due to sampling decisions) may ignore operations + * like setAttribute() to optimize performance. This method allows conditional + * logic to avoid expensive operations on non-recording spans. + * + * @returns True if the span is actively recording information, false otherwise + * + * @remarks + * - Recording status is typically determined at span creation time + * - Non-recording spans still provide valid span context for propagation + * - Use this check to avoid expensive attribute calculations for non-recording spans + * - Recording status remains constant throughout the span's lifetime + * + * @example + * ```typescript + * const span = tracer.startSpan('data-processing'); + * + * // Only perform expensive operations if span is recording + * if (span.isRecording()) { + * const metadata = await expensiveMetadataCalculation(); + * span.setAttributes({ + * 'process.metadata': JSON.stringify(metadata), + * 'process.complexity': metadata.complexity, + * 'process.estimated_duration': metadata.estimatedMs + * }); + * } + * + * // Always safe to set basic attributes + * span.setAttribute('process.started', true); + * ``` + */ + isRecording(): boolean; + + /** + * Records an exception as a span event with automatic error status handling. + * + * This method captures exception information and automatically creates a span event + * with standardized exception attributes. It's the recommended way to handle errors + * within spans, providing consistent error reporting across the application. + * + * @param exception - The exception to record; accepts string messages or Error objects + * @param time - Optional timestamp for when the exception occurred; defaults to current time + * + * @remarks + * - Automatically extracts exception type, message, and stack trace when available + * - Creates a standardized span event with exception details + * - Does NOT automatically set span status to ERROR - call setStatus() explicitly if needed + * - Exception events are useful for debugging and error analysis + * + * @example + * ```typescript + * const span = tracer.startSpan('risky-operation'); + * + * try { + * await performRiskyOperation(); + * span.setStatus({ code: SpanStatusCode.OK }); + * + * } catch (error) { + * // Record the exception details + * span.recordException(error); + * + * // Explicitly set error status + * span.setStatus({ + * code: SpanStatusCode.ERROR, + * message: 'Operation failed due to: ' + error.message + * }); + * + * // Re-throw if needed + * throw error; + * } finally { + * span.end(); + * } + * + * // Recording string exceptions + * span.recordException('Custom error message occurred'); + * + * // Recording with custom timestamp + * const errorTime = Date.now() * 1000000; // nanoseconds + * span.recordException(error, errorTime); + * ``` + */ + recordException(exception: OTelException, time?: OTelTimeInput): void; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanContext.ts b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanContext.ts new file mode 100644 index 000000000..9da8f4b90 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanContext.ts @@ -0,0 +1,23 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IDistributedTraceInit } from "../../ai/IDistributedTraceContext"; + +/** + * A SpanContext represents the portion of a {@link IOTelSpan} which must be + * serialized and propagated along side of a {@link IOTelBaggage}. + */ +export interface IOTelSpanContext extends IDistributedTraceInit { + + /** + * Trace flags to propagate. + * + * It is represented as 1 byte (bitmap). Bit to represent whether trace is + * sampled or not. When set, the least significant bit documents that the + * caller may have recorded trace data. A caller who does not record trace + * data out-of-band leaves this flag unset. + * + * see {@link eW3CTraceFlags} for valid flag values. + */ + traceFlags: number; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanCtx.ts b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanCtx.ts new file mode 100644 index 000000000..162b76b33 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanCtx.ts @@ -0,0 +1,75 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { OTelException } from "../../IException"; +import { OTelTimeInput } from "../../IOTelHrTime"; +import { IDistributedTraceContext } from "../../ai/IDistributedTraceContext"; +import { IOTelApi } from "../IOTelApi"; +import { IOTelAttributes } from "../IOTelAttributes"; +import { IReadableSpan } from "./IReadableSpan"; + +/** + * The context to use for creating a Span + */ +export interface IOTelSpanCtx { + /** + * The current {@link IOTelApi} instance that is being used. + */ + api: IOTelApi; + + // /** + // * The current {@link IOTelResource} instance to use for this Span Context + // */ + // resource?: IOTelResource; + + // /** + // * The current {@link IOTelInstrumentationScope} instrumentationScope instance to + // * use for this Span Context + // */ + // instrumentationScope?: IOTelInstrumentationScope; + + // /** + // * The context for the current instance (not currently used) + // */ + // context?: IOTelContext; + + /* + * The current {@link IDistributedTraceContext} instance to associated with the span + * used to create the span. + */ + spanContext: IDistributedTraceContext; + + /** + * Identifies the user provided start time of the span + */ + startTime?: OTelTimeInput; + + parentSpanContext?: IDistributedTraceContext; + + attributes?: IOTelAttributes; + + // links?: IOTelLink[]; + + isRecording?: boolean; + + /** + * When the span ends this callback will be called to process the specified Span and end it + * @param span - The span to end + * @param endTime - The end time of the span + * @param duration - The duration of the span + * @returns + */ + onEnd?: (span: IReadableSpan) => void; + + /** + * When an exception is recorded via the span's recordException API this callback will be called + * to process the exception. Unlike the OpenTelemetry spec this callback also provides the span instance + * to allow implementations to associate the exception with the span as needed. + * It is also called immediately when recordException is called rather than waiting until the span ends. + * @param span - The span associated with the exception + * @param exception - The exception to process + * @param time - The time the exception occurred + * @returns + */ + onException?: (span: IReadableSpan, exception: OTelException, time?: OTelTimeInput) => void; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanOptions.ts b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanOptions.ts new file mode 100644 index 000000000..bc4d550cb --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanOptions.ts @@ -0,0 +1,34 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { OTelSpanKind } from "../../../enums/otel/OTelSpanKind"; +import { OTelTimeInput } from "../../IOTelHrTime"; +import { IOTelAttributes } from "../IOTelAttributes"; + +/** + * Provides an OpenTelemetry like Interface for the Open Telemetry Api (1.9.0) SpanOptions + * type. Where SpanOptions are options that can be used to configure a span. + */ +export interface IOTelSpanOptions { + /** + * The SpanKind of a span of this span, this is used to specify + * the relationship between the span and its parent span. + * @see {@link eOTelSpanKind} for possible values. + * @default eOTelSpanKind.INTERNAL + */ + kind?: OTelSpanKind; + + /** + * A span's attributes + */ + attributes?: IOTelAttributes; + + /** A manually specified start time for the created `Span` object. */ + startTime?: OTelTimeInput; + + /** The new span should be a root span. (Ignore parent from context). */ + root?: boolean; + + /** Specify whether the span should be a recording span, default is true */ + recording?: boolean; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanStatus.ts b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanStatus.ts new file mode 100644 index 000000000..4bfdeb170 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelSpanStatus.ts @@ -0,0 +1,16 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { eOTelSpanStatusCode } from "../../../enums/otel/OTelSpanStatus"; + +export interface IOTelSpanStatus { + /** + * The status code of this message. + */ + code: eOTelSpanStatusCode; + + /** + * A developer-facing error message. + */ + message?: string; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTraceApi.ts b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTraceApi.ts new file mode 100644 index 000000000..3eab1e73f --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTraceApi.ts @@ -0,0 +1,51 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IDistributedTraceContext, IDistributedTraceInit } from "../../ai/IDistributedTraceContext"; +import { ISpanScope } from "../../ai/ITraceProvider"; +import { IOTelSpanContext } from "./IOTelSpanContext"; +import { IOTelTracer } from "./IOTelTracer"; +import { IOTelTracerOptions } from "./IOTelTracerOptions"; +import { IReadableSpan } from "./IReadableSpan"; + +/** + * ITraceApi provides an interface definition which is simular to the OpenTelemetry TraceAPI + */ +export interface ITraceApi { + /** + * Returns a Tracer, creating one if one with the given name and version + * if one has not already created. + * + * @param name - The name of the tracer or instrumentation library. + * @param version - The version of the tracer or instrumentation library. + * @param options - The options of the tracer or instrumentation library. + * @returns Tracer A Tracer with the given name and version + */ + getTracer(name: string, version?: string, options?: IOTelTracerOptions): IOTelTracer; + + /** + * Wrap the given {@link IDistributedTraceContext} in a new non-recording {@link IReadableSpan} + * + * @param spanContext - The {@link IDistributedTraceContext} to be wrapped + * @returns a new non-recording {@link IReadableSpan} with the provided context + */ + wrapSpanContext(spanContext: IDistributedTraceContext | IDistributedTraceInit | IOTelSpanContext): IReadableSpan; + + /** + * Returns true if this {@link IDistributedTraceContext} is valid. + * @return true if this {@link IDistributedTraceContext} is valid. + */ + isSpanContextValid(spanContext: IDistributedTraceContext | IDistributedTraceInit | IOTelSpanContext): boolean; + + /** + * Gets the span from the current context, if one exists. + */ + getActiveSpan(): IReadableSpan | undefined | null; + + /** + * Set or clear the current active span. + * @param span - The span to set as the active span, or null/undefined to clear the active span. + * @return An ISpanScope instance returned by the host, or void if there is no defined host. + */ + setActiveSpan(span: IReadableSpan | undefined | null): ISpanScope | undefined | null; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTraceState.ts b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTraceState.ts new file mode 100644 index 000000000..35838ae12 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTraceState.ts @@ -0,0 +1,62 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +/** + * Provides an OpenTelemetry compatible Interface for the Open Telemetry Api (1.9.0) TraceState type. + * + * The TraceState is a list of key/value pairs that are used to propagate + * vendor-specific trace information across different distributed tracing systems. + * The TraceState is used to store the state of a trace across different + * distributed tracing systems, and it is used to ensure that the trace information + * is consistent across different systems. + * + * Instances of TraceState are immutable, and the methods on this interface + * return a new instance of TraceState with the updated values. + */ +export interface IOTelTraceState { + /** + * Create a new TraceState which inherits from this TraceState and has the + * given key set. + * The new entry will always be added in the front of the list of states. + * + * @param key - key of the TraceState entry. + * @param value - value of the TraceState entry. + */ + set(key: string, value: string): IOTelTraceState; + + /** + * Return a new TraceState which inherits from this TraceState but does not + * contain the given key. + * + * @param key - the key for the TraceState entry to be removed. + */ + unset(key: string): IOTelTraceState; + + /** + * Returns the value to which the specified key is mapped, or `undefined` if + * this map contains no mapping for the key. + * + * @param key - with which the specified value is to be associated. + * @returns the value to which the specified key is mapped, or `undefined` if + * this map contains no mapping for the key. + */ + get(key: string): string | undefined; + + /** + * Serializes the TraceState to a `list` as defined below. The `list` is a series of `list-members` + * separated by commas `,`, and a list-member is a key/value pair separated by an equals sign `=`. + * Spaces and horizontal tabs surrounding `list-members` are ignored. There can be a maximum of 32 + * `list-members` in a `list`. + * + * If the resulting serialization is limited to no longer than 512 bytes, if the combination of + * keys and values exceeds this limit, the serialization will be truncated to the last key/value pair + * that fits within the limit. The serialization will be returned as a string. + * + * This is different from the {@link IW3cTraceState} serialization which returns an array of strings where each + * string is limited to 512 bytes and the array is limited to 32 strings. Thus the OpenTelemetry serialization + * will only return the first single string that fits within the limie. + * + * @returns the serialized string. + */ + serialize(): string; +} \ No newline at end of file diff --git a/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracer.ts b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracer.ts new file mode 100644 index 000000000..a4cf84eea --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracer.ts @@ -0,0 +1,147 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IOTelSpanOptions } from "./IOTelSpanOptions"; +import { IReadableSpan } from "./IReadableSpan"; + +/** + * OpenTelemetry tracer interface for creating and managing spans within a trace. + * + * A tracer is responsible for creating spans that represent units of work within a distributed system. + * Each tracer is typically associated with a specific instrumentation library or component, + * allowing for fine-grained control over how different parts of an application generate telemetry. + * + * @example + * ```typescript + * // Get a tracer instance + * const tracer = otelApi.getTracer('my-service'); + * + * // Create a simple span + * const span = tracer.startSpan('database-query'); + * span.setAttribute('db.operation', 'SELECT'); + * span.end(); + * + * // Create an active span with automatic context management + * tracer.startActiveSpan('process-request', (span) => { + * span.setAttribute('request.id', '12345'); + * + * // Any spans created within this block will be children of this span + * processRequest(); + * + * span.end(); + * }); + * ``` + * + * @see {@link IReadableSpan} - Interface for individual spans + * @see {@link IOTelSpanOptions} - Configuration options for span creation + * + * @since 3.4.0 + */ +export interface IOTelTracer { + /** + * Creates and starts a new span without setting it as the active span in the current context. + * + * This method creates a span but does NOT modify the current execution context. + * The caller is responsible for managing the span's lifecycle, including calling `end()` + * when the operation completes. + * + * @param name - The name of the span, should be descriptive of the operation being traced + * @param options - Optional configuration for span creation (parent context, attributes, etc.) + * + * @returns The newly created span, or null if span creation failed + * + * @remarks + * - The returned span must be manually ended by calling `span.end()` + * - This span will not automatically become the parent for spans created in nested operations + * - Use `startActiveSpan` if you want automatic context management + * + * @example + * ```typescript + * const span = tracer.startSpan('database-operation'); + * if (span) { + * try { + * span.setAttribute('db.table', 'users'); + * span.setAttribute('db.operation', 'SELECT'); + * + * // Perform database operation + * const result = await db.query('SELECT * FROM users'); + * + * span.setAttributes({ + * 'db.rows_affected': result.length, + * 'operation.success': true + * }); + * } catch (error) { + * span.setStatus({ + * code: SpanStatusCode.ERROR, + * message: error.message + * }); + * span.recordException(error); + * } finally { + * span.end(); // Always end the span + * } + * } + * ``` + */ + startSpan(name: string, options?: IOTelSpanOptions): IReadableSpan | null; + + /** + * Creates and starts a new span, sets it as the active span in the current context, + * and executes a provided function within this context. + * + * This method creates a span, makes it active during the execution of the provided + * function, and automatically ends the span when the function completes (or throws). + * This provides automatic span lifecycle management and context propagation. + * + * @param name - The name of the span, should be descriptive of the operation being traced + * @param options - Optional configuration for span creation (parent context, attributes, etc.) + * @param fn - The function to execute within the span's active context + * + * @returns The result of executing the provided function + * + * @remarks + * - The span is automatically ended when the function completes or throws an exception + * - The span becomes the active parent for any spans created within the function + * - If the function throws an error, the span status is automatically set to ERROR + * - This is the recommended method for most tracing scenarios due to automatic lifecycle management + * - Multiple overloads available for different parameter combinations + * + * @example + * ```typescript + * // Synchronous operation with just name and function + * const result = tracer.startActiveSpan('user-service', (span) => { + * span.setAttribute('operation', 'get-user-details'); + * return { user: getUserData(), timestamp: new Date().toISOString() }; + * }); + * + * // With options + * const result2 = tracer.startActiveSpan('database-query', + * { attributes: { 'db.table': 'users' } }, + * (span) => { + * span.setAttribute('db.operation', 'SELECT'); + * return database.getUser('123'); + * } + * ); + * + * // With full context control + * const result3 = tracer.startActiveSpan('external-api', + * { attributes: { 'service.name': 'payment-api' } }, + * currentContext, + * async (span) => { + * try { + * const response = await fetch('/api/payment'); + * span.setAttributes({ + * 'http.status_code': response.status, + * 'operation.success': response.ok + * }); + * return response.json(); + * } catch (error) { + * span.setAttribute('error.type', error.constructor.name); + * throw error; // Error automatically recorded + * } + * } + * ); + * ``` + */ + startActiveSpan unknown>(name: string, fn: F): ReturnType; + startActiveSpan unknown>(name: string, options: IOTelSpanOptions,fn: F ): ReturnType; + } diff --git a/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracerOptions.ts b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracerOptions.ts new file mode 100644 index 000000000..348d64ed2 --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracerOptions.ts @@ -0,0 +1,9 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +export interface IOTelTracerOptions { + /** + * The schemaUrl of the tracer or instrumentation library + */ + schemaUrl?: string; +} diff --git a/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracerProvider.ts b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracerProvider.ts new file mode 100644 index 000000000..c986902ee --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/trace/IOTelTracerProvider.ts @@ -0,0 +1,39 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IPromise } from "@nevware21/ts-async"; +import { IOTelTracer } from "./IOTelTracer"; +import { IOTelTracerOptions } from "./IOTelTracerOptions"; + +/** + * OpenTelemetry Trace API for getting tracers. + * This provides the standard OpenTelemetry trace API entry point. + */ +export interface IOTelTracerProvider { + /** + * Returns a Tracer, creating one if one with the given name and version is + * not already created. This may return + * - The same Tracer instance if one has already been created with the same name and version + * - A new Tracer instance if one has not already been created with the same name and version + * - A non-operational Tracer if the provider is not operational + * + * @param name - The name of the tracer or instrumentation library. + * @param version - The version of the tracer or instrumentation library. + * @param options - The options of the tracer or instrumentation library. + * @returns Tracer A Tracer with the given name and version + */ + getTracer(name: string, version?: string, options?: IOTelTracerOptions): IOTelTracer; + + /** + * Forces the tracer provider to flush any buffered data. + * @returns A promise that resolves when the flush is complete. + */ + forceFlush?: () => IPromise | void; + + /** + * Shuts down the tracer provider and releases any resources. + * @returns A promise that resolves when the shutdown is complete. + */ + shutdown?: () => IPromise | void; +} + diff --git a/shared/AppInsightsCore/src/interfaces/otel/trace/IReadableSpan.ts b/shared/AppInsightsCore/src/interfaces/otel/trace/IReadableSpan.ts new file mode 100644 index 000000000..1c9a0810d --- /dev/null +++ b/shared/AppInsightsCore/src/interfaces/otel/trace/IReadableSpan.ts @@ -0,0 +1,56 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { OTelSpanKind } from "../../../enums/otel/OTelSpanKind"; +import { IOTelHrTime } from "../../IOTelHrTime"; +import { IDistributedTraceContext } from "../../ai/IDistributedTraceContext"; +import { IOTelAttributes } from "../IOTelAttributes"; +import { IOTelSpan } from "./IOTelSpan"; +import { IOTelSpanStatus } from "./IOTelSpanStatus"; + +/** + * Provides an OpenTelemetry compatible Interface for the Open Telemetry Sdk-Trace-Base (1.8.0 and 2.0.0) ReadableSpan type. + * + * The IReadableSpan interface is used to represent a span that can be read and exported, while the OpenTelemetry + * specification defines a ReadableSpan as a Span that has been ended and is ready to be exported. By default all + * spans created by this library implement the IReadableSpan interface which also extends the {@link IOTelSpan} interface. + * + * This interface is defined to provide compatibility with exporters defined by the OpenTelemetry Trace SDK. + * @since 3.4.0 + */ +export interface IReadableSpan extends IOTelSpan { + + /** + * The span's unique identifier. + */ + readonly name: string; + + /** + * Identifies the type (or kind) that this span is representing. + */ + readonly kind: OTelSpanKind; + readonly spanContext: () => IDistributedTraceContext; + readonly parentSpanId?: string; + readonly parentSpanContext?: IDistributedTraceContext; + readonly startTime: IOTelHrTime; + readonly endTime: IOTelHrTime; + readonly status: IOTelSpanStatus; + + /** + * Provides a snapshot of the span's attributes at the time this span was ended. + * @returns A read-only snapshot of the span's attributes + * @remarks + * It is recommended that you only access this property sparingly due to the + * performance cost of taking a snapshot of all attributes. + */ + readonly attributes: IOTelAttributes; + // readonly links: IOTelLink[]; + // readonly events: IOTelTimedEvent[]; + readonly duration: IOTelHrTime; + readonly ended: boolean; + // readonly resource: IOTelResource; + // readonly instrumentationScope: IOTelInstrumentationScope; + readonly droppedAttributesCount: number; + // readonly droppedEventsCount: number; + // readonly droppedLinksCount: number; + } diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/EventHelpers.ts b/shared/AppInsightsCore/src/internal/EventHelpers.ts similarity index 89% rename from shared/AppInsightsCore/src/JavaScriptSDK/EventHelpers.ts rename to shared/AppInsightsCore/src/internal/EventHelpers.ts index 1c58ab99c..19af8c335 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/EventHelpers.ts +++ b/shared/AppInsightsCore/src/internal/EventHelpers.ts @@ -1,8 +1,10 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { arrForEach, arrIndexOf, getDocument, getWindow, isArray, objForEachKey, objKeys } from "@nevware21/ts-utils"; -import { createElmNodeData, createUniqueNamespace } from "./DataCacheHelper"; -import { STR_EMPTY } from "./InternalConstants"; +import { + ICachedValue, arrForEach, arrIndexOf, createCachedValue, getDocument, getWindow, isArray, objForEachKey, objKeys +} from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { createElmNodeData, createUniqueNamespace } from "../utils/DataCacheHelper"; // Added to help with minfication const strOnPrefix = "on"; @@ -10,15 +12,33 @@ const strAttachEvent = "attachEvent"; const strAddEventHelper = "addEventListener"; const strDetachEvent = "detachEvent"; const strRemoveEventListener = "removeEventListener"; -const strEvents = "events" +const strEvents = "events"; const strVisibilityChangeEvt: string = "visibilitychange"; const strPageHide: string = "pagehide"; const strPageShow: string = "pageshow"; const strUnload: string = "unload"; const strBeforeUnload: string = "beforeunload"; -const strPageHideNamespace = createUniqueNamespace("aiEvtPageHide"); -const strPageShowNamespace = createUniqueNamespace("aiEvtPageShow"); +let _strPageHideNamespace: ICachedValue; +let _strPageShowNamespace: ICachedValue; + +/*#__NO_SIDE_EFFECTS__*/ +function _getPageHideNamespace() { + // Note: Using a cached value instead of a lazy value as we want to ensure that the namespace is consistent + // across multiple calls as the `getLazy()` supports runtime invalidation via `setBypassLazyCache()` which would + // result in different namespaces being returned. + !_strPageHideNamespace && (_strPageHideNamespace = createCachedValue(createUniqueNamespace("aiEvtPageHide"))); + return _strPageHideNamespace.v; +} + +/*#__NO_SIDE_EFFECTS__*/ +function _getPageShowNamespace() { + // Note: Using a cached value instead of a lazy value as we want to ensure that the namespace is consistent + // across multiple calls as the `getLazy()` supports runtime invalidation via `setBypassLazyCache()` which would + // result in different namespaces being returned. + !_strPageShowNamespace && (_strPageShowNamespace = createCachedValue(createUniqueNamespace("aiEvtPageShow"))); + return _strPageShowNamespace.v; +} const rRemoveEmptyNs = /\.[\.]+/g; const rRemoveTrailingEmptyNs = /[\.]+$/; @@ -41,9 +61,20 @@ interface IAiEvents { [name: string]: IRegisteredEvent[] } -const elmNodeData = createElmNodeData("events"); -const eventNamespace = /^([^.]*)(?:\.(.+)|)/ +let _elmNodeData: ICachedValue>; + +/*#__NO_SIDE_EFFECTS__*/ +function _getElmNodeData() { + // Note: Using a cached value instead of a lazy value as we want to ensure that the namespace is consistent + // across multiple calls as the `getLazy()` supports runtime invalidation via `setBypassLazyCache()` which would + // result in different namespaces being returned. + !_elmNodeData && (_elmNodeData = createCachedValue(createElmNodeData("events"))); + return _elmNodeData.v; +} + +const eventNamespace = /^([^.]*)(?:\.(.+)|)/; +/*#__NO_SIDE_EFFECTS__*/ function _normalizeNamespace(name: string) { if (name && name.replace) { return name.replace(/^[\s\.]+|(?=[\s\.])[\.\s]+$/g, STR_EMPTY); @@ -52,6 +83,7 @@ function _normalizeNamespace(name: string) { return name; } +/*#__NO_SIDE_EFFECTS__*/ function _getEvtNamespace(eventName: string | undefined, evtNamespace?: string | string[] | null): IEventDetails { if (evtNamespace) { let theNamespace: string = STR_EMPTY; @@ -103,9 +135,10 @@ export interface _IRegisteredEvents { * @param evtNamespace - [Optional] Additional namespace(s) to append to the event listeners so they can be uniquely identified and removed based on this namespace, * if the eventName also includes a namespace the namespace(s) are merged into a single namespace */ +/*#__NO_SIDE_EFFECTS__*/ export function __getRegisteredEvents(target: any, eventName?: string, evtNamespace?: string | string[]): _IRegisteredEvents[] { let theEvents: _IRegisteredEvents[] = []; - let eventCache = elmNodeData.get(target, strEvents, {}, false); + let eventCache = _getElmNodeData().get(target, strEvents, {}, false); let evtName = _getEvtNamespace(eventName, evtNamespace); objForEachKey(eventCache, (evtType, registeredEvents) => { @@ -125,8 +158,9 @@ export function __getRegisteredEvents(target: any, eventName?: string, evtNamesp } // Exported for internal unit testing only +/*#__NO_SIDE_EFFECTS__*/ function _getRegisteredEvents(target: any, evtName: string, addDefault: boolean = true): IRegisteredEvent[] { - let aiEvts = elmNodeData.get(target, strEvents, {}, addDefault); + let aiEvts = _getElmNodeData().get(target, strEvents, {}, addDefault); let registeredEvents = aiEvts[evtName]; if (!registeredEvents) { registeredEvents = aiEvts[evtName] = []; @@ -183,18 +217,19 @@ function _unregisterEvents(target: any, evtName: IEventDetails, unRegFn: (regEve if (evtName.type) { _doUnregister(target, _getRegisteredEvents(target, evtName.type), evtName, unRegFn); } else { - let eventCache = elmNodeData.get(target, strEvents, {}); + let eventCache = _getElmNodeData().get(target, strEvents, {}); objForEachKey(eventCache, (evtType, events) => { _doUnregister(target, events, evtName, unRegFn); }); // Cleanup if (objKeys(eventCache).length === 0) { - elmNodeData.kill(target, strEvents); + _getElmNodeData().kill(target, strEvents); } } } +/*#__NO_SIDE_EFFECTS__*/ export function mergeEvtNamespace(theNamespace: string, namespaces?: string | string[] | null): string | string[] { let newNamespaces: string | string[]; @@ -233,7 +268,7 @@ export function eventOn(target: T, eventName: string, handlerRef: any, evtNam let evtName = _getEvtNamespace(eventName, evtNamespace); result = _doAttach(target, evtName, handlerRef, useCapture); - if (result && elmNodeData.accept(target)) { + if (result && _getElmNodeData().accept(target)) { let registeredEvent: IRegisteredEvent = { guid: _guid++, evtName: evtName, @@ -472,7 +507,7 @@ export function addPageHideEventListener(listener: any, excludeEvents?: string[] } // add the unique page show namespace to any provided namespace so we can only remove the ones added by "pagehide" - let newNamespaces = mergeEvtNamespace(strPageHideNamespace, evtNamespace); + let newNamespaces = mergeEvtNamespace(_getPageHideNamespace(), evtNamespace); let pageUnloadAdded = _addEventListeners([strPageHide], listener, excludeEvents, newNamespaces); if (!excludeEvents || arrIndexOf(excludeEvents, strVisibilityChangeEvt) === -1) { @@ -497,7 +532,7 @@ export function addPageHideEventListener(listener: any, excludeEvents?: string[] export function removePageHideEventListener(listener: any, evtNamespace?: string | string[] | null) { // add the unique page show namespace to any provided namespace so we only remove the ones added by "pagehide" - let newNamespaces = mergeEvtNamespace(strPageHideNamespace, evtNamespace); + let newNamespaces = mergeEvtNamespace(_getPageHideNamespace(), evtNamespace); removeEventListeners([strPageHide], listener, newNamespaces); removeEventListeners([strVisibilityChangeEvt], null, newNamespaces); } @@ -523,7 +558,7 @@ export function addPageShowEventListener(listener: any, excludeEvents?: string[] } // add the unique page show namespace to any provided namespace so we can only remove the ones added by "pageshow" - let newNamespaces = mergeEvtNamespace(strPageShowNamespace, evtNamespace); + let newNamespaces = mergeEvtNamespace(_getPageShowNamespace(), evtNamespace); let pageShowAdded = _addEventListeners([strPageShow], listener, excludeEvents, newNamespaces); pageShowAdded = _addEventListeners([strVisibilityChangeEvt], _handlePageVisibility, excludeEvents, newNamespaces) || pageShowAdded; @@ -544,7 +579,7 @@ export function addPageShowEventListener(listener: any, excludeEvents?: string[] */ export function removePageShowEventListener(listener: any, evtNamespace?: string | string[] | null) { // add the unique page show namespace to any provided namespace so we only remove the ones added by "pageshow" - let newNamespaces = mergeEvtNamespace(strPageShowNamespace, evtNamespace); + let newNamespaces = mergeEvtNamespace(_getPageShowNamespace(), evtNamespace); removeEventListeners([strPageShow], listener, newNamespaces); removeEventListeners([strVisibilityChangeEvt], null, newNamespaces); } diff --git a/shared/AppInsightsCore/src/internal/attributeHelpers.ts b/shared/AppInsightsCore/src/internal/attributeHelpers.ts new file mode 100644 index 000000000..2d6bdb2de --- /dev/null +++ b/shared/AppInsightsCore/src/internal/attributeHelpers.ts @@ -0,0 +1,93 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { arrForEach, arrSlice, isArray, isObject, isString, objForEachKey } from "@nevware21/ts-utils"; +import { IOTelApi } from "../interfaces/otel/IOTelApi"; +import { IOTelAttributes, OTelAttributeValue } from "../interfaces/otel/IOTelAttributes"; +import { createAttributeContainer } from "../otel/attribute/attributeContainer"; +import { handleWarn } from "./handleErrors"; + +/*#__NO_SIDE_EFFECTS__*/ +function _isSupportedType(theType: string): boolean { + return theType === "number" || theType === "boolean" || theType === "string"; +} + +/*#__NO_SIDE_EFFECTS__*/ +function _isHomogeneousArray(arr: unknown[]): boolean { + let type: string | undefined; + let result = true; + + arrForEach(arr, (element) => { + // null/undefined elements are allowed + if (element !== null) { + let elType = typeof element; + + if (!type) { + result = _isSupportedType(elType); + type = elType; + } else { + result = (type === elType); + } + } + + if (!result) { + return -1; + } + }); + + return result; +} + +/** + * Helper to determine if the provided key is a valid attribute key + * @param key - The key to check + * @returns true if the key is a valid attribute key + */ +/*#__NO_SIDE_EFFECTS__*/ +export function isAttributeKey(key: unknown): key is string { + return isString(key) && !!key; +} + +/** + * Helper to determine if the provided value is a valid attribute value + * @param val - The value to check + * @returns true if the value is a valid attribute value + */ +/*#__NO_SIDE_EFFECTS__*/ +export function isAttributeValue(val: unknown): val is OTelAttributeValue { + let result = (val === null || _isSupportedType(typeof val)); + if (val && isArray(val)) { + result = _isHomogeneousArray(val); + } + + return result; +} + +/** + * Sanitize the provided attributes to ensure they conform to OTel attribute requirements + * @param otelApi - The OpenTelemetry API instance + * @param attributes - The attributes to sanitize + * @returns The sanitized attributes + */ +/*#__NO_SIDE_EFFECTS__*/ +export function sanitizeAttributes(otelApi: IOTelApi, attributes: unknown): IOTelAttributes { + let container = createAttributeContainer(otelApi.cfg); + + if (!isObject(attributes) || attributes == null) { + return {}; + } + + objForEachKey(attributes, (key: string, val: unknown) => { + if (!isAttributeKey(key)) { + handleWarn(otelApi.cfg.errorHandlers, "Invalid attribute key: " + key); + } else if (!isAttributeValue(val)) { + handleWarn(otelApi.cfg.errorHandlers, "Invalid attribute value set for : " + key); + } else if (isArray(val)) { + container.set(key, arrSlice(val as any) as OTelAttributeValue); + } else { + container.set(key, val); + } + }); + + return container.attributes; +} diff --git a/shared/AppInsightsCore/src/internal/commonUtils.ts b/shared/AppInsightsCore/src/internal/commonUtils.ts new file mode 100644 index 000000000..c0047d7a4 --- /dev/null +++ b/shared/AppInsightsCore/src/internal/commonUtils.ts @@ -0,0 +1,329 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IPromise, createRacePromise, createTimeoutPromise } from "@nevware21/ts-async"; +import { ILazyValue, asString, dumpObj, isError, isObject, isPrimitive, safe, safeGetLazy } from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { OTelAttributeValue } from "../interfaces/otel/IOTelAttributes"; +import { IAttributeContainer } from "../interfaces/otel/attribute/IAttributeContainer"; +import { IOTelErrorHandlers } from "../interfaces/otel/config/IOTelErrorHandlers"; +import { + DBSYSTEMVALUES_DB2, DBSYSTEMVALUES_DERBY, DBSYSTEMVALUES_H2, DBSYSTEMVALUES_HSQLDB, DBSYSTEMVALUES_MARIADB, DBSYSTEMVALUES_MSSQL, + DBSYSTEMVALUES_ORACLE, DBSYSTEMVALUES_OTHER_SQL, DBSYSTEMVALUES_SQLITE +} from "../otel/attribute/SemanticConventions"; +import { getJSON } from "../utils/EnvUtils"; +import { handleError } from "./handleErrors"; + +const _hasJsonStringify: ILazyValue = (/*#__PURE__*/ safeGetLazy(() => !!getJSON().stringify, null)); + +const SYNTHETIC_TYPE = "user_agent.synthetic.type"; +const CLIENT_DOT = "client."; +const HTTP_DOT = "http."; +const NET_DOT = "net."; +const PEER_DOT = "peer."; +const ATTR_NETWORK_PEER_ADDRESS = "network.peer.address"; +const SEMATTRS_NET_PEER_IP = NET_DOT + PEER_DOT + "ip"; +const ATTR_CLIENT_ADDRESS = CLIENT_DOT + "address"; +const SEMATTRS_HTTP_CLIENT_IP = HTTP_DOT + "client_ip"; +const ATTR_USER_AGENT_ORIGINAL = "user_agent.original"; +const SEMATTRS_HTTP_USER_AGENT = HTTP_DOT + "user_agent"; +const ATTR_URL_FULL = "url.full"; +const SEMATTRS_HTTP_URL = HTTP_DOT + "url"; +const ATTR_HTTP_REQUEST_METHOD = HTTP_DOT + "request.method"; +const SEMATTRS_HTTP_METHOD = HTTP_DOT + "method"; +const ATTR_HTTP_RESPONSE_STATUS_CODE = HTTP_DOT + "response.status_code"; +const SEMATTRS_HTTP_STATUS_CODE = HTTP_DOT + "status_code"; +const ATTR_URL_SCHEME = "url.scheme"; +const SEMATTRS_HTTP_SCHEME = HTTP_DOT + "scheme"; +const ATTR_URL_PATH = "url.path"; +const ATTR_URL_QUERY = "url.query"; +const SEMATTRS_HTTP_TARGET = HTTP_DOT + "target"; +const ATTR_SERVER_ADDRESS = "server.address"; +const SEMATTRS_HTTP_HOST = HTTP_DOT + "host"; +const SEMATTRS_NET_PEER_NAME = NET_DOT + PEER_DOT + "name"; +const ATTR_CLIENT_PORT = CLIENT_DOT + "port"; +const ATTR_SERVER_PORT = "server.port"; +const SEMATTRS_NET_PEER_PORT = NET_DOT + PEER_DOT + "port"; +const SEMATTRS_PEER_SERVICE = PEER_DOT + "service"; + +/** + * Get the URL from the attribute container + * @param container - The attribute container to extract the URL from + * @returns The constructed URL string + */ +/* #__NO_SIDE_EFFECTS__ */ +export function getUrl(container: IAttributeContainer): string { + let result = ""; + if (container) { + const httpMethod = getHttpMethod(container); + if (httpMethod) { + const httpUrl = getHttpUrl(container); + if (httpUrl) { + result = asString(httpUrl); + } else { + const httpScheme = getHttpScheme(container); + const httpTarget = getHttpTarget(container); + if (httpScheme && httpTarget) { + const httpHost = getHttpHost(container); + if (httpHost) { + result = httpScheme + "://" + httpHost + httpTarget; + } else { + const netPeerPort = getNetPeerPort(container); + if (netPeerPort) { + const netPeerName = getNetPeerName(container); + if (netPeerName) { + result = httpScheme + "://" + netPeerName + ":" + netPeerPort + httpTarget; + } else { + const netPeerIp = getPeerIp(container); + if (netPeerIp) { + result = httpScheme + "://" + netPeerIp + ":" + netPeerPort + httpTarget; + } + } + } + } + } + } + } + } + + return result; +} + +/** + * Gets the synthetic type from the attribute container + * @param container - The attribute container to extract the synthetic type from + * @returns The synthetic type value + */ +/* #__NO_SIDE_EFFECTS__ */ +export function getSyntheticType(container: IAttributeContainer): OTelAttributeValue { + return container.get(SYNTHETIC_TYPE); +} + +/** + * Determine if the attribute container represents a synthetic source + * @param container - The attribute container to check + * @returns True if the attribute container is from a synthetic source + */ +/* #__NO_SIDE_EFFECTS__ */ +export function isSyntheticSource(container: IAttributeContainer): boolean { + return !!getSyntheticType(container); +} + +/** + * Serialize an attribute value to a string value + * @param value - The attribute value to serialize + * @returns The serialized string value + */ +/* #__NO_SIDE_EFFECTS__ */ +export function serializeAttribute(value: any): string | number | bigint | boolean | undefined | symbol | null { + let result: string | number | boolean | null | undefined; + if (isError(value)) { + result = dumpObj(value); + } else if ((!value.toString || isObject(value)) && _hasJsonStringify.v) { + result = safe(getJSON().stringify, [value]).v; + } else if (isPrimitive(value)) { + // We keep primitives as-is, so that the standard attribute types are preserved + // These are converted as required in the sending channel(s) + result = value as any; + } else { + result = asString(value); + } + + // Return scalar and undefined values + return result; +} + +/** + * Peer address of the network connection - IP address or Unix domain socket name. + * + * @example 10.1.2.80 + * @example /tmp/my.sock + * @since 3.4.0 + */ +/* #__NO_SIDE_EFFECTS__ */ +export function getPeerIp(container: IAttributeContainer): OTelAttributeValue | undefined { + if (container) { + return container.get(ATTR_NETWORK_PEER_ADDRESS) || container.get(SEMATTRS_NET_PEER_IP); + } +} + +/* #__NO_SIDE_EFFECTS__ */ +export function getLocationIp(container: IAttributeContainer): OTelAttributeValue | undefined { + let result: OTelAttributeValue | undefined; + if (container) { + const httpClientIp = getHttpClientIp(container); + if (httpClientIp) { + result = asString(httpClientIp); + } + + if (!result) { + const netPeerIp = getPeerIp(container); + if (netPeerIp) { + result = asString(netPeerIp); + } + } + } + + return result; +} + +/* #__NO_SIDE_EFFECTS__ */ +export function getHttpClientIp(container: IAttributeContainer): OTelAttributeValue | undefined { + if (container) { + return container.get(ATTR_CLIENT_ADDRESS) || container.get(SEMATTRS_HTTP_CLIENT_IP); + } +} + +/* #__NO_SIDE_EFFECTS__ */ +export function getUserAgent(container: IAttributeContainer): OTelAttributeValue | undefined { + if (container) { + return container.get(ATTR_USER_AGENT_ORIGINAL) || container.get(SEMATTRS_HTTP_USER_AGENT); + } +} + +/* #__NO_SIDE_EFFECTS__ */ +export function getHttpUrl(container: IAttributeContainer): OTelAttributeValue | undefined { + // Stable sem conv only supports populating url from `url.full` + if (container) { + return container.get(ATTR_URL_FULL) || container.get(SEMATTRS_HTTP_URL); + } +} + +/* #__NO_SIDE_EFFECTS__ */ +export function getHttpMethod(container: IAttributeContainer): OTelAttributeValue | undefined { + if (container) { + return container.get(ATTR_HTTP_REQUEST_METHOD) || container.get(SEMATTRS_HTTP_METHOD); + } +} + +/* #__NO_SIDE_EFFECTS__ */ +export function getHttpStatusCode(container: IAttributeContainer): OTelAttributeValue | undefined { + if (container) { + return container.get(ATTR_HTTP_RESPONSE_STATUS_CODE) || container.get(SEMATTRS_HTTP_STATUS_CODE); + } +} + +/* #__NO_SIDE_EFFECTS__ */ +export function getHttpScheme(container: IAttributeContainer): OTelAttributeValue | undefined { + if (container) { + return container.get(ATTR_URL_SCHEME) || container.get(SEMATTRS_HTTP_SCHEME); + } +} + +/* #__NO_SIDE_EFFECTS__ */ +export function getHttpTarget(container: IAttributeContainer): OTelAttributeValue | undefined { + if (container) { + return container.get(ATTR_URL_PATH) || container.get(ATTR_URL_QUERY) || container.get(SEMATTRS_HTTP_TARGET); + } +} + +/* #__NO_SIDE_EFFECTS__ */ +export function getHttpHost(container: IAttributeContainer): OTelAttributeValue | undefined { + if (container) { + return container.get(ATTR_SERVER_ADDRESS) || container.get(SEMATTRS_HTTP_HOST); + } +} + +/* #__NO_SIDE_EFFECTS__ */ +export function getNetPeerName(container: IAttributeContainer): OTelAttributeValue | undefined { + if (container) { + return container.get(ATTR_CLIENT_ADDRESS) || container.get(SEMATTRS_NET_PEER_NAME); + } +} + +/* #__NO_SIDE_EFFECTS__ */ +export function getNetPeerPort(container: IAttributeContainer): OTelAttributeValue | undefined { + if (container) { + return ( + container.get(ATTR_CLIENT_PORT) || + container.get(ATTR_SERVER_PORT) || + container.get(SEMATTRS_NET_PEER_PORT) + ); + } +} + +export function getDependencyTarget(container: IAttributeContainer): string { + let result: string; + if (container) { + const peerService = container.get(SEMATTRS_PEER_SERVICE); + if (peerService) { + result = asString(peerService); + } + + if (!result) { + const httpHost = getHttpHost(container); + if (httpHost) { + result = asString(httpHost); + } + } + + if (!result) { + const httpUrl = getHttpUrl(container); + if (httpUrl) { + result = asString(httpUrl); + } + } + + if (!result) { + const netPeerName = getNetPeerName(container); + if (netPeerName) { + result = asString(netPeerName); + } + } + if (!result) { + const netPeerIp = getPeerIp(container); + if (netPeerIp) { + result = asString(netPeerIp); + } + } + } + + return result || STR_EMPTY; +} + +/** #__NO_SIDE_EFFECTS__ */ +export function isSqlDB(dbSystem: string): boolean { + return ( + dbSystem === DBSYSTEMVALUES_DB2 || + dbSystem === DBSYSTEMVALUES_DERBY || + dbSystem === DBSYSTEMVALUES_MARIADB || + dbSystem === DBSYSTEMVALUES_MSSQL || + dbSystem === DBSYSTEMVALUES_ORACLE || + dbSystem === DBSYSTEMVALUES_SQLITE || + dbSystem === DBSYSTEMVALUES_OTHER_SQL || + dbSystem === DBSYSTEMVALUES_HSQLDB || + dbSystem === DBSYSTEMVALUES_H2 + ); +} + +/** + * Adds a timeout to a promise and rejects if the specified timeout has elapsed. + * Reports the timeout through the configured error handlers before rejecting. + * + * @param handlers - The configured error handlers to notify. + * @param promise - The promise to guard with the timeout. + * @param timeout - Timeout in milliseconds before the promise is rejected. + */ +export function callWithTimeout( + handlers: IOTelErrorHandlers, + promise: Promise, + timeout: number +): IPromise { + const timeoutMessage = "Operation timed out."; + const timeoutError = new Error(timeoutMessage); + timeoutError.name = "TimeoutError"; + (timeoutError as { __otelTimeout?: boolean }).__otelTimeout = true; + + const racedPromise = createRacePromise([ + promise, + createTimeoutPromise(timeout, false, timeoutError) as unknown as PromiseLike + ]); + + return racedPromise.catch((error) => { + if (error && (error === timeoutError || (error as { __otelTimeout?: boolean }).__otelTimeout)) { + handleError(handlers, timeoutMessage); + } + + throw error; + }); +} diff --git a/shared/AppInsightsCore/src/internal/handleErrors.ts b/shared/AppInsightsCore/src/internal/handleErrors.ts new file mode 100644 index 000000000..c035a7698 --- /dev/null +++ b/shared/AppInsightsCore/src/internal/handleErrors.ts @@ -0,0 +1,107 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { dumpObj, fnApply } from "@nevware21/ts-utils"; +import { IOTelErrorHandlers } from "../interfaces/otel/config/IOTelErrorHandlers"; + +/** + * Handle / report an error. + * When not provided the default is to generally throw an {@link OTelInvalidAttributeError} + * @param handlers - The error handlers configuration + * @param message - The error message to report + * @param key - The attribute key that caused the error + * @param value - The attribute value that caused the error + */ +/*#__NO_SIDE_EFFECTS__*/ +export function handleAttribError(handlers: IOTelErrorHandlers, message: string, key: string, value: any) { + if (handlers.attribError) { + handlers.attribError(message, key, value); + } else { + handleWarn(handlers, message + " for [" + key + "]: " + dumpObj(value)); + } +} + +/** + * There was an error with the span. + * @param handlers - The error handlers configuration + * @param message - The message to report + * @param spanName - The name of the span + */ +/*#__NO_SIDE_EFFECTS__*/ +export function handleSpanError(handlers: IOTelErrorHandlers, message: string, spanName: string) { + if (handlers.spanError) { + handlers.spanError(message, spanName); + } else { + handleWarn(handlers, "Span [" + spanName + "]: " + message); + } +} + +/** + * Report a general debug message, should not be treated as fatal + * @param handlers - The error handlers configuration + * @param message - The debug message to report + */ +/*#__NO_SIDE_EFFECTS__*/ +export function handleDebug(handlers: IOTelErrorHandlers, message: string) { + if (handlers.debug) { + handlers.debug(message); + } else { + if (console) { + let fn = console.log; + fnApply(fn, console, [message]); + } + } +} + +/** + * Report a general warning, should not be treated as fatal + * @param handlers - The error handlers configuration + * @param message - The warning message to report + */ +/*#__NO_SIDE_EFFECTS__*/ +export function handleWarn(handlers: IOTelErrorHandlers, message: string) { + if (handlers.warn) { + handlers.warn(message); + } else { + if (console) { + let fn = console.warn || console.log; + fnApply(fn, console, [message]); + } + } +} + +/** + * Report a general error, should not be treated as fatal + * @param handlers - The error handlers configuration + * @param message - The error message to report + */ +/*#__NO_SIDE_EFFECTS__*/ +export function handleError(handlers: IOTelErrorHandlers, message: string) { + if (handlers.error) { + handlers.error(message); + } else if (handlers.warn) { + handlers.warn(message); + } else { + if (console) { + let fn = console.error || console.warn || console.log; + fnApply(fn, console, [message]); + } + } +} + +/** + * A general error handler for not implemented methods. + * @param handlers - The error handlers configuration + * @param message - The message to report + */ +/*#__NO_SIDE_EFFECTS__*/ +export function handleNotImplemented(handlers: IOTelErrorHandlers, message: string) { + if (handlers.notImplemented) { + handlers.notImplemented(message); + } else { + if (console) { + let fn = console.error || console.log; + fnApply(fn, console, [message]); + } + } +} diff --git a/shared/AppInsightsCore/src/internal/noopHelpers.ts b/shared/AppInsightsCore/src/internal/noopHelpers.ts new file mode 100644 index 000000000..d675ef456 --- /dev/null +++ b/shared/AppInsightsCore/src/internal/noopHelpers.ts @@ -0,0 +1,16 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +/** + * A simple function that does nothing and returns the current this (if any). + * @returns + */ +export function _noopThis(this: T): T { + return this; +} + +/** + * A simple function that does nothing and returns undefined. + */ +export function _noopVoid(this: T): void { +} \ No newline at end of file diff --git a/shared/AppInsightsCore/src/internal/timeHelpers.ts b/shared/AppInsightsCore/src/internal/timeHelpers.ts new file mode 100644 index 000000000..818fb4d5a --- /dev/null +++ b/shared/AppInsightsCore/src/internal/timeHelpers.ts @@ -0,0 +1,371 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { + ICachedValue, ObjDefinePropDescriptor, createCachedValue, getDeferred, getPerformance, isArray, isDate, isNullOrUndefined, isNumber, + mathFloor, mathRound, objDefineProps, objFreeze, perfNow, strLeft, strRight, throwTypeError +} from "@nevware21/ts-utils"; +import { IOTelHrTime, OTelTimeInput } from "../interfaces/IOTelHrTime"; +import { setObjStringTag, toISOString } from "../utils/HelperFuncs"; +import { INVALID_TRACE_ID } from "../utils/TraceParent"; + +const NANOSECOND_DIGITS = 9; + +// Constants for time unit conversions and manipulation +const NANOS_IN_MILLIS = 1000000; // Number of nanoseconds in a millisecond +const NANOS_IN_SECOND = 1000000000; // Number of nanoseconds in a second +const MICROS_IN_SECOND = 1000000; // Number of microseconds in a second +const MICROS_IN_MILLIS = 1000; // Number of microseconds in a millisecond +const MILLIS_IN_SECOND = 1000; // Number of milliseconds in a second +interface IOriginHrTime { + to: number; + hr: IOTelHrTime +} + +let cSecondsToNanos: ICachedValue; +let cTimeOrigin: ICachedValue; +let cNanoPadding: ICachedValue; + +function _notMutable() { + throwTypeError("HrTime is not mutable") +} + +/** + * Initialize the cached value for converting milliseconds to nanoseconds. + * @returns + */ +/*#__NO_SIDE_EFFECTS__*/ +function _initSecondsToNanos(): ICachedValue { + if (!cSecondsToNanos) { + cSecondsToNanos = createCachedValue(NANOS_IN_SECOND); + } + return cSecondsToNanos; +} + +/** + * Initialize the time origin. + * @returns + */ +/*#__NO_SIDE_EFFECTS__*/ +function _initTimeOrigin(): ICachedValue { + if (!cTimeOrigin) { + let timeOrigin = 0; + let perf = getPerformance(); + if (perf) { + timeOrigin = perf.timeOrigin; + if (!isNumber(timeOrigin)) { + timeOrigin = (perf as any).timing && (perf as any).timing.fetchStart; + } + + if (!isNumber(timeOrigin) && perf.now) { + timeOrigin = perf.now(); + } + } + + cTimeOrigin = createCachedValue({ + to: timeOrigin, + hr: millisToHrTime(timeOrigin) + }); + } + return cTimeOrigin; +} + +/*#__NO_SIDE_EFFECTS__*/ +function _finalizeHrTime(hrTime: IOTelHrTime) { + function _toString() { + return "[" + hrTime[0] + ", " + hrTime[1] + "]"; + } + + setObjStringTag(hrTime, _toString); + + return objFreeze(hrTime); +} + +/*#__NO_SIDE_EFFECTS__*/ +function _createUnixNanoHrTime(unixNano: number): IOTelHrTime { + // Create array with initial length of 2 + const hrTime = [0, 0] as any as IOTelHrTime; + const immutable: ObjDefinePropDescriptor = { v: _notMutable, w: false, e: false }; + + // Define the array elements and other properties (avoid redefining length) + objDefineProps(hrTime, { + 0: { + l: getDeferred(() => mathFloor(unixNano / NANOS_IN_SECOND)) + }, + 1: { + l: getDeferred(() => unixNano % NANOS_IN_SECOND) + }, + // unixNano: { + // v: unixNano, + // e: false, + // w: false + // }, + // Override array mutating methods with single _notMutable function + push: immutable, + pop: immutable, + shift: immutable, + unshift: immutable, + splice: immutable, + sort: immutable, + reverse: immutable, + fill: immutable, + copyWithin: immutable + }); + + return _finalizeHrTime(hrTime); +} + +/*#__NO_SIDE_EFFECTS__*/ +function _createHrTime(seconds: number, nanoseconds: number): IOTelHrTime { + const hrTime = [seconds, nanoseconds] as IOTelHrTime; + + // objDefine(hrTime, "unixNano", { + // v: (seconds * NANOS_IN_SECOND) + nanoseconds, + // w: false, + // e: false + // }); + + return _finalizeHrTime(hrTime); +} + +/** + * Returns a new HrTime object with zero values for seconds and nanoseconds. + * @returns A HrTime object representing zero time. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function zeroHrTime(): IOTelHrTime { + return _createUnixNanoHrTime(0); +} + +/** + * Converts a number of milliseconds from epoch to HrTime([seconds, remainder in nanoseconds]). + * @param epochMillis - The number of milliseconds since the epoch (January 1, 1970). + * @returns A HrTime object representing the converted time. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function millisToHrTime(epochMillis: number): IOTelHrTime { + let result: IOTelHrTime; + + if (epochMillis > 0) { + // Handle whole and fractional parts separately for maximum precision + const wholeMillis = mathFloor(epochMillis); + const fractionalMillis = epochMillis - wholeMillis; + + // Handle whole milliseconds using integer arithmetic + const seconds = mathFloor(wholeMillis / MILLIS_IN_SECOND); + const millisFromSeconds = wholeMillis % MILLIS_IN_SECOND; + const nanosFromWholeMillis = millisFromSeconds * NANOS_IN_MILLIS; + + // Convert fractional milliseconds to nanoseconds with proper rounding + // Use Math.round to properly handle cases where we need to round up + const nanosFromFraction = mathRound(fractionalMillis * NANOS_IN_MILLIS); + + // Combine the nanoseconds parts and handle any potential overflow + let totalNanos = nanosFromWholeMillis + nanosFromFraction; + let adjustedSeconds = seconds; + + // Check if we need to increment the seconds + if (totalNanos >= NANOS_IN_SECOND) { + adjustedSeconds++; + totalNanos -= NANOS_IN_SECOND; + } + + result = _createHrTime(adjustedSeconds, totalNanos); + } + + return result || zeroHrTime(); +} + +/** + * Converts a number of nanoseconds to HrTime([seconds, remainder in nanoseconds]). + * @param nanos - The number of nanoseconds since the epoch (January 1, 1970). + * @returns A HrTime object representing the converted time. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function nanosToHrTime(nanos: number): IOTelHrTime { + let result: IOTelHrTime; + if (nanos > 0) { + result = _createUnixNanoHrTime(nanos); + } + return result || zeroHrTime(); +} + +// /** +// * Converts a HrTime object to a number representing nanoseconds since epoch. +// * Note: Due to JavaScript number limitations, values greater than Number.MAX_SAFE_INTEGER +// * may lose precision. For very large time values, consider using string representation +// * or splitting into separate second/nanosecond components. +// * @param hrTime - The HrTime object to convert. +// * @returns The number of nanoseconds represented by the HrTime object. +// */ +// export function hrTimeToUnixNanos(hrTime: IOTelHrTime): number { +// let value = hrTime.unixNano; +// if (isNullOrUndefined(value)) { +// // Handle legacy HRTime format using standard number operations +// // First calculate seconds contribution to nanoseconds +// const secondsInNanos = hrTime[0] * NANOS_IN_MILLIS; +// // Add the additional nanoseconds +// value = secondsInNanos + hrTime[1]; +// +// // // Add warning if we're approaching number precision limits +// // if (Math.abs(value) > Number.MAX_SAFE_INTEGER) { +// // console.warn("Time value exceeds safe integer limits, precision may be lost"); +// // } +// } +// +// return value; +// } + +/** + * Returns an hrtime calculated via performance component. + * @param performanceNow - The current time in milliseconds since the epoch. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function hrTime(performanceNow?: number): IOTelHrTime { + let result = millisToHrTime(isNumber(performanceNow) ? performanceNow : perfNow()); + const perf = getPerformance(); + if (perf) { + const timeOrigin = cTimeOrigin || _initTimeOrigin(); + result = addHrTimes(timeOrigin.v.hr, result); + } + + return result; +} + +/** + * Converts a TimeInput to an HrTime, defaults to _hrtime(). + * @param time - The time input to convert. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function timeInputToHrTime(time: OTelTimeInput): IOTelHrTime { + let result: IOTelHrTime; + + if (!isTimeInputHrTime(time)) { + if (isNumber(time)) { + const timeOrigin = cTimeOrigin || _initTimeOrigin(); + // Must be a performance.now() if it's smaller than process start time + result = (time < timeOrigin.v.to) ? hrTime(time) : millisToHrTime(time); + } else if (isDate(time)) { + result = millisToHrTime((time as Date).getTime()); + } else { + throwTypeError("Invalid input type"); + } + } else { + // Convert HrTime array to IOTelHrTime + result = _createHrTime(time[0], time[1]); + } + + return result; +} + +/** + * Returns a duration of two hrTime. + * @param startTime - The start time of the duration + * @param endTime - The end time of the duration + * @returns The duration between startTime and endTime as an IOTelHrTime + */ +/*#__NO_SIDE_EFFECTS__*/ +export function hrTimeDuration(startTime: IOTelHrTime, endTime: IOTelHrTime): IOTelHrTime { + const seconds = endTime[0] - startTime[0]; + let nanos = endTime[1] - startTime[1]; + + // overflow + if (nanos < 0) { + const adjustedSeconds = seconds - 1; + // negate + const nanoSeconds = cSecondsToNanos || _initSecondsToNanos(); + nanos += nanoSeconds.v; + return _createHrTime(adjustedSeconds, nanos); + } + + return _createHrTime(seconds, nanos); +} + +/** + * Convert hrTime to timestamp, for example "2019-05-14T17:00:00.000123456Z" + * @param time - The hrTime to convert. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function hrTimeToTimeStamp(time: IOTelHrTime): string { + if (!cNanoPadding) { + cNanoPadding = createCachedValue(strLeft(INVALID_TRACE_ID, NANOSECOND_DIGITS)); + } + + const date = toISOString(new Date(time[0] * 1000)); + return date.replace("000Z", strRight(cNanoPadding.v + time[1] + "Z", NANOSECOND_DIGITS + 1)); +} + +/** + * Convert hrTime to nanoseconds. + * @param time - The hrTime to convert. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function hrTimeToNanoseconds(time: IOTelHrTime): number { + let nanoSeconds = cSecondsToNanos || _initSecondsToNanos(); + return time[0] * nanoSeconds.v + time[1]; +} + +/** + * Convert hrTime to milliseconds. + * @param time - The hrTime to convert. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function hrTimeToMilliseconds(time: IOTelHrTime): number { + // Use integer math for the seconds part to avoid floating point precision loss + const millisFromSeconds = time[0] * MILLIS_IN_SECOND; + // Convert nanoseconds to milliseconds with proper rounding + const millisFromNanos = Math.round(time[1] / NANOS_IN_MILLIS); + return millisFromSeconds + millisFromNanos; +} + +/** + * Convert hrTime to microseconds. + * @param time - The hrTime to convert. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function hrTimeToMicroseconds(time: IOTelHrTime): number { + // Use integer math for the seconds part to avoid floating point precision loss + const microsFromSeconds = time[0] * MICROS_IN_SECOND; + // Convert nanoseconds to microseconds with proper rounding + const microsFromNanos = Math.round(time[1] / MICROS_IN_MILLIS); + return microsFromSeconds + microsFromNanos; +} + +/** + * check if time is HrTime + * @param value - The value to check. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function isTimeInputHrTime(value: unknown): value is IOTelHrTime { + return isArray(value) && value.length === 2 && isNumber(value[0]) && isNumber(value[1]); +} + +/** + * check if input value is a correct types.TimeInput + * @param value - The value to check. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function isTimeInput(value: unknown): value is OTelTimeInput { + return !isNullOrUndefined(value) && (isTimeInputHrTime(value) || isNumber(value) || isDate(value)); +} + +/** + * Given 2 HrTime formatted times, return their sum as an HrTime. + * @param time1 - The first HrTime to add + * @param time2 - The second HrTime to add + * @returns The sum of the two HrTime values as an IOTelHrTime + */ +/*#__NO_SIDE_EFFECTS__*/ +export function addHrTimes(time1: IOTelHrTime, time2: IOTelHrTime): IOTelHrTime { + const seconds = time1[0] + time2[0]; + let nanos = time1[1] + time2[1]; + const nanoSeconds = cSecondsToNanos || _initSecondsToNanos(); + + // Nanoseconds overflow check + if (nanos >= nanoSeconds.v) { + nanos -= nanoSeconds.v; + return _createHrTime(seconds + 1, nanos); + } + + return _createHrTime(seconds, nanos); +} diff --git a/shared/AppInsightsCore/src/otel/api/OTelApi.ts b/shared/AppInsightsCore/src/otel/api/OTelApi.ts new file mode 100644 index 000000000..b379cebb8 --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/OTelApi.ts @@ -0,0 +1,25 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { ILazyValue, objDefineProps } from "@nevware21/ts-utils"; +import { IOTelApi } from "../../interfaces/otel/IOTelApi"; +import { IOTelApiCtx } from "../../interfaces/otel/IOTelApiCtx"; +import { ITraceApi } from "../../interfaces/otel/trace/IOTelTraceApi"; +import { setProtoTypeName } from "../../utils/HelperFuncs"; +import { _createTraceApi } from "./trace/traceApi"; +import { _createTracerProvider } from "./trace/tracerProvider"; + +/*#__NO_SIDE_EFFECTS__*/ +export function createOTelApi(otelApiCtx: IOTelApiCtx): IOTelApi { + let _traceApi: ILazyValue; + + let otelApi = setProtoTypeName(objDefineProps(_createTracerProvider(otelApiCtx.host) as IOTelApi, { + cfg: { g: () => otelApiCtx.host.config }, + trace: { g: () => _traceApi.v }, + host: { g: () => otelApiCtx.host } + }), "OTelApi"); + + _traceApi = _createTraceApi(otelApi); + + return otelApi +} diff --git a/shared/AppInsightsCore/src/otel/api/errors/OTelError.ts b/shared/AppInsightsCore/src/otel/api/errors/OTelError.ts new file mode 100644 index 000000000..7ce1d8868 --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/errors/OTelError.ts @@ -0,0 +1,32 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { CustomErrorConstructor, createCustomError } from "@nevware21/ts-utils"; + +let otelErrorType: OpenTelemetryErrorConstructor; + +export interface OpenTelemetryError extends Error { + +} + +export interface OpenTelemetryErrorConstructor extends CustomErrorConstructor { + + new (message?: string): T; + + (message?: string): T; +} + + +/*#__NO_SIDE_EFFECTS__*/ +export function getOpenTelemetryError(): OpenTelemetryErrorConstructor { + if (!otelErrorType) { + otelErrorType = createCustomError("OpenTelemetryError", function (self, args) { + }); + } + + return otelErrorType; +} + +export function throwOTelError(message: string): never { + throw new (getOpenTelemetryError())(message); +} \ No newline at end of file diff --git a/shared/AppInsightsCore/src/otel/api/errors/OTelInvalidAttributeError.ts b/shared/AppInsightsCore/src/otel/api/errors/OTelInvalidAttributeError.ts new file mode 100644 index 000000000..ce726dc81 --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/errors/OTelInvalidAttributeError.ts @@ -0,0 +1,34 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { createCustomError } from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../../../constants/InternalConstants"; +import { OpenTelemetryError, OpenTelemetryErrorConstructor, getOpenTelemetryError } from "./OTelError"; + +let otelInvalidAttributeErrorType: OTelInvalidAttributeErrorConstructor; + +export interface OTelInvalidAttributeError extends OpenTelemetryError { + readonly attribName: string; + + readonly value: any; +} + +interface OTelInvalidAttributeErrorConstructor extends OpenTelemetryErrorConstructor { + + new (message?: string, attribName?: string, value?: any): OTelInvalidAttributeError; + + (message?: string, attribName?: string, value?: any): OTelInvalidAttributeError; +} + +export function throwOTelInvalidAttributeError(message: string, attribName: string, value: any): void { + if (!otelInvalidAttributeErrorType) { + otelInvalidAttributeErrorType = createCustomError("OTelInvalidAttributeError", function (self, args) { + let len = args.length; + + self.attribName = len > 1 ? args[1] : STR_EMPTY; + self.value = len > 2 ? args[2] : STR_EMPTY; + }, getOpenTelemetryError()); + } + + throw new otelInvalidAttributeErrorType(message, attribName, value); +} diff --git a/shared/AppInsightsCore/src/otel/api/errors/OTelSpanError.ts b/shared/AppInsightsCore/src/otel/api/errors/OTelSpanError.ts new file mode 100644 index 000000000..473cbe549 --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/errors/OTelSpanError.ts @@ -0,0 +1,31 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { createCustomError } from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../../../constants/InternalConstants"; +import { OpenTelemetryError, OpenTelemetryErrorConstructor, getOpenTelemetryError } from "./OTelError"; + +let otelSpanErrorType: OTelSpanErrorConstructor; + +export interface OTelSpanError extends OpenTelemetryError { + readonly spanName: string; +} + +interface OTelSpanErrorConstructor extends OpenTelemetryErrorConstructor { + + new (message?: string, spanName?: string): OTelSpanError; + + (message?: string, spanName?: string): OTelSpanError; +} + +export function throwOTelSpanError(message: string, spanName: string): never { + if (!otelSpanErrorType) { + otelSpanErrorType = createCustomError("OTelSpanError", (self, args) => { + let len = args.length; + + self.spanName = len > 1 ? args[1] : STR_EMPTY; + }, getOpenTelemetryError()); + } + + throw new otelSpanErrorType(message, spanName); +} diff --git a/shared/AppInsightsCore/src/otel/api/trace/span.ts b/shared/AppInsightsCore/src/otel/api/trace/span.ts new file mode 100644 index 000000000..52be737eb --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/trace/span.ts @@ -0,0 +1,290 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { + ILazyValue, dumpObj, getDeferred, isNullOrUndefined, isString, objDefineProps, objFreeze, objIs, objKeys, perfNow +} from "@nevware21/ts-utils"; +import { STR_EMPTY, UNDEFINED_VALUE } from "../../../constants/InternalConstants"; +import { OTelSpanKind, eOTelSpanKind } from "../../../enums/otel/OTelSpanKind"; +import { eOTelSpanStatusCode } from "../../../enums/otel/OTelSpanStatus"; +import { OTelException } from "../../../interfaces/IException"; +import { IOTelHrTime, OTelTimeInput } from "../../../interfaces/IOTelHrTime"; +import { IOTelAttributes } from "../../../interfaces/otel/IOTelAttributes"; +import { IAttributeContainer } from "../../../interfaces/otel/attribute/IAttributeContainer"; +import { IOTelSpanCtx } from "../../../interfaces/otel/trace/IOTelSpanCtx"; +import { IOTelSpanStatus } from "../../../interfaces/otel/trace/IOTelSpanStatus"; +import { IReadableSpan } from "../../../interfaces/otel/trace/IReadableSpan"; +import { isAttributeValue } from "../../../internal/attributeHelpers"; +import { handleAttribError, handleNotImplemented, handleSpanError, handleWarn } from "../../../internal/handleErrors"; +import { + hrTime, hrTimeDuration, hrTimeToMilliseconds, millisToHrTime, timeInputToHrTime, zeroHrTime +} from "../../../internal/timeHelpers"; +import { setProtoTypeName, updateProtoTypeName } from "../../../utils/HelperFuncs"; +import { addAttributes, createAttributeContainer } from "../../attribute/attributeContainer"; + +export function createSpan(spanCtx: IOTelSpanCtx, orgName: string, kind: OTelSpanKind): IReadableSpan { + let otelCfg = spanCtx.api.cfg; + let perfStartTime: number = perfNow(); + let spanContext = spanCtx.spanContext; + let attributes: ILazyValue; + let isEnded = false; + let errorHandlers = otelCfg.errorHandlers || {}; + let spanStartTime: ILazyValue = getDeferred(() => { + if (isNullOrUndefined(spanCtx.startTime)) { + return hrTime(perfStartTime); + } + + return timeInputToHrTime(spanCtx.startTime); + }); + let spanEndTime: IOTelHrTime | undefined; + let spanDuration: IOTelHrTime; + let spanStatus: IOTelSpanStatus | undefined; + // let links: IOTelLink[] = []; + // let events: IOTelTimedEvent[] = []; + let localDroppedAttributes = 0; + let localContainer: IAttributeContainer = null; + // let droppedEvents = 0; + // let droppedLinks = 0; + let isRecording = spanCtx.isRecording !== false; + + if (otelCfg.traceCfg && otelCfg.traceCfg.suppressTracing) { + // Automatically disable the span from recording + isRecording = false; + } + + let spanName = orgName || STR_EMPTY; + if (isRecording) { + attributes = getDeferred(() => createAttributeContainer(otelCfg, spanName, spanCtx.attributes)); + } + + function _handleIsEnded(operation: string, extraMsg?: string): boolean { + if (isEnded) { + handleSpanError(errorHandlers, "Span {traceID: " + spanContext.traceId + ", spanId: " + spanContext.spanId + "} has ended - operation [" + operation + "] unsuccessful" + (extraMsg ? (" - " + extraMsg) : STR_EMPTY) + ".", spanName); + } + + return isEnded; + } + function _toString() { + return "ReadableSpan (\"" + spanName + "\")" + } + + let theSpan: IReadableSpan = setProtoTypeName({ + spanContext: () => spanContext, + setAttribute: (key: string, value: any) => { + let message: string; + + if (value !== null && !_handleIsEnded("setAttribute") && isRecording) { + if (!key || key.length === 0) { + message = "Invalid attribute key: " + dumpObj(key); + } else if (!isAttributeValue(value)) { + message = "Invalid attribute value: " + dumpObj(value); + } + + if (message) { + handleAttribError(errorHandlers, message, key, value); + localDroppedAttributes++; + } else if (attributes){ + attributes.v.set(key, value); + } else { + localDroppedAttributes++; + } + } else { + localDroppedAttributes++; + } + + return theSpan; + }, + setAttributes: (attrs: IOTelAttributes) => { + if (!_handleIsEnded("setAttributes") && isRecording && attributes) { + addAttributes(attributes.v, attrs); + } else { + localDroppedAttributes += (objKeys(attrs).length || 0); + } + + return theSpan; + }, + // addEvent: (name: string, attributesOrStartTime?: IOTelAttributes | OTelTimeInput, startTime?: OTelTimeInput) => { + // droppedEvents++; + // if(!_handleIsEnded("addEvent") && isRecording) { + // handleWarn(errorHandlers, "Span.addEvent: " + name + " not added - No events allowed"); + // } + + // return theSpan; + // }, + // addLink: (link: any) => { + // droppedLinks++; + // if(!_handleIsEnded("addEvent") && isRecording) { + // handleWarn(errorHandlers, "Span.addLink: " + link + " not added - No links allowed"); + // } + + // return theSpan; + // }, + // addLinks: (links: any[]) => { + // droppedLinks += links.length; + // if (!_handleIsEnded("addLinks") && isRecording) { + // handleWarn(errorHandlers, "Span.addLinks: " + links + " not added - No links allowed"); + // } + + // return theSpan; + // }, + setStatus: (newStatus: IOTelSpanStatus) => { + if (!_handleIsEnded("setStatus")) { + spanStatus = newStatus; + if (!isNullOrUndefined(spanStatus) && !isNullOrUndefined(spanStatus.message) && !isString(spanStatus.message)) { + spanStatus.message = dumpObj(spanStatus.message); + } + } + + return theSpan; + }, + updateName: (name: string) => { + if (!_handleIsEnded("updateName") && !objIs(spanName, name)) { + spanName = name; + updateProtoTypeName(theSpan, _toString()); + } + + return theSpan; + }, + end: (endTime?: OTelTimeInput) => { + let calcDuration: number; + if (!_handleIsEnded("end", "You can only call end once")) { + try { + if (!isNullOrUndefined(endTime)) { + // User provided an end time + spanEndTime = timeInputToHrTime(endTime); + spanDuration = hrTimeDuration(spanStartTime.v, spanEndTime); + calcDuration = hrTimeToMilliseconds(spanDuration); + } else { + let perfEndTime = perfNow(); + calcDuration = perfEndTime - perfStartTime; + spanDuration = millisToHrTime(calcDuration); + spanEndTime = hrTime(perfEndTime); + } + + if (calcDuration < 0) { + handleWarn(errorHandlers, "Span.end: duration is negative - startTime > endTime. Setting duration to 0 ms"); + spanDuration = zeroHrTime(); + spanEndTime = spanStartTime.v; + } + + // if (droppedEvents > 0) { + // handleWarn(errorHandlers, "Droped " + droppedEvents + " events"); + // } + + // We don't mark as ended until after the onEnd callback to ensure that it can + // still read / change the span if required as well as ensuring that the returned + // value for isRecording is correct. + spanCtx.onEnd && spanCtx.onEnd(theSpan); + } finally { + // Ensure we mark as ended even if the onEnd throws + isEnded = true; + } + } + }, + isRecording: () => isRecording && !isEnded, + recordException: (exception: OTelException, time?: OTelTimeInput) => { + if (!_handleIsEnded("recordException")) { + if (spanCtx.onException) { + spanCtx.onException(theSpan, exception, time); + } else { + handleNotImplemented(errorHandlers, "Span.recordException: " + dumpObj(exception) + " not handled"); + } + } + } + } as IReadableSpan, _toString()); + + // Make the relevant properties dynamic (and read-only) + objDefineProps(theSpan, { + name: { + g: () => spanName + }, + kind: { + v: kind || eOTelSpanKind.INTERNAL + }, + startTime: { + g: () => { + return spanStartTime.v; + } + }, + endTime: { + g: () => { + return spanEndTime; + } + }, + status: { + g: () => { + return spanStatus || { + code: eOTelSpanStatusCode.UNSET + }; + } + }, + attributes: { + g: () => { + return attributes ? attributes.v.attributes : objFreeze({}); + } + }, + attribContainer: { + g: () => { + if (!attributes && !localContainer) { + // Create an empty container and cache it for future use (for performance only) + localContainer = createAttributeContainer(otelCfg, spanName); + } + + return attributes ? attributes.v : localContainer; + } + }, + links: { + g: () => { + return []; + } + }, + events: { + g: () => { + return []; + } + }, + duration: { + g: () => { + return spanDuration || zeroHrTime(); + } + }, + ended: { + g: () => { + return isEnded; + } + }, + droppedAttributesCount: { + g: () => { + return attributes ? attributes.v.droppedAttributes : localDroppedAttributes; + } + }, + // droppedEventsCount: { + // g: () => { + // return droppedEvents; + // } + // }, + // droppedLinksCount: { + // g: () => { + // return droppedLinks; + // } + // }, + parentSpanContext: { + l: getDeferred(() => { + return spanCtx ? spanCtx.parentSpanContext : UNDEFINED_VALUE; + }) + }, + parentSpanId: { + l: getDeferred(() => { + let parentSpanId = UNDEFINED_VALUE; + if (spanCtx) { + let parentSpanCtx = spanCtx.parentSpanContext; + parentSpanId = parentSpanCtx ? parentSpanCtx.spanId : UNDEFINED_VALUE; + } + + return parentSpanId; + }) + } + }); + + return theSpan; +} diff --git a/shared/AppInsightsCore/src/otel/api/trace/spanContext.ts b/shared/AppInsightsCore/src/otel/api/trace/spanContext.ts new file mode 100644 index 000000000..123f708fe --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/trace/spanContext.ts @@ -0,0 +1,65 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { isNullOrUndefined, isNumber, isObject, isString, objDefineProps } from "@nevware21/ts-utils"; +import { eW3CTraceFlags } from "../../../enums/W3CTraceFlags"; +import { IDistributedTraceContext } from "../../../interfaces/ai/IDistributedTraceContext"; +import { IOTelSpanContext } from "../../../interfaces/otel/trace/IOTelSpanContext"; +import { IOTelTraceState } from "../../../interfaces/otel/trace/IOTelTraceState"; +import { INVALID_SPAN_ID, INVALID_TRACE_ID, isValidSpanId, isValidTraceId } from "../../../utils/TraceParent"; +import { createOTelTraceState } from "./traceState"; + +export function createOTelSpanContext(traceContext: IDistributedTraceContext | IOTelSpanContext): IOTelSpanContext { + + let traceId = isValidTraceId(traceContext.traceId) ? traceContext.traceId : INVALID_TRACE_ID; + let spanId = isValidSpanId(traceContext.spanId) ? traceContext.spanId : INVALID_SPAN_ID; + let isRemote = traceContext.isRemote; + let traceFlags = (!isNullOrUndefined(traceContext.traceFlags) ? traceContext.traceFlags : eW3CTraceFlags.Sampled); + let otTraceState: IOTelTraceState | null = null; + + let traceContextObj: IOTelSpanContext = { + traceId, + spanId, + traceFlags + }; + + return objDefineProps(traceContextObj, { + traceId: { + g: () => traceId, + s: (value: string) => traceId = isValidTraceId(value) ? value : INVALID_TRACE_ID + }, + spanId: { + g: () => spanId, + s: (value: string) => spanId = isValidSpanId(value) ? value : INVALID_SPAN_ID + }, + isRemote: { + g: () => isRemote + }, + traceFlags: { + g: () => traceFlags, + s: (value: number) => traceFlags = value + }, + traceState: { + g: () => { + if (!otTraceState) { + // The Trace State has changed, update the local copy + otTraceState = createOTelTraceState(traceContext.traceState); + } + + return otTraceState; + }, + s: (value: IOTelTraceState) => { + // The Trace State has changed, update the local copy + otTraceState = value; + } + } + }); +} + +export function isSpanContext(spanContext: any): spanContext is IOTelSpanContext { + return spanContext && isObject(spanContext) && isString(spanContext.traceId) && isString(spanContext.spanId) && isNumber(spanContext.traceFlags); +} + +export function wrapDistributedTrace(traceContext: IDistributedTraceContext): IOTelSpanContext { + return createOTelSpanContext(traceContext); +} diff --git a/shared/AppInsightsCore/src/otel/api/trace/traceApi.ts b/shared/AppInsightsCore/src/otel/api/trace/traceApi.ts new file mode 100644 index 000000000..3aff40129 --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/trace/traceApi.ts @@ -0,0 +1,46 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { ICachedValue, fnBind, getDeferred } from "@nevware21/ts-utils"; +import { UNDEFINED_VALUE } from "../../../constants/InternalConstants"; +import { IDistributedTraceContext } from "../../../interfaces/ai/IDistributedTraceContext"; +import { IOTelApi } from "../../../interfaces/otel/IOTelApi"; +import { ITraceApi } from "../../../interfaces/otel/trace/IOTelTraceApi"; +import { IOTelTracerProvider } from "../../../interfaces/otel/trace/IOTelTracerProvider"; +import { IReadableSpan } from "../../../interfaces/otel/trace/IReadableSpan"; +import { setProtoTypeName } from "../../../utils/HelperFuncs"; +import { throwOTelError } from "../errors/OTelError"; +import { isSpanContextValid, wrapSpanContext } from "./utils"; + +/** + * @internal + * Create a new instance of the OpenTelemetry Trace API, this is bound to the + * provided instance of the traceProvider (the {@link IOTelApi} instance), + * to "change" (setGlobalTraceProvider) you MUST create a new instance of this API. + * @param otelApi - The IOTelApi instance associated with this instance + * @returns A new instance of the ITraceApi for the provided ITelApi + */ +export function _createTraceApi(otelApi: IOTelApi): ICachedValue { + let traceProvider: IOTelTracerProvider = otelApi; + if (!traceProvider) { + throwOTelError("Must provide an otelApi instance"); + } + + return getDeferred(() => { + return setProtoTypeName({ + getTracer: fnBind(traceProvider.getTracer, traceProvider), + + // We use fnBind to automatically inject the "otelApi" argument as the first argument to the wrapSpanContext function + wrapSpanContext: fnBind(wrapSpanContext, UNDEFINED_VALUE, [otelApi]) as unknown as (spanContext: IDistributedTraceContext) => IReadableSpan, + + isSpanContextValid: isSpanContextValid, + + getActiveSpan: (): IReadableSpan | undefined | null => { + return otelApi.host ? otelApi.host.getActiveSpan() : null; + }, + setActiveSpan(span: IReadableSpan | undefined | null) { + return otelApi.host ? otelApi.host.setActiveSpan(span) : null; + } + }, "TraceApi"); + }); +} diff --git a/shared/AppInsightsCore/src/otel/api/trace/traceProvider.ts b/shared/AppInsightsCore/src/otel/api/trace/traceProvider.ts new file mode 100644 index 000000000..fc3e258f0 --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/trace/traceProvider.ts @@ -0,0 +1,76 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { objDefine, strSubstr } from "@nevware21/ts-utils"; +import { createDistributedTraceContext } from "../../../core/TelemetryHelpers"; +import { eOTelSpanKind } from "../../../enums/otel/OTelSpanKind"; +import { OTelTimeInput } from "../../../interfaces/IOTelHrTime"; +import { IDistributedTraceContext } from "../../../interfaces/ai/IDistributedTraceContext"; +import { ITraceHost, ITraceProvider } from "../../../interfaces/ai/ITraceProvider"; +import { IOTelApi } from "../../../interfaces/otel/IOTelApi"; +import { IOTelSpanCtx } from "../../../interfaces/otel/trace/IOTelSpanCtx"; +import { IOTelSpanOptions } from "../../../interfaces/otel/trace/IOTelSpanOptions"; +import { IReadableSpan } from "../../../interfaces/otel/trace/IReadableSpan"; +import { generateW3CId } from "../../../utils/CoreUtils"; +import { createSpan } from "./span"; + +/** + * @internal + * Creates a new trace provider adapter + * @param host - The trace host instance (typically IAppInsightsCore). + * @param traceName - The name of the trace provider. + * @param api - The OpenTelemetry API instance (as a lazy value). + * @param onEnd - Optional callback to be invoked when a span ends. + * @param onException - Optional callback to be invoked when an exception is recorded on a span. + * @returns The created trace provider. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function createTraceProvider(host: ITraceHost, traceName: string, api: IOTelApi, onEnd?: (span: IReadableSpan) => void, onException?: (span: IReadableSpan, exception: any, time?: OTelTimeInput) => void): ITraceProvider { + let provider: ITraceProvider = { + api: null, + createSpan: (name: string, options?: IOTelSpanOptions, parent?: IDistributedTraceContext): IReadableSpan => { + let newCtx: IDistributedTraceContext; + let parentCtx: IDistributedTraceContext | undefined; + + if (options && options.root) { + newCtx = createDistributedTraceContext(); + } else { + newCtx = createDistributedTraceContext(parent || host.getTraceCtx()); + if (newCtx.parentCtx) { + parentCtx = newCtx.parentCtx; + } + } + + // Always generate a new spanId + newCtx.spanId = strSubstr(generateW3CId(), 0, 16); + + let spanCtx: IOTelSpanCtx = { + api: api, + spanContext: newCtx, + attributes: options ? options.attributes : undefined, + startTime: options ? options.startTime : undefined, + isRecording: options ? options.recording !== false : true, + onEnd: onEnd, + onException: onException + }; + + if (parentCtx) { + objDefine(spanCtx, "parentSpanContext", { + v: parentCtx, + w: false + }); + } + + return createSpan(spanCtx, name, options?.kind || eOTelSpanKind.INTERNAL); + }, + getProviderId: () => traceName, + isAvailable: () => !!onEnd + }; + + objDefine(provider, "api", { + v: api, + w: false + }); + + return provider; +} diff --git a/shared/AppInsightsCore/src/otel/api/trace/traceState.ts b/shared/AppInsightsCore/src/otel/api/trace/traceState.ts new file mode 100644 index 000000000..6e733687a --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/trace/traceState.ts @@ -0,0 +1,109 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { ICachedValue, createCachedValue, isFunction, isString, objDefine, symbolFor } from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../../../constants/InternalConstants"; +import { IW3cTraceState } from "../../../interfaces/ai/IW3cTraceState"; +import { IOTelTraceState } from "../../../interfaces/otel/trace/IOTelTraceState"; +import { createW3cTraceState, isW3cTraceState } from "../../../telemetry/W3cTraceState"; + +let _otelTraceState: ICachedValue; + +function _initOTelTraceStateSymbol() { + if (!_otelTraceState) { + _otelTraceState = createCachedValue(symbolFor("otTraceState")); + } + + return _otelTraceState; +} + +/*#__NO_SIDE_EFFECTS__*/ +function _createOTelTraceState(traceState: IW3cTraceState): IOTelTraceState { + if (!_otelTraceState) { + _otelTraceState = _initOTelTraceStateSymbol(); + } + + let otTraceState = { + set: (key: string, value: string): IOTelTraceState => { + let newState = createW3cTraceState(STR_EMPTY, traceState); + newState.set(key, value); + return _createOTelTraceState(newState); + }, + unset: (key: string): IOTelTraceState => { + let newState = createW3cTraceState(STR_EMPTY, traceState); + newState.del(key); + return _createOTelTraceState(newState); + }, + get: (key: string): string | undefined => { + return traceState.get(key); + }, + serialize: (): string => { + let headers = traceState.hdrs(1); + if (headers.length > 0) { + return headers[0]; + } + + return STR_EMPTY; + } + }; + + objDefine(otTraceState as any, _otelTraceState.v, { g: () => traceState }); + + return otTraceState; +} + +/** + * Identifies if the provided value is an OpenTelemetry TraceState object + * @param value - The value to check + * @returns true if the value is an OpenTelemetry TraceState object otherwise false + * @remarks The OpenTelemetry TraceState is an immutable object, meaning that any changes made to the trace state will + * @since 3.4.0 + */ +/*#__NO_SIDE_EFFECTS__*/ +export function isOTelTraceState(value: any): value is IOTelTraceState { + if (!_otelTraceState) { + _otelTraceState = _initOTelTraceStateSymbol(); + } + + if (value && value[_otelTraceState.v]) { + return true; + } + + return value && isFunction(value.serialize) && isFunction(value.unset) && isFunction(value.get) && isFunction(value.set); +} + +/** + * Returns an OpenTelemetry compatible instance of the trace state, an important distinction + * between an {@link IW3cTraceState} and an {@link IOTelTraceState} is that the OpenTelemetry version + * is immutable, meaning that any changes made to the trace state will create and return a new + * instance for the {@link IOTelTraceState.set} and {@link IOTelTraceState.unset} methods. + * @param value - The current trace state value + * @returns An OpenTelemetry compatible instance of the trace state + * @remarks The OpenTelemetry TraceState is an immutable object, meaning that any changes made to the trace state will + * @since 3.4.0 + */ +/*#__NO_SIDE_EFFECTS__*/ +export function createOTelTraceState(value?: string | IW3cTraceState | IOTelTraceState | null): IOTelTraceState { + let traceState: IW3cTraceState | null = null; + if (isOTelTraceState(value)) { + let parentTraceState: IW3cTraceState; + if (_otelTraceState) { + // Only attempt the lookup if the symbol has been created and therefore possibly + // assigned to a parent trace state + parentTraceState = (value as any)[_otelTraceState.v] as IW3cTraceState; + } + if (parentTraceState) { + // Reuse the existing trace state as a parent to avoid copying objects + traceState = createW3cTraceState(STR_EMPTY, parentTraceState); + } else { + // fallback to creating a new trace state + traceState = createW3cTraceState(value.serialize()); + } + } else if (isW3cTraceState(value)) { + traceState = value; + } else { + traceState = createW3cTraceState(isString(value) ? value : STR_EMPTY); + } + + return _createOTelTraceState(traceState); +} diff --git a/shared/AppInsightsCore/src/otel/api/trace/tracer.ts b/shared/AppInsightsCore/src/otel/api/trace/tracer.ts new file mode 100644 index 000000000..1fdd313a2 --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/trace/tracer.ts @@ -0,0 +1,52 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { fnApply, isFunction } from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../../../constants/InternalConstants"; +import { ISpanScope, ITraceHost } from "../../../interfaces/ai/ITraceProvider"; +import { IOTelSpanOptions } from "../../../interfaces/otel/trace/IOTelSpanOptions"; +import { IOTelTracer } from "../../../interfaces/otel/trace/IOTelTracer"; +import { IReadableSpan } from "../../../interfaces/otel/trace/IReadableSpan"; +import { setProtoTypeName } from "../../../utils/HelperFuncs"; +import { startActiveSpan } from "./utils"; + +/** + * @internal + * Create a tracer implementation. + * @param host - The ApplicationInsights core instance + * @returns A tracer object + */ +export function _createTracer(host: ITraceHost, name?: string): IOTelTracer { + let tracer: IOTelTracer = setProtoTypeName({ + startSpan(spanName: string, options?: IOTelSpanOptions): IReadableSpan | null { + // Note: context is not used / needed for Application Insights / 1DS + if (host) { + return host.startSpan(spanName, options); + } + + return null; + }, + startActiveSpan) => ReturnType>(name: string, fnOrOptions?: F | IOTelSpanOptions, fn?: F): ReturnType { + // Figure out which parameter order was passed + let theFn: F | null = null; + let opts: IOTelSpanOptions | null = null; + + if (isFunction(fnOrOptions)) { + // startActiveSpan unknown>(name: string, fn: F): ReturnType; + theFn = fnOrOptions; + } else { + // startActiveSpan unknown>(name: string, options: IOTelSpanOptions, fn: F): ReturnType; or + opts = fnOrOptions as IOTelSpanOptions; + theFn = fn; + } + + if (theFn) { + return startActiveSpan(host, name, opts, (spanScope: ISpanScope) => { + return fnApply(theFn, spanScope, [spanScope.span, spanScope]); + }); + } + } + }, "OTelTracer" + (name ? (" (" + name + ")") : STR_EMPTY)); + + return tracer; +} diff --git a/shared/AppInsightsCore/src/otel/api/trace/tracerProvider.ts b/shared/AppInsightsCore/src/otel/api/trace/tracerProvider.ts new file mode 100644 index 000000000..25993f8e4 --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/trace/tracerProvider.ts @@ -0,0 +1,40 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { IPromise } from "@nevware21/ts-async"; +import { ITraceHost } from "../../../interfaces/ai/ITraceProvider"; +import { IOTelTracer } from "../../../interfaces/otel/trace/IOTelTracer"; +import { IOTelTracerProvider } from "../../../interfaces/otel/trace/IOTelTracerProvider"; +import { _createTracer } from "./tracer"; + +/** + * @internal + * Create a trace implementation with tracer caching. + * @param core - The ApplicationInsights core instance + * @returns A trace object + */ +export function _createTracerProvider(host: ITraceHost): IOTelTracerProvider { + let tracers: { [key: string]: IOTelTracer } = {}; + + return { + getTracer(name: string, version?: string): IOTelTracer { + const tracerKey = (name|| "ai-web") + "@" + (version || "unknown"); + + if (!tracers[tracerKey]) { + tracers[tracerKey] = _createTracer(host); + } + + return tracers[tracerKey]; + }, + forceFlush(): IPromise | void { + // Nothing to flush + return; + }, + shutdown(): IPromise | void { + // Just clear the locally cached IOTelTracer instances so they can be garbage collected + tracers = {}; + host = null; + return; + } + }; +} diff --git a/shared/AppInsightsCore/src/otel/api/trace/utils.ts b/shared/AppInsightsCore/src/otel/api/trace/utils.ts new file mode 100644 index 000000000..0b537ac6d --- /dev/null +++ b/shared/AppInsightsCore/src/otel/api/trace/utils.ts @@ -0,0 +1,383 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { doAwait, doFinally } from "@nevware21/ts-async"; +import { arrSlice, fnApply, isFunction, isObject, isPromiseLike } from "@nevware21/ts-utils"; +import { createDistributedTraceContext, isDistributedTraceContext } from "../../../core/TelemetryHelpers"; +import { eOTelSpanKind } from "../../../enums/otel/OTelSpanKind"; +import { eOTelSpanStatusCode } from "../../../enums/otel/OTelSpanStatus"; +import { IAppInsightsCore } from "../../../interfaces/ai/IAppInsightsCore"; +import { IConfiguration } from "../../../interfaces/ai/IConfiguration"; +import { IDistributedTraceContext, IDistributedTraceInit } from "../../../interfaces/ai/IDistributedTraceContext"; +import { ISpanScope, ITraceHost } from "../../../interfaces/ai/ITraceProvider"; +import { IOTelApi } from "../../../interfaces/otel/IOTelApi"; +import { ITraceCfg } from "../../../interfaces/otel/config/IOTelTraceCfg"; +import { IOTelSpanContext } from "../../../interfaces/otel/trace/IOTelSpanContext"; +import { IOTelSpanCtx } from "../../../interfaces/otel/trace/IOTelSpanCtx"; +import { IOTelSpanOptions } from "../../../interfaces/otel/trace/IOTelSpanOptions"; +import { IReadableSpan } from "../../../interfaces/otel/trace/IReadableSpan"; +import { isValidSpanId, isValidTraceId } from "../../../utils/TraceParent"; +import { createSpan } from "./span"; + +/** + * Internal helper to execute a callback function with a span set as the active span. + * Handles both synchronous and asynchronous (Promise-based) callbacks, ensuring + * the previous active span is properly restored after execution. + * @param scope - The span scope instance + * @param fn - The callback function to execute + * @param thisArg - The `this` context for the callback + * @param args - Array of arguments to pass to the callback + * @returns The result of the callback function + */ +function _executeWithActiveSpan( + scope: S, + fn: (...args: any) => any, + thisArg: any, + args: any[] +): R { + let isAsync = false; + try { + let result = fnApply(fn, thisArg || scope, args); + if (isPromiseLike(result)) { + isAsync = true; + return doFinally(result, function () { + // Restore previous active span after promise settles (resolves or rejects) + if (scope) { + scope.restore(); + } + }) as any; + } + return result; + } finally { + // Restore previous active span only if result is not a promise + // (promises handle restoration in their callbacks) + if (scope && !isAsync) { + scope.restore(); + } + } +} + +/** + * Execute the callback `fn` function with the passed span as the active span + * Note: The callback will be executed even if the span is null. + * @param traceHost - The current trace host instance (core or AISKU instance) + * @param span - The span to set as the active span during the execution of the callback + * @param fn - the callback function + * @param thisArg - the `this` argument for the callback. If not provided, ISpanScope is used as `this` + * @param _args - Additional arguments to be passed to the function + */ +export function withSpan | ISpanScope, ...args: A) => ReturnType>(traceHost: T, span: IReadableSpan, fn: F, thisArg?: ThisParameterType, ..._args: A) : ReturnType; + +/** + * Execute the callback `fn` function with the passed span as the active span + * Note: The callback will be executed even if the span is null. + * @param traceHost - The current trace host instance (core or AISKU instance) + * @param span - The span to set as the active span during the execution of the callback + * @param fn - the callback function + * @param thisArg - the `this` argument for the callback. If not provided, ISpanScope is used as `this` + * @returns the result of the function + */ +export function withSpan | ISpanScope,...args: A) => ReturnType>(traceHost: T, span: IReadableSpan, fn: F, thisArg?: ThisParameterType): ReturnType { + const scope = traceHost.setActiveSpan(span); + return _executeWithActiveSpan(scope, fn, thisArg, arrSlice(arguments, 4)); +} + +/** + * Execute the callback `fn` function with the passed span as the active span. The callback receives + * an ISpanScope object as its first parameter and the `this` context (when no thisArg is provided). + * Note: The callback will be executed even if the span is null. + * @param traceHost - The current trace host instance (core or AISKU instance) + * @param span - The span to set as the active span during the execution of the callback + * @param fn - the callback function that receives an ISpanScope + * @param thisArg - the `this` argument for the callback. If not provided, ISpanScope is used as `this` + * @returns The result of the function + */ +export function useSpan | ISpanScope, scope: ISpanScope) => ReturnType>(traceHost: T, span: IReadableSpan, fn: F, thisArg?: ThisParameterType) : ReturnType; + +/** + * Execute the callback `fn` function with the passed span as the active span. The callback receives + * an ISpanScope object as its first parameter and the `this` context (when no thisArg is provided). + * Note: The callback will be executed even if the span is null. + * @param traceHost - The current trace host instance (core or AISKU instance) + * @param span - The span to set as the active span during the execution of the callback + * @param fn - the callback function that receives an ISpanScope and additional arguments + * @param thisArg - the `this` argument for the callback. If not provided, ISpanScope is used as `this` + * @param _args - Additional arguments to be passed to the function + * @returns The result of the function + */ +export function useSpan | ISpanScope, scope: ISpanScope, ...args: A) => ReturnType>(traceHost: T, span: IReadableSpan, fn: F, thisArg?: ThisParameterType, ..._args: A) : ReturnType; + +/** + * Execute the callback `fn` function with the passed span as the active span. The callback receives + * an ISpanScope object as its first parameter and the `this` context (when no thisArg is provided). + * Note: The callback will be executed even if the span is null. + * @param traceHost - The current trace host instance (core or AISKU instance) + * @param span - The span to set as the active span during the execution of the callback + * @param fn - the callback function that receives an ISpanScope and additional arguments + * @param thisArg - the `this` argument for the callback. If not provided, ISpanScope is used as `this` + * @param _args - Additional arguments to be passed to the function + */ +export function useSpan | ISpanScope, scope: ISpanScope, ...args: A) => ReturnType>(traceHost: T, span: IReadableSpan, fn: F, thisArg?: ThisParameterType): ReturnType { + let scope = traceHost.setActiveSpan(span); + return _executeWithActiveSpan(scope, fn, thisArg, [scope].concat(arrSlice(arguments, 4))); +} + +/** + * Creates and starts a new span, sets it as the active span in the current context, + * and executes a provided function within this context. + * + * This method creates a span, makes it active during the execution of the provided + * function, and automatically ends the span when the function completes (or throws). + * This provides automatic span lifecycle management and context propagation. If the function + * is asynchronous the span will be ended when the returned Promise resolves or rejects. + * Note: The callback will be executed even if the traceHost is unable to create a span (returns null). + * @param name - The name of the span, should be descriptive of the operation being traced + * @param fn - The function to execute within the span's active context + * @param thisArg - The `this` argument for the callback. If not provided, ISpanScope is used as `this` + * @returns The result of executing the provided function + * @remarks + * - The span is automatically ended when the function completes or throws an exception + * - The span becomes the active parent for any spans created within the function + * - If the function throws an error, the span status is automatically set to ERROR + * - This is the recommended method for most tracing scenarios due to automatic lifecycle management + * - Multiple overloads available for different parameter combinations + */ +export function startActiveSpan | ISpanScope, scope?: ISpanScope) => ReturnType>(traceHost: T, name: string, fn: F, thisArg?: ThisParameterType): ReturnType; + +/** + * Creates and starts a new span, sets it as the active span in the current context, + * and executes a provided function within this context. + * + * This method creates a span, makes it active during the execution of the provided + * function, and automatically ends the span when the function completes (or throws). + * This provides automatic span lifecycle management and context propagation. If the function + * is asynchronous the span will be ended when the returned Promise resolves or rejects. + * Note: The callback will be executed even if the traceHost is unable to create a span (returns null). + * @param name - The name of the span, should be descriptive of the operation being traced + * @param options - Optional configuration for span creation (parent context, attributes, etc.) + * @param fn - The function to execute within the span's active context + * @param thisArg - The `this` argument for the callback. If not provided, ISpanScope is used as `this` + * @returns The result of executing the provided function + * @remarks + * - The span is automatically ended when the function completes or throws an exception + * - The span becomes the active parent for any spans created within the function + * - If the function throws an error, the span status is automatically set to ERROR + * - This is the recommended method for most tracing scenarios due to automatic lifecycle management + * - Multiple overloads available for different parameter combinations + */ +export function startActiveSpan | ISpanScope, scope?: ISpanScope) => ReturnType>(traceHost: T, name: string, options: IOTelSpanOptions, fn: F, thisArg?: ThisParameterType): ReturnType; + +/** + * Creates and starts a new span, sets it as the active span in the current context, + * and executes a provided function within this context. + * + * This method creates a span, makes it active during the execution of the provided + * function, and automatically ends the span when the function completes (or throws). + * This provides automatic span lifecycle management and context propagation. If the function + * is asynchronous the span will be ended when the returned Promise resolves or rejects. + * Note: The callback will be executed even if the traceHost is unable to create a span (returns null). + * @remarks + * This overloaded version supports both optional span creation options and the `this` argument for the callback. + * @param name - The name of the span, should be descriptive of the operation being traced + * @param optionsOrFn - Optional configuration for span creation (parent context, attributes, etc.) or the function to execute within the span's active context + * @param maybeFnOrThis - The function to execute within the span's active context or the `this` argument for the callback + * @param thisArg - The `this` argument for the callback. If not provided, ISpanScope is used as `this` + * @returns The result of executing the provided function + * @remarks + * - The span is automatically ended when the function completes or throws an exception + * - The span becomes the active parent for any spans created within the function + * - If the function throws an error, the span status is automatically set to ERROR + * - This is the recommended method for most tracing scenarios due to automatic lifecycle management + * - Multiple overloads available for different parameter combinations + */ +export function startActiveSpan | ISpanScope, scope?: ISpanScope) => ReturnType>(traceHost: T, name: string, optionsOrFn: IOTelSpanOptions | F, maybeFnOrThis?: F | ThisParameterType, thisArg?: ThisParameterType): ReturnType { + let options: IOTelSpanOptions = null; + let fn: F; + let that: ThisParameterType = thisArg; + + if (isFunction(optionsOrFn)) { + fn = optionsOrFn as F; + } else { + options = optionsOrFn as IOTelSpanOptions; + fn = maybeFnOrThis as F; + that = thisArg || (maybeFnOrThis as ThisParameterType); + } + + let span = traceHost.startSpan(name, options); + let useAsync = false; + + try { + let result = useSpan(traceHost, span, (scope: ISpanScope): ReturnType => { + return fnApply(fn, that, [scope]); + }); + + if (isPromiseLike(result)) { + useAsync = true; + + return doAwait(result, + (value) => value, + (reason) => { + if (span) { + span.setStatus({ code: reason ? eOTelSpanStatusCode.ERROR : eOTelSpanStatusCode.OK, message: reason ? reason.message || reason : undefined }); + } + }, + () => { + if (span) { + span.end(); + } + }) as ReturnType; + } + + return result; + } catch (e) { + if (span) { + span.setStatus({ code: e ? eOTelSpanStatusCode.ERROR : eOTelSpanStatusCode.OK, message: e ? e.message : undefined }); + } + throw e; + } finally { + // If the function returned a promise, we need to end the span when the promise resolves/rejects + if (!useAsync && span) { + span.end(); + } + } +} + +/** + * Returns true if the passed spanContext of type {@link IDistributedTraceContext} or {@link IDistributedTraceInit} is valid. + * @return true if this {@link IDistributedTraceContext} is valid. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function isSpanContextValid(spanContext: IDistributedTraceContext | IDistributedTraceInit | IOTelSpanContext): boolean { + return spanContext ? (isValidTraceId(spanContext.traceId) && isValidSpanId(spanContext.spanId)) : false; +} + +/** + * Wrap the given {@link IDistributedTraceContext} in a new non-recording {@link IReadableSpan} + * + * @param spanContext - span context to be wrapped + * @returns a new non-recording {@link IReadableSpan} with the provided context + */ +export function wrapSpanContext(otelApi: IOTelApi, spanContext: IDistributedTraceContext | IDistributedTraceInit | IOTelSpanContext): IReadableSpan { + if (!isDistributedTraceContext(spanContext)) { + spanContext = createDistributedTraceContext(spanContext); + } + + // Return a non-recording span + return createNonRecordingSpan(otelApi, "wrapped(\"" + spanContext.spanId + "\")", spanContext); +} + +/** + * Return a non-recording span based on the provided spanContext using the otelApi instance as the + * owning instance. + * @param otelApi - The otelApi to use for creating the non-Recording Span + * @param spanName - The span name to associated with the span + * @param spanContext - The Span context to use for the span + * @returns A new span that is marked as a non-recording span + */ +export function createNonRecordingSpan(otelApi: IOTelApi, spanName: string, spanContext: IDistributedTraceContext | IDistributedTraceInit | IOTelSpanContext): IReadableSpan { + // Return a non-recording span + let spanCtx: IOTelSpanCtx = { + api: otelApi, + spanContext: isDistributedTraceContext(spanContext) ? spanContext : createDistributedTraceContext(spanContext), + isRecording: false + }; + + return createSpan(spanCtx, spanName, eOTelSpanKind.INTERNAL); +} + +/** + * Identifies whether the span is an {@link IReadableSpan} or not + * @param span - The span to check + * @returns true if the span is an {@link IReadableSpan} otherwise false + */ +/*#__NO_SIDE_EFFECTS__*/ +export function isReadableSpan(span: any): span is IReadableSpan { + return !!span && + isObject(span) && + "name" in span && + "kind" in span && + isFunction(span.spanContext) && + "duration" in span && + "ended" in span && + "startTime" in span && + "endTime" in span && + "attributes" in span && + "links" in span && + "events" in span && + "status" in span && + // "resource" in span && + // "instrumentationScope" in span && + "droppedAttributesCount" in span && + isFunction(span.isRecording) && + isFunction(span.setStatus) && + isFunction(span.updateName) && + isFunction(span.setAttribute) && + isFunction(span.setAttributes) && + isFunction(span.end) && + isFunction(span.recordException); +} + +function _getTraceCfg(context: IOTelApi | ITraceHost | IConfiguration): ITraceCfg { + let traceCfg: ITraceCfg = null; + if (context) { + if ((context as IOTelApi).cfg && (context as IOTelApi).host) { + traceCfg = (context as IOTelApi).cfg.traceCfg; + } else if (isFunction((context as IAppInsightsCore).initialize) && (context as IAppInsightsCore).config) { + traceCfg = (context as IAppInsightsCore).config.traceCfg; + } else if ((context as IConfiguration).traceCfg) { + traceCfg = (context as IConfiguration).traceCfg; + } + } + + return traceCfg; +} + +/** + * Set the suppress tracing flag on the context + * @param context - The context to set the suppress tracing flag on + * @returns The context with the suppress tracing flag set + * @remarks This is used to suppress tracing for the current context and all child contexts + */ +/*#__NO_SIDE_EFFECTS__*/ +export function suppressTracing(context: T): T { + let traceCfg = _getTraceCfg(context); + if (traceCfg) { + traceCfg.suppressTracing = true; + } + + return context; +} + +/** + * Remove the suppress tracing flag from the context + * @param context - The context to remove the suppress tracing flag from + * @returns The context with the suppress tracing flag removed + * @remarks This is used to remove the suppress tracing flag from the current context and all child + * contexts. This is used to restore tracing after it has been suppressed + */ +/*#__NO_SIDE_EFFECTS__*/ +export function unsuppressTracing(context: T): T { + let traceCfg = _getTraceCfg(context); + if (traceCfg) { + traceCfg.suppressTracing = false; + } + + return context; +} + +/** + * Check if the tracing is suppressed for the current context + * @param context - The context to check for the suppress tracing flag + * @returns true if tracing is suppressed for the current context otherwise false + * @remarks This is used to check if tracing is suppressed for the current context and all child + */ +/*#__NO_SIDE_EFFECTS__*/ +export function isTracingSuppressed(context: T): boolean { + let result = false; + let traceCfg = _getTraceCfg(context); + if (traceCfg) { + result = !!traceCfg.suppressTracing; + } + + return result; +} diff --git a/shared/AppInsightsCore/src/otel/attribute/SemanticConventions.ts b/shared/AppInsightsCore/src/otel/attribute/SemanticConventions.ts new file mode 100644 index 000000000..c81afe16d --- /dev/null +++ b/shared/AppInsightsCore/src/otel/attribute/SemanticConventions.ts @@ -0,0 +1,106 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +const MICROSOFT_APPLICATIONINSIGHTS_NAME = "Microsoft.ApplicationInsights."; +const MICROSOFT_DOT = "microsoft."; +const CLIENT_DOT = "client."; +const HTTP_DOT = "http."; +const NET_DOT = "net."; +const PEER_DOT = "peer."; +const EXCEPTION_DOT = "exception."; +const ENDUSER_DOT = "enduser."; +const URL_DOT = "url."; +const DB_DOT = "db."; +const NETWORK_DOT = "network."; + +export const AzureMonitorSampleRate = MICROSOFT_DOT + "sample_rate"; +export const ApplicationInsightsCustomEventName = MICROSOFT_DOT + "custom_event.name"; +export const MicrosoftClientIp = MICROSOFT_DOT + CLIENT_DOT + "ip"; + +export const ApplicationInsightsMessageName = MICROSOFT_APPLICATIONINSIGHTS_NAME + "Message"; +export const ApplicationInsightsExceptionName = MICROSOFT_APPLICATIONINSIGHTS_NAME + "Exception"; +export const ApplicationInsightsPageViewName = MICROSOFT_APPLICATIONINSIGHTS_NAME + "PageView"; +export const ApplicationInsightsAvailabilityName = MICROSOFT_APPLICATIONINSIGHTS_NAME + "Availability"; +export const ApplicationInsightsEventName = MICROSOFT_APPLICATIONINSIGHTS_NAME + "Event"; + +export const ApplicationInsightsBaseType = "_MS.baseType"; +export const ApplicationInsightsMessageBaseType = "MessageData"; +export const ApplicationInsightsExceptionBaseType = "ExceptionData"; +export const ApplicationInsightsPageViewBaseType = "PageViewData"; +export const ApplicationInsightsAvailabilityBaseType = "AvailabilityData"; +export const ApplicationInsightsEventBaseType = "EventData"; + +export const ATTR_ENDUSER_ID = ENDUSER_DOT + "id"; +export const ATTR_ENDUSER_PSEUDO_ID = ENDUSER_DOT + "pseudo_id"; +export const ATTR_HTTP_ROUTE = HTTP_DOT + "route"; + +export const SEMATTRS_NET_PEER_IP = NET_DOT + PEER_DOT + "ip"; +export const SEMATTRS_NET_PEER_NAME = NET_DOT + PEER_DOT + "name"; +export const SEMATTRS_NET_HOST_IP = NET_DOT + "host.ip"; +export const SEMATTRS_PEER_SERVICE = PEER_DOT + "service"; +export const SEMATTRS_HTTP_USER_AGENT = HTTP_DOT + "user_agent"; +export const SEMATTRS_HTTP_METHOD = HTTP_DOT + "method"; +export const SEMATTRS_HTTP_URL = HTTP_DOT + "url"; +export const SEMATTRS_HTTP_STATUS_CODE = HTTP_DOT + "status_code"; +export const SEMATTRS_HTTP_ROUTE = HTTP_DOT + "route"; +export const SEMATTRS_HTTP_HOST = HTTP_DOT + "host"; +export const SEMATTRS_DB_SYSTEM = DB_DOT + "system"; +export const SEMATTRS_DB_STATEMENT = DB_DOT + "statement"; +export const SEMATTRS_DB_OPERATION = DB_DOT + "operation"; +export const SEMATTRS_DB_NAME = DB_DOT + "name"; +export const SEMATTRS_RPC_SYSTEM = "rpc.system"; +export const SEMATTRS_RPC_GRPC_STATUS_CODE = "rpc.grpc.status_code"; +export const SEMATTRS_EXCEPTION_TYPE = EXCEPTION_DOT + "type"; +export const SEMATTRS_EXCEPTION_MESSAGE = EXCEPTION_DOT + "message"; +export const SEMATTRS_EXCEPTION_STACKTRACE = EXCEPTION_DOT + "stacktrace"; +export const SEMATTRS_HTTP_SCHEME = HTTP_DOT + "scheme"; +export const SEMATTRS_HTTP_TARGET = HTTP_DOT + "target"; +export const SEMATTRS_HTTP_FLAVOR = HTTP_DOT + "flavor"; +export const SEMATTRS_NET_TRANSPORT = NET_DOT + "transport"; +export const SEMATTRS_NET_HOST_NAME = NET_DOT + "host.name"; +export const SEMATTRS_NET_HOST_PORT = NET_DOT + "host.port"; +export const SEMATTRS_NET_PEER_PORT = NET_DOT + PEER_DOT + "port"; +export const SEMATTRS_HTTP_CLIENT_IP = HTTP_DOT + "client_ip"; +export const SEMATTRS_ENDUSER_ID = ENDUSER_DOT + "id"; + +export const ATTR_CLIENT_ADDRESS = CLIENT_DOT + "address"; +export const ATTR_CLIENT_PORT = CLIENT_DOT + "port"; +export const ATTR_SERVER_ADDRESS = "server.address"; +export const ATTR_SERVER_PORT = "server.port"; +export const ATTR_URL_FULL = URL_DOT + "full"; +export const ATTR_URL_PATH = URL_DOT + "path"; +export const ATTR_URL_QUERY = URL_DOT + "query"; +export const ATTR_URL_SCHEME = URL_DOT + "scheme"; +export const ATTR_ERROR_TYPE = "error.type"; +export const ATTR_NETWORK_LOCAL_ADDRESS = NETWORK_DOT + "local.address"; +export const ATTR_NETWORK_LOCAL_PORT = NETWORK_DOT + "local.port"; +export const ATTR_NETWORK_PROTOCOL_NAME = NETWORK_DOT + "protocol.name"; +export const ATTR_NETWORK_PEER_ADDRESS = NETWORK_DOT + PEER_DOT + "address"; +export const ATTR_NETWORK_PEER_PORT = NETWORK_DOT + PEER_DOT + "port"; +export const ATTR_NETWORK_PROTOCOL_VERSION = NETWORK_DOT + "protocol.version"; +export const ATTR_NETWORK_TRANSPORT = NETWORK_DOT + "transport"; +export const ATTR_USER_AGENT_ORIGINAL = "user_agent.original"; +export const ATTR_HTTP_REQUEST_METHOD = HTTP_DOT + "request.method"; +export const ATTR_HTTP_RESPONSE_STATUS_CODE = HTTP_DOT + "response.status_code"; +export const ATTR_EXCEPTION_TYPE = EXCEPTION_DOT + "type"; +export const ATTR_EXCEPTION_MESSAGE = EXCEPTION_DOT + "message"; +export const ATTR_EXCEPTION_STACKTRACE = EXCEPTION_DOT + "stacktrace"; +export const EXP_ATTR_ENDUSER_ID = ENDUSER_DOT + "id"; +export const EXP_ATTR_ENDUSER_PSEUDO_ID = ENDUSER_DOT + "pseudo_id"; +export const EXP_ATTR_SYNTHETIC_TYPE = "synthetic.type"; + + +export const DBSYSTEMVALUES_MONGODB = "mongodb"; +export const DBSYSTEMVALUES_COSMOSDB = "cosmosdb"; +export const DBSYSTEMVALUES_MYSQL = "mysql"; +export const DBSYSTEMVALUES_POSTGRESQL = "postgresql"; +export const DBSYSTEMVALUES_REDIS = "redis"; +export const DBSYSTEMVALUES_DB2 = "db2"; +export const DBSYSTEMVALUES_DERBY = "derby"; +export const DBSYSTEMVALUES_MARIADB = "mariadb"; +export const DBSYSTEMVALUES_MSSQL = "mssql"; +export const DBSYSTEMVALUES_ORACLE = "oracle"; +export const DBSYSTEMVALUES_SQLITE = "sqlite"; +export const DBSYSTEMVALUES_OTHER_SQL = "other_sql"; +export const DBSYSTEMVALUES_HSQLDB = "hsqldb"; +export const DBSYSTEMVALUES_H2 = "h2"; diff --git a/shared/AppInsightsCore/src/otel/attribute/attributeContainer.ts b/shared/AppInsightsCore/src/otel/attribute/attributeContainer.ts new file mode 100644 index 000000000..c77d303b2 --- /dev/null +++ b/shared/AppInsightsCore/src/otel/attribute/attributeContainer.ts @@ -0,0 +1,1136 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { + CreateIteratorContext, ICachedValue, arrForEach, arrIndexOf, createCachedValue, createIterator, getLength, isFunction, isObject, + isUndefined, iterForOf, objCreate, objDefine, objDefineProps, objForEachKey, objIs, objKeys, safe, strSplit +} from "@nevware21/ts-utils"; +import { STR_EMPTY, UNDEFINED_VALUE } from "../../constants/InternalConstants"; +import { eAttributeChangeOp } from "../../enums/otel/eAttributeChangeOp"; +import { IUnloadHook } from "../../interfaces/ai/IUnloadHook"; +import { IOTelAttributes, OTelAttributeValue } from "../../interfaces/otel/IOTelAttributes"; +import { IAttributeChangeInfo, IAttributeContainer, eAttributeFilter } from "../../interfaces/otel/attribute/IAttributeContainer"; +import { IOTelAttributeLimits } from "../../interfaces/otel/config/IOTelAttributeLimits"; +import { IOTelConfig } from "../../interfaces/otel/config/IOTelConfig"; +import { ITraceCfg } from "../../interfaces/otel/config/IOTelTraceCfg"; +import { handleAttribError } from "../../internal/handleErrors"; + +const _inheritedKey = "~[[inherited]]"; +const _deletedKey = "~[[deleted]]"; +let _containerId = 0; + +type IAttributeBranch = { [key: string]: IAttributeNode }; + +const enum eAttributeSource { + Local = 0, + + Inherited = 1 +} + +interface _AttributeFindDetail { + /** + * The hosting container + */ + c: IAttributeContainer; + + /** + * The found key + */ + k: string; + + /** + * The local attribute node + */ + n: IAttributeNode; + + /** + * The found value + */ + v: V; + + /** + * The attribute source (local or inherited) + */ + s: eAttributeSource; + + /** + * Identifies if this key should be considered to "exist" (is if present) + */ + e: boolean; +} + +interface IAttributeNode { + /** + * The value of the attribute tree + */ + v?: V; + + /** + * Identifies that this is a leaf node in the attribute tree + */ + n?: IAttributeBranch; + + /** + * Identifies that this node has been locally deleted + */ + d?: boolean; +} + +interface IAttributeIteratorState { + p: string; + n: IAttributeBranch; + k: string[]; + i: number; +} + +const enum AddAttributeResult { + Success = 0, + MaxAttribsExceeded = 1, + BranchNodeExists = 2, + LeafNodeExists = 3, + EmptyKey = 4 +} + +interface IAddAttributeDetails { + r: AddAttributeResult; // result + a?: boolean; // added (true if new attribute was added, false if existing was replaced) + p?: V; // prev (the previous value if one existed) +} + +function _noOpFunc() { + // No-op function +} + +function _addValue(container: IAttributeContainer, target: IAttributeBranch, theKey: string, value: V, maxAttribs: number, cfg: IOTelConfig): IAddAttributeDetails { + let errorHandlers = cfg.errorHandlers || {}; + if (theKey && getLength(theKey) > 0) { + let key = theKey.split("."); + let parts: string[] = []; + let keyIndex = 0; + + let keyLen = getLength(key); + while (keyIndex < keyLen) { + let part = key[keyIndex]; + parts.push(part); + + if (keyIndex === keyLen - 1) { + // last part + if (target[part] && target[part].n) { + // This node already exists as a branch node + handleAttribError(errorHandlers, "Attribute key [" + parts.join(".") + "] already exists as a branch node", theKey, value); + return { r: AddAttributeResult.BranchNodeExists }; + } + + if (!target[part] || target[part].d) { + // Node doesn't exist or was deleted + if (container.size >= maxAttribs) { + // If the key is not already present, we have exceeded the limit + // But if it does exist in the hierarchy, we can replace it locally + if (!container.has(theKey)) { + handleAttribError(errorHandlers, "Maximum allowed attributes exceeded [" + maxAttribs + "]", theKey, value); + return { r: AddAttributeResult.MaxAttribsExceeded }; + } + } + + // Add new leaf node or restore deleted node + target[part] = { + v: value + }; + return { r: AddAttributeResult.Success, a: true }; + } else { + // replace the value - capture the previous value + let previousValue = target[part].v; + target[part].v = value; + target[part].d = false; // Clear any deleted flag when setting value + return { r: AddAttributeResult.Success, a: false, p: previousValue }; + } + } + + if (!target[part]) { + target[part] = { + n: {} + }; + } + + if (!target[part].n) { + handleAttribError(errorHandlers, "Attribute key [" + parts.join(".") + "] already exists as a leaf node", theKey, value); + return { r: AddAttributeResult.LeafNodeExists }; + } + + target = target[part].n; + keyIndex++; + } + } + + handleAttribError(errorHandlers, "Attribute key is empty", theKey, value); + return { r: AddAttributeResult.EmptyKey }; +} + +/** + * Delete a specific attribute from the tree + * @param target - The target branch to delete from + * @param key - The key parts to delete + * @returns true if the attribute was deleted, false if it didn't exist + */ +function _deleteValue(target: IAttributeBranch, key: string[]): { d: boolean; p?: V } { + if (key && getLength(key) > 0) { + let keyIndex = 0; + let keyLen = getLength(key); + + // Navigate to the parent of the target node, creating path if needed + while (keyIndex < keyLen - 1) { + let part = key[keyIndex++]; + + if (!target[part]) { + target[part] = { + n: {} + }; + } + + if (!target[part].n) { + // Path is blocked by a leaf node, can't delete + return { d: false }; + } + + target = target[part].n; + } + + // Now we're at the parent of the target node + let lastPart = key[keyLen - 1]; + + if (target[lastPart] && !target[lastPart].d && !target[lastPart].n) { + // Node exists, is not already deleted, and is a leaf - delete it + let prev = target[lastPart].v; + target[lastPart].d = true; // Mark as deleted + target[lastPart].v = UNDEFINED_VALUE; // Clear the value + return { d: true, p: prev }; + } else if (!target[lastPart] || !target[lastPart].d) { + // Node doesn't exist or is not already deleted - create a deleted marker + // This is important for tracking inherited keys that are locally deleted + // and for the edge case where inherited keys might be added later + target[lastPart] = { + d: true // Mark as deleted without a value + }; + return { d: true }; + } + } + + return { d: false }; +} + +/** + * Find a local specific node (including if it's been deleted) in the attribute tree, or if it exists in an inherited source + * @param container - The target attribute container to search in + * @param key - The key to search for + * @param filter - The filter to apply + * @param cb - The callback function to execute when the node is found + * @param nodes - The attribute branches to search in + * @param inheritContainer - The inherited attribute container + * @param inheritAttribObj - The inherited attribute object + * @param otelCfg - Optional OpenTelemetry configuration + * @returns The result of the callback function + */ +function _findDetail(container: IAttributeContainer, key: string, filter: eAttributeFilter | undefined, cb: (detail?: _AttributeFindDetail) => T, nodes: IAttributeBranch, inheritContainer?: IAttributeContainer | null, inheritAttribObj?: IOTelAttributes | null, otelCfg?: IOTelConfig) { + let theDetail: _AttributeFindDetail; + + // Find the local node + if (key && getLength(key) > 0) { + let theNode: IAttributeNode | undefined; + let target = nodes; + let keys = strSplit(key, "."); + let keysLen = getLength(keys); + let idx = 0; + + if (filter == eAttributeFilter.Local || filter === eAttributeFilter.LocalOrDeleted || isUndefined(filter)) { + // Find any local node (if it exists) + while (target && idx < keysLen) { + let part = keys[idx++]; + let node = target[part]; + if (node && idx >= keysLen) { + // last part + theNode = node; + break; + } + + if (!node || !node.n) { + // TODO - support wildcard(s) (when no node is found), likely need to refactor this to support multiple keys + break; + } + + target = node.n; + } + + // So if the caller didn't specify a filter then always return the found "value" (which will be undefined for a deleted key) + // If they have specified a filter, then we need to respect that filter where Local and LocalOrDeleted are treated the same + if (theNode) { + theDetail = { + c: container, + k: key, + n: theNode, + v: theNode.d ? UNDEFINED_VALUE : theNode.v, + s: eAttributeSource.Local, + e: !theNode.d + }; + } + } + + // So we get here if + // - We didn't find a node (no local overrides) + // - The filter doesn't match (undefined, Local, LocalOrDeleted), so we didn't look for a node + // So we just need to check if the filter is undefined or inherited and if so we are free + // to look in the inherited sources + if (!theDetail && (filter === eAttributeFilter.Inherited || isUndefined(filter))) { + const inheritedValue = _getInheritedValue(key, inheritContainer, inheritAttribObj); + if (inheritedValue) { + theDetail = { + c: container, + k: key, + n: null, + v: inheritedValue.v as V, + s: eAttributeSource.Inherited, + e: true + }; + } + } + } + + return cb(theDetail); +} + +function _size(target: IAttributeBranch, inheritAttrib?: IOTelAttributes, inheritContainer?: IAttributeContainer): number { + let count = 0; + let seenKeys: string[] = []; + + // Use the iterator to count all unique attributes including inheritance + let iter = _iterator(target, (prefix, key, _node) => { + return prefix + key; + }, undefined, inheritContainer || inheritAttrib); + + iterForOf(iter, (key) => { + if (arrIndexOf(seenKeys, key) === -1) { + seenKeys.push(key); + count++; + } + }); + + return count; +} + +function _getStackEntry(theNode: IAttributeBranch, prefix: string): IAttributeIteratorState | undefined { + let entry: IAttributeIteratorState | undefined; + + if (theNode) { + let allKeys: string[] = objKeys(theNode); + if (getLength(allKeys) > 0) { + entry = { + p: prefix, + n: theNode, + k: allKeys, + i: -1 + }; + } + } + + return entry; +} + +function _iterator(target: IAttributeBranch, cb: (prefix: string, key: string, node: IAttributeNode, source: eAttributeFilter) => T, otelCfg?: IOTelConfig, parentAttribs?: IOTelAttributes | IAttributeContainer, inclDeleted?: boolean): Iterator { + let stack: IAttributeIteratorState[] = []; + let visitedKeys: string[] | undefined = parentAttribs ? [] : UNDEFINED_VALUE; + let inheritState: IAttributeIteratorState; + let inheritContainerIter: Iterator<[string, any, eAttributeFilter]>; + let current = _getStackEntry(target, STR_EMPTY); // Start with the root branch + let inheritContainer: IAttributeContainer | undefined; + let inheritAttribs: IOTelAttributes | undefined; + + // Used as the initializer for the iterator and the flag to indicate that the iterator is done + let ctx: CreateIteratorContext | null = { + v: undefined, + n: _moveNext + }; + + if (parentAttribs) { + if (isAttributeContainer(parentAttribs)) { + inheritContainer = parentAttribs; + } else if (isObject(parentAttribs)) { + inheritAttribs = parentAttribs; + } + } + + function _moveNext(): boolean { + let thePrefix = STR_EMPTY; + let theKey: string | undefined; + let theNode: IAttributeNode | undefined; + let theSource = eAttributeFilter.Local; // Default to local + + // Process entries from the current node + while (current) { + current.i++; + if (current.i < getLength(current.k)) { + // We have at least 1 key + let key = current.k[current.i]; + // The key is "real" (we don't support null or undefined keys) + let node = current.n[key]; + if (node) { + if (node.n) { + // We are at a branch node, so add the current node to the stack so we can continue traversing it later + stack.push(current); + current = _getStackEntry(node.n, current.p + key + "."); + + // If the branch is empty, continue with the next item from stack + if (!current) { + current = stack.pop(); + } + } else { + // Leaf node + let fullKey = current.p + key; + if (visitedKeys && arrIndexOf(visitedKeys, fullKey) === -1) { + visitedKeys.push(fullKey); + } + + // If we are not deleted, then return this key + if (!node.d || inclDeleted) { + // Leaf node with value - not deleted + thePrefix = current.p; + theKey = key; + theNode = node; + theSource = node.d ? eAttributeFilter.LocalOrDeleted : eAttributeFilter.Local; + break; + } + } + } + } else { + // No more keys to process at this level + current = stack.pop(); + } + } + + // We get here only after we have processed all keys of the current instance + // Switch to processing inherited attributes if we have them + if (!theNode && inheritAttribs) { + if (!inheritState) { + // Lazy initialize the inherited state + inheritState = { + p: STR_EMPTY, + n: UNDEFINED_VALUE, + k: objKeys(inheritAttribs), + i: -1 + }; + } + + // Process inherited attributes that haven't been overridden or deleted + while (inheritState.i < getLength(inheritState.k) - 1) { + inheritState.i++; + let key = inheritState.k[inheritState.i]; + if (arrIndexOf(visitedKeys, key) === -1) { + visitedKeys.push(key); + theKey = key; + theNode = { v: inheritAttribs[key] as V }; + theSource = eAttributeFilter.Inherited; + break; + } + } + } + + // Process inherited container if we have one and no other node was found + if (!theNode && inheritContainer) { + if (!inheritContainerIter) { + // Initialize the container's iterator only once - this will include the full inheritance chain + inheritContainerIter = inheritContainer.entries(); + } + + // Use the container's iterator to get next entry + iterForOf(inheritContainerIter, (entry) => { + let key = entry[0]; + if (arrIndexOf(visitedKeys, key) === -1) { + visitedKeys.push(key); + theKey = key; + theNode = { v: entry[1] as V }; + theSource = eAttributeFilter.Inherited; // Always return Inherited for inherited container entries + return -1; + } + }); + } + + if (theNode) { + ctx.v = cb(thePrefix, theKey, theNode, theSource); + } + + return !theNode; + } + + return createIterator(ctx); +} + +function _generateAttributes(container: IAttributeContainer, target: IAttributeBranch, otelCfg?: IOTelConfig, inheritAttrib?: IOTelAttributes | IAttributeContainer, showTree?: boolean): IOTelAttributes { + // TODO: Look at making this return a proxy instead of a new object + // This should be more effecient as it would make the property lookups lazy + let attribs = objCreate(null) as { [key: string]: V | undefined }; + let deletedKeys: { [key: string]: V | undefined }; + + if (showTree) { + // Add Node Id + objDefine(attribs, "#id", { + v: container.id, + w: false + }); + } + + // Use the iterator properly - collect the key-value pairs + let iter = _iterator(target, (prefix, key, node, source) => { + let name = prefix + key; + return { n: name, v: node.v as V, d: node.d, s: source }; + }, otelCfg, showTree ? null : inheritAttrib, showTree ? eAttributeFilter.LocalOrDeleted : UNDEFINED_VALUE); + + iterForOf(iter, (entry) => { + let theNode = attribs; + let theName = entry.n; + + if (entry.d) { + if (!deletedKeys) { + // Create separate objects for inherited and deleted attributes + deletedKeys = objCreate(null); + objDefine(attribs, _deletedKey, { + v: deletedKeys, + w: false + }); + } + + theNode = deletedKeys; + } + + if (theNode && !(theName in theNode)) { + theNode[theName] = entry.v; + } + }); + + if (inheritAttrib && showTree) { + // Create separate objects for inherited and deleted attributes + let inheritedKeys: any; + + // Add a tree showing the inherited attributes + if (isAttributeContainer(inheritAttrib)) { + inheritedKeys = (inheritAttrib as any)._attributes; + } else { + // Create separate objects for inherited and deleted attributes + inheritedKeys = objCreate(null) as any; + objForEachKey(inheritAttrib, (key, value) => { + inheritedKeys[key] = value as V; + }); + } + + objDefine(attribs, _inheritedKey, { + v: inheritedKeys, + w: false + }); + } + + return attribs; +} + +function _notifyListeners(listeners: ({ cb: (changeInfo: IAttributeChangeInfo) => void })[], changeInfo: IAttributeChangeInfo) { + // Notify all registered listeners of changes + if (listeners) { + arrForEach(listeners, (listener) => { + safe(listener.cb, [changeInfo]); + }); + } +} + +function _getInheritedValue(key: string, inheritContainer?: IAttributeContainer | null, inheritAttribObj?: IOTelAttributes | null ): { v: V } | undefined { + // Check inherited container first + if (inheritContainer && inheritContainer.has(key)) { + return { v: inheritContainer.get(key) as V }; + } + + // Then check inherited attributes object + if (inheritAttribObj && key in inheritAttribObj) { + return { v: inheritAttribObj[key] as V }; + } +} + +function _createUnloadHook(listeners: { cb: (changeInfo: IAttributeChangeInfo) => void }[], callback: (changeInfo: IAttributeChangeInfo) => void): IUnloadHook { + let cbInst = { + cb: callback + } + + listeners.push(cbInst); + + let unloadHook = { + rm: () => { + if (listeners && cbInst) { + let index = arrIndexOf(listeners, cbInst); + if (index >= 0) { + // Remove the current listener + listeners.splice(index, 1); + } + + // Clear the cached values + cbInst.cb = null; + cbInst = null; + + // Optimization to drop all references and shortcut any future lookups + unloadHook.rm = _noOpFunc; + } + } + }; + + return unloadHook; +} + +/** + * Creates a new attribute container with only configuration. + * + * @param otelCfg - The OpenTelemetry configuration containing trace configuration and limits + * @returns A new IAttributeContainer instance with auto-generated container ID + * @since 3.4.0 + * @example + * ```typescript + * const config = { traceCfg: { generalLimits: { attributeCountLimit: 64 } } }; + * const container = createAttributeContainer(config); + * console.log(container.id); // ".0" (auto-generated) + * ``` + */ +export function createAttributeContainer(otelCfg: IOTelConfig): IAttributeContainer; + +/** + * Creates a new attribute container with configuration and a name. + * + * @param otelCfg - The OpenTelemetry configuration containing trace configuration and limits + * @param name - The name for the container (used in the container ID) + * @returns A new IAttributeContainer instance with the specified name + * @since 3.4.0 + * @example + * ```typescript + * const config = { traceCfg: { generalLimits: { attributeCountLimit: 64 } } }; + * const container = createAttributeContainer(config, "my-container"); + * console.log(container.id); // "my-container.0" + * container.set("service.name", "my-service"); + * ``` + */ +export function createAttributeContainer(otelCfg: IOTelConfig, name: string): IAttributeContainer; + +/** + * Creates a new attribute container with configuration, name, and inheritance. + * + * @param otelCfg - The OpenTelemetry configuration containing trace configuration and limits + * @param name - The name for the container (used in the container ID) + * @param inheritAttrib - Parent attributes or container to inherit from + * @returns A new IAttributeContainer instance that inherits from the specified parent + * @since 3.4.0 + * @example + * ```typescript + * const config = { traceCfg: { generalLimits: { attributeCountLimit: 64 } } }; + * const parent = { "environment": "production", "region": "us-east-1" }; + * const child = createAttributeContainer(config, "child-container", parent); + * console.log(child.get("environment")); // "production" (inherited) + * child.set("service.name", "my-service"); // local attribute + * ``` + */ +export function createAttributeContainer(otelCfg: IOTelConfig, name: string, inheritAttrib: IOTelAttributes | IAttributeContainer): IAttributeContainer; + +/** + * Creates a new attribute container with full configuration options. + * + * @param otelCfg - The OpenTelemetry configuration containing trace configuration and limits + * @param name - The name for the container (used in the container ID) + * @param inheritAttrib - Parent attributes or container to inherit from + * @param attribLimits - Specific attribute limits to override configuration defaults + * @returns A new IAttributeContainer instance with custom limits and inheritance + * @since 3.4.0 + * @example + * ```typescript + * const config = { traceCfg: { generalLimits: { attributeCountLimit: 64 } } }; + * const parent = { "environment": "production" }; + * const customLimits = { attributeCountLimit: 32, attributeValueLengthLimit: 256 }; + * const container = createAttributeContainer(config, "limited-container", parent, customLimits); + * // This container has stricter limits than the default configuration + * ``` + */ +export function createAttributeContainer(otelCfg: IOTelConfig, name: string, inheritAttrib: IOTelAttributes | IAttributeContainer, attribLimits: IOTelAttributeLimits): IAttributeContainer; + +/** + * Creates a new attribute container that provides an efficient, observable key-value storage + * for OpenTelemetry attributes with support for inheritance, limits, and change notifications. + * + * The container supports inherited attributes from parent containers or plain objects, + * enforces attribute count and value size limits, and provides efficient iteration and access patterns. + * + * @param otelCfg - The OpenTelemetry configuration containing trace configuration and limits + * @param name - Optional name for the container (used in the container ID) + * @param inheritAttrib - Optional parent attributes or container to inherit from + * @param attribLimits - Optional specific attribute limits to override configuration defaults + * @returns A new IAttributeContainer instance with the specified configuration + * @since 3.4.0 + * @example + * ```typescript + * const config = { traceCfg: { generalLimits: { attributeCountLimit: 64 } } }; + * const container = createAttributeContainer(config, "my-container"); + * container.set("service.name", "my-service"); + * + * // With inheritance + * const parent = { "environment": "production" }; + * const child = createAttributeContainer(config, "child-container", parent); + * console.log(child.get("environment")); // "production" + * ``` + */ +export function createAttributeContainer(otelCfg: IOTelConfig, name?: string | null | undefined, inheritAttrib?: IOTelAttributes | IAttributeContainer, attribLimits?: IOTelAttributeLimits): IAttributeContainer { + let traceCfg: ITraceCfg = otelCfg.traceCfg || {}; + let nodes: { [key: string]: IAttributeNode } | null = null; + let theSize: ICachedValue | null = null; + let theDropped: ICachedValue | null = null; + let limits: IOTelAttributeLimits = traceCfg.generalLimits || {}; + let maxAttribs: number = limits.attributeCountLimit || 128; + // let maxValueLen: number = limits.attributeValueLengthLimit; + let theAttributes: ICachedValue; + let localAttributes: ICachedValue; + let droppedAttribs = 0; + let inheritContainer: IAttributeContainer | null = null; + let inheritAttribObj: IOTelAttributes | null = null; + let listeners: ({ cb: (changeInfo: IAttributeChangeInfo) => void })[] | null = null; + let parentListenerHook: IUnloadHook | null = null; + let containerName: string = name || STR_EMPTY; + + if (attribLimits) { + maxAttribs = attribLimits.attributeCountLimit || maxAttribs; + // maxValueLen = attribLimits.attributeValueLengthLimit || maxValueLen; + } + + // Determine if inheritAttrib is a container or plain attributes object + if (isAttributeContainer(inheritAttrib)) { + inheritContainer = inheritAttrib; + } else if (isObject(inheritAttrib)) { + inheritAttribObj = inheritAttrib; + } + + inheritAttrib = null; + + let inheritSrc = inheritAttribObj || inheritContainer; + let container: IAttributeContainer = { + id: (containerName || STR_EMPTY) + "." + (_containerId++), + size: 0, + droppedAttributes: 0, + attributes: UNDEFINED_VALUE, + clear: () => { + // Remove parent listener if exists + if (parentListenerHook) { + parentListenerHook.rm(); + parentListenerHook = null; + } + + // Only inform children if we appear to have any keys or could possible inherit keys + if (nodes || inheritContainer || inheritAttribObj) { + // Inform any children (Synchronously) that we are about to clear all attributes + // Called prior to clearing the nodes, so that children still have full access + // to the current and inherited attributes + _notifyListeners(listeners, { frm: container.id, op: eAttributeChangeOp.Clear }); + } + + nodes = null; + theSize = null; + theDropped = null; + theAttributes = null; + localAttributes = null; + droppedAttribs = 0; + inheritSrc = null; // Clear the inherited attributes + inheritContainer = null; // Clear inherited container as well + inheritAttribObj = null; // Clear inherited attributes as well + }, + get: (key: string, source?: eAttributeFilter) => { + return _findDetail(container, key, source, (detail) => { + // Just return the value (which will include deleted ones (as undefined)if the filter is set) + return detail ? detail.v : UNDEFINED_VALUE; + }, nodes, inheritContainer, inheritAttribObj, otelCfg); + }, + has: (key: string, source?: eAttributeFilter) => { + return _findDetail(container, key, source, (detail) => { + // Note: We may still have a detail object if the key was deleted, so we need to + // - Check if the detail is considered to "exist" + // - If the source filter is LocalOrDeleted, then return that is "exists" + return detail ? (detail.e || source === eAttributeFilter.LocalOrDeleted) : false; + }, nodes, inheritContainer, inheritAttribObj, otelCfg); + }, + set: (key: string, value: any) => { + if (!nodes) { + // Lazily create a container object + nodes = objCreate(null); + } + + let addResult = _addValue(container, nodes, key, value, maxAttribs, otelCfg); + if (addResult.r === AddAttributeResult.Success) { + theSize = null; // invalidate any previously cached size + theAttributes = null; // invalidate any previously cached attributes + localAttributes = null; // invalidate any previously cached local attributes + + // Determine operation type based on whether this was a new attribute or set + let op = addResult.a ? eAttributeChangeOp.Add : eAttributeChangeOp.Set; + let prevValue = addResult.p; + + if (op === eAttributeChangeOp.Add) { + // Special case, if we just added/changed it locally we need to lookup our parents to see if we are replacing (hiding) + // a previously inherited value and if so we need to notify our children that this was a set not an add (if the value changed) + const inheritedValue = _getInheritedValue(key, inheritContainer, inheritAttribObj); + if (inheritedValue) { + op = eAttributeChangeOp.Set; + prevValue = inheritedValue.v; + } + } + + // Only notify children if the value has changed + if (!objIs(value, prevValue)) { + + // Inform any children (Synchronously) that we have "changed" a value + // Unlike clear, this is called after the change is made, so children + // will need to "handle" the change accordingly + _notifyListeners(listeners, { + frm: container.id, + op: op, + k: key, + prev: prevValue, + val: value + }); + } + } else if (addResult.r === AddAttributeResult.MaxAttribsExceeded) { + theDropped = null; + droppedAttribs++; + _notifyListeners(listeners, { + frm: container.id, + op: eAttributeChangeOp.DroppedAttributes, + k: key, + val: value + }); + } + + return addResult.r === AddAttributeResult.Success; + }, + del: (key: string) => { + + // Find the node / value for this container + return _findDetail(container, key, UNDEFINED_VALUE, (detail) => { + // Check if the key exists locally (incl marked as deleted) or is inherited + if (detail) { + // So the key exists either locally or in an inherited source + // - It may also exists but marked as deleted + if (!nodes) { + // Always create nodes structure if it doesn't exist since we need to track deleted keys + nodes = objCreate(null); + } + + // Now lets mark this key as deleted locally + let deleteResult = _deleteValue(nodes, key.split(".")); + if (deleteResult.d) { + theSize = null; // invalidate any previously cached size + theAttributes = null; // invalidate any previously cached attributes + localAttributes = null; // invalidate any previously cached local attributes + } + + // Inform any children (Synchronously) that we have deleted a value + // This is important even if the key didn't exist locally, as it existed in inheritance + // chain and we need children (like snapshot containers) to know about the deletion to + // ignore any future inherited values. + _notifyListeners(listeners, { + frm: container.id, + op: eAttributeChangeOp.Delete, + k: key, + prev: detail.v + }); + + // Return true if we successfully marked it as deleted + return deleteResult.d; + } + + return false; + }, nodes, inheritContainer, inheritAttribObj, otelCfg); + }, + keys: () => { + return _iterator(nodes, (prefix, key, node, source) => { + return prefix + key; + }, otelCfg, inheritSrc); + }, + entries: () => { + return _iterator(nodes, (prefix, key, node, source) => { + return [prefix + key, node.v, source] as [string, V, eAttributeFilter]; + }, otelCfg, inheritSrc); + }, + values: () => { + return _iterator(nodes, (_prefix, _key, node, _source) => { + return node.v; + }, otelCfg, inheritSrc); + }, + forEach: (cb: (key: string, value: V, source?: eAttributeFilter) => void) => { + let iter = _iterator(nodes, (prefix, key, node, source) => { + cb(prefix + key, node.v, source); + return true; // Return true to indicate the callback was executed + }, otelCfg, inheritSrc); + + // Iterate over the entire container + iterForOf(iter, _noOpFunc); + }, + child: (name?: string, snapshot?: boolean) => { + const childName = (name ? name : "child") + (snapshot ? "<-@[" : "<=[") + container.id + "]"; + return snapshot ? _createSnapshotContainer(otelCfg, childName, container, attribLimits) : createAttributeContainer(otelCfg, childName, container, attribLimits); + }, + listen: (callback: (changeInfo: IAttributeChangeInfo) => void): IUnloadHook => { + if (!listeners) { + listeners = []; + } + + return _createUnloadHook(listeners, callback); + } + }; + + function _listener(changeInfo: IAttributeChangeInfo) { + // Invalidate caches when parent changes + let shouldNotify = true; + + // If a parent adds a new key, then if this instance has not replaced it + // then we need to also inform our children + if (changeInfo.op === eAttributeChangeOp.Add || changeInfo.op === eAttributeChangeOp.Set || changeInfo.op === eAttributeChangeOp.Delete) { + theSize = null; + theAttributes = null; + localAttributes = null; + + // If we already have (or deleted) this key locally, then we don't need to notify children + _findDetail(container, changeInfo.k, eAttributeFilter.LocalOrDeleted, (detail) => { + // If we have a node (which also might be that it was deleted) then don't notify + // If we didn't find a node then we still need to notify our children + shouldNotify = !detail || !detail.n; + }, nodes); + } else if (changeInfo.op === eAttributeChangeOp.Clear) { + theSize = null; + theAttributes = null; + localAttributes = null; + + // If the parent clears all of it's keys, we need to inform our children + shouldNotify = true; + } else if (changeInfo.op === eAttributeChangeOp.DroppedAttributes) { + shouldNotify = true; + theDropped = null; + } + + if (shouldNotify) { + _notifyListeners(listeners, changeInfo); + } + } + + // If we have a parent container, register a listener to stay connected to parent changes + if (inheritContainer && isFunction(inheritContainer.listen)) { + parentListenerHook = inheritContainer.listen(_listener); + } + + return objDefineProps(container, { + size: { + g: () => { + if (!theSize) { + theSize = createCachedValue(_size(nodes, inheritAttribObj, inheritContainer)); + } + + return theSize.v; + } + }, + droppedAttributes: { + g: () => { + if (!theDropped) { + theDropped = createCachedValue((inheritContainer ? inheritContainer.droppedAttributes : 0) + droppedAttribs); + } + + return theDropped.v; + } + }, + attributes: { + g: () => { + if (!theAttributes) { + theAttributes = createCachedValue(_generateAttributes(container, nodes, otelCfg, inheritSrc)); + } + + return theAttributes.v; + } + }, + _attributes: { + g: () => { + if (!localAttributes) { + localAttributes = createCachedValue(_generateAttributes(container, nodes, otelCfg, inheritSrc, true)); + } + + return localAttributes.v; + } + } + }); +} + +/** + * Add all attributes from the source attributes or container to the target container. + * This function has performance and memory implications as it immediately copies all key-value pairs + * from the source to the target container, handling both plain attribute objects and other attribute + * containers. + * + * @param container - The target container to add attributes to + * @param attributes - The source attributes or container to copy from + * @since 3.4.0 + * @example + * ```typescript + * const target = createAttributeContainer(config, "target"); + * const source = { key1: "value1", key2: "value2" }; + * addAttributes(target, source); + * + * // Or from another container + * const sourceContainer = createAttributeContainer(config, "source"); + * sourceContainer.set("key3", "value3"); + * addAttributes(target, sourceContainer); + * ``` + */ +export function addAttributes(container: IAttributeContainer, attributes: IOTelAttributes | IAttributeContainer): void { + if (isAttributeContainer(attributes)) { + // Use the container's forEach for direct processing - more efficient than entries() iterator + attributes.forEach((key, value, source) => { + container.set(key, value); + }); + } else if (attributes) { + if (isFunction(attributes.entries)) { + // Handle any type that has an entries function + iterForOf((attributes as any).entries(), function(entry: [string, any]) { + container.set(entry[0], entry[1]); + }); + } else { + // Handle as plain attributes object + objForEachKey(attributes, function(key, value) { + container.set(key, value); + }); + } + } +} + +function _createSnapshotContainer(otelCfg: IOTelConfig, name: string | undefined, sourceContainer: IAttributeContainer, attribLimits?: IOTelAttributeLimits): IAttributeContainer { + let newContainer = createAttributeContainer(otelCfg, name, sourceContainer, attribLimits); + + sourceContainer.listen((changeInfo) => { + if (changeInfo.op === eAttributeChangeOp.Clear) { + // Copy all parent keys to this container if not already present + iterForOf((sourceContainer as any).entries(), function(entry: [string, any]) { + let key = entry[0]; + // If the Key has not been set or explicitly deleted from the new Container, and the source Container + // has the key (directly or inherited), then copy the value locally so we have a copy because our parent + // some grand parent is clearing their collection of values. Note ideal as if it's a grand parent or higher + // we end up duplicating the keys multiple times, but we only know about our direct ancestor + // Note: Deleting locally hides any inherited value + if (!newContainer.has(key, eAttributeFilter.LocalOrDeleted) && sourceContainer.has(key)) { + newContainer.set(key, entry[1]); + } + }); + } else if (changeInfo.op === eAttributeChangeOp.Add) { + // For add operations add), we need to ensure that when the key is not already present on the current instance + // that we hide the key so it's "not" inherited, by marking it as deleted. + let key = changeInfo.k; + if (!newContainer.has(key, eAttributeFilter.Local)) { + newContainer.del(key); + } + } else if (changeInfo.op === eAttributeChangeOp.Set || changeInfo.op === eAttributeChangeOp.Delete) { + // For set operations (change), preserve the previous value if we don't have it locally + // For delete operations, if we don't have the key locally, preserve it with the deleted value + let key = changeInfo.k; + if (!newContainer.has(key, eAttributeFilter.LocalOrDeleted)) { + newContainer.set(key, changeInfo.prev); + } + } + }); + + return newContainer; +} + +/** + * Creates a snapshot container based on the passed IOTelAttributes or IAttributeContainer. + * The returned container effectively treats the source attributes as immutable as at the time of creation, you + * may still add / update existing attributes without affecting the original source. And changes to the source + * attributes / container will not be reflected from the new snapshot container, only changes made to the returned + * container itself. + * + * Note: It implements this in a lazy manner, so changes made to the original source after the snapshot is taken will + * cause the changed attributes to be copied into this snapshot container with it's original value and cause a local + * version of the changed key (if not already been present) to be added, so when using the {@link IAttributeContainer#has} or + * {@link IAttributeContainer#get} methods, with the optional source filter as {@link eAttributeFilter.Inherited} it will + * only return attributes that were present and unchanged at the time of the snapshot. This means for those attributes you must + * use the {@link eAttributeFilter.Local} as source filter (or leave it undefined) to access the local version. + * + * It is recommended that you always use {@link IAttributeContainer} instances for better memory and performance overheads, + * for this specific function when you pass a {@link IOTelAttributes} instance it will create a copy of all present attributes + * at the point of creation (not lazily). + * + * @param otelCfg - The OpenTelemetry configuration to use for the snapshot container + * @param source - The source attributes or container to create a snapshot view from + * @param attribLimits - Optional attribute limits to apply to the snapshot container + * @returns An IAttributeContainer instance that will preserves the attributes and values from the source attributes / container + * at creation time. The snapshot container will be named "snapshot(xxxx)" where xxxx is the source container ID, or "snapshot(...)" for non-container sources. + * @since 3.4.0 + * @example + * ```typescript + * // Immediate copy for plain attributes + * const attrs = { key1: "value1", key2: "value2" }; + * const snapshot = createAttributeSnapshot(attrs); + * // snapshot.id will be "snapshot(...).N" + * attrs.key1 = "changed"; // snapshot.get("key1") is still "value1" + * + * // Lazy copy-on-change for containers + * const container = createAttributeContainer(config, "my-container"); + * container.set("key1", "value1"); + * const snapshot2 = createAttributeSnapshot(container); + * // snapshot2.id will be "snapshot(my-container.N).M" + * container.set("key1", "changed"); // snapshot2.get("key1") remains "value1" (previous value copied) + * ``` + */ +/*#__NO_SIDE_EFFECTS__*/ +export function createAttributeSnapshot(otelCfg: IOTelConfig, name: string, source: IOTelAttributes | IAttributeContainer, attribLimits?: IOTelAttributeLimits): IAttributeContainer { + let newContainer: IAttributeContainer; + + if (isAttributeContainer(source)) { + newContainer = _createSnapshotContainer(otelCfg, (name || "child") + "<-@[" + source.id + "]", source, attribLimits); + } else { + newContainer = createAttributeContainer(otelCfg, (name || "child") + "<-@[(...)]", null, attribLimits); + addAttributes(newContainer, source); + } + + return newContainer; +} + +/** + * Helper function to identify if a passed argument is or implements the IAttributeContainer interface + * @param value - The value to check + * @returns true if the value implements IAttributeContainer interface, false otherwise + * @since 3.4.0 + * @example + * ```typescript + * const container = createAttributeContainer(config, "test-container"); + * if (isAttributeContainer(container)) { + * // TypeScript now knows container is IAttributeContainer + * console.log(container.size); + * container.set("key", "value"); + * } + * + * // Check unknown object + * function processContainer(obj: unknown) { + * if (isAttributeContainer(obj)) { + * obj.forEach((key, value) => console.log(key, value)); + * } + * } + * ``` + */ +/*#__NO_SIDE_EFFECTS__*/ +export function isAttributeContainer(value: any): value is IAttributeContainer { + return value && + isFunction(value.clear) && + isFunction(value.get) && + isFunction(value.has) && + isFunction(value.set) && + isFunction(value.del) && + isFunction(value.keys) && + isFunction(value.entries) && + isFunction(value.forEach) && + isFunction(value.values) && + isFunction(value.child) && + isFunction(value.listen) && + (isObject(value) || isFunction(value)) && + // Check for existence of the required properties, but don't cause them to be processed ("executed") + ("id" in value) && + ("size" in value) && + ("droppedAttributes" in value) && + ("attributes" in value); +} diff --git a/shared/AppInsightsCommon/src/ConnectionStringParser.ts b/shared/AppInsightsCore/src/telemetry/ConnectionStringParser.ts similarity index 87% rename from shared/AppInsightsCommon/src/ConnectionStringParser.ts rename to shared/AppInsightsCore/src/telemetry/ConnectionStringParser.ts index 4170f9d5c..3f277a841 100644 --- a/shared/AppInsightsCommon/src/ConnectionStringParser.ts +++ b/shared/AppInsightsCore/src/telemetry/ConnectionStringParser.ts @@ -1,9 +1,9 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { arrReduce, objKeys, strEndsWith } from "@microsoft/applicationinsights-core-js"; -import { DEFAULT_BREEZE_ENDPOINT } from "./Constants"; -import { ConnectionString, ConnectionStringKey } from "./Interfaces/ConnectionString"; +import { arrReduce, objKeys, strEndsWith } from "@nevware21/ts-utils"; +import { DEFAULT_BREEZE_ENDPOINT } from "../constants/Constants"; +import { ConnectionString, ConnectionStringKey } from "../interfaces/ai/ConnectionString"; const _FIELDS_SEPARATOR = ";"; const _FIELD_KEY_VALUE_SEPARATOR = "="; diff --git a/shared/AppInsightsCommon/src/RequestResponseHeaders.ts b/shared/AppInsightsCore/src/telemetry/RequestResponseHeaders.ts similarity index 91% rename from shared/AppInsightsCommon/src/RequestResponseHeaders.ts rename to shared/AppInsightsCore/src/telemetry/RequestResponseHeaders.ts index c8b659053..b2e40faec 100644 --- a/shared/AppInsightsCommon/src/RequestResponseHeaders.ts +++ b/shared/AppInsightsCore/src/telemetry/RequestResponseHeaders.ts @@ -1,7 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { createValueMap } from "@microsoft/applicationinsights-core-js"; +import { createValueMap } from "../enums/EnumHelperFuncs"; export interface IRequestHeaders { /** @@ -55,7 +55,7 @@ export const enum eRequestHeaders { requestContextAppIdFormat = 2, requestIdHeader = 3, traceParentHeader = 4, - traceStateHeader = 5, // currently not used + traceStateHeader = 5, sdkContextHeader = 6, sdkContextHeaderAppIdRequest = 7, requestContextHeaderLowerCase = 8 @@ -68,7 +68,7 @@ export const RequestHeaders = (/* @__PURE__ */ createValueMap(item: T, } let iKey = ""; - if (item[strIkey]) { - iKey = item[strIkey]; - delete item[strIkey]; + if ((item as any)[strIkey]) { + iKey = (item as any)[strIkey]; + delete (item as any)[strIkey]; } const telemetryItem: ITelemetryItem = { diff --git a/shared/AppInsightsCore/src/telemetry/W3cTraceState.ts b/shared/AppInsightsCore/src/telemetry/W3cTraceState.ts new file mode 100644 index 000000000..02546ac63 --- /dev/null +++ b/shared/AppInsightsCore/src/telemetry/W3cTraceState.ts @@ -0,0 +1,458 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { + ICachedValue, arrForEach, arrIndexOf, createCachedValue, isArray, isFunction, isNullOrUndefined, isString, objDefineProps, + safeGetDeferred, strSplit +} from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { IW3cTraceState } from "../interfaces/ai/IW3cTraceState"; +import { findMetaTags, findNamedServerTimings } from "../utils/EnvUtils"; +import { setObjStringTag } from "../utils/HelperFuncs"; + +const MAX_TRACE_STATE_MEMBERS = 32; +const MAX_TRACE_STATE_LEN = 512; + +// https://www.w3.org/TR/trace-context-1/#key +const LCALPHA = "[a-z]"; +const LCALPHA_DIGIT = "[a-z\\d]"; +const LCALPHA_DIGIT_UNDERSCORE_DASH_STAR_SLASH = "[a-z\\d_\\-*\\/]"; +const SIMPLE_KEY = "(" + LCALPHA + LCALPHA_DIGIT_UNDERSCORE_DASH_STAR_SLASH + "{0,255})"; +const TENANT_ID = "(" + LCALPHA_DIGIT + LCALPHA_DIGIT_UNDERSCORE_DASH_STAR_SLASH + "{0,240})"; +const SYSTEM_ID = "(" + LCALPHA + LCALPHA_DIGIT_UNDERSCORE_DASH_STAR_SLASH + "{0,13})"; +const MULTI_TENANT_KEY = "(" + TENANT_ID + "@" + SYSTEM_ID + ")"; + +// https://www.w3.org/TR/trace-context-1/#value +const NBLK_CHAR = "\x21-\x2B\\--\x3C\x3E-\x7E"; +const TRACESTATE_VALUE = "[\x20" + NBLK_CHAR + "]{0,255}[" + NBLK_CHAR + "]"; + +// https://www.w3.org/TR/trace-context-1/#tracestate-header +const TRACESTATE_KVP_REGEX = (/*#__PURE__*/ new RegExp("^\\s*((?:" + SIMPLE_KEY + "|" + MULTI_TENANT_KEY + ")=(" + TRACESTATE_VALUE + "))\\s*$")); + +/** + * @internal + * @ignore + * Identifies the components of a multi-tenant key + */ +interface ITraceStateMultiTenantKey { + tenantId: string; + systemId: string; +} + +/** + * @internal + * @ignore + * Identifies the member entry type + */ +const enum eTraceStateKeyType { + simple = 0, + multiTenant = 1, + + /** + * Internal flag the identifies that the associated key has been deleted + */ + deleted = 2 +} + +/** + * @internal + * @ignore + * Represents the parsed trace state member + */ +interface ITraceStateMember { + /** + * Identifies the type of identified key, simple or multi-tenant + */ + readonly type: eTraceStateKeyType; + + /** + * The full key of the trace state member + */ + readonly key: string; + + /** + * When the {@link #type} is {@link eTraceStateKeyType.multiTenant}, the tenantId and systemId will be populated + * with the values from the key, otherwise this entry will be undefined or null and should be ignored. + */ + readonly multiTenant?: ITraceStateMultiTenantKey; + + /** + * The value associated with the trace state member. + * If the type is {@link eTraceStateKeyType.deleted} then the value should be ignored + * and will likely be undefined. + */ + readonly value?: string; +} + +/** + * @internal + * Parse a trace state key/value pair + * @param value - the key/value pair as a string + * @returns The trace state member if valid, otherwise null + */ +function _parseListMember(value: string): ITraceStateMember | null { + if (value) { + TRACESTATE_KVP_REGEX.lastIndex = 0; // Reset the regex to ensure we start from the beginning + let match = TRACESTATE_KVP_REGEX.exec(value); + if (match && match.length >= 7 && match[1] && match[6]) { + let type = match[3] ? eTraceStateKeyType.multiTenant : eTraceStateKeyType.simple; + let multiTenant: ITraceStateMultiTenantKey = null; + if (type === eTraceStateKeyType.multiTenant) { + multiTenant = { + tenantId: match[4], + systemId: match[5] + }; + } + let parts: ITraceStateMember = { + type: type, + key: match[2], + multiTenant: multiTenant, + value: match[6] + }; + + return parts; + } + } + + return null; +} + +/** + * @internal + * Parse the trace state list from a string + * @param value - the list of trace states as a string + * @returns An array of trace state members + */ +function _parseTraceStateList(value?: string): ITraceStateMember[] { + let items: ITraceStateMember[] = []; + + if (value) { + let addedKeys: string[] = []; + arrForEach(strSplit(value, ","), (member) => { + let parts = _parseListMember(member); + if (parts) { + // As per the spec, the first occurrence of a key is the one that should be used + // as all new entries are added to the front (left) of the list + if (arrIndexOf(addedKeys, parts.key) === -1) { + items.push(parts); + addedKeys.push(parts.key); + + if (items.length >= MAX_TRACE_STATE_MEMBERS) { + // The trace state list should not exceed 32 members + return -1; + } + } + } + }); + } + + return items; +} + +function _indexOf(items: ITraceStateMember[], key: string): number { + for (let lp = 0; lp < items.length; lp++) { + if (items[lp].key === key) { + return lp; + } + } + + return -1; +} + +function _keys(items: ITraceStateMember[], parent?: IW3cTraceState | null): string[] { + let keys: string[] = []; + let delKeys: string[] = []; + arrForEach(items, (member) => { + if (member.value != null) { + keys.push(member.key); + } else { + delKeys.push(member.key); + } + }); + + if (parent) { + // Get and add parent keys that are not in the current list or marked as deleted + arrForEach(parent.keys, (key) => { + if (arrIndexOf(keys, key) === -1 && arrIndexOf(delKeys, key) === -1) { + keys.push(key); + } + }); + } + + return keys; +} + +/** + * @internal + * Identifies if the provided items are empty, meaning it has no keys or values. + * @param items - The items to check + * @param parent - The parent trace state to check for keys + * @returns true if the items are empty, false otherwise + */ +/*#__NO_SIDE_EFFECTS__*/ +function _isEmpty(items: ITraceStateMember[], parent?: IW3cTraceState | null): boolean { + let delKeys: string[]; + let isEmpty = true; + + if (items && items.length > 0) { + arrForEach(items, (member) => { + if (member.value != null) { + isEmpty = false; + } else { + if (!delKeys) { + delKeys = []; + } + + // If the value is null then this is a deleted key, so we can ignore it + delKeys.push(member.key); + } + }); + } + + if (isEmpty && parent) { + isEmpty = parent.isEmpty; + if (!isEmpty && delKeys && delKeys.length > 0) { + // If the parent is not empty then we need to check if any of the keys are in the deleted list + isEmpty = true; + arrForEach(parent.keys, (key) => { + if (arrIndexOf(delKeys, key) === -1) { + isEmpty = false; + return -1; // Break out of the loop + } + }); + } + } + + return isEmpty; +} + +/** + * Identifies if the provided value looks like a distributed trace state instance + * @param value - The value to check + * @returns - True if the value looks like a distributed trace state instance + */ +/*#__NO_SIDE_EFFECTS__*/ +export function isW3cTraceState(value: any): value is IW3cTraceState { + return !!(value && isArray(value.keys) && isFunction(value.get) && isFunction(value.set) && isFunction(value.del) && isFunction(value.hdrs)); +} + +/** + * Creates a new mutable {@link IW3cTraceState} instance, optionally inheriting from the parent trace state + * and optionally using the provided encoded string value as the initial trace state. + * Calls to {@link IW3cTraceState.set} and {@link IW3cTraceState.del} will modify the current instance + * which means that any child instance that is using this instance as a parent will also be indirectly + * modified unless the child instance has overridden the value associated with the modified key. + * @since 3.4.0 + * @param value - The string value for the trace state + * @param parent - The parent trace state to inherit any existing keys from. + * @returns - A new distributed trace state instance + */ +/*#__NO_SIDE_EFFECTS__*/ +export function createW3cTraceState(value?: string | null, parent?: IW3cTraceState | null): IW3cTraceState { + let cachedItems: ICachedValue = safeGetDeferred(_parseTraceStateList, [], [value || STR_EMPTY]); + + function _get(key: string): string | undefined { + let value: string | undefined; + let theItems = cachedItems.v; + let idx = _indexOf(theItems, key); + if (idx !== -1) { + let itmValue = theItems[idx].value; + if (itmValue != null) { + // Special case for the value being null, which means the key was deleted + value = itmValue; + } + } else if (parent) { + // Get the value from the parent if it exists + value = parent.get(key); + } + + return value; + } + + function _setMember(member: ITraceStateMember): number { + if (member) { + let theItems = cachedItems.v; + let idx = _indexOf(theItems, member.key); + if (idx !== -1) { + // Move the item to the front of the list, removing the previous instance + theItems.splice(idx, 1); + } + + theItems.unshift(member); + // We need to re-create the cached value as during testing the cached lazy value + // may get re-evaluated resetting the items to the original value + cachedItems = createCachedValue(theItems); + + return 0; + } + + return -1; + } + + function _set(key: string, value: string | null): number { + let member: ITraceStateMember | null; + if (key && isString(key) && !isNullOrUndefined(value) && isString(value)) { + member = _parseListMember(key + "=" + value); // Validate the key/value pair before adding it to the state + } + + return _setMember(member); + } + + function _del(key: string) { + _setMember({ + type: eTraceStateKeyType.deleted, + key: key + }); + } + + function _headers(maxHeaders?: number, maxKeys?: number, maxLen?: number): string[] { + let results: string[] = []; + let result = STR_EMPTY; + let numKeys = 0; + let len = 0; + + // Default to the max values if not provided + maxKeys = maxKeys || MAX_TRACE_STATE_MEMBERS; + + // Default to the max length if not provided + maxLen = maxLen || MAX_TRACE_STATE_LEN; + + let theItems = cachedItems.v; + arrForEach(_keys(theItems, parent), (key) => { + let value = _get(key); + if (!isNullOrUndefined(value) && isString(value)) { + numKeys++; + let val = key + "=" + value; + let valLen = val.length; + if (len + 1 + valLen >= maxLen) { + // Don't exceed the max length for any single combined header value + results.push(result); + + if (maxHeaders && results.length <= maxHeaders) { + // Don't exceed the max number of entries + return -1; + } + + result = STR_EMPTY; + len = 0; + } + + if (result.length > 0) { + result += ","; + len++; + } + + result += val; + len += valLen; + + if (numKeys >= maxKeys) { + // Only allow the first maxKeys members + return -1; + } + } + }); + + if (result) { + results.push(result); + } + + return results; + } + + let traceStateList: IW3cTraceState = { + keys: [], + isEmpty: false, + get: _get, + set: _set, + del: _del, + hdrs: _headers, + child: () => createW3cTraceState(null, traceStateList) + }; + + function _toString() { + let headers = traceStateList.hdrs(1); + return headers.length > 0 ? headers[0] : STR_EMPTY; + } + + objDefineProps(traceStateList, { + "keys": { + g: () => _keys(cachedItems.v, parent) + }, + "isEmpty": { + g: () => _isEmpty(cachedItems.v, parent) + }, + "toString": { + v: _toString, + e: false // Do not allow the toString to be enumerated + }, + "_p": { + v: parent, + e: false // Do not allow the parent to be enumerated + } + }); + + setObjStringTag(traceStateList, _toString); + + return traceStateList; +} + +/** + * Create a new independent instance of IW3cTraceState that contains a snapshot of all current key/value pairs + * from the provided instance and any parent instances. The returned instance will have no parent and will be completely + * independent from any future changes to the original instance or its parent chain. + * This is useful when you need to capture the current state and ensure it remains unchanged regardless of + * future modifications to the parent instances. + * @since 3.4.0 + * @param traceState - The trace state instance to snapshot + * @returns A new independent instance of IW3cTraceState with all current key/value pairs captured + */ +/*#__NO_SIDE_EFFECTS__*/ +export function snapshotW3cTraceState(traceState: IW3cTraceState): IW3cTraceState { + // Create a new independent instance with no parent + // This ensures the returned instance is completely independent from future changes + let snapshot = createW3cTraceState(null, null); + + if (traceState) { + let theKeys = traceState.keys; + + // Iterate over the keys in reverse order to maintain correct precedence + // Since set() adds items to the front, we need to add them in reverse order + // to preserve the original key ordering where newer keys take precedence + for (let i = theKeys.length - 1; i >= 0; i--) { + let key = theKeys[i]; + let value = traceState.get(key); + if (!isNullOrUndefined(value) && isString(value)) { + // Use the set function to add the key/value pair to the snapshot + // This leverages the existing validation and formatting logic + snapshot.set(key, value); + } + } + } + + return snapshot; +} + +/** + * Helper function to fetch the passed traceparent from the page, looking for it as a meta-tag or a Server-Timing header. + * @since 3.4.0 + * @param selectIdx - If the found value is comma separated which is the preferred entry to select, defaults to the first + * @returns + */ +/*#__NO_SIDE_EFFECTS__*/ +export function findW3cTraceState(): IW3cTraceState { + const name = "tracestate"; + let traceState: IW3cTraceState = null; + let metaTags = findMetaTags(name); + if (metaTags.length > 0) { + traceState = createW3cTraceState(metaTags.join(",")); + } + + if (!traceState) { + let serverTimings = findNamedServerTimings(name); + if (serverTimings.length > 0) { + traceState = createW3cTraceState(serverTimings.join(",")); + } + } + + return traceState; +} diff --git a/shared/AppInsightsCommon/src/Telemetry/Common/Data.ts b/shared/AppInsightsCore/src/telemetry/ai/Common/Data.ts similarity index 68% rename from shared/AppInsightsCommon/src/Telemetry/Common/Data.ts rename to shared/AppInsightsCore/src/telemetry/ai/Common/Data.ts index 753f87d62..3049f0131 100644 --- a/shared/AppInsightsCommon/src/Telemetry/Common/Data.ts +++ b/shared/AppInsightsCore/src/telemetry/ai/Common/Data.ts @@ -1,10 +1,14 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { FieldType } from "../../Enums"; -import { IData } from "../../Interfaces/Contracts/IData"; -import { ISerializable } from "../../Interfaces/Telemetry/ISerializable"; +import { FieldType } from "../../../enums/ai/Enums"; +import { IData } from "../../../interfaces/ai/contracts/IData"; +import { ISerializable } from "../../../interfaces/ai/telemetry/ISerializable"; +/** + * @deprecated - will be removed in future releases as this was only used by the applicationinsights-channel-js package. + * And it no longer uses this class. + */ export class Data implements IData, ISerializable { /** diff --git a/shared/AppInsightsCommon/src/Telemetry/Common/DataPoint.ts b/shared/AppInsightsCore/src/telemetry/ai/Common/DataPoint.ts similarity index 82% rename from shared/AppInsightsCommon/src/Telemetry/Common/DataPoint.ts rename to shared/AppInsightsCore/src/telemetry/ai/Common/DataPoint.ts index ca2233792..93268ee4b 100644 --- a/shared/AppInsightsCommon/src/Telemetry/Common/DataPoint.ts +++ b/shared/AppInsightsCore/src/telemetry/ai/Common/DataPoint.ts @@ -1,10 +1,10 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { FieldType } from "../../Enums"; -import { DataPointType } from "../../Interfaces/Contracts/DataPointType"; -import { IDataPoint } from "../../Interfaces/Contracts/IDataPoint"; -import { ISerializable } from "../../Interfaces/Telemetry/ISerializable"; +import { FieldType } from "../../../enums/ai/Enums"; +import { DataPointType } from "../../../interfaces/ai/contracts/DataPointType"; +import { IDataPoint } from "../../../interfaces/ai/contracts/IDataPoint"; +import { ISerializable } from "../../../interfaces/ai/telemetry/ISerializable"; export class DataPoint implements IDataPoint, ISerializable { diff --git a/shared/AppInsightsCommon/src/Telemetry/Common/DataSanitizer.ts b/shared/AppInsightsCore/src/telemetry/ai/Common/DataSanitizer.ts similarity index 88% rename from shared/AppInsightsCommon/src/Telemetry/Common/DataSanitizer.ts rename to shared/AppInsightsCore/src/telemetry/ai/Common/DataSanitizer.ts index daae63b91..efce32458 100644 --- a/shared/AppInsightsCommon/src/Telemetry/Common/DataSanitizer.ts +++ b/shared/AppInsightsCore/src/telemetry/ai/Common/DataSanitizer.ts @@ -1,11 +1,12 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { - IConfiguration, IDiagnosticLogger, _eInternalMessageId, _throwInternal, eLoggingSeverity, fieldRedaction, getJSON, hasJSON, isObject, - objForEachKey, strTrim -} from "@microsoft/applicationinsights-core-js"; -import { asString, isString, strSubstr, strSubstring } from "@nevware21/ts-utils"; +import { asString, isObject, isString, objForEachKey, strSubstr, strSubstring, strTrim } from "@nevware21/ts-utils"; +import { _throwInternal } from "../../../diagnostics/DiagnosticLogger"; +import { _eInternalMessageId, eLoggingSeverity } from "../../../enums/ai/LoggingEnums"; +import { IConfiguration } from "../../../interfaces/ai/IConfiguration"; +import { IDiagnosticLogger } from "../../../interfaces/ai/IDiagnosticLogger"; +import { fieldRedaction, getJSON, hasJSON } from "../../../utils/EnvUtils"; export const enum DataSanitizerValues { /** @@ -44,6 +45,7 @@ export const enum DataSanitizerValues { MAX_EXCEPTION_LENGTH = 32768 } +/*#__NO_SIDE_EFFECTS__*/ export function dataSanitizeKeyAndAddUniqueness(logger: IDiagnosticLogger, key: any, map: any) { const origLength = key.length; let field = dataSanitizeKey(logger, key); @@ -61,6 +63,7 @@ export function dataSanitizeKeyAndAddUniqueness(logger: IDiagnosticLogger, key: return field; } +/*#__NO_SIDE_EFFECTS__*/ export function dataSanitizeKey(logger: IDiagnosticLogger, name: any) { let nameTrunc: String; if (name) { @@ -81,6 +84,7 @@ export function dataSanitizeKey(logger: IDiagnosticLogger, name: any) { return nameTrunc || name; } +/*#__NO_SIDE_EFFECTS__*/ export function dataSanitizeString(logger: IDiagnosticLogger, value: any, maxLength: number = DataSanitizerValues.MAX_STRING_LENGTH) { let valueTrunc : String; if (value) { @@ -99,6 +103,7 @@ export function dataSanitizeString(logger: IDiagnosticLogger, value: any, maxLen return valueTrunc || value; } +/*#__NO_SIDE_EFFECTS__*/ export function dataSanitizeUrl(logger: IDiagnosticLogger, url: any, config?: IConfiguration) { if (isString(url)) { url = fieldRedaction(url, config); @@ -106,6 +111,7 @@ export function dataSanitizeUrl(logger: IDiagnosticLogger, url: any, config?: IC return dataSanitizeInput(logger, url, DataSanitizerValues.MAX_URL_LENGTH, _eInternalMessageId.UrlTooLong); } +/*#__NO_SIDE_EFFECTS__*/ export function dataSanitizeMessage(logger: IDiagnosticLogger, message: any) { let messageTrunc : String; if (message) { @@ -122,6 +128,7 @@ export function dataSanitizeMessage(logger: IDiagnosticLogger, message: any) { return messageTrunc || message; } +/*#__NO_SIDE_EFFECTS__*/ export function dataSanitizeException(logger: IDiagnosticLogger, exception: any) { let exceptionTrunc : String; if (exception) { @@ -138,9 +145,10 @@ export function dataSanitizeException(logger: IDiagnosticLogger, exception: any) return exceptionTrunc || exception; } +/*#__NO_SIDE_EFFECTS__*/ export function dataSanitizeProperties(logger: IDiagnosticLogger, properties: any) { if (properties) { - const tempProps = {}; + const tempProps: any = {}; objForEachKey(properties, (prop, value) => { if (isObject(value) && hasJSON()) { // Stringify any part C properties @@ -160,9 +168,10 @@ export function dataSanitizeProperties(logger: IDiagnosticLogger, properties: an return properties; } +/*#__NO_SIDE_EFFECTS__*/ export function dataSanitizeMeasurements(logger: IDiagnosticLogger, measurements: any) { if (measurements) { - const tempMeasurements = {}; + const tempMeasurements: any = {}; objForEachKey(measurements, (measure, value) => { measure = dataSanitizeKeyAndAddUniqueness(logger, measure, tempMeasurements); tempMeasurements[measure] = value; @@ -174,10 +183,12 @@ export function dataSanitizeMeasurements(logger: IDiagnosticLogger, measurements return measurements; } +/*#__NO_SIDE_EFFECTS__*/ export function dataSanitizeId(logger: IDiagnosticLogger, id: string): string { return id ? dataSanitizeInput(logger, id, DataSanitizerValues.MAX_ID_LENGTH, _eInternalMessageId.IdTooLong).toString() : id; } +/*#__NO_SIDE_EFFECTS__*/ export function dataSanitizeInput(logger: IDiagnosticLogger, input: any, maxLength: number, _msgId: _eInternalMessageId) { let inputTrunc : String; if (input) { @@ -196,6 +207,7 @@ export function dataSanitizeInput(logger: IDiagnosticLogger, input: any, maxLeng return inputTrunc || input; } +/*#__NO_SIDE_EFFECTS__*/ export function dsPadNumber(num: number) { const s = "00" + num; return strSubstr(s, s.length - 3); diff --git a/shared/AppInsightsCommon/src/Telemetry/Common/Envelope.ts b/shared/AppInsightsCore/src/telemetry/ai/Common/Envelope.ts similarity index 87% rename from shared/AppInsightsCommon/src/Telemetry/Common/Envelope.ts rename to shared/AppInsightsCore/src/telemetry/ai/Common/Envelope.ts index 3918904a9..c08e68abe 100644 --- a/shared/AppInsightsCommon/src/Telemetry/Common/Envelope.ts +++ b/shared/AppInsightsCore/src/telemetry/ai/Common/Envelope.ts @@ -1,11 +1,12 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IDiagnosticLogger, toISOString } from "@microsoft/applicationinsights-core-js"; -import { strNotSpecified } from "../../Constants"; -import { FieldType } from "../../Enums"; -import { IBase } from "../../Interfaces/Contracts/IBase"; -import { IEnvelope } from "../../Interfaces/Telemetry/IEnvelope"; +import { strNotSpecified } from "../../../constants/Constants"; +import { FieldType } from "../../../enums/ai/Enums"; +import { IDiagnosticLogger } from "../../../interfaces/ai/IDiagnosticLogger"; +import { IBase } from "../../../interfaces/ai/contracts/IBase"; +import { IEnvelope } from "../../../interfaces/ai/telemetry/IEnvelope"; +import { toISOString } from "../../../utils/HelperFuncs"; import { dataSanitizeString } from "./DataSanitizer"; export class Envelope implements IEnvelope { diff --git a/shared/AppInsightsCore/src/telemetry/ai/DataTypes.ts b/shared/AppInsightsCore/src/telemetry/ai/DataTypes.ts new file mode 100644 index 000000000..13e6ae95e --- /dev/null +++ b/shared/AppInsightsCore/src/telemetry/ai/DataTypes.ts @@ -0,0 +1,11 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +export const EventDataType = "EventData"; +export const ExceptionDataType = "ExceptionData"; +export const MetricDataType = "MetricData"; +export const PageViewDataType = "PageviewData"; +export const PageViewPerformanceDataType = "PageviewPerformanceData"; +export const RemoteDependencyDataType = "RemoteDependencyData"; +export const RequestDataType = "RequestData"; +export const TraceDataType = "MessageData"; diff --git a/shared/AppInsightsCore/src/telemetry/ai/EnvelopeTypes.ts b/shared/AppInsightsCore/src/telemetry/ai/EnvelopeTypes.ts new file mode 100644 index 000000000..f13a3a5fc --- /dev/null +++ b/shared/AppInsightsCore/src/telemetry/ai/EnvelopeTypes.ts @@ -0,0 +1,16 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +/* #__NO_SIDE_EFFECTS__# */ +function _AddPrefix(name: string) { + return "Microsoft.ApplicationInsights.{0}." + name; +} + +export const EventEnvelopeType = (/*#__PURE__*/ _AddPrefix("Event")); +export const ExceptionEnvelopeType = (/*#__PURE__*/ _AddPrefix("Exception")); +export const MetricEnvelopeType = (/*#__PURE__*/ _AddPrefix("Metric")); +export const PageViewEnvelopeType = (/*#__PURE__*/ _AddPrefix("Pageview")); +export const PageViewPerformanceEnvelopeType = (/*#__PURE__*/ _AddPrefix("PageviewPerformance")); +export const RemoteDependencyEnvelopeType = (/*#__PURE__*/ _AddPrefix("RemoteDependency")); +export const RequestEnvelopeType = (/*#__PURE__*/ _AddPrefix("Request")); +export const TraceEnvelopeType = (/*#__PURE__*/ _AddPrefix("Message")); \ No newline at end of file diff --git a/shared/AppInsightsCommon/src/Telemetry/Event.ts b/shared/AppInsightsCore/src/telemetry/ai/Event.ts similarity index 65% rename from shared/AppInsightsCommon/src/Telemetry/Event.ts rename to shared/AppInsightsCore/src/telemetry/ai/Event.ts index 8dce98f97..769fa3e39 100644 --- a/shared/AppInsightsCommon/src/Telemetry/Event.ts +++ b/shared/AppInsightsCore/src/telemetry/ai/Event.ts @@ -1,16 +1,25 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IDiagnosticLogger } from "@microsoft/applicationinsights-core-js"; -import { strNotSpecified } from "../Constants"; -import { FieldType } from "../Enums"; -import { IEventData } from "../Interfaces/Contracts/IEventData"; -import { ISerializable } from "../Interfaces/Telemetry/ISerializable"; +import { strNotSpecified } from "../../constants/Constants"; +import { FieldType } from "../../enums/ai/Enums"; +import { IDiagnosticLogger } from "../../interfaces/ai/IDiagnosticLogger"; +import { IEventData } from "../../interfaces/ai/contracts/IEventData"; +import { ISerializable } from "../../interfaces/ai/telemetry/ISerializable"; import { dataSanitizeMeasurements, dataSanitizeProperties, dataSanitizeString } from "./Common/DataSanitizer"; +import { EventDataType } from "./DataTypes"; +import { EventEnvelopeType } from "./EnvelopeTypes"; export class Event implements IEventData, ISerializable { - public static envelopeType = "Microsoft.ApplicationInsights.{0}.Event"; - public static dataType = "EventData"; + /** + * @deprecated Use the constant EventEnvelopeType instead. + */ + public static envelopeType = EventEnvelopeType; + + /** + * @deprecated Use the constant EventDataType instead. + */ + public static dataType = EventDataType; public aiDataContract = { ver: FieldType.Required, diff --git a/shared/AppInsightsCommon/src/Telemetry/Exception.ts b/shared/AppInsightsCore/src/telemetry/ai/Exception.ts similarity index 95% rename from shared/AppInsightsCommon/src/Telemetry/Exception.ts rename to shared/AppInsightsCore/src/telemetry/ai/Exception.ts index c45da9bf9..1514fa009 100644 --- a/shared/AppInsightsCommon/src/Telemetry/Exception.ts +++ b/shared/AppInsightsCore/src/telemetry/ai/Exception.ts @@ -2,20 +2,25 @@ // Licensed under the MIT License. import { - IDiagnosticLogger, arrForEach, arrMap, isArray, isError, isFunction, isNullOrUndefined, isObject, isString, strTrim -} from "@microsoft/applicationinsights-core-js"; -import { asString, getWindow, objFreeze, strIndexOf } from "@nevware21/ts-utils"; -import { strNotSpecified } from "../Constants"; -import { FieldType } from "../Enums"; -import { IExceptionData } from "../Interfaces/Contracts/IExceptionData"; -import { IExceptionDetails } from "../Interfaces/Contracts/IExceptionDetails"; -import { IStackFrame } from "../Interfaces/Contracts/IStackFrame"; -import { SeverityLevel } from "../Interfaces/Contracts/SeverityLevel"; -import { IAutoExceptionTelemetry, IExceptionDetailsInternal, IExceptionInternal, IStackDetails } from "../Interfaces/IExceptionTelemetry"; -import { ISerializable } from "../Interfaces/Telemetry/ISerializable"; + arrForEach, arrMap, asString, getWindow, isArray, isError, isFunction, isNullOrUndefined, isObject, isString, objFreeze, strIndexOf, + strTrim +} from "@nevware21/ts-utils"; +import { strNotSpecified } from "../../constants/Constants"; +import { FieldType } from "../../enums/ai/Enums"; +import { IDiagnosticLogger } from "../../interfaces/ai/IDiagnosticLogger"; +import { + IAutoExceptionTelemetry, IExceptionDetailsInternal, IExceptionInternal, IStackDetails +} from "../../interfaces/ai/IExceptionTelemetry"; +import { IExceptionData } from "../../interfaces/ai/contracts/IExceptionData"; +import { IExceptionDetails } from "../../interfaces/ai/contracts/IExceptionDetails"; +import { IStackFrame } from "../../interfaces/ai/contracts/IStackFrame"; +import { SeverityLevel } from "../../interfaces/ai/contracts/SeverityLevel"; +import { ISerializable } from "../../interfaces/ai/telemetry/ISerializable"; import { dataSanitizeException, dataSanitizeMeasurements, dataSanitizeMessage, dataSanitizeProperties, dataSanitizeString } from "./Common/DataSanitizer"; +import { ExceptionDataType } from "./DataTypes"; +import { ExceptionEnvelopeType } from "./EnvelopeTypes"; // These Regex covers the following patterns // 1. Chrome/Firefox/IE/Edge: @@ -368,7 +373,7 @@ function _getStackFromErrorObj(errorObj:any): IStackDetails { details = errorObj; } else if (_isStackDetails(errorObj[strStackDetails])) { details = errorObj[strStackDetails]; - } else if (getWindow() && getWindow()["opera"] && errorObj[strMessage]) { + } else if (getWindow() && (getWindow() as any)["opera"] && errorObj[strMessage]) { // Opera details = _getOperaStack(errorObj.message); } else if (errorObj["reason"] && errorObj.reason[strStack]) { @@ -535,9 +540,15 @@ export function _formatErrorCode(errorObj:any) { } export class Exception implements IExceptionData, ISerializable { + /** + * @deprecated Use the constant ExceptionEnvelopeType instead. + */ + public static envelopeType = ExceptionEnvelopeType; - public static envelopeType = "Microsoft.ApplicationInsights.{0}.Exception"; - public static dataType = "ExceptionData"; + /** + * @deprecated Use the constant ExceptionDataType instead. + */ + public static dataType = ExceptionDataType; public id?: string; public problemGroup?: string; @@ -694,7 +705,7 @@ export class Exception implements IExceptionData, ISerializable { public static formatError = _formatErrorCode; } -const exDetailsAiDataContract = objFreeze({ +const exDetailsAiDataContract = (/*#__PURE__*/ objFreeze({ id: FieldType.Default, outerId: FieldType.Default, typeName: FieldType.Required, @@ -702,7 +713,7 @@ const exDetailsAiDataContract = objFreeze({ hasFullStack: FieldType.Default, stack: FieldType.Default, parsedStack: FieldType.Array -}); +})); interface _IExceptionDetails extends IExceptionDetails, ISerializable { toInterface: () => IExceptionDetailsInternal; @@ -745,7 +756,7 @@ export function _createExceptionDetails(logger: IDiagnosticLogger, exception: Er typeName = dataSanitizeString(logger, _getErrorType(error)) || strNotSpecified; message = dataSanitizeMessage(logger, _formatMessage(exception || error, typeName)) || strNotSpecified; - const stack = exception[strStackDetails] || _getStackFromErrorObj(exception); + const stack = (exception as any)[strStackDetails] || _getStackFromErrorObj(exception); parsedStack = _parseStack(stack); // after parsedStack is inited, iterate over each frame object, sanitize its assembly field @@ -842,13 +853,13 @@ export interface _IParsedStackFrame extends IStackFrame, ISerializable { sizeInBytes: number; } -const stackFrameAiDataContract = objFreeze({ +const stackFrameAiDataContract = (/*#__PURE__*/ objFreeze({ level: FieldType.Required, method: FieldType.Required, assembly: FieldType.Default, fileName: FieldType.Default, line: FieldType.Default -}); +})); export function _extractStackFrame(frame: string, level: number): _IParsedStackFrame | undefined { let theFrame: _IParsedStackFrame; diff --git a/shared/AppInsightsCommon/src/Telemetry/Metric.ts b/shared/AppInsightsCore/src/telemetry/ai/Metric.ts similarity index 73% rename from shared/AppInsightsCommon/src/Telemetry/Metric.ts rename to shared/AppInsightsCore/src/telemetry/ai/Metric.ts index 45f08de9b..a302e5b22 100644 --- a/shared/AppInsightsCommon/src/Telemetry/Metric.ts +++ b/shared/AppInsightsCore/src/telemetry/ai/Metric.ts @@ -1,18 +1,26 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IDiagnosticLogger } from "@microsoft/applicationinsights-core-js"; -import { strNotSpecified } from "../Constants"; -import { FieldType } from "../Enums"; -import { IMetricData } from "../Interfaces/Contracts/IMetricData"; -import { ISerializable } from "../Interfaces/Telemetry/ISerializable"; +import { strNotSpecified } from "../../constants/Constants"; +import { FieldType } from "../../enums/ai/Enums"; +import { IDiagnosticLogger } from "../../interfaces/ai/IDiagnosticLogger"; +import { IMetricData } from "../../interfaces/ai/contracts/IMetricData"; +import { ISerializable } from "../../interfaces/ai/telemetry/ISerializable"; import { DataPoint } from "./Common/DataPoint"; import { dataSanitizeMeasurements, dataSanitizeProperties, dataSanitizeString } from "./Common/DataSanitizer"; +import { MetricDataType } from "./DataTypes"; +import { MetricEnvelopeType } from "./EnvelopeTypes"; export class Metric implements IMetricData, ISerializable { + /** + * @deprecated Use the constant MetricEnvelopeType instead. + */ + public static envelopeType = MetricEnvelopeType; - public static envelopeType = "Microsoft.ApplicationInsights.{0}.Metric"; - public static dataType = "MetricData"; + /** + * @deprecated Use the constant MetricDataType instead. + */ + public static dataType = MetricDataType; public aiDataContract = { ver: FieldType.Required, diff --git a/shared/AppInsightsCommon/src/Telemetry/PageView.ts b/shared/AppInsightsCore/src/telemetry/ai/PageView.ts similarity index 74% rename from shared/AppInsightsCommon/src/Telemetry/PageView.ts rename to shared/AppInsightsCore/src/telemetry/ai/PageView.ts index de4df7f27..9c101f09c 100644 --- a/shared/AppInsightsCommon/src/Telemetry/PageView.ts +++ b/shared/AppInsightsCore/src/telemetry/ai/PageView.ts @@ -1,20 +1,28 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IDiagnosticLogger } from "@microsoft/applicationinsights-core-js"; -import { strNotSpecified } from "../Constants"; -import { FieldType } from "../Enums"; -import { msToTimeSpan } from "../HelperFuncs"; -import { IPageViewData } from "../Interfaces/Contracts/IPageViewData"; -import { ISerializable } from "../Interfaces/Telemetry/ISerializable"; +import { strNotSpecified } from "../../constants/Constants"; +import { FieldType } from "../../enums/ai/Enums"; +import { IDiagnosticLogger } from "../../interfaces/ai/IDiagnosticLogger"; +import { IPageViewData } from "../../interfaces/ai/contracts/IPageViewData"; +import { ISerializable } from "../../interfaces/ai/telemetry/ISerializable"; +import { msToTimeSpan } from "../../utils/HelperFuncs"; import { dataSanitizeId, dataSanitizeMeasurements, dataSanitizeProperties, dataSanitizeString, dataSanitizeUrl } from "./Common/DataSanitizer"; +import { PageViewDataType } from "./DataTypes"; +import { PageViewEnvelopeType } from "./EnvelopeTypes"; export class PageView implements IPageViewData, ISerializable { + /** + * @deprecated Use the constant PageViewEnvelopeType instead. + */ + public static envelopeType = PageViewEnvelopeType; - public static envelopeType = "Microsoft.ApplicationInsights.{0}.Pageview"; - public static dataType = "PageviewData"; + /** + * @deprecated Use the constant PageViewDataType instead. + */ + public static dataType = PageViewDataType; public aiDataContract = { ver: FieldType.Required, diff --git a/shared/AppInsightsCommon/src/Telemetry/PageViewPerformance.ts b/shared/AppInsightsCore/src/telemetry/ai/PageViewPerformance.ts similarity index 79% rename from shared/AppInsightsCommon/src/Telemetry/PageViewPerformance.ts rename to shared/AppInsightsCore/src/telemetry/ai/PageViewPerformance.ts index 29f9776d4..0b11d83ca 100644 --- a/shared/AppInsightsCommon/src/Telemetry/PageViewPerformance.ts +++ b/shared/AppInsightsCore/src/telemetry/ai/PageViewPerformance.ts @@ -1,18 +1,26 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IDiagnosticLogger } from "@microsoft/applicationinsights-core-js"; -import { strNotSpecified } from "../Constants"; -import { FieldType } from "../Enums"; -import { IPageViewPerfData } from "../Interfaces/Contracts/IPageViewPerfData"; -import { IPageViewPerformanceTelemetry } from "../Interfaces/IPageViewPerformanceTelemetry"; -import { ISerializable } from "../Interfaces/Telemetry/ISerializable"; +import { strNotSpecified } from "../../constants/Constants"; +import { FieldType } from "../../enums/ai/Enums"; +import { IDiagnosticLogger } from "../../interfaces/ai/IDiagnosticLogger"; +import { IPageViewPerformanceTelemetry } from "../../interfaces/ai/IPageViewPerformanceTelemetry"; +import { IPageViewPerfData } from "../../interfaces/ai/contracts/IPageViewPerfData"; +import { ISerializable } from "../../interfaces/ai/telemetry/ISerializable"; import { dataSanitizeMeasurements, dataSanitizeProperties, dataSanitizeString, dataSanitizeUrl } from "./Common/DataSanitizer"; +import { PageViewPerformanceDataType } from "./DataTypes"; +import { PageViewPerformanceEnvelopeType } from "./EnvelopeTypes"; export class PageViewPerformance implements IPageViewPerfData, ISerializable { + /** + * @deprecated Use the constant PageViewPerformanceEnvelopeType instead. + */ + public static envelopeType = PageViewPerformanceEnvelopeType; - public static envelopeType = "Microsoft.ApplicationInsights.{0}.PageviewPerformance"; - public static dataType = "PageviewPerformanceData"; + /** + * @deprecated Use the constant PageViewPerformanceDataType instead. + */ + public static dataType = PageViewPerformanceDataType; public aiDataContract = { ver: FieldType.Required, diff --git a/shared/AppInsightsCommon/src/Telemetry/RemoteDependencyData.ts b/shared/AppInsightsCore/src/telemetry/ai/RemoteDependencyData.ts similarity index 78% rename from shared/AppInsightsCommon/src/Telemetry/RemoteDependencyData.ts rename to shared/AppInsightsCore/src/telemetry/ai/RemoteDependencyData.ts index 4ba3d1c78..f27c2db3e 100644 --- a/shared/AppInsightsCommon/src/Telemetry/RemoteDependencyData.ts +++ b/shared/AppInsightsCore/src/telemetry/ai/RemoteDependencyData.ts @@ -1,18 +1,30 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IDiagnosticLogger } from "@microsoft/applicationinsights-core-js"; -import { FieldType } from "../Enums"; -import { msToTimeSpan } from "../HelperFuncs"; -import { IRemoteDependencyData } from "../Interfaces/Contracts/IRemoteDependencyData"; -import { ISerializable } from "../Interfaces/Telemetry/ISerializable"; -import { AjaxHelperParseDependencyPath } from "../Util"; +import { FieldType } from "../../enums/ai/Enums"; +import { IDiagnosticLogger } from "../../interfaces/ai/IDiagnosticLogger"; +import { IRemoteDependencyData } from "../../interfaces/ai/contracts/IRemoteDependencyData"; +import { ISerializable } from "../../interfaces/ai/telemetry/ISerializable"; +import { msToTimeSpan } from "../../utils/HelperFuncs"; +import { AjaxHelperParseDependencyPath } from "../../utils/Util"; import { dataSanitizeMeasurements, dataSanitizeProperties, dataSanitizeString, dataSanitizeUrl } from "./Common/DataSanitizer"; +import { RemoteDependencyDataType } from "./DataTypes"; +import { RemoteDependencyEnvelopeType } from "./EnvelopeTypes"; +/** + * @deprecated - will be removed in future releases as this was only used by the applicationinsights-channel-js package. + * And it no longer uses this class. + */ export class RemoteDependencyData implements IRemoteDependencyData, ISerializable { + /** + * @deprecated Use the constant RemoteDependencyEnvelopeType instead. + */ + public static envelopeType = RemoteDependencyEnvelopeType; - public static envelopeType = "Microsoft.ApplicationInsights.{0}.RemoteDependency"; - public static dataType = "RemoteDependencyData"; + /** + * @deprecated Use the constant RemoteDependencyDataType instead. + */ + public static dataType = RemoteDependencyDataType; public aiDataContract = { id: FieldType.Required, @@ -104,7 +116,7 @@ export class RemoteDependencyData implements IRemoteDependencyData, ISerializabl _self.id = id; _self.duration = msToTimeSpan(value); _self.success = success; - _self.resultCode = resultCode + ""; + _self.resultCode = "" + resultCode; _self.type = dataSanitizeString(logger, requestAPI); diff --git a/shared/AppInsightsCommon/src/Telemetry/Trace.ts b/shared/AppInsightsCore/src/telemetry/ai/Trace.ts similarity index 66% rename from shared/AppInsightsCommon/src/Telemetry/Trace.ts rename to shared/AppInsightsCore/src/telemetry/ai/Trace.ts index f74c78ef9..f04bfb890 100644 --- a/shared/AppInsightsCommon/src/Telemetry/Trace.ts +++ b/shared/AppInsightsCore/src/telemetry/ai/Trace.ts @@ -1,18 +1,26 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IDiagnosticLogger } from "@microsoft/applicationinsights-core-js"; -import { strNotSpecified } from "../Constants"; -import { FieldType } from "../Enums"; -import { IMessageData } from "../Interfaces/Contracts/IMessageData"; -import { SeverityLevel } from "../Interfaces/Contracts/SeverityLevel"; -import { ISerializable } from "../Interfaces/Telemetry/ISerializable"; +import { strNotSpecified } from "../../constants/Constants"; +import { FieldType } from "../../enums/ai/Enums"; +import { IDiagnosticLogger } from "../../interfaces/ai/IDiagnosticLogger"; +import { IMessageData } from "../../interfaces/ai/contracts/IMessageData"; +import { SeverityLevel } from "../../interfaces/ai/contracts/SeverityLevel"; +import { ISerializable } from "../../interfaces/ai/telemetry/ISerializable"; import { dataSanitizeMeasurements, dataSanitizeMessage, dataSanitizeProperties } from "./Common/DataSanitizer"; +import { TraceDataType } from "./DataTypes"; +import { TraceEnvelopeType } from "./EnvelopeTypes"; export class Trace implements IMessageData, ISerializable { + /** + * @deprecated Use the constant TraceEnvelopeType instead. + */ + public static envelopeType = TraceEnvelopeType; - public static envelopeType = "Microsoft.ApplicationInsights.{0}.Message"; - public static dataType = "MessageData"; + /** + * @deprecated Use the constant TraceDataType instead. + */ + public static dataType = TraceDataType; public aiDataContract = { ver: FieldType.Required, diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/CoreUtils.ts b/shared/AppInsightsCore/src/utils/CoreUtils.ts similarity index 97% rename from shared/AppInsightsCore/src/JavaScriptSDK/CoreUtils.ts rename to shared/AppInsightsCore/src/utils/CoreUtils.ts index 78d3b4c64..856fdda23 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/CoreUtils.ts +++ b/shared/AppInsightsCore/src/utils/CoreUtils.ts @@ -3,7 +3,7 @@ "use strict"; import { strShimUndefined } from "@microsoft/applicationinsights-shims"; import { strSubstr, strSubstring } from "@nevware21/ts-utils"; -import { STR_EMPTY } from "./InternalConstants"; +import { STR_EMPTY } from "../constants/InternalConstants"; import { random32 } from "./RandomHelper"; // Added to help with minfication diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/DataCacheHelper.ts b/shared/AppInsightsCore/src/utils/DataCacheHelper.ts similarity index 97% rename from shared/AppInsightsCore/src/JavaScriptSDK/DataCacheHelper.ts rename to shared/AppInsightsCore/src/utils/DataCacheHelper.ts index 79f2e2bac..cd02749a2 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/DataCacheHelper.ts +++ b/shared/AppInsightsCore/src/utils/DataCacheHelper.ts @@ -2,8 +2,8 @@ // Licensed under the MIT License. import { objDefine } from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../constants/InternalConstants"; import { normalizeJsName } from "./HelperFuncs"; -import { STR_EMPTY } from "./InternalConstants"; import { newId } from "./RandomHelper"; const version = "#version#"; diff --git a/shared/AppInsightsCommon/src/DomHelperFuncs.ts b/shared/AppInsightsCore/src/utils/DomHelperFuncs.ts similarity index 86% rename from shared/AppInsightsCommon/src/DomHelperFuncs.ts rename to shared/AppInsightsCore/src/utils/DomHelperFuncs.ts index d13b28267..a4acf42c8 100644 --- a/shared/AppInsightsCommon/src/DomHelperFuncs.ts +++ b/shared/AppInsightsCore/src/utils/DomHelperFuncs.ts @@ -1,6 +1,7 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { getDocument, isFunction } from "@microsoft/applicationinsights-core-js"; + +import { getDocument, isFunction } from "@nevware21/ts-utils"; export function createDomEvent(eventName: string): Event { let event: Event = null; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/EnvUtils.ts b/shared/AppInsightsCore/src/utils/EnvUtils.ts similarity index 75% rename from shared/AppInsightsCore/src/JavaScriptSDK/EnvUtils.ts rename to shared/AppInsightsCore/src/utils/EnvUtils.ts index 4ab66e2cf..ed2382901 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/EnvUtils.ts +++ b/shared/AppInsightsCore/src/utils/EnvUtils.ts @@ -4,11 +4,12 @@ import { getGlobal, strShimObject, strShimPrototype, strShimUndefined } from "@microsoft/applicationinsights-shims"; import { - getDocument, getInst, getNavigator, getPerformance, hasNavigator, isFunction, isString, isUndefined, mathMax, strIndexOf, strSubstring + ICachedValue, arrForEach, createCachedValue, getDeferred, getDocument, getInst, getLazy, getNavigator, getPerformance, hasNavigator, + isFunction, isNullOrUndefined, isString, isUndefined, mathMax, strIndexOf, strSubstring } from "@nevware21/ts-utils"; -import { IConfiguration } from "../applicationinsights-core-js"; +import { DEFAULT_SENSITIVE_PARAMS, STR_EMPTY, STR_REDACTED } from "../constants/InternalConstants"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; import { strContains } from "./HelperFuncs"; -import { DEFAULT_SENSITIVE_PARAMS, STR_EMPTY, STR_REDACTED } from "./InternalConstants"; // TypeScript removed this interface so we need to declare the global so we can check for it's existence. declare var XDomainRequest: any; @@ -31,11 +32,13 @@ const strMsie = "msie"; const strTrident = "trident/"; const strXMLHttpRequest = "XMLHttpRequest"; -let _isTrident: boolean = null; -let _navUserAgentCheck: string = null; +// Local cached properties which are used to avoid multiple checks within the module +let _isTrident: ICachedValue; +let _navUserAgentCheck: string; let _enableMocks = false; -let _useXDomainRequest: boolean | null = null; -let _beaconsSupported: boolean | null = null; +let _useXDomainRequest: ICachedValue; +let _beaconsSupported: ICachedValue; +let _userAgent: ICachedValue; function _hasProperty(theClass: any, property: string) { let supported = false; @@ -65,6 +68,19 @@ function _hasProperty(theClass: any, property: string) { return supported; } +/*#__NO_SIDE_EFFECTS__*/ +export function getUserAgentString(): string { + if (!_userAgent) { + // Use lazy to allow mocking + _userAgent = getLazy(() => { + let nav = getNavigator() || {} as Navigator; + return nav.userAgent || STR_EMPTY; + }); + } + + return _userAgent.v; +} + /** * Enable the lookup of test mock objects if requested * @param enabled - A flag to enable or disable the mock @@ -78,6 +94,7 @@ export function setEnableEnvMocks(enabled: boolean) { * This helper is used to access the location object without causing an exception * "Uncaught ReferenceError: location is not defined" */ +/*#__NO_SIDE_EFFECTS__*/ export function getLocation(checkForMock?: boolean): Location | null { if (checkForMock && _enableMocks) { let mockLocation = getInst("__mockLocation") as Location; @@ -96,6 +113,7 @@ export function getLocation(checkForMock?: boolean): Location | null { /** * Returns the global console object */ +/*#__NO_SIDE_EFFECTS__*/ export function getConsole(): Console | null { if (typeof console !== strShimUndefined) { return console; @@ -111,6 +129,7 @@ export function getConsole(): Console | null { * exception will be thrown. * Defined as a function to support lazy / late binding environments. */ +/*#__NO_SIDE_EFFECTS__*/ export function hasJSON(): boolean { return Boolean((typeof JSON === strShimObject && JSON) || getInst(strJSON) !== null); } @@ -120,6 +139,7 @@ export function hasJSON(): boolean { * This helper is used to access the JSON object without causing an exception * "Uncaught ReferenceError: JSON is not defined" */ +/*#__NO_SIDE_EFFECTS__*/ export function getJSON(): JSON | null { if (hasJSON()) { return JSON || getInst(strJSON); @@ -133,6 +153,7 @@ export function getJSON(): JSON | null { * This helper is used to access the crypto object from the current * global instance which could be window or globalThis for a web worker */ +/*#__NO_SIDE_EFFECTS__*/ export function getCrypto(): Crypto | null { return getInst(strCrypto); } @@ -142,6 +163,7 @@ export function getCrypto(): Crypto | null { * This helper is used to access the crypto object from the current * global instance which could be window or globalThis for a web worker */ +/*#__NO_SIDE_EFFECTS__*/ export function getMsCrypto(): Crypto | null { return getInst(strMsCrypto); } @@ -149,6 +171,7 @@ export function getMsCrypto(): Crypto | null { /** * Returns whether the environment is reporting that we are running in a React Native Environment */ +/*#__NO_SIDE_EFFECTS__*/ export function isReactNative(): boolean { // If running in React Native, navigator.product will be populated var nav = getNavigator(); @@ -162,25 +185,26 @@ export function isReactNative(): boolean { /** * Identifies whether the current environment appears to be IE */ +/*#__NO_SIDE_EFFECTS__*/ export function isIE() { - let nav = getNavigator(); - if (nav && (nav.userAgent !== _navUserAgentCheck || _isTrident === null)) { + let userAgent = getUserAgentString(); + if (!_isTrident || userAgent !== _navUserAgentCheck) { // Added to support test mocking of the user agent - _navUserAgentCheck = nav.userAgent; - let userAgent = (_navUserAgentCheck || STR_EMPTY).toLowerCase(); - _isTrident = (strContains(userAgent, strMsie) || strContains(userAgent, strTrident)); + _navUserAgentCheck = userAgent; + let lwrUserAgent = _navUserAgentCheck.toLowerCase(); + _isTrident = createCachedValue( strContains(lwrUserAgent, strMsie) || strContains(lwrUserAgent, strTrident)); } - return _isTrident; + return _isTrident.v; } /** * Gets IE version returning the document emulation mode if we are running on IE, or null otherwise */ +/*#__NO_SIDE_EFFECTS__*/ export function getIEVersion(userAgentStr: string = null): number { if (!userAgentStr) { - let navigator = getNavigator() || ({} as Navigator); - userAgentStr = navigator ? (navigator.userAgent || STR_EMPTY).toLowerCase() : STR_EMPTY; + userAgentStr = getUserAgentString(); } var ua = (userAgentStr || STR_EMPTY).toLowerCase(); @@ -198,10 +222,10 @@ export function getIEVersion(userAgentStr: string = null): number { return null; } +/*#__NO_SIDE_EFFECTS__*/ export function isSafari(userAgentStr ?: string) { if (!userAgentStr || !isString(userAgentStr)) { - let navigator = getNavigator() || ({} as Navigator); - userAgentStr = navigator ? (navigator.userAgent || STR_EMPTY).toLowerCase() : STR_EMPTY; + userAgentStr = getUserAgentString().toLowerCase(); } var ua = (userAgentStr || STR_EMPTY).toLowerCase(); @@ -209,17 +233,19 @@ export function isSafari(userAgentStr ?: string) { } /** - * Checks if HTML5 Beacons are supported in the current environment. + * Checks if HTML5 Beacons are supported in the current environment. There is a side effect (for testing) + * when `useCached` is set to `false`, it will reset the cached value causing all future calls to + * use the new re-evaluated value for all future calls. * @param useCached - [Optional] used for testing to bypass the cached lookup, when `true` this will * cause the cached global to be reset. * @returns True if supported, false otherwise. */ export function isBeaconsSupported(useCached?: boolean): boolean { - if (_beaconsSupported === null || useCached === false) { - _beaconsSupported = hasNavigator() && Boolean(getNavigator().sendBeacon); + if (!_beaconsSupported || useCached === false) { + _beaconsSupported = createCachedValue(hasNavigator() && !!(getNavigator().sendBeacon)); } - return _beaconsSupported; + return _beaconsSupported.v; } /** @@ -227,6 +253,7 @@ export function isBeaconsSupported(useCached?: boolean): boolean { * @param withKeepAlive - [Optional] If True, check if fetch is available and it supports the keepalive feature, otherwise only check if fetch is supported * @returns True if supported, otherwise false */ +/*#__NO_SIDE_EFFECTS__*/ export function isFetchSupported(withKeepAlive?: boolean): boolean { let isSupported = false; try { @@ -242,21 +269,27 @@ export function isFetchSupported(withKeepAlive?: boolean): boolean { return isSupported; } +/*#__NO_SIDE_EFFECTS__*/ export function useXDomainRequest(): boolean | undefined { - if (_useXDomainRequest === null) { - _useXDomainRequest = (typeof XDomainRequest !== strShimUndefined); - if (_useXDomainRequest && isXhrSupported()) { - _useXDomainRequest = _useXDomainRequest && !_hasProperty(getInst(strXMLHttpRequest), "withCredentials"); - } + if (!_useXDomainRequest) { + _useXDomainRequest = getDeferred(() => { + let useXDomainRequest = (typeof XDomainRequest !== strShimUndefined); + if (useXDomainRequest && isXhrSupported()) { + useXDomainRequest = useXDomainRequest && !_hasProperty(getInst(strXMLHttpRequest), "withCredentials"); + } + + return !!useXDomainRequest; + }); } - return _useXDomainRequest; + return _useXDomainRequest.v; } /** * Checks if XMLHttpRequest is supported * @returns True if supported, otherwise false */ +/*#__NO_SIDE_EFFECTS__*/ export function isXhrSupported(): boolean { let isSupported = false; try { @@ -269,50 +302,94 @@ export function isXhrSupported(): boolean { return isSupported; } - -function _getNamedValue(values: any, name: string) { +/*#__NO_SIDE_EFFECTS__*/ +function _getNamedValue(values: any, name: string): T[] { + let items: T[] = []; if (values) { - for (var i = 0; i < values.length; i++) { - var value = values[i] as any; + arrForEach(values, (value) => { if (value.name) { if(value.name === name) { - return value; + items.push(value); } } - } + }); } - return {}; + return items; } /** * Helper function to fetch the named meta-tag from the page. * @param name - The name of the meta-tag to find. */ +/*#__NO_SIDE_EFFECTS__*/ export function findMetaTag(name: string): any { + let tags = findMetaTags(name); + if (tags.length > 0) { + return tags[0]; + } + + return null; +} + +/** + * Helper function to fetch all named meta-tag from the page. + * @since 3.4.0 + * @param name - The name of the meta-tag to find. + * @returns - An array of meta-tag values. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function findMetaTags(name: string): string[] { + let tags: string[] = []; let doc = getDocument(); if (doc && name) { // Look for a meta-tag - return _getNamedValue(doc.querySelectorAll("meta"), name).content; + arrForEach(_getNamedValue(doc.querySelectorAll("meta"), name), (item) => { + tags.push(item.content); + }); } - return null; + return tags; } /** * Helper function to fetch the named server timing value from the page response (first navigation event). * @param name - The name of the server timing value to find. */ +/*#__NO_SIDE_EFFECTS__*/ export function findNamedServerTiming(name: string): any { let value: any; + let serverTimings = findNamedServerTimings(name); + if (serverTimings.length > 0) { + value = serverTimings[0]; + } + + return value; +} + +/** + * Helper function to fetch the named server timing value from the page response (first navigation event). + * @since 3.4.0 + * @param name - The name of the server timing value to find. + * @returns - An array of server timing values. + */ +/*#__NO_SIDE_EFFECTS__*/ +export function findNamedServerTimings(name: string): string[] { + let values: string[] = []; let perf = getPerformance(); - if (perf) { + if (perf && perf.getEntriesByType) { // Try looking for a server-timing header - let navPerf = perf.getEntriesByType("navigation") || []; - value = _getNamedValue((navPerf.length > 0 ? navPerf[0] : {} as any).serverTiming, name).description; + arrForEach(perf.getEntriesByType("navigation") || [], (navPerf: any) => { + arrForEach(_getNamedValue(navPerf.serverTiming, name), (value: any) => { + let desc = value.description; + if (!isNullOrUndefined(desc)) { + values.push(desc); + } + }); + }); } - return value; + return values; } // TODO: should reuse this method for analytics plugin @@ -325,6 +402,7 @@ export function dispatchEvent(target:EventTarget, evnt: Event | CustomEvent): bo } +/*#__NO_SIDE_EFFECTS__*/ export function createCustomDomEvent(eventName: string, details?: any): CustomEvent { let event: CustomEvent = null; let detail = {detail: details || null } as CustomEventInit; @@ -341,8 +419,6 @@ export function createCustomDomEvent(eventName: string, details?: any): CustomEv return event; } - - export function sendCustomEvent(evtName: string, cfg?: any, customDetails?: any): boolean { let global = getGlobal(); if (global && (global as any).CustomEvent) { @@ -361,6 +437,7 @@ export function sendCustomEvent(evtName: string, cfg?: any, customDetails?: any) * @param url - The URL string to redact * @returns The URL with user information redacted */ +/*#__NO_SIDE_EFFECTS__*/ function redactUserInfo(url: string): string { return url.replace(/^([a-zA-Z][a-zA-Z0-9+.-]*:\/\/)([^:@]{1,200}):([^@]{1,200})@(.*)$/, "$1REDACTED:REDACTED@$4"); } @@ -370,6 +447,7 @@ function redactUserInfo(url: string): string { * @param url - The URL string to redact * @returns The URL with sensitive query parameters redacted */ +/*#__NO_SIDE_EFFECTS__*/ function redactQueryParameters(url: string, config?: IConfiguration): string { let sensitiveParams: string[]; const questionMarkIndex = strIndexOf(url, "?"); @@ -460,8 +538,9 @@ function redactQueryParameters(url: string, config?: IConfiguration): string { * @param config - Configuration object that contain redactUrls setting. * @returns The redacted URL string or the original string if no redaction was needed or possible. */ +/*#__NO_SIDE_EFFECTS__*/ export function fieldRedaction(input: string, config: IConfiguration): string { - if (!input || !isString(input) || input.indexOf(" ") !== -1) { + if (!input || !isString(input) || strIndexOf(input, " ") !== -1) { return input; } const isRedactionDisabled = config && config.redactUrls === false; diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/HelperFuncs.ts b/shared/AppInsightsCore/src/utils/HelperFuncs.ts similarity index 71% rename from shared/AppInsightsCore/src/JavaScriptSDK/HelperFuncs.ts rename to shared/AppInsightsCore/src/utils/HelperFuncs.ts index 2d9ca9362..f28c6aa2c 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/HelperFuncs.ts +++ b/shared/AppInsightsCore/src/utils/HelperFuncs.ts @@ -1,15 +1,17 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { ObjAssign, ObjClass } from "@microsoft/applicationinsights-shims"; +import { ObjAssign, ObjClass, ObjProto } from "@microsoft/applicationinsights-shims"; import { - arrForEach, asString as asString21, isArray, isBoolean, isError, isFunction, isNullOrUndefined, isNumber, isObject, isPlainObject, - isString, isUndefined, objDeepFreeze, objDefine, objForEachKey, objHasOwn, strIndexOf, strTrim + ICachedValue, WellKnownSymbols, arrForEach, asString as asString21, createCachedValue, getKnownSymbol, isArray, isBoolean, isError, + isFunction, isNullOrUndefined, isNumber, isObject, isPlainObject, isString, isUndefined, mathFloor, mathRound, newSymbol, objCreate, + objDeepFreeze, objDefine, objForEachKey, objGetPrototypeOf, objHasOwn, objSetPrototypeOf, safe, strIndexOf, strSplit, strTrim } from "@nevware21/ts-utils"; -import { FeatureOptInMode } from "../JavaScriptSDK.Enums/FeatureOptInEnums"; -import { TransportType } from "../JavaScriptSDK.Enums/SendRequestReason"; -import { IConfiguration } from "../JavaScriptSDK.Interfaces/IConfiguration"; -import { IXDomainRequest } from "../JavaScriptSDK.Interfaces/IXDomainRequest"; -import { STR_EMPTY } from "./InternalConstants"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { FeatureOptInMode } from "../enums/ai/FeatureOptInEnums"; +import { TransportType } from "../enums/ai/SendRequestReason"; +import { IConfiguration } from "../interfaces/ai/IConfiguration"; +import { IPlugin } from "../interfaces/ai/ITelemetryPlugin"; +import { IXDomainRequest } from "../interfaces/ai/IXDomainRequest"; // RESTRICT and AVOID circular dependencies you should not import other contained modules or export the contents of this file directly @@ -20,12 +22,16 @@ const rCamelCase = /-([a-z])/g; const rNormalizeInvalid = /([^\w\d_$])/g; const rLeadingNumeric = /^(\d+[\w\d_$])/; +let _ProtoNameTag: ICachedValue; + export let _getObjProto = Object[strGetPrototypeOf]; +/*#__NO_SIDE_EFFECTS__*/ export function isNotUndefined(value: T): value is T { return !isUndefined(value); } +/*#__NO_SIDE_EFFECTS__*/ export function isNotNullOrUndefined(value: T): value is T { return !isNullOrUndefined(value); } @@ -37,6 +43,7 @@ export function isNotNullOrUndefined(value: T): value is T { * This is a simplified version * @param name - The name to validate */ +/*#__NO_SIDE_EFFECTS__*/ export function normalizeJsName(name: string): string { let value = name; @@ -60,6 +67,7 @@ export function normalizeJsName(name: string): string { * @param value - The string value to check for the existence of the search value * @param search - The value search within the value */ +/*#__NO_SIDE_EFFECTS__*/ export function strContains(value: string, search: string): boolean { if (value && search) { return strIndexOf(value, search) !== -1; @@ -71,8 +79,9 @@ export function strContains(value: string, search: string): boolean { /** * Convert a date to I.S.O. format in IE8 */ +/*#__NO_SIDE_EFFECTS__*/ export function toISOString(date: Date) { - return date && date.toISOString() || ""; + return date && date.toISOString() || STR_EMPTY; } export const deepFreeze: (obj: T) => T = objDeepFreeze; @@ -80,6 +89,7 @@ export const deepFreeze: (obj: T) => T = objDeepFreeze; /** * Returns the name of object if it's an Error. Otherwise, returns empty string. */ +/*#__NO_SIDE_EFFECTS__*/ export function getExceptionName(object: any): string { if (isError(object)) { return object.name; @@ -246,6 +256,7 @@ export function proxyFunctions(target: T, source: S | (() => S), functions * Only instance properties (hasOwnProperty) values are copied from the defaults to the new instance * @param defaults - Simple helper */ +/*#__NO_SIDE_EFFECTS__*/ export function createClassFromInterface(defaults?: T) { return class { constructor() { @@ -258,6 +269,96 @@ export function createClassFromInterface(defaults?: T) { } as new () => T; } +/** + * Set the type of the object by defining the toStringTag well known symbol, this will impact the + * Object.prototype.toString.call() results for the object. And can be used to identify the type + * in the debug output and also in the DevTools watchers window when inspecting the object etc. + * @param target - The object to set the toStringTag symbol on + * @param nameOrFunc - The name or function to use for the toStringTag + * @returns The target object + */ +export function setObjStringTag(target: T, nameOrFunc: string | (() => string)): T { + safe(objDefine, [target, getKnownSymbol(WellKnownSymbols.toStringTag), isFunction(nameOrFunc) ? { g: nameOrFunc, e: false } : { v: nameOrFunc, w: false, e: false }]); + + return target; +} + +/** + * Introduce an intermediate prototype to the target object and that sets the toStringTag on that prototype, + * this avoids directly modifying the target object and also allows multiple different "types" to be + * applied to the same target object if required. + * This is done as a best effort approach and may not always succeed due to security / object model restrictions + * So if it fails then it will fallback to just defining the toStringTag on the target object, which also may fail + * resulting in no change. + * @param target - The object to set the toStringTag symbol on + * @param name - The name or function to use for the toStringTag + * @returns The target object + */ +export function setProtoTypeName(target: T, name: string): T { + if (target) { + let proto = _getObjProto(target); + let done = false; + if (proto) { + // Set the target's prototype to the new intermediate prototype + safe(() => { + // Create a new intermediate prototype that extends the current prototype + let newProto = setObjStringTag(objCreate(proto), name); + if (!_ProtoNameTag) { + // Note: Using a cached value instead of a lazy value as we want to ensure that the namespace is consistent + // across multiple calls as the `getLazy()` supports runtime invalidation via `setBypassLazyCache()` which would + // result in different namespaces being returned. + _ProtoNameTag = createCachedValue(newSymbol("ai$ProtoName")); + } + + // Tag this new prototype + objDefine(newProto, _ProtoNameTag.v as any, { + v: true, + w: false, + e: false + }); + + objSetPrototypeOf(target, newProto); + done = true; + }); + } + + if (!done) { + // Either no prototype or we failed to set it so just define the toStringTag on the target + safe(setObjStringTag, [target, name]); + } + } + + return target; +} + +/** + * Update the introduced intermediate prototype name of the target object. + * @param target - The object to look for the prototype name and update + * @param name - The updated name to apply + * @returns The target Object + */ +export function updateProtoTypeName(target: T, name: string): T { + if (_ProtoNameTag) { + // Find the Parent Proto + while (target && target !== ObjProto && !(target as any)[_ProtoNameTag.v]) { + let protoTarget = objGetPrototypeOf(target); + if (target === protoTarget) { + // Break out of any recursive case (happens on some runtimes) where the prototype of an + // object is the same prototype + break; + } + target = protoTarget; + } + + if ((target as any)[_ProtoNameTag.v]) { + // Found it so update + safe(setObjStringTag, [target, name]); + } + } + + return target; +} + /** * A helper function to assist with JIT performance for objects that have properties added / removed dynamically * this is primarily for chromium based browsers and has limited effects on Firefox and none of IE. Only call this @@ -265,6 +366,7 @@ export function createClassFromInterface(defaults?: T) { * This helps when iterating using for..in, objKeys() and objForEach() * @param theObject - The object to be optimized if possible */ +/*#__NO_SIDE_EFFECTS__*/ export function optimizeObject(theObject: T): T { // V8 Optimization to cause the JIT compiler to create a new optimized object for looking up the own properties // primarily for object with <= 19 properties for >= 20 the effect is reduced or non-existent @@ -360,6 +462,7 @@ export const asString = asString21; * @param sdkDefaultState - Optional default state to return if the feature is not defined * @returns True if the feature is enabled, false if the feature is disabled, or undefined if the feature is not defined and no default state is provided. */ +/*#__NO_SIDE_EFFECTS__*/ export function isFeatureEnabled(feature?: string, cfg?: T, sdkDefaultState?: boolean): boolean | undefined { let ft = cfg && cfg.featureOptIn && cfg.featureOptIn[feature]; if (feature && ft) { @@ -376,6 +479,7 @@ export function isFeatureEnabled(feat return sdkDefaultState; } +/*#__NO_SIDE_EFFECTS__*/ export function getResponseText(xhr: XMLHttpRequest | IXDomainRequest) { try { return xhr.responseText; @@ -386,22 +490,25 @@ export function getResponseText(xhr: XMLHttpRequest | IXDomainRequest) { return null; } +/*#__NO_SIDE_EFFECTS__*/ export function formatErrorMessageXdr(xdr: IXDomainRequest, message?: string): string { if (xdr) { - return "XDomainRequest,Response:" + getResponseText(xdr) || ""; + return "XDomainRequest,Response:" + getResponseText(xdr) || STR_EMPTY; } return message; } +/*#__NO_SIDE_EFFECTS__*/ export function formatErrorMessageXhr(xhr: XMLHttpRequest, message?: string): string { if (xhr) { - return "XMLHttpRequest,Status:" + xhr.status + ",Response:" + getResponseText(xhr) || xhr.response || ""; + return "XMLHttpRequest,Status:" + xhr.status + ",Response:" + getResponseText(xhr) || xhr.response || STR_EMPTY; } return message; } +/*#__NO_SIDE_EFFECTS__*/ export function prependTransports(theTransports: TransportType[], newTransports: TransportType | TransportType[]) { if (newTransports) { if (isNumber(newTransports)) { @@ -475,13 +582,14 @@ export function openXhr(method: string, urlString: string, withCredentials?: boo * @internal */ // tslint:disable-next-line: align +/*#__NO_SIDE_EFFECTS__*/ export function convertAllHeadersToMap(headersString: string): { [headerName: string]: string } { - let headers = {}; + let headers:any = {}; if (isString(headersString)) { let headersArray = strTrim(headersString).split(/[\r\n]+/); arrForEach(headersArray, (headerEntry) => { if (headerEntry) { - let idx = headerEntry.indexOf(": "); + let idx = strIndexOf(headerEntry, ": "); if (idx !== -1) { // The new spec has the headers returning all as lowercase -- but not all browsers do this yet let header = strTrim(headerEntry.substring(0, idx)).toLowerCase(); @@ -538,3 +646,94 @@ export function _getAllResponseHeaders(xhr: XMLHttpRequest, isOneDs?: boolean) { return theHeaders; } + +/*#__NO_SIDE_EFFECTS__*/ +export function stringToBoolOrDefault(str: any, defaultValue = false): boolean { + if (str === undefined || str === null) { + return defaultValue; + } + + return str.toString().toLowerCase() === "true"; +} + +/** + * Convert ms to c# time span format + */ +/*#__NO_SIDE_EFFECTS__*/ +export function msToTimeSpan(totalms: number | string): string { + if (isTimeSpan(totalms)) { + // Already in time span format + return totalms; + } + + if (isNaN(totalms) || totalms < 0) { + totalms = 0; + } + + totalms = mathRound(totalms); + + let ms = STR_EMPTY + totalms % 1000; + let sec = STR_EMPTY + mathFloor(totalms / 1000) % 60; + let min = STR_EMPTY + mathFloor(totalms / (1000 * 60)) % 60; + let hour = STR_EMPTY + mathFloor(totalms / (1000 * 60 * 60)) % 24; + const days = mathFloor(totalms / (1000 * 60 * 60 * 24)); + + ms = ms.length === 1 ? "00" + ms : ms.length === 2 ? "0" + ms : ms; + sec = sec.length < 2 ? "0" + sec : sec; + min = min.length < 2 ? "0" + min : min; + hour = hour.length < 2 ? "0" + hour : hour; + + return (days > 0 ? days + "." : STR_EMPTY) + hour + ":" + min + ":" + sec + "." + ms; +} + +/*#__NO_SIDE_EFFECTS__*/ +export function getExtensionByName(extensions: IPlugin[], identifier: string): IPlugin | null { + let extension: IPlugin = null; + arrForEach(extensions, (value) => { + if (value.identifier === identifier) { + extension = value; + return -1; + } + }); + + return extension; +} + +/*#__NO_SIDE_EFFECTS__*/ +export function isCrossOriginError(message: string|Event, url: string, lineNumber: number, columnNumber: number, error: Error | Event): boolean { + return !error && isString(message) && (message === "Script error." || message === "Script error"); +} + +/** + * A helper method to determine whether the provided value is in a ISO time span format (DD.HH:MM:SS.MMMMMM) + * @param value - The value to check + * @returns True if the value is in a time span format; false otherwise + */ +/*#__NO_SIDE_EFFECTS__*/ +export function isTimeSpan(value: any): value is string { + let result = false; + + if (isString(value)) { + const parts = strSplit(value, ":"); + if (parts.length === 3) { + // Looks like a candidate, now validate each part + const daysHours = strSplit(parts[0], "."); + if (daysHours.length === 2) { + result = !isNaN(parseInt(daysHours[0] || "0")) && !isNaN(parseInt(daysHours[1] || "0")); + } else { + result = !isNaN(parseInt(daysHours[0] || "0")); + } + + result = result && !isNaN(parseInt(parts[1] || "0")); + + const secondsParts = strSplit(parts[2], "."); + if (secondsParts.length === 2) { + result = result && !isNaN(parseInt(secondsParts[0] || "0")) && !isNaN(parseInt(secondsParts[1] || "0")); + } else { + result = result && !isNaN(parseInt(secondsParts[0] || "0")); + } + } + } + + return result; +} diff --git a/shared/AppInsightsCommon/src/Offline.ts b/shared/AppInsightsCore/src/utils/Offline.ts similarity index 92% rename from shared/AppInsightsCommon/src/Offline.ts rename to shared/AppInsightsCore/src/utils/Offline.ts index b52a61a6a..cdc90ce59 100644 --- a/shared/AppInsightsCommon/src/Offline.ts +++ b/shared/AppInsightsCore/src/utils/Offline.ts @@ -1,7 +1,10 @@ -import { - IUnloadHook, arrForEach, createUniqueNamespace, eventOff, eventOn, getDocument, getNavigator, getWindow, isNullOrUndefined, isUndefined, - mergeEvtNamespace -} from "@microsoft/applicationinsights-core-js"; +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { arrForEach, arrIndexOf, getDocument, getNavigator, getWindow, isNullOrUndefined, isUndefined } from "@nevware21/ts-utils"; +import { IUnloadHook } from "../interfaces/ai/IUnloadHook"; +import { eventOff, eventOn, mergeEvtNamespace } from "../internal/EventHelpers"; +import { createUniqueNamespace } from "./DataCacheHelper"; /** * this is the callback that will be called when the network status changes @@ -164,7 +167,7 @@ export function createOfflineListener(parentEvtNamespace?: string | string[]): I // Define rm as an instance of IUnloadHook return { rm: () => { - let index = listenerList.indexOf(callback); + let index = arrIndexOf(listenerList, callback); if (index > -1){ return listenerList.splice(index, 1); } else { diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/RandomHelper.ts b/shared/AppInsightsCore/src/utils/RandomHelper.ts similarity index 94% rename from shared/AppInsightsCore/src/JavaScriptSDK/RandomHelper.ts rename to shared/AppInsightsCore/src/utils/RandomHelper.ts index 4661d4547..8a0124117 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/RandomHelper.ts +++ b/shared/AppInsightsCore/src/utils/RandomHelper.ts @@ -1,13 +1,13 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { mathFloor, utcNow } from "@nevware21/ts-utils"; +import { mathFloor, mathRandom, utcNow } from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../constants/InternalConstants"; import { getCrypto, getMsCrypto, isIE } from "./EnvUtils"; -import { STR_EMPTY } from "./InternalConstants"; const UInt32Mask = 0x100000000; const MaxUInt32 = 0xffffffff; -const SEED1 = 123456789 -const SEED2 = 987654321 +const SEED1 = 123456789; +const SEED2 = 987654321; // MWC based Random generator (for IE) let _mwcSeeded = false; @@ -31,7 +31,7 @@ function _autoSeedMwc() { // and bitwise XOR with the current milliseconds try { const now = utcNow() & 0x7fffffff; - _mwcSeed(((Math.random() * UInt32Mask) ^ now) + now); + _mwcSeed(((mathRandom() * UInt32Mask) ^ now) + now); } catch (e) { // Don't crash if something goes wrong } @@ -76,7 +76,7 @@ export function random32(signed?: boolean) { if (value === 0) { // Make sure the number is converted into the specified range (-0x80000000..0x7FFFFFFF) - value = mathFloor((UInt32Mask * Math.random()) | 0); + value = mathFloor((UInt32Mask * mathRandom()) | 0); } if (!signed) { diff --git a/shared/AppInsightsCommon/src/StorageHelperFuncs.ts b/shared/AppInsightsCore/src/utils/StorageHelperFuncs.ts similarity index 92% rename from shared/AppInsightsCommon/src/StorageHelperFuncs.ts rename to shared/AppInsightsCore/src/utils/StorageHelperFuncs.ts index 0399de91c..45380cba3 100644 --- a/shared/AppInsightsCommon/src/StorageHelperFuncs.ts +++ b/shared/AppInsightsCore/src/utils/StorageHelperFuncs.ts @@ -1,15 +1,17 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { - IDiagnosticLogger, _eInternalMessageId, _throwInternal, dumpObj, eLoggingSeverity, getExceptionName, getGlobal, getGlobalInst, - isNullOrUndefined, objForEachKey -} from "@microsoft/applicationinsights-core-js"; -import { StorageType } from "./Enums"; +import { dumpObj, getGlobal, getInst as getGlobalInst, isNullOrUndefined, objForEachKey } from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { _throwInternal } from "../diagnostics/DiagnosticLogger"; +import { StorageType } from "../enums/ai/Enums"; +import { _eInternalMessageId, eLoggingSeverity } from "../enums/ai/LoggingEnums"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { getExceptionName } from "./HelperFuncs"; let _canUseLocalStorage: boolean = undefined; let _canUseSessionStorage: boolean = undefined; -let _storagePrefix: string = ""; +let _storagePrefix: string = STR_EMPTY; /** * Gets the localStorage object if available @@ -71,7 +73,7 @@ export function utlDisableStorage() { } export function utlSetStoragePrefix(storagePrefix: string) { - _storagePrefix = storagePrefix || ""; + _storagePrefix = storagePrefix || STR_EMPTY; } /** diff --git a/shared/AppInsightsCore/src/JavaScriptSDK/W3cTraceParent.ts b/shared/AppInsightsCore/src/utils/TraceParent.ts similarity index 89% rename from shared/AppInsightsCore/src/JavaScriptSDK/W3cTraceParent.ts rename to shared/AppInsightsCore/src/utils/TraceParent.ts index 5faafaacc..3b0b27ec4 100644 --- a/shared/AppInsightsCore/src/JavaScriptSDK/W3cTraceParent.ts +++ b/shared/AppInsightsCore/src/utils/TraceParent.ts @@ -1,15 +1,19 @@ -import { arrForEach, isArray, isString, strLeft, strTrim } from "@nevware21/ts-utils"; -import { ITraceParent } from "../JavaScriptSDK.Interfaces/ITraceParent"; +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +import { arrForEach, isArray, isNullOrUndefined, isString, strIndexOf, strLeft, strTrim } from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { eW3CTraceFlags } from "../enums/W3CTraceFlags"; +import { ITraceParent } from "../interfaces/ai/ITraceParent"; import { generateW3CId } from "./CoreUtils"; import { findMetaTag, findNamedServerTiming } from "./EnvUtils"; -import { STR_EMPTY } from "./InternalConstants"; // using {0,16} for leading and trailing whitespace just to constrain the possible runtime of a random string const TRACE_PARENT_REGEX = /^([\da-f]{2})-([\da-f]{32})-([\da-f]{16})-([\da-f]{2})(-[^\s]{1,64})?$/i; const DEFAULT_VERSION = "00"; const INVALID_VERSION = "ff"; -const INVALID_TRACE_ID = "00000000000000000000000000000000"; -const INVALID_SPAN_ID = "0000000000000000"; +export const INVALID_TRACE_ID = "00000000000000000000000000000000"; +export const INVALID_SPAN_ID = "0000000000000000"; const SAMPLED_FLAG = 0x01; function _isValid(value: string, len: number, invalidValue?: string): boolean { @@ -49,13 +53,13 @@ function _formatFlags(value: number): string { * @param version - The version to used, defaults to version "01" if not supplied or invalid. * @returns */ +/*#__NO_SIDE_EFFECTS__*/ export function createTraceParent(traceId?: string, spanId?: string, flags?: number, version?: string): ITraceParent { - return { version: _isValid(version, 2, INVALID_VERSION) ? version : DEFAULT_VERSION, traceId: isValidTraceId(traceId) ? traceId : generateW3CId(), spanId: isValidSpanId(spanId) ? spanId : strLeft(generateW3CId(), 16), - traceFlags: flags >= 0 && flags <= 0xFF ? flags : 1 + traceFlags: (!isNullOrUndefined(flags) && flags >= 0 && flags <= 0xFF ? flags : eW3CTraceFlags.Sampled) }; } @@ -66,6 +70,7 @@ export function createTraceParent(traceId?: string, spanId?: string, flags?: num * @param selectIdx - If the found value is comma separated which is the preferred entry to select, defaults to the first * @returns */ +/*#__NO_SIDE_EFFECTS__*/ export function parseTraceParent(value: string, selectIdx?: number): ITraceParent { if (!value) { // Don't pass a null/undefined or empty string @@ -74,7 +79,7 @@ export function parseTraceParent(value: string, selectIdx?: number): ITraceParen if (isArray(value)) { // The value may have been encoded on the page into an array so handle this automatically - value = value[0] || ""; + value = value[0] || STR_EMPTY; } if (!value || !isString(value) || value.length > 8192) { @@ -82,12 +87,13 @@ export function parseTraceParent(value: string, selectIdx?: number): ITraceParen return null; } - if (value.indexOf(",") !== -1) { + if (strIndexOf(value, ",") !== -1) { let values = value.split(","); value = values[selectIdx > 0 && values.length > selectIdx ? selectIdx : 0]; } // See https://www.w3.org/TR/trace-context/#versioning-of-traceparent + TRACE_PARENT_REGEX.lastIndex = 0; const match = TRACE_PARENT_REGEX.exec(strTrim(value)); if (!match || // No match match[1] === INVALID_VERSION || // version ff is forbidden @@ -101,7 +107,7 @@ export function parseTraceParent(value: string, selectIdx?: number): ITraceParen traceId: (match[2] || STR_EMPTY).toLowerCase(), spanId: (match[3] || STR_EMPTY).toLowerCase(), traceFlags: parseInt(match[4], 16) - } + }; } /** @@ -111,6 +117,7 @@ export function parseTraceParent(value: string, selectIdx?: number): ITraceParen * @param value - The W3c trace Id to be validated * @returns true if valid otherwise false */ +/*#__NO_SIDE_EFFECTS__*/ export function isValidTraceId(value: string): boolean { return _isValid(value, 32, INVALID_TRACE_ID); } @@ -122,6 +129,7 @@ export function isValidTraceId(value: string): boolean { * @param value - The W3c span id to be validated * @returns true if valid otherwise false */ +/*#__NO_SIDE_EFFECTS__*/ export function isValidSpanId(value: string): boolean { return _isValid(value, 16, INVALID_SPAN_ID); } @@ -131,6 +139,7 @@ export function isValidSpanId(value: string): boolean { * @param value - The parsed traceParent value * @returns */ +/*#__NO_SIDE_EFFECTS__*/ export function isValidTraceParent(value: ITraceParent) { if (!value || !_isValid(value.version, 2, INVALID_VERSION) || @@ -150,6 +159,7 @@ export function isValidTraceParent(value: ITraceParent) { * @param value - The parsed traceParent value * @returns */ +/*#__NO_SIDE_EFFECTS__*/ export function isSampledFlag(value: ITraceParent) { if (isValidTraceParent(value)) { return (value.traceFlags & SAMPLED_FLAG) === SAMPLED_FLAG; @@ -166,6 +176,7 @@ export function isSampledFlag(value: ITraceParent) { * @param value - The parsed traceParent value * @returns */ +/*#__NO_SIDE_EFFECTS__*/ export function formatTraceParent(value: ITraceParent) { if (value) { // Special Note: This only supports formatting as version 00, future versions should encode any known supported version @@ -185,7 +196,7 @@ export function formatTraceParent(value: ITraceParent) { return `${version.toLowerCase()}-${_formatValue(value.traceId, 32, INVALID_TRACE_ID).toLowerCase()}-${_formatValue(value.spanId, 16, INVALID_SPAN_ID).toLowerCase()}-${flags.toLowerCase()}`; } - return ""; + return STR_EMPTY; } /** @@ -216,7 +227,7 @@ export interface scriptsInfo { * @param doc - The document to search for script tags * @returns */ -export function findAllScripts(doc: any) { +export function findAllScripts(doc: any): scriptsInfo[] { let scripts = doc.getElementsByTagName("script"); let result: scriptsInfo[] = []; arrForEach(scripts, (script: any) => { @@ -242,5 +253,6 @@ export function findAllScripts(doc: any) { result.push(info); } }); + return result; } diff --git a/shared/AppInsightsCommon/src/UrlHelperFuncs.ts b/shared/AppInsightsCore/src/utils/UrlHelperFuncs.ts similarity index 85% rename from shared/AppInsightsCommon/src/UrlHelperFuncs.ts rename to shared/AppInsightsCore/src/utils/UrlHelperFuncs.ts index c0b9e838c..e76ef62e1 100644 --- a/shared/AppInsightsCommon/src/UrlHelperFuncs.ts +++ b/shared/AppInsightsCore/src/utils/UrlHelperFuncs.ts @@ -1,7 +1,8 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { getDocument, isString } from "@microsoft/applicationinsights-core-js"; +import { getDocument, isString } from "@nevware21/ts-utils"; +import { STR_EMPTY } from "../constants/InternalConstants"; let _document: any = getDocument() || {}; @@ -65,11 +66,11 @@ export function urlGetCompleteUrl(method: string, absoluteUrl: string) { // Fallback method to grab host from url if document.createElement method is not available export function urlParseHost(url: string, inclPort?: boolean) { - let fullHost = urlParseFullHost(url, inclPort) || ""; + let fullHost = urlParseFullHost(url, inclPort) || STR_EMPTY; if (fullHost) { const match = fullHost.match(/(www\d{0,5}\.)?([^\/:]{1,256})(:\d{1,20})?/i); if (match != null && match.length > 3 && isString(match[2]) && match[2].length > 0) { - return match[2] + (match[3] || ""); + return match[2] + (match[3] || STR_EMPTY); } } @@ -81,15 +82,15 @@ export function urlParseFullHost(url: string, inclPort?: boolean) { if (url) { const match = url.match(/(\w{1,150}):\/\/([^\/:]{1,256})(:\d{1,20})?/i); if (match != null && match.length > 2 && isString(match[2]) && match[2].length > 0) { - result = match[2] || ""; + result = match[2] || STR_EMPTY; if (inclPort && match.length > 2) { - const protocol = (match[1] || "").toLowerCase(); - let port = match[3] || ""; + const protocol = (match[1] || STR_EMPTY).toLowerCase(); + let port = match[3] || STR_EMPTY; // IE includes the standard port so pass it off if it's the same as the protocol if (protocol === "http" && port === ":80") { - port = ""; + port = STR_EMPTY; } else if (protocol === "https" && port === ":443") { - port = ""; + port = STR_EMPTY; } result += port; diff --git a/shared/AppInsightsCommon/src/Util.ts b/shared/AppInsightsCore/src/utils/Util.ts similarity index 79% rename from shared/AppInsightsCommon/src/Util.ts rename to shared/AppInsightsCore/src/utils/Util.ts index 2d8a76ed5..02fa5a641 100644 --- a/shared/AppInsightsCommon/src/Util.ts +++ b/shared/AppInsightsCore/src/utils/Util.ts @@ -1,16 +1,16 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { - IDiagnosticLogger, IDistributedTraceContext, arrForEach, arrIndexOf, dateNow, getPerformance, isNullOrUndefined, isValidSpanId, - isValidTraceId -} from "@microsoft/applicationinsights-core-js"; -import { strIndexOf } from "@nevware21/ts-utils"; -import { DEFAULT_BREEZE_ENDPOINT, DEFAULT_BREEZE_PATH } from "./Constants"; -import { ITelemetryTrace } from "./Interfaces/Context/ITelemetryTrace"; -import { ICorrelationConfig } from "./Interfaces/ICorrelationConfig"; -import { RequestHeaders, eRequestHeaders } from "./RequestResponseHeaders"; -import { dataSanitizeString } from "./Telemetry/Common/DataSanitizer"; +import { arrForEach, arrIndexOf, getPerformance, isNullOrUndefined, strIndexOf, utcNow as dateNow } from "@nevware21/ts-utils"; +import { DEFAULT_BREEZE_ENDPOINT, DEFAULT_BREEZE_PATH } from "../constants/Constants"; +import { STR_EMPTY } from "../constants/InternalConstants"; +import { createDistributedTraceContext } from "../core/TelemetryHelpers"; +import { ICorrelationConfig } from "../interfaces/ai/ICorrelationConfig"; +import { IDiagnosticLogger } from "../interfaces/ai/IDiagnosticLogger"; +import { IDistributedTraceContext } from "../interfaces/ai/IDistributedTraceContext"; +import { ITelemetryTrace } from "../interfaces/ai/context/ITelemetryTrace"; +import { RequestHeaders, eRequestHeaders } from "../telemetry/RequestResponseHeaders"; +import { dataSanitizeString } from "../telemetry/ai/Common/DataSanitizer"; import { urlParseFullHost, urlParseUrl } from "./UrlHelperFuncs"; // listing only non-geo specific locations @@ -180,42 +180,17 @@ export function dateTimeUtilsDuration(start: number, end: number): number { * @param telemetryTrace - The telemetryTrace instance that is being wrapped * @param parentCtx - An optional parent distributed trace instance, almost always undefined as this scenario is only used in the case of multiple property handlers. * @returns A new IDistributedTraceContext instance that is backed by the telemetryTrace or temporary object + * @deprecated This function is deprecated and will be removed in a future version. Use the createDistributedTraceContext function instead and set the necessary properties + * on the context object directly. */ export function createDistributedTraceContextFromTrace(telemetryTrace?: ITelemetryTrace, parentCtx?: IDistributedTraceContext): IDistributedTraceContext { - let trace: ITelemetryTrace = telemetryTrace || {}; + let traceCtx: IDistributedTraceContext = createDistributedTraceContext(parentCtx); + if (telemetryTrace) { + traceCtx.pageName = telemetryTrace.name || traceCtx.pageName || STR_EMPTY; // The name of the page + traceCtx.traceId = telemetryTrace.traceID || traceCtx.traceId || STR_EMPTY; // 16 byte hex string + traceCtx.spanId = telemetryTrace.parentID || traceCtx.spanId || STR_EMPTY; // 8 byte hex string + traceCtx.traceFlags = (!isNullOrUndefined(telemetryTrace.traceFlags) ? telemetryTrace.traceFlags : traceCtx.traceFlags) || 0; // 1 byte hex string + } - return { - getName: (): string => { - return trace.name; - }, - setName: (newValue: string): void => { - parentCtx && parentCtx.setName(newValue); - trace.name = newValue; - }, - getTraceId: (): string => { - return trace.traceID; - }, - setTraceId: (newValue: string): void => { - parentCtx && parentCtx.setTraceId(newValue); - if (isValidTraceId(newValue)) { - trace.traceID = newValue - } - }, - getSpanId: (): string => { - return trace.parentID; - }, - setSpanId: (newValue: string): void => { - parentCtx && parentCtx.setSpanId(newValue); - if (isValidSpanId(newValue)) { - trace.parentID = newValue - } - }, - getTraceFlags: (): number => { - return trace.traceFlags; - }, - setTraceFlags: (newTraceFlags?: number): void => { - parentCtx && parentCtx.setTraceFlags(newTraceFlags); - trace.traceFlags = newTraceFlags - } - }; + return traceCtx } diff --git a/shared/AppInsightsCore/tsconfig.json b/shared/AppInsightsCore/tsconfig.json index fa0e78410..abfefa5e0 100644 --- a/shared/AppInsightsCore/tsconfig.json +++ b/shared/AppInsightsCore/tsconfig.json @@ -21,11 +21,9 @@ "removeComments": false }, "include": [ - "./src/applicationinsights-core-js.ts", - "./src/Config/**/*.ts", - "./src/JavaScriptSDK/**/*.ts", - "./src/JavaScriptSDK.Enums/**/*.ts", - "./src/JavaScriptSDK.Interfaces/**/*.ts" + "./src/**/*.ts" ], - "exclude": ["node_modules/"] + "exclude": [ + "node_modules/" + ] } diff --git a/shared/AppInsightsCore/typedoc.json b/shared/AppInsightsCore/typedoc.json index 843eccf94..ffdff254e 100644 --- a/shared/AppInsightsCore/typedoc.json +++ b/shared/AppInsightsCore/typedoc.json @@ -1,6 +1,6 @@ { "$schema": "https://typedoc.org/schema.json", - "entryPoints": [ "./src/applicationinsights-core-js.ts" ], + "entryPoints": [ "./src/index.ts" ], "exclude": [ "**/internal/**/*.ts", "node_modules/**" ], "externalPattern": [ "**/node_modules/**", diff --git a/tools/applicationinsights-web-snippet/package.json b/tools/applicationinsights-web-snippet/package.json index c31638a8e..b7972ce06 100644 --- a/tools/applicationinsights-web-snippet/package.json +++ b/tools/applicationinsights-web-snippet/package.json @@ -35,7 +35,7 @@ "@types/qunit": "^2.19.3", "@microsoft/ai-test-framework": "0.0.1", "@microsoft/applicationinsights-web": "3.3.11", - "@microsoft/applicationinsights-common": "3.3.11", + "@microsoft/applicationinsights-core-js": "3.3.11", "@microsoft/applicationinsights-rollup-plugin-uglify3-js": "1.0.0", "@microsoft/applicationinsights-rollup-es5": "1.0.2", "@microsoft/dynamicproto-js": "^2.0.3", diff --git a/tools/applicationinsights-web-snippet/src/aiSupport.ts b/tools/applicationinsights-web-snippet/src/aiSupport.ts index b43e30fb8..71bb811a9 100644 --- a/tools/applicationinsights-web-snippet/src/aiSupport.ts +++ b/tools/applicationinsights-web-snippet/src/aiSupport.ts @@ -1,6 +1,6 @@ // Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT License. -import { IEnvelope } from "@microsoft/applicationinsights-common"; +import { IEnvelope } from "@microsoft/applicationinsights-core-js"; import { _getTime } from "./support"; let strEmpty = ""; diff --git a/tools/applicationinsights-web-snippet/src/snippet.ts b/tools/applicationinsights-web-snippet/src/snippet.ts index 5fa812f17..059e8bff0 100644 --- a/tools/applicationinsights-web-snippet/src/snippet.ts +++ b/tools/applicationinsights-web-snippet/src/snippet.ts @@ -3,7 +3,7 @@ /* eslint-disable no-constant-condition */ import { Fields, ISnippetConfig } from "./type"; -import { IConfig, IEnvelope } from "@microsoft/applicationinsights-common"; +import { IConfig, IEnvelope } from "@microsoft/applicationinsights-core-js"; import { IConfiguration, Snippet } from "@microsoft/applicationinsights-web"; import { oneDsEnvelope } from "./1dsType"; import { _createAiEnvelope, aiMethod } from "./aiSupport"; diff --git a/tools/applicationinsights-web-snippet/src/type.ts b/tools/applicationinsights-web-snippet/src/type.ts index 20f7adc7d..2b36ef123 100644 --- a/tools/applicationinsights-web-snippet/src/type.ts +++ b/tools/applicationinsights-web-snippet/src/type.ts @@ -50,7 +50,7 @@ export interface ISnippetConfig { */ ttp?: TrustedTypePolicy; /** - * Custom optional value to specify the nounce tag value that would be applied on the script when we drop it on the page + * Custom optional value to specify the nonce tag value that would be applied on the script when we drop it on the page */ nt?: string; } diff --git a/tools/chrome-debug-extension/package.json b/tools/chrome-debug-extension/package.json index 44cc76f31..360dbfa47 100644 --- a/tools/chrome-debug-extension/package.json +++ b/tools/chrome-debug-extension/package.json @@ -44,7 +44,6 @@ "sideEffects": false, "dependencies": { "@microsoft/applicationinsights-core-js": "3.3.11", - "@microsoft/applicationinsights-common": "3.3.11", "@microsoft/applicationinsights-shims": "3.0.1", "@microsoft/dynamicproto-js": "^2.0.3", "@nevware21/ts-async": ">= 0.5.5 < 2.x", diff --git a/tools/chrome-debug-extension/scripts/publishAzReleaseToCdn.ps1 b/tools/chrome-debug-extension/scripts/publishAzReleaseToCdn.ps1 index 9b96d2002..0bdc2dd93 100644 --- a/tools/chrome-debug-extension/scripts/publishAzReleaseToCdn.ps1 +++ b/tools/chrome-debug-extension/scripts/publishAzReleaseToCdn.ps1 @@ -127,7 +127,7 @@ elseif ($version.type -eq "dev" -or $version.type -eq "beta") { # Publish to release type folder folder PublishFiles $releaseFiles "$($version.type)/tools" $cacheControl1Year $contentType $overwrite } -elseif ($version.type -eq "nightly" -or $version.type -eq "nightly3") { +elseif ($version.type -like "nightly*") { # Publish to release nightly folder folder PublishFiles $releaseFiles "nightly/tools" $cacheControl1Year $contentType $overwrite } diff --git a/tools/chrome-debug-extension/scripts/setAzActiveCdnVersion.ps1 b/tools/chrome-debug-extension/scripts/setAzActiveCdnVersion.ps1 index f2412bbb4..91cc80821 100644 --- a/tools/chrome-debug-extension/scripts/setAzActiveCdnVersion.ps1 +++ b/tools/chrome-debug-extension/scripts/setAzActiveCdnVersion.ps1 @@ -87,7 +87,7 @@ Function Validate-Params Write-LogFailure "Container [$container] is not valid for version type [$($version.type)]" } } - elseif ($version.type -eq "nightly" -or $version.type -eq "nightly3") { + elseif ($version.type -like "nightly*") { if ("nightly" -ne $container) { Write-LogFailure "Container [$container] is not valid for version type [$($version.type)]" } diff --git a/tools/chrome-debug-extension/src/dataSources/dataSources.ts b/tools/chrome-debug-extension/src/dataSources/dataSources.ts index a637f98d0..71e56c746 100644 --- a/tools/chrome-debug-extension/src/dataSources/dataSources.ts +++ b/tools/chrome-debug-extension/src/dataSources/dataSources.ts @@ -9,7 +9,7 @@ import { NoOpDataSource } from "./noOpDataSource"; export function createDataSource(configuration: IConfiguration): IDataSource { // If on localhost, assume we are doing local testing (e.g. for accessibility issues) and use the NoOpDataSource - if (window.location.host.indexOf("localhost") === 0) { + if (window.location.host.indexOf("localhost:9001") === 0) { return new NoOpDataSource(); } diff --git a/tools/config/scripts/publishAzReleaseToCdn.ps1 b/tools/config/scripts/publishAzReleaseToCdn.ps1 index 99f0db08c..404dcf5b2 100644 --- a/tools/config/scripts/publishAzReleaseToCdn.ps1 +++ b/tools/config/scripts/publishAzReleaseToCdn.ps1 @@ -153,7 +153,7 @@ elseif ($version.type -eq "dev" -or $version.type -eq "beta") { # PublishFiles $releaseFiles "$($version.type)" $cacheControl1Year $contentType $overwrite PublishFiles $testFiles "$($version.type)" $cacheControl1Year $contentType $overwrite } -elseif ($version.type -eq "nightly" -or $version.type -eq "nightly3") { +elseif ($version.type -like "nightly*") { # Publish to release nightly folder folder # PublishFiles $releaseFiles "nightly" $cacheControl1Year $contentType $overwrite PublishFiles $testFiles "nightly" $cacheControl1Year $contentType $overwrite diff --git a/tools/config/scripts/setAzActiveCdnVersion.ps1 b/tools/config/scripts/setAzActiveCdnVersion.ps1 index 4e5684ddb..afdcdad40 100644 --- a/tools/config/scripts/setAzActiveCdnVersion.ps1 +++ b/tools/config/scripts/setAzActiveCdnVersion.ps1 @@ -87,7 +87,7 @@ Function Validate-Params Write-LogFailure "Container [$container] is not valid for version type [$($version.type)]" } } - elseif ($version.type -eq "nightly" -or $version.type -eq "nightly3") { + elseif ($version.type -like "nightly*") { if ("nightly" -ne $container) { Write-LogFailure "Container [$container] is not valid for version type [$($version.type)]" } diff --git a/tools/grunt-tasks/minifyNames.js b/tools/grunt-tasks/minifyNames.js index 021f6279c..cdbe96751 100644 --- a/tools/grunt-tasks/minifyNames.js +++ b/tools/grunt-tasks/minifyNames.js @@ -29,7 +29,7 @@ var defaultInternalConstants = [ var defaultIgnoreConst = [ ]; var defaultIgnoreNames = [ - "_e[A-Z]*", "e[A-Z]*", "[A-Z]*Type", "[A-Z]*Value", "[A-Z]*Reason", "chrome.*" + "_e[A-Z]*", "e[A-Z]*", "[A-Z]*Type", "[A-Z]*Value", "[A-Z]*Reason", "chrome.*", "WellKnownSymbols" ]; function _createEnumWildcardRegEx(value) { @@ -167,9 +167,9 @@ function readSourceValueMap(theReplacements, src, from) { } } - function readInternalConstants(theReplacements, src, from) { - const getConstStrings = /export\sconst\s(\w+)\s=\s(["\w \t\+]*as\s)?\"(\w*)";/gm + // eslint-disable-next-line security/detect-unsafe-regex + const getConstStrings = /export\sconst\s(\w+)\s=\s(?:\(?\/\*\s*[#@]__PURE__\s*\*\/\s*){0,1}(["\w \t\+]*as\s)?\"(\w*)\"\)?;/gm var matches = getConstStrings.exec(src); while (matches != null) { @@ -1217,7 +1217,8 @@ function minifyNamesFn(grunt) { } catch (e) { console.log(" ** Sleeping -- Error: " + e); retry ++; - child_process.execSync("sleep 1"); + // Cross-platform sleep: use timeout on Windows, sleep on Unix + child_process.execSync(process.platform === "win32" ? "timeout /t 1 /nobreak >nul" : "sleep 1", { shell: true, stdio: "ignore" }); } } } diff --git a/tools/grunt-tasks/qunit.js b/tools/grunt-tasks/qunit.js index b63916341..bd641d7b7 100644 --- a/tools/grunt-tasks/qunit.js +++ b/tools/grunt-tasks/qunit.js @@ -321,12 +321,54 @@ module.exports = function(grunt) { // The final tasks to run before terminating the task function finishTask(success) { - // Close the puppeteer browser - if (browser) { - browser.close(); + function closeBrowserWithTimeout(timeoutMs) { + return new Promise(function(resolve) { + var resolved = false; + + // Set a timeout to force resolve if browser.close() hangs + var timeout = setTimeout(function() { + if (!resolved) { + resolved = true; + grunt.log.warn('Browser close timed out after ' + timeoutMs + 'ms, forcing completion'); + // Try to force kill the browser process + if (browser && browser.process()) { + try { + browser.process().kill('SIGKILL'); + } catch (e) { + // Ignore kill errors + } + } + resolve(); + } + }, timeoutMs); + + if (browser) { + browser.close().then(function() { + if (!resolved) { + resolved = true; + clearTimeout(timeout); + resolve(); + } + }).catch(function(e) { + if (!resolved) { + resolved = true; + clearTimeout(timeout); + grunt.log.error('Error closing browser: ' + e); + resolve(); + } + }); + } else { + resolved = true; + clearTimeout(timeout); + resolve(); + } + }); } - // Finish the task - done(success); + + // Close browser with 10 second timeout + closeBrowserWithTimeout(10000).then(function() { + done(success); + }); } function appendToUrls (queryParam, value) { diff --git a/tools/release-tools/setVersion.js b/tools/release-tools/setVersion.js index dde02a3f3..c750c1479 100644 --- a/tools/release-tools/setVersion.js +++ b/tools/release-tools/setVersion.js @@ -406,7 +406,7 @@ function updatePublishConfig(package, newVersion) { } // Set the publishing tag - if (details.type === "nightly" || details.type === "dev" || details.type === "beta" || details.type === "alpha") { + if (details.type === "nightly" || details.type === "nightlybeta" || details.type === "dev" || details.type === "beta" || details.type === "alpha") { console.log(` Type - [${details.type}] - ${majorVersion}`); package.publishConfig.tag = details.type + (majorVersion !== "0" ? majorVersion : ""); } else { diff --git a/tools/rollup-es5/Tests/UnitTests.html b/tools/rollup-es5/Tests/UnitTests.html index b1138d827..3e015a230 100644 --- a/tools/rollup-es5/Tests/UnitTests.html +++ b/tools/rollup-es5/Tests/UnitTests.html @@ -5,17 +5,16 @@ Tests for Es5 Rollup Plugin - - - - - + + + + - - - -
-
-
- - - \ No newline at end of file diff --git a/tools/rollup-es5/src/ImportCheck.ts b/tools/rollup-es5/src/ImportCheck.ts index 30cc6fecb..296fc705c 100644 --- a/tools/rollup-es5/src/ImportCheck.ts +++ b/tools/rollup-es5/src/ImportCheck.ts @@ -24,7 +24,10 @@ export function importCheck(options:IImportCheckRollupOptions = {}) { checkOptions.keywords.push({ funcNames: [ /(\w[\d\w]*)\[\1\.(\w[\w\d]*)\]/g ], errorMsg: "Incorrect usage of an indexed map lookup detected - [%funcName%] you should use the enum name value as the lookup not the map name -- eg. Name[eName.xxxx]", - errorTitle: "Incorrect usage of indexed map lookup" + errorTitle: "Incorrect usage of indexed map lookup", + ignoreIds: [ + "tslib.es6" // tslib.es6 library has a pre existence check before usage + ] }); diff --git a/tools/shims/Tests/UnitTests.html b/tools/shims/Tests/UnitTests.html index 96605b214..99d6eed35 100644 --- a/tools/shims/Tests/UnitTests.html +++ b/tools/shims/Tests/UnitTests.html @@ -22,7 +22,7 @@ modules.add("qunit"); loadCommonModules(modules, function() { - var testModule = modules.add("Tests/Unit/src/shims.tests", "./Unit/dist/shimstests.js") + var testModule = modules.add("Tests/Unit/src/shims.tests", "./Unit/dist/shimstests.js"); testModule.run = function (tests) { console && console.log("Starting tests"); QUnit.start(); diff --git a/tsdoc.json b/tsdoc.json index 47181a276..04892f4ca 100644 --- a/tsdoc.json +++ b/tsdoc.json @@ -21,6 +21,14 @@ "tagName": "@default", "syntaxKind": "block" }, + { + "tagName": "@defaultBehavior", + "syntaxKind": "block" + }, + { + "tagName": "@function", + "syntaxKind": "block" + }, { "tagName": "@property", "syntaxKind": "block" @@ -57,6 +65,8 @@ "@since": true, "@returns": true, "@default": true, + "@defaultBehavior": true, + "@function": true, "@property": true, "@type": true, "@export": true, diff --git a/version.json b/version.json index 3d6aa43c9..61db7beea 100644 --- a/version.json +++ b/version.json @@ -2,7 +2,7 @@ "description": "The release value identifies the base version that will be applied via the tools/release-tools/setVersion.js", "usage": "When creating a new release you should update this value directly or via the eg. 'npm run setVersion -- 3.2.0' or 'npm run setVersion -- -patch' or 'npm run setVersion -- -minor'", "release": "3.3.11", - "next": "patch", + "next": "minor", "pkgs": { "@microsoft/applicationinsights-web": { "package": "package.json", @@ -103,6 +103,10 @@ "@microsoft/applicationinsights-web-snippet": { "package": "tools/applicationinsights-web-snippet/package.json", "release": "1.2.4" + }, + "@microsoft/applicationinsights-example-startspan": { + "package": "examples/startSpan/package.json", + "release": "3.3.11" } } }