From ec6bdd84da12576cd0b1a2994118772e844784b1 Mon Sep 17 00:00:00 2001
From: David Pine <david.pine.7@gmail.com>
Date: Sun, 26 Nov 2023 10:50:54 -0600
Subject: [PATCH] Fix test

---
 .../GlobalUsings.cs                           |  2 ++
 .../ServiceCollectionExtensionsTests.cs       | 27 ++++++++++++-------
 2 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/tests/ProfanityFilter.Action.Tests/GlobalUsings.cs b/tests/ProfanityFilter.Action.Tests/GlobalUsings.cs
index 3dffae4..ba371e4 100644
--- 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
index d4bb1c7..565eff4 100644
--- 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);
+        }
     }
 }