Fix test

IEvangelist · Nov 26, 2023 · ec6bdd8 · ec6bdd8
1 parent 84cd743
commit ec6bdd8
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 9 deletions.
diff --git a/tests/ProfanityFilter.Action.Tests/GlobalUsings.cs b/tests/ProfanityFilter.Action.Tests/GlobalUsings.cs
@@ -3,6 +3,8 @@
 
 global using Microsoft.Extensions.DependencyInjection;
 
+global using Actions.Octokit.EnvironmentVariables;
+
 global using ProfanityFilter.Action.Extensions;
 
 global using ProfanityFilter.Services;

diff --git a/tests/ProfanityFilter.Action.Tests/ServiceCollectionExtensionsTests.cs b/tests/ProfanityFilter.Action.Tests/ServiceCollectionExtensionsTests.cs
@@ -8,17 +8,26 @@ public class ServiceCollectionExtensionsTests
     [Fact]
     public void AddProfanityFilter_AddsServices()
     {
-        // Arrange
-        var services = new ServiceCollection();
+        Environment.SetEnvironmentVariable(Keys.GITHUB_TOKEN, "TEST");
 
-        // Act
-        services.AddActionProcessorServices();
+        try
+        {
+            // Arrange
+            var services = new ServiceCollection();
 
-        // Assert
-        var provider = services.BuildServiceProvider();
-        var censorService = provider.GetService<IProfaneContentCensorService>();
+            // Act
+            services.AddActionProcessorServices();
 
-        Assert.NotNull(censorService);
-        Assert.IsType<DefaultProfaneContentCensorService>(censorService);
+            // Assert
+            var provider = services.BuildServiceProvider();
+            var censorService = provider.GetService<IProfaneContentCensorService>();
+
+            Assert.NotNull(censorService);
+            Assert.IsType<DefaultProfaneContentCensorService>(censorService);
+        }
+        finally
+        {
+            Environment.SetEnvironmentVariable(Keys.GITHUB_TOKEN, null);
+        }
     }
 }