Added tests for FindAndReturnChildrenAsItems

2025-07-19 11:51:33 +01:00
parent d7f83e1b7d
commit 2be4f22773
1 changed files with 162 additions and 0 deletions
@@ -0,0 +1,162 @@
+using SPTarkov.Server.Core.Extensions;
+using SPTarkov.Server.Core.Models.Common;
+using SPTarkov.Server.Core.Models.Eft.Common.Tables;
+
+namespace UnitTests.Tests.Extensions
+{
+    [TestClass]
+    public class ItemTests
+    {
+        [TestInitialize]
+        public void Initialize() { }
+
+        [TestMethod]
+        public void FindAndReturnChildrenAsItems_one_child_mods_only()
+        {
+            var testData = new List<Item>();
+            var rootItem = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMOBOX_127X33_COPPER_20RND,
+            };
+            var childItem = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMO_127X33_COPPER,
+                ParentId = rootItem.Id,
+            };
+            testData.Add(rootItem);
+            testData.Add(childItem);
+
+            var result = testData.FindAndReturnChildrenAsItems(rootItem.Id, true);
+
+            Assert.AreEqual(result[1].Id, childItem.Id);
+        }
+
+        [TestMethod]
+        public void FindAndReturnChildrenAsItems_mods_only_one_inventory_item()
+        {
+            var testData = new List<Item>();
+            var rootItem = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMOBOX_127X33_COPPER_20RND,
+            };
+            var childItem = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMO_127X33_COPPER,
+                ParentId = rootItem.Id,
+                Location = 1,
+            };
+            var childItem2 = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMO_26X75_GREEN,
+                ParentId = rootItem.Id,
+            };
+            testData.Add(rootItem);
+            testData.Add(childItem);
+            testData.Add(childItem2);
+
+            var result = testData.FindAndReturnChildrenAsItems(rootItem.Id, true);
+
+            Assert.AreEqual(result[1].Id, childItem2.Id);
+            Assert.AreEqual(result.Count, 2);
+        }
+
+        [TestMethod]
+        public void FindAndReturnChildrenAsItems_mods_and_inventory_item()
+        {
+            var testData = new List<Item>();
+            var rootItem = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMOBOX_127X33_COPPER_20RND,
+            };
+            var childItem = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMO_127X33_COPPER,
+                ParentId = rootItem.Id,
+                Location = 1,
+            };
+            var childItem2 = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMO_26X75_GREEN,
+                ParentId = rootItem.Id,
+            };
+            testData.Add(rootItem);
+            testData.Add(childItem);
+            testData.Add(childItem2);
+
+            var result = testData.FindAndReturnChildrenAsItems(rootItem.Id, false);
+
+            Assert.AreEqual(result[1].Id, childItem.Id);
+            Assert.AreEqual(result.Count, 3);
+        }
+
+        [TestMethod]
+        public void FindAndReturnChildrenAsItems_mod_with_child()
+        {
+            var testData = new List<Item>();
+            var rootItem = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMOBOX_127X33_COPPER_20RND,
+            };
+            var childItem = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMO_127X33_COPPER,
+                ParentId = rootItem.Id,
+            };
+            var childOfChild = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMO_26X75_GREEN,
+                ParentId = childItem.Id,
+            };
+            testData.Add(rootItem);
+            testData.Add(childItem);
+            testData.Add(childOfChild);
+
+            var result = testData.FindAndReturnChildrenAsItems(rootItem.Id, true);
+
+            Assert.AreEqual(result[1].Id, childItem.Id);
+            Assert.AreEqual(result.Count, 3);
+        }
+
+        [TestMethod]
+        public void FindAndReturnChildrenAsItems_no_matching_children()
+        {
+            var testData = new List<Item>();
+            var rootItem = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMOBOX_127X33_COPPER_20RND,
+            };
+            var childItem = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMO_127X33_COPPER,
+                ParentId = new MongoId(),
+            };
+            var childOfChild = new Item
+            {
+                Id = new MongoId(),
+                Template = ItemTpl.AMMO_26X75_GREEN,
+                ParentId = childItem.Id,
+            };
+            testData.Add(rootItem);
+            testData.Add(childItem);
+            testData.Add(childOfChild);
+
+            var result = testData.FindAndReturnChildrenAsItems(rootItem.Id, true);
+
+            Assert.AreEqual(result[0].Id, rootItem.Id);
+            Assert.AreEqual(result.Count, 1);
+        }
+    }
+}