diff --git a/libraries/tests/AWS.Lambda.Powertools.BatchProcessing.Tests/Internal/BatchProcessingInternalTests.cs b/libraries/tests/AWS.Lambda.Powertools.BatchProcessing.Tests/Internal/BatchProcessingInternalTests.cs index daf430826..95e879c25 100644 --- a/libraries/tests/AWS.Lambda.Powertools.BatchProcessing.Tests/Internal/BatchProcessingInternalTests.cs +++ b/libraries/tests/AWS.Lambda.Powertools.BatchProcessing.Tests/Internal/BatchProcessingInternalTests.cs @@ -35,7 +35,7 @@ public void BatchProcessing_Set_Execution_Environment_Context_SQS() var sqsBatchProcessor = new SqsBatchProcessor(conf); // Assert - Assert.Equal($"{Constants.FeatureContextIdentifier}/BatchProcessing/{env.GetAssemblyVersion(this)}", + Assert.Contains($"{Constants.FeatureContextIdentifier}/BatchProcessing/", env.GetEnvironmentVariable("AWS_EXECUTION_ENV")); Assert.NotNull(sqsBatchProcessor); @@ -52,7 +52,7 @@ public void BatchProcessing_Set_Execution_Environment_Context_Kinesis() var KinesisEventBatchProcessor = new KinesisEventBatchProcessor(conf); // Assert - Assert.Equal($"{Constants.FeatureContextIdentifier}/BatchProcessing/{env.GetAssemblyVersion(this)}", + Assert.Contains($"{Constants.FeatureContextIdentifier}/BatchProcessing/", env.GetEnvironmentVariable("AWS_EXECUTION_ENV")); Assert.NotNull(KinesisEventBatchProcessor); @@ -69,7 +69,7 @@ public void BatchProcessing_Set_Execution_Environment_Context_DynamoDB() var dynamoDbStreamBatchProcessor = new DynamoDbStreamBatchProcessor(conf); // Assert - Assert.Equal($"{Constants.FeatureContextIdentifier}/BatchProcessing/{env.GetAssemblyVersion(this)}", + Assert.Contains($"{Constants.FeatureContextIdentifier}/BatchProcessing/", env.GetEnvironmentVariable("AWS_EXECUTION_ENV")); Assert.NotNull(dynamoDbStreamBatchProcessor); diff --git a/libraries/tests/AWS.Lambda.Powertools.Idempotency.Tests/Internal/IdempotentAspectTests.cs b/libraries/tests/AWS.Lambda.Powertools.Idempotency.Tests/Internal/IdempotentAspectTests.cs index 8cc7e7f99..7a2e02e88 100644 --- a/libraries/tests/AWS.Lambda.Powertools.Idempotency.Tests/Internal/IdempotentAspectTests.cs +++ b/libraries/tests/AWS.Lambda.Powertools.Idempotency.Tests/Internal/IdempotentAspectTests.cs @@ -272,7 +272,7 @@ public void Idempotency_Set_Execution_Environment_Context() var xRayRecorder = new Idempotency(conf); // Assert - Assert.Equal($"{Constants.FeatureContextIdentifier}/Idempotency/{env.GetAssemblyVersion(this)}", + Assert.Contains($"{Constants.FeatureContextIdentifier}/Idempotency/", env.GetEnvironmentVariable("AWS_EXECUTION_ENV")); Assert.NotNull(xRayRecorder); diff --git a/libraries/tests/AWS.Lambda.Powertools.Logging.Tests/PowertoolsLoggerTest.cs b/libraries/tests/AWS.Lambda.Powertools.Logging.Tests/PowertoolsLoggerTest.cs index 099e8a2a2..756e7234b 100644 --- a/libraries/tests/AWS.Lambda.Powertools.Logging.Tests/PowertoolsLoggerTest.cs +++ b/libraries/tests/AWS.Lambda.Powertools.Logging.Tests/PowertoolsLoggerTest.cs @@ -1369,7 +1369,7 @@ public void Log_Set_Execution_Environment_Context() logger.LogInformation("Test"); // Assert - Assert.Equal($"{Constants.FeatureContextIdentifier}/Logging/{env.GetAssemblyVersion(this)}", + Assert.Contains($"{Constants.FeatureContextIdentifier}/Logging/", env.GetEnvironmentVariable("AWS_EXECUTION_ENV")); } diff --git a/libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/MetricsTests.cs b/libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/MetricsTests.cs index a1c1ee515..b710236ab 100644 --- a/libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/MetricsTests.cs +++ b/libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/MetricsTests.cs @@ -24,7 +24,7 @@ public void Metrics_Set_Execution_Environment_Context() _ = new Metrics(conf); // Assert - Assert.Equal($"{Constants.FeatureContextIdentifier}/Metrics/{env.GetAssemblyVersion(this)}", + Assert.Contains($"{Constants.FeatureContextIdentifier}/Metrics/", env.GetEnvironmentVariable("AWS_EXECUTION_ENV")); } diff --git a/libraries/tests/AWS.Lambda.Powertools.Tracing.Tests/XRayRecorderTests.cs b/libraries/tests/AWS.Lambda.Powertools.Tracing.Tests/XRayRecorderTests.cs index a4e9e515c..2e3f1a80a 100644 --- a/libraries/tests/AWS.Lambda.Powertools.Tracing.Tests/XRayRecorderTests.cs +++ b/libraries/tests/AWS.Lambda.Powertools.Tracing.Tests/XRayRecorderTests.cs @@ -40,7 +40,7 @@ public void Tracing_Set_Execution_Environment_Context() var xRayRecorder = new XRayRecorder(awsXray, conf); // Assert - Assert.Equal($"{Constants.FeatureContextIdentifier}/Tracing/{env.GetAssemblyVersion(this)}", + Assert.Contains($"{Constants.FeatureContextIdentifier}/Tracing/", env.GetEnvironmentVariable("AWS_EXECUTION_ENV")); Assert.NotNull(xRayRecorder);