From 5eaedccdfaadc9f41c5040532c86e2626fd3ad20 Mon Sep 17 00:00:00 2001
From: Chomp <dev@dev.sp-tarkov.com>
Date: Mon, 21 Jul 2025 12:33:00 +0100
Subject: [PATCH] Added tests

---
 .../Tests/Helpers/BotGeneratorHelperTests.cs  | 32 +++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/UnitTests/Tests/Helpers/BotGeneratorHelperTests.cs b/UnitTests/Tests/Helpers/BotGeneratorHelperTests.cs
index 077b2043..f9d48252 100644
--- a/UnitTests/Tests/Helpers/BotGeneratorHelperTests.cs
+++ b/UnitTests/Tests/Helpers/BotGeneratorHelperTests.cs
@@ -24,6 +24,8 @@ namespace UnitTests.Tests.Helpers
             ;
         }
 
+        #region AddItemWithChildrenToEquipmentSlot
+
         [TestMethod]
         public void AddItemWithChildrenToEquipmentSlot_fit_vertical()
         {
@@ -461,5 +463,35 @@ namespace UnitTests.Tests.Helpers
 
             Assert.AreEqual(ItemAddedResult.NO_SPACE, result);
         }
+
+        #endregion
+
+        #region GetBotEquipmentRole
+
+        [TestMethod]
+        public void GetBotEquipmentRole_assault()
+        {
+            var result = _botGeneratorHelper.GetBotEquipmentRole("assault");
+
+            Assert.AreEqual("assault", result);
+        }
+
+        [TestMethod]
+        public void GetBotEquipmentRole_pmcBEAR()
+        {
+            var result = _botGeneratorHelper.GetBotEquipmentRole("pmcBEAR");
+
+            Assert.AreEqual("pmc", result);
+        }
+
+        [TestMethod]
+        public void GetBotEquipmentRole_pmcBEAR_lowercase()
+        {
+            var result = _botGeneratorHelper.GetBotEquipmentRole("pmcbear");
+
+            Assert.AreEqual("pmc", result);
+        }
+
+        #endregion
     }
 }