Added BotWeaponGeneratorTests

2025-07-21 17:44:14 +01:00
parent 2700273de1
commit c760cbb6bf
1 changed files with 75 additions and 0 deletions
@@ -0,0 +1,75 @@
+using SPTarkov.Server.Core.Generators;
+using SPTarkov.Server.Core.Helpers;
+using SPTarkov.Server.Core.Models.Common;
+using SPTarkov.Server.Core.Services;
+using SPTarkov.Server.Core.Utils;
+
+namespace UnitTests.Tests.Generators
+{
+    [TestClass]
+    public class BotWeaponGeneratorTests
+    {
+        private BotWeaponGenerator _botWeaponGenerator;
+        private DatabaseService _databaseService;
+        private InventoryHelper _inventoryHelper;
+
+        [TestInitialize]
+        public void Initialize()
+        {
+            _botWeaponGenerator = DI.GetService<BotWeaponGenerator>();
+            var databaseImporter = DI.GetService<DatabaseImporter>();
+            _databaseService = DI.GetService<DatabaseService>();
+            _inventoryHelper = DI.GetService<InventoryHelper>();
+            Task.Factory.StartNew(() =>
+            {
+                databaseImporter.OnLoad();
+            });
+        }
+
+        [TestMethod]
+        public void GenerateWeaponByTpl_generate_m4_pmc()
+        {
+            var usecTemplate = _databaseService.GetBots().Types["usec"];
+            var botTemplateInventory = usecTemplate.BotInventory;
+
+            var sessionId = new MongoId();
+            var weaponTpl = ItemTpl.ASSAULTRIFLE_COLT_M4A1_556X45_ASSAULT_RIFLE;
+            const string slotName = "FirstPrimaryWeapon";
+            var weaponModChances = usecTemplate.BotChances.WeaponModsChances;
+            foreach (var (key, _) in weaponModChances)
+            {
+                // Set all mods to 100%
+                weaponModChances[key] = 100d;
+            }
+
+            var weaponParentId = new MongoId();
+
+            for (var i = 0; i < 100; i++)
+            {
+                var result = _botWeaponGenerator.GenerateWeaponByTpl(
+                    sessionId,
+                    weaponTpl,
+                    slotName,
+                    botTemplateInventory,
+                    weaponParentId,
+                    weaponModChances,
+                    "pmcUSEC",
+                    true,
+                    69
+                );
+
+                var itemSize = _inventoryHelper.GetItemSize(
+                    weaponTpl,
+                    result.Weapon[0].Id,
+                    result.Weapon
+                );
+
+                Assert.AreEqual(weaponTpl, result.WeaponTemplate.Id);
+
+                // Ensure it's bigger than just weapon lower
+                Assert.AreNotEqual(2, itemSize.Item1);
+                Assert.AreNotEqual(1, itemSize.Item2);
+            }
+        }
+    }
+}