test: Fix flagd test shutdown (#404)

Signed-off-by: André Silva <[email protected]>

Author: askpt

test/OpenFeature.Contrib.Providers.Flagd.Test/FlagdProviderTest.cs

@@ -205,6 +205,8 @@ public async Task TestResolveBooleanValueAsync()
         var val = await flagdProvider.ResolveBooleanValueAsync("my-key", false);
 
         Assert.True(val.Value);
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -231,6 +233,8 @@ public async Task TestResolveStringValue()
         var val = await flagdProvider.ResolveStringValueAsync("my-key", "");
 
         Assert.Equal("my-value", val.Value);
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -258,6 +262,8 @@ public async Task TestResolveIntegerValue()
         var val = await flagdProvider.ResolveIntegerValueAsync("my-key", 0);
 
         Assert.Equal(10, val.Value);
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -285,6 +291,8 @@ public async Task TestResolveDoubleValue()
         var val = await flagdProvider.ResolveDoubleValueAsync("my-key", 0.0);
 
         Assert.Equal(10.0, val.Value);
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -315,6 +323,8 @@ public async Task TestResolveStructureValue()
         var val = await flagdProvider.ResolveStructureValueAsync("my-key", null);
 
         Assert.True(val.Value.AsStructure.ContainsKey("my-key"));
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -347,6 +357,8 @@ public async Task TestResolveFlagNotFound()
             Assert.Equal(ErrorType.FlagNotFound, ex.ErrorType);
             Assert.Equal(ErrorType.FlagNotFound.ToString(), ex.Message);
         });
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -380,6 +392,8 @@ public async Task TestResolveGrpcHostUnavailable()
             Assert.Equal(ErrorType.ProviderNotReady, ex.ErrorType);
             Assert.Equal(ErrorType.ProviderNotReady.ToString(), ex.Message);
         });
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -413,6 +427,8 @@ public async Task TestResolveTypeMismatch()
             Assert.Equal(ErrorType.TypeMismatch, ex.ErrorType);
             Assert.Equal(ErrorType.TypeMismatch.ToString(), ex.Message);
         });
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -446,6 +462,8 @@ public async Task TestResolveUnknownError()
             Assert.Equal(ErrorType.General, ex.ErrorType);
             Assert.Equal(ErrorType.General.ToString(), ex.Message);
         });
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -523,6 +541,8 @@ public async Task TestCacheAsync()
         mockCache.Received(1).TryGet(Arg.Is<string>(s => s == "my-key"));
         mockCache.Received(1).Add(Arg.Is<string>(s => s == "my-key"), Arg.Any<object>());
         mockGrpcClient.Received(Quantity.AtLeastOne()).EventStream(Arg.Any<EventStreamRequest>(), null, null, CancellationToken.None);
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -587,6 +607,8 @@ public async Task TestCacheHitAsync()
         Assert.True(_autoResetEvent.WaitOne(10000));
         mockCache.Received(1).TryGet("my-key");
         mockGrpcClient.Received(Quantity.AtLeastOne()).EventStream(Arg.Any<EventStreamRequest>(), null, null, CancellationToken.None);
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -691,6 +713,8 @@ public async Task TestCacheInvalidationAsync()
         mockCache.Received(2).Add("my-key", Arg.Any<object>());
         mockCache.Received().Delete("my-key");
         mockGrpcClient.Received(Quantity.AtLeastOne()).EventStream(Arg.Any<EventStreamRequest>(), null, null, CancellationToken.None);
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -749,6 +773,8 @@ await Utils.AssertUntilAsync(
             });
 
         mockGrpcClient.Received(Quantity.AtLeastOne()).SyncFlags(Arg.Is<SyncFlagsRequest>(req => req.Selector == "source-selector"), null, null, CancellationToken.None);
+
+        await flagdProvider.ShutdownAsync();
     }
 
     [Fact]
@@ -807,5 +833,7 @@ await Utils.AssertUntilAsync(
             });
 
         mockGrpcClient.Received(Quantity.AtLeastOne()).SyncFlags(Arg.Is<SyncFlagsRequest>(req => req.Selector == "source-selector"), null, null, CancellationToken.None);
+
+        await flagdProvider.ShutdownAsync();
     }
 }