Replaced the context.

Signed-off-by: André Silva <[email protected]>
open-feature · Jan 2, 2025 · 1769be8 · 1769be8
1 parent 7a60997
commit 1769be8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/test/OpenFeature.Tests/AsyncLocalTransactionContextPropagatorTests.cs b/test/OpenFeature.Tests/AsyncLocalTransactionContextPropagatorTests.cs
@@ -23,14 +23,17 @@ public void SetTransactionContext_SetsAndGetsContextCorrectly()
     {
         // Arrange
         var propagator = new AsyncLocalTransactionContextPropagator();
-        var evaluationContext = EvaluationContext.Empty;
+        var evaluationContext = EvaluationContext.Builder()
+            .Set("initial", "yes")
+            .Build();
 
         // Act
         propagator.SetTransactionContext(evaluationContext);
         var context = propagator.GetTransactionContext();
 
         // Assert
         Assert.Equal(evaluationContext, context);
+        Assert.Equal(evaluationContext.GetValue("initial"), context.GetValue("initial"));
     }
 
     [Fact]

diff --git a/test/OpenFeature.Tests/OpenFeatureTests.cs b/test/OpenFeature.Tests/OpenFeatureTests.cs
@@ -285,7 +285,9 @@ public void SetTransactionContext_ShouldSetTransactionContext_WhenValidEvaluatio
         {
             // Arrange
             var api = Api.Instance;
-            var evaluationContext = EvaluationContext.Empty;
+            var evaluationContext = EvaluationContext.Builder()
+                .Set("initial", "yes")
+                .Build();
             var mockPropagator = Substitute.For<ITransactionContextPropagator>();
             mockPropagator.GetTransactionContext().Returns(evaluationContext);
             api.SetTransactionContextPropagator(mockPropagator);
@@ -298,6 +300,7 @@ public void SetTransactionContext_ShouldSetTransactionContext_WhenValidEvaluatio
             // Assert
             mockPropagator.Received().SetTransactionContext(evaluationContext);
             Assert.Equal(evaluationContext, result);
+            Assert.Equal(evaluationContext.GetValue("initial"), result.GetValue("initial"));
         }
 
         [Fact]