From 9a3f81632c3536b8a2145823ac23e2f3cad60d9d Mon Sep 17 00:00:00 2001
From: MSWS <imodmaker@gmail.com>
Date: Sun, 1 Sep 2024 21:27:02 -0700
Subject: [PATCH] Add player output tests

---
 GangsAPI/Data/PlayerWrapper.cs      | 19 +++++++++++++------
 GangsTest/API/PlayerWrapperTests.cs | 16 ++++++++++++++++
 GangsTest/GangsTest.csproj          | 19 ++++++++++---------
 GangsTest/Startup.cs                |  6 +++++-
 4 files changed, 44 insertions(+), 16 deletions(-)

diff --git a/GangsAPI/Data/PlayerWrapper.cs b/GangsAPI/Data/PlayerWrapper.cs
index 4e45aa5..eb1a303 100644
--- a/GangsAPI/Data/PlayerWrapper.cs
+++ b/GangsAPI/Data/PlayerWrapper.cs
@@ -1,4 +1,5 @@
-using CounterStrikeSharp.API.Core;
+using System.Diagnostics;
+using CounterStrikeSharp.API.Core;
 using CounterStrikeSharp.API.Modules.Admin;
 
 namespace GangsAPI.Data;
@@ -8,6 +9,9 @@ public class PlayerWrapper {
   public readonly CCSPlayerController? Player;
   public readonly ulong Steam;
   public AdminData? Data;
+  private readonly List<string> chatOutput = [], consoleOutput = [];
+  public IReadOnlyList<string> ChatOutput => chatOutput;
+  public IReadOnlyList<string> ConsoleOutput => consoleOutput;
 
   public PlayerWrapper(CCSPlayerController player) {
     Player = player;
@@ -21,7 +25,8 @@ public PlayerWrapper(ulong steam, string? name) {
     Steam = steam;
     Name  = name;
 
-    Data = new AdminData { Identity = Steam.ToString() };
+    chatOutput = new();
+    Data       = new AdminData { Identity = Steam.ToString() };
   }
 
   private static char USER_CHAR => PermissionCharacters.UserPermissionChar;
@@ -99,19 +104,21 @@ public PlayerWrapper WithGroups(params string[] groups) {
 
   public void PrintToChat(string message) {
     if (Player == null) {
-      Console.WriteLine($"{Steam} {Player} received chat: {message}");
+      Debug.WriteLine($"{Steam} {Name} received chat: {message}");
+      chatOutput.Add(message);
       return;
     }
 
-    Player?.PrintToChat(message);
+    Player.PrintToChat(message);
   }
 
   public void PrintToConsole(string message) {
     if (Player == null) {
-      Console.WriteLine($"{Steam} {Player} received console: {message}");
+      Console.WriteLine($"{Steam} {Name} received chat: {message}");
+      consoleOutput.Add(message);
       return;
     }
 
-    Player?.PrintToConsole(message);
+    Player.PrintToConsole(message);
   }
 }
\ No newline at end of file
diff --git a/GangsTest/API/PlayerWrapperTests.cs b/GangsTest/API/PlayerWrapperTests.cs
index 4f737df..a694d70 100644
--- a/GangsTest/API/PlayerWrapperTests.cs
+++ b/GangsTest/API/PlayerWrapperTests.cs
@@ -26,6 +26,22 @@ public void PlayerWrapper_Init_Flags() {
     Assert.True(player.Data.Flags.ContainsKey("test"));
   }
 
+  [Fact]
+  public void PlayerWrapper_Print_Chat() {
+    Assert.Empty(testPlayer.ChatOutput);
+    testPlayer.PrintToChat("Test Message");
+    Assert.Single(testPlayer.ChatOutput);
+    Assert.Equal("Test Message", testPlayer.ChatOutput[0]);
+  }
+
+  [Fact]
+  public void PlayerWrapper_Print_Console() {
+    Assert.Empty(testPlayer.ConsoleOutput);
+    testPlayer.PrintToConsole("Test Message");
+    Assert.Single(testPlayer.ConsoleOutput);
+    Assert.Equal("Test Message", testPlayer.ConsoleOutput[0]);
+  }
+
   [Theory]
   [InlineData("test/flag")]
   [InlineData("test/flag/child")]
diff --git a/GangsTest/GangsTest.csproj b/GangsTest/GangsTest.csproj
index 7241a51..9b49c5b 100644
--- a/GangsTest/GangsTest.csproj
+++ b/GangsTest/GangsTest.csproj
@@ -10,15 +10,16 @@
     </PropertyGroup>
 
     <ItemGroup>
-        <PackageReference Include="coverlet.collector" Version="6.0.2"/>
-        <PackageReference Include="FuzzDotNet.MSTest" Version="0.1.0"/>
-        <PackageReference Include="Microsoft.Extensions.DependencyInjection.Abstractions" Version="8.0.1"/>
-        <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.11.0"/>
-        <PackageReference Include="Xunit.DependencyInjection" Version="9.3.1"/>
+        <PackageReference Include="coverlet.collector" Version="6.0.2" />
+        <PackageReference Include="FuzzDotNet.MSTest" Version="0.1.0" />
+        <PackageReference Include="Microsoft.Extensions.DependencyInjection.Abstractions" Version="8.0.1" />
+        <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.11.0" />
+        <PackageReference Include="Xunit.DependencyInjection" Version="9.3.1" />
+        <PackageReference Include="Xunit.DependencyInjection.Logging" Version="9.0.0" />
     </ItemGroup>
 
     <ItemGroup>
-        <Using Include="Xunit"/>
+        <Using Include="Xunit" />
     </ItemGroup>
 
     <ItemGroup>
@@ -28,9 +29,9 @@
     </ItemGroup>
 
     <ItemGroup>
-        <ProjectReference Include="..\GangsImpl\Mock\Mock.csproj"/>
-        <ProjectReference Include="..\GangsImpl\SQLite\SQLite.csproj"/>
-        <ProjectReference Include="..\GangsImpl\SQL\SQL.csproj"/>
+        <ProjectReference Include="..\GangsImpl\Mock\Mock.csproj" />
+        <ProjectReference Include="..\GangsImpl\SQLite\SQLite.csproj" />
+        <ProjectReference Include="..\GangsImpl\SQL\SQL.csproj" />
         <ProjectReference Include="..\Core\Core.csproj" />
     </ItemGroup>
 
diff --git a/GangsTest/Startup.cs b/GangsTest/Startup.cs
index 782009d..3ccbb30 100644
--- a/GangsTest/Startup.cs
+++ b/GangsTest/Startup.cs
@@ -1,7 +1,11 @@
-using GangsAPI.Services;
+using System.Text;
+using GangsAPI.Services;
 using GangsAPI.Services.Commands;
 using Microsoft.Extensions.DependencyInjection;
+using Microsoft.VisualBasic.CompilerServices;
 using Mock;
+using Xunit.Abstractions;
+using Xunit.DependencyInjection.Logging;
 
 namespace GangsTest;