Add rough draft of Botgen tests

2025-07-20 19:43:53 +01:00
parent 850441fcda
commit d3be2b8b03
1 changed files with 166 additions and 0 deletions
@@ -0,0 +1,166 @@
+using SPTarkov.Server.Core.Helpers;
+using SPTarkov.Server.Core.Models.Common;
+using SPTarkov.Server.Core.Models.Eft.Common.Tables;
+using SPTarkov.Server.Core.Models.Enums;
+using SPTarkov.Server.Core.Utils;
+
+namespace UnitTests.Tests.Helpers
+{
+    [TestClass]
+    public class BotGeneratorHelperTests
+    {
+        private BotGeneratorHelper _botGeneratorHelper;
+
+        [TestInitialize]
+        public void Initialize()
+        {
+            _botGeneratorHelper = DI.GetService<BotGeneratorHelper>();
+            var databaseImporter = DI.GetService<DatabaseImporter>();
+            _ = databaseImporter.OnLoad();
+        }
+
+        [TestMethod]
+        public void AddItemWithChildrenToEquipmentSlot_fit_vertical()
+        {
+            var stashId = new MongoId();
+            var equipmentId = new MongoId();
+            var botInventory = new BotBaseInventory
+            {
+                Items = [],
+                Stash = stashId,
+                Equipment = equipmentId,
+            };
+
+            // Create backpack on player
+            var backpack = new Item
+            {
+                Id = new MongoId(),
+                // Has a 3grids, first is a 3hx5v grid
+                Template = ItemTpl.BACKPACK_EBERLESTOCK_G2_GUNSLINGER_II_BACKPACK_DRY_EARTH,
+                ParentId = equipmentId,
+                SlotId = "Backpack",
+            };
+            botInventory.Items.Add(backpack);
+
+            var weaponWithChildren = new List<Item>();
+            var weaponRoot = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.SHOTGUN_MP18_762X54R_SINGLESHOT_RIFLE,
+            };
+            weaponWithChildren.Add(weaponRoot);
+
+            var result = _botGeneratorHelper.AddItemWithChildrenToEquipmentSlot(
+                [EquipmentSlots.Backpack],
+                weaponRoot.Id,
+                weaponRoot.Template,
+                weaponWithChildren,
+                botInventory
+            );
+
+            Assert.Equals(result, ItemAddedResult.SUCCESS);
+            Assert.Equals((weaponRoot.Location as ItemLocation).X, 0);
+            Assert.Equals((weaponRoot.Location as ItemLocation).Y, 0);
+            Assert.Equals((weaponRoot.Location as ItemLocation).R, ItemRotation.Vertical);
+        }
+
+        [TestMethod]
+        public void AddItemWithChildrenToEquipmentSlot_fit_horizontal()
+        {
+            var stashId = new MongoId();
+            var equipmentId = new MongoId();
+            var botInventory = new BotBaseInventory
+            {
+                Items = [],
+                Stash = stashId,
+                Equipment = equipmentId,
+            };
+
+            // Create backpack on player
+            var backpack = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.BACKPACK_ANA_TACTICAL_BETA_2_BATTLE_BACKPACK_OLIVE_DRAB,
+                ParentId = equipmentId,
+                SlotId = "Backpack",
+            };
+            botInventory.Items.Add(backpack);
+
+            var weaponWithChildren = new List<Item>();
+            var weaponRoot = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.SHOTGUN_MP18_762X54R_SINGLESHOT_RIFLE,
+            };
+            weaponWithChildren.Add(weaponRoot);
+
+            var result = _botGeneratorHelper.AddItemWithChildrenToEquipmentSlot(
+                [EquipmentSlots.Backpack],
+                weaponRoot.Id,
+                weaponRoot.Template,
+                weaponWithChildren,
+                botInventory
+            );
+
+            Assert.Equals(result, ItemAddedResult.SUCCESS);
+            Assert.Equals((weaponRoot.Location as ItemLocation).X, 0);
+            Assert.Equals((weaponRoot.Location as ItemLocation).Y, 0);
+            Assert.Equals((weaponRoot.Location as ItemLocation).R, ItemRotation.Horizontal);
+        }
+
+        /// <summary>
+        /// Backpack with one bullet in top row, blocking gun from being placed at 0,0
+        /// </summary>
+        [TestMethod]
+        public void AddItemWithChildrenToEquipmentSlot_fit_vertical_with_items_in_backpack()
+        {
+            var botInventory = new BotBaseInventory { Items = [] };
+            var backpack = new Item
+            {
+                Id = new MongoId(),
+                // Has a 3hx5v grid first
+                Template = ItemTpl.BACKPACK_EBERLESTOCK_G2_GUNSLINGER_II_BACKPACK_DRY_EARTH,
+                SlotId = "Backpack",
+            };
+            botInventory.Items.Add(backpack);
+
+            botInventory.Items.Add(
+                new Item
+                {
+                    Id = new MongoId(),
+                    Template = ItemTpl.AMMO_762X25TT_AKBS,
+                    ParentId = backpack.Id,
+                    SlotId = "main",
+                    Location = new ItemLocation
+                    {
+                        X = 1,
+                        Y = 0,
+                        R = ItemRotation.Horizontal,
+                    },
+                    Upd = new Upd { StackObjectsCount = 1 },
+                }
+            );
+
+            var weaponWithChildren = new List<Item>();
+            var weaponRoot = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.SHOTGUN_MP18_762X54R_SINGLESHOT_RIFLE,
+            };
+            weaponWithChildren.Add(weaponRoot);
+
+            var result = _botGeneratorHelper.AddItemWithChildrenToEquipmentSlot(
+                [EquipmentSlots.Backpack],
+                weaponRoot.Id,
+                weaponRoot.Template,
+                weaponWithChildren,
+                botInventory
+            );
+
+            Assert.Equals(result, ItemAddedResult.SUCCESS);
+            Assert.Equals((weaponRoot.Location as ItemLocation).X, 0);
+            Assert.Equals((weaponRoot.Location as ItemLocation).Y, 1);
+            Assert.Equals((weaponRoot.Location as ItemLocation).R, ItemRotation.Vertical);
+        }
+    }
+}