From ff2e5b87369db2e6dd970bf22e5773c695cf4494 Mon Sep 17 00:00:00 2001
From: Mister_Nebula <41904486+misternebula@users.noreply.github.com>
Date: Fri, 12 Feb 2021 13:29:01 +0000
Subject: [PATCH] add tests

---
 QSB.sln                             |  6 +++
 QSB/Events/QSBEvent.cs              |  4 ++
 QSB/Events/QSBEventManager.cs       |  5 ++
 QSB/QSB.csproj                      |  1 +
 QSB/Utility/UnitTestDetector.cs     | 17 +++++++
 QSBTests/EventTests.cs              | 41 ++++++++++++++++
 QSBTests/Properties/AssemblyInfo.cs | 20 ++++++++
 QSBTests/QSBTests.csproj            | 75 +++++++++++++++++++++++++++++
 QSBTests/UtilityTests.cs            | 13 +++++
 QSBTests/packages.config            |  5 ++
 10 files changed, 187 insertions(+)
 create mode 100644 QSB/Utility/UnitTestDetector.cs
 create mode 100644 QSBTests/EventTests.cs
 create mode 100644 QSBTests/Properties/AssemblyInfo.cs
 create mode 100644 QSBTests/QSBTests.csproj
 create mode 100644 QSBTests/UtilityTests.cs
 create mode 100644 QSBTests/packages.config

diff --git a/QSB.sln b/QSB.sln
index 95d71f00..51984e1c 100644
--- a/QSB.sln
+++ b/QSB.sln
@@ -7,6 +7,8 @@ Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "QSB", "QSB\QSB.csproj", "{1
 EndProject
 Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "QuantumUNET", "QuantumUNET\QuantumUNET.csproj", "{C8C53004-1508-4F86-A419-4292C188DC2A}"
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "QSBTests", "QSBTests\QSBTests.csproj", "{2FE8256C-0CB6-48F1-A4C0-5FA18A1846A3}"
+EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|Any CPU = Debug|Any CPU
@@ -21,6 +23,10 @@ Global
 		{C8C53004-1508-4F86-A419-4292C188DC2A}.Debug|Any CPU.Build.0 = Debug|Any CPU
 		{C8C53004-1508-4F86-A419-4292C188DC2A}.Release|Any CPU.ActiveCfg = Release|Any CPU
 		{C8C53004-1508-4F86-A419-4292C188DC2A}.Release|Any CPU.Build.0 = Release|Any CPU
+		{2FE8256C-0CB6-48F1-A4C0-5FA18A1846A3}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{2FE8256C-0CB6-48F1-A4C0-5FA18A1846A3}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{2FE8256C-0CB6-48F1-A4C0-5FA18A1846A3}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{2FE8256C-0CB6-48F1-A4C0-5FA18A1846A3}.Release|Any CPU.Build.0 = Release|Any CPU
 	EndGlobalSection
 	GlobalSection(SolutionProperties) = preSolution
 		HideSolutionNode = FALSE
diff --git a/QSB/Events/QSBEvent.cs b/QSB/Events/QSBEvent.cs
index b11b8415..0d9ef042 100644
--- a/QSB/Events/QSBEvent.cs
+++ b/QSB/Events/QSBEvent.cs
@@ -17,6 +17,10 @@ namespace QSB.Events
 
 		protected QSBEvent()
 		{
+			if (UnitTestDetector.IsInUnitTest)
+			{
+				return;
+			}
 			_eventHandler = new MessageHandler<T>(Type);
 			_eventHandler.OnClientReceiveMessage += message => OnReceive(false, message);
 			_eventHandler.OnServerReceiveMessage += message => OnReceive(true, message);
diff --git a/QSB/Events/QSBEventManager.cs b/QSB/Events/QSBEventManager.cs
index 1d7e7559..be487eee 100644
--- a/QSB/Events/QSBEventManager.cs
+++ b/QSB/Events/QSBEventManager.cs
@@ -64,6 +64,11 @@ namespace QSB.Events
 				new IdentifySignalEvent()
 			};
 
+			if (UnitTestDetector.IsInUnitTest)
+			{
+				return;
+			}
+
 			_eventList.ForEach(ev => ev.SetupListener());
 
 			Ready = true;
diff --git a/QSB/QSB.csproj b/QSB/QSB.csproj
index 08027e97..ee3539ca 100644
--- a/QSB/QSB.csproj
+++ b/QSB/QSB.csproj
@@ -222,6 +222,7 @@
     <Compile Include="Utility\Popcron.Gizmos\Drawers\PolygonDrawer.cs" />
     <Compile Include="Utility\Popcron.Gizmos\Drawers\SquareDrawer.cs" />
     <Compile Include="Utility\Tuple.cs" />
+    <Compile Include="Utility\UnitTestDetector.cs" />
     <Compile Include="WorldSync\Events\BoolWorldObjectMessage.cs" />
     <Compile Include="WorldSync\Events\WorldObjectMessage.cs" />
     <Compile Include="Tools\QSBFlashlight.cs" />
diff --git a/QSB/Utility/UnitTestDetector.cs b/QSB/Utility/UnitTestDetector.cs
new file mode 100644
index 00000000..439f3584
--- /dev/null
+++ b/QSB/Utility/UnitTestDetector.cs
@@ -0,0 +1,17 @@
+using System;
+using System.Linq;
+
+namespace QSB.Utility
+{
+    public static class UnitTestDetector
+    {
+        static UnitTestDetector()
+        {
+            var testAssemblyName = "Microsoft.VisualStudio.TestPlatform.TestFramework";
+            IsInUnitTest = AppDomain.CurrentDomain.GetAssemblies()
+                .Any(a => a.FullName.StartsWith(testAssemblyName));
+        }
+
+        public static bool IsInUnitTest { get; private set; }
+    }
+}
diff --git a/QSBTests/EventTests.cs b/QSBTests/EventTests.cs
new file mode 100644
index 00000000..bfb4d7c2
--- /dev/null
+++ b/QSBTests/EventTests.cs
@@ -0,0 +1,41 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Reflection;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using QSB.Events;
+
+namespace QSBTests
+{
+	[TestClass]
+	public class EventTests
+	{
+		[TestMethod]
+		public void CheckUnreferencedEvents()
+		{
+			var qsbAssembly = Assembly.Load("QSB");
+			var allEventTypes = qsbAssembly
+				.GetTypes()
+				.Where(x => typeof(IQSBEvent).IsAssignableFrom(x) && !x.IsInterface && !x.IsAbstract);
+
+			QSBEventManager.Init();
+			var eventInstances = (List<IQSBEvent>)typeof(QSBEventManager)
+				.GetField("_eventList", BindingFlags.NonPublic | BindingFlags.Static)
+				.GetValue(typeof(QSBEventManager));
+
+			var failedTypes = new List<Type>();
+			foreach (var type in allEventTypes)
+			{
+				if (!eventInstances.Any(x => x.GetType() == type))
+				{
+					failedTypes.Add(type);
+				}
+			}
+
+			if (failedTypes.Count > 0)
+			{
+				Assert.Fail(string.Join(", ", failedTypes.Select(x => x.Name)));
+			}
+		}
+	}
+}
diff --git a/QSBTests/Properties/AssemblyInfo.cs b/QSBTests/Properties/AssemblyInfo.cs
new file mode 100644
index 00000000..825ccfea
--- /dev/null
+++ b/QSBTests/Properties/AssemblyInfo.cs
@@ -0,0 +1,20 @@
+using System.Reflection;
+using System.Runtime.CompilerServices;
+using System.Runtime.InteropServices;
+
+[assembly: AssemblyTitle("QSBTests")]
+[assembly: AssemblyDescription("")]
+[assembly: AssemblyConfiguration("")]
+[assembly: AssemblyCompany("")]
+[assembly: AssemblyProduct("QSBTests")]
+[assembly: AssemblyCopyright("Copyright ©  2021")]
+[assembly: AssemblyTrademark("")]
+[assembly: AssemblyCulture("")]
+
+[assembly: ComVisible(false)]
+
+[assembly: Guid("2fe8256c-0cb6-48f1-a4c0-5fa18a1846a3")]
+
+// [assembly: AssemblyVersion("1.0.*")]
+[assembly: AssemblyVersion("1.0.0.0")]
+[assembly: AssemblyFileVersion("1.0.0.0")]
diff --git a/QSBTests/QSBTests.csproj b/QSBTests/QSBTests.csproj
new file mode 100644
index 00000000..4507a8a3
--- /dev/null
+++ b/QSBTests/QSBTests.csproj
@@ -0,0 +1,75 @@
+<?xml version="1.0" encoding="utf-8"?>
+<Project ToolsVersion="15.0" DefaultTargets="Build" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
+  <Import Project="..\packages\MSTest.TestAdapter.1.3.2\build\net45\MSTest.TestAdapter.props" Condition="Exists('..\packages\MSTest.TestAdapter.1.3.2\build\net45\MSTest.TestAdapter.props')" />
+  <Import Project="$(MSBuildExtensionsPath)\$(MSBuildToolsVersion)\Microsoft.Common.props" Condition="Exists('$(MSBuildExtensionsPath)\$(MSBuildToolsVersion)\Microsoft.Common.props')" />
+  <PropertyGroup>
+    <Configuration Condition=" '$(Configuration)' == '' ">Debug</Configuration>
+    <Platform Condition=" '$(Platform)' == '' ">AnyCPU</Platform>
+    <ProjectGuid>{2FE8256C-0CB6-48F1-A4C0-5FA18A1846A3}</ProjectGuid>
+    <OutputType>Library</OutputType>
+    <AppDesignerFolder>Properties</AppDesignerFolder>
+    <RootNamespace>QSBTests</RootNamespace>
+    <AssemblyName>QSBTests</AssemblyName>
+    <TargetFrameworkVersion>v4.7.2</TargetFrameworkVersion>
+    <FileAlignment>512</FileAlignment>
+    <ProjectTypeGuids>{3AC096D0-A1C2-E12C-1390-A8335801FDAB};{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}</ProjectTypeGuids>
+    <VisualStudioVersion Condition="'$(VisualStudioVersion)' == ''">15.0</VisualStudioVersion>
+    <VSToolsPath Condition="'$(VSToolsPath)' == ''">$(MSBuildExtensionsPath32)\Microsoft\VisualStudio\v$(VisualStudioVersion)</VSToolsPath>
+    <ReferencePath>$(ProgramFiles)\Common Files\microsoft shared\VSTT\$(VisualStudioVersion)\UITestExtensionPackages</ReferencePath>
+    <IsCodedUITest>False</IsCodedUITest>
+    <TestProjectType>UnitTest</TestProjectType>
+    <NuGetPackageImportStamp>
+    </NuGetPackageImportStamp>
+  </PropertyGroup>
+  <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug|AnyCPU' ">
+    <DebugSymbols>true</DebugSymbols>
+    <DebugType>full</DebugType>
+    <Optimize>false</Optimize>
+    <OutputPath>bin\Debug\</OutputPath>
+    <DefineConstants>DEBUG;TRACE</DefineConstants>
+    <ErrorReport>prompt</ErrorReport>
+    <WarningLevel>4</WarningLevel>
+  </PropertyGroup>
+  <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Release|AnyCPU' ">
+    <DebugType>pdbonly</DebugType>
+    <Optimize>true</Optimize>
+    <OutputPath>bin\Release\</OutputPath>
+    <DefineConstants>TRACE</DefineConstants>
+    <ErrorReport>prompt</ErrorReport>
+    <WarningLevel>4</WarningLevel>
+  </PropertyGroup>
+  <ItemGroup>
+    <Reference Include="Microsoft.VisualStudio.TestPlatform.TestFramework, Version=14.0.0.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a, processorArchitecture=MSIL">
+      <HintPath>..\packages\MSTest.TestFramework.1.3.2\lib\net45\Microsoft.VisualStudio.TestPlatform.TestFramework.dll</HintPath>
+    </Reference>
+    <Reference Include="Microsoft.VisualStudio.TestPlatform.TestFramework.Extensions, Version=14.0.0.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a, processorArchitecture=MSIL">
+      <HintPath>..\packages\MSTest.TestFramework.1.3.2\lib\net45\Microsoft.VisualStudio.TestPlatform.TestFramework.Extensions.dll</HintPath>
+    </Reference>
+    <Reference Include="System" />
+    <Reference Include="System.Core" />
+  </ItemGroup>
+  <ItemGroup>
+    <Compile Include="EventTests.cs" />
+    <Compile Include="Properties\AssemblyInfo.cs" />
+    <Compile Include="UtilityTests.cs" />
+  </ItemGroup>
+  <ItemGroup>
+    <None Include="packages.config" />
+  </ItemGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\QSB\QSB.csproj">
+      <Project>{1F00090A-C697-4C55-B401-192F3CFB9DC2}</Project>
+      <Name>QSB</Name>
+    </ProjectReference>
+  </ItemGroup>
+  <Import Project="$(VSToolsPath)\TeamTest\Microsoft.TestTools.targets" Condition="Exists('$(VSToolsPath)\TeamTest\Microsoft.TestTools.targets')" />
+  <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
+  <Target Name="EnsureNuGetPackageBuildImports" BeforeTargets="PrepareForBuild">
+    <PropertyGroup>
+      <ErrorText>This project references NuGet package(s) that are missing on this computer. Use NuGet Package Restore to download them.  For more information, see http://go.microsoft.com/fwlink/?LinkID=322105. The missing file is {0}.</ErrorText>
+    </PropertyGroup>
+    <Error Condition="!Exists('..\packages\MSTest.TestAdapter.1.3.2\build\net45\MSTest.TestAdapter.props')" Text="$([System.String]::Format('$(ErrorText)', '..\packages\MSTest.TestAdapter.1.3.2\build\net45\MSTest.TestAdapter.props'))" />
+    <Error Condition="!Exists('..\packages\MSTest.TestAdapter.1.3.2\build\net45\MSTest.TestAdapter.targets')" Text="$([System.String]::Format('$(ErrorText)', '..\packages\MSTest.TestAdapter.1.3.2\build\net45\MSTest.TestAdapter.targets'))" />
+  </Target>
+  <Import Project="..\packages\MSTest.TestAdapter.1.3.2\build\net45\MSTest.TestAdapter.targets" Condition="Exists('..\packages\MSTest.TestAdapter.1.3.2\build\net45\MSTest.TestAdapter.targets')" />
+</Project>
\ No newline at end of file
diff --git a/QSBTests/UtilityTests.cs b/QSBTests/UtilityTests.cs
new file mode 100644
index 00000000..8ed47ceb
--- /dev/null
+++ b/QSBTests/UtilityTests.cs
@@ -0,0 +1,13 @@
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using QSB.Utility;
+
+namespace QSBTests
+{
+	[TestClass]
+	public class UtilityTests
+	{
+		[TestMethod]
+		public void IsInUnitTest() 
+			=> Assert.IsTrue(UnitTestDetector.IsInUnitTest, "UnitTestDetector is not working.");
+	}
+}
diff --git a/QSBTests/packages.config b/QSBTests/packages.config
new file mode 100644
index 00000000..2f7c5a18
--- /dev/null
+++ b/QSBTests/packages.config
@@ -0,0 +1,5 @@
+<?xml version="1.0" encoding="utf-8"?>
+<packages>
+  <package id="MSTest.TestAdapter" version="1.3.2" targetFramework="net472" />
+  <package id="MSTest.TestFramework" version="1.3.2" targetFramework="net472" />
+</packages>
\ No newline at end of file