diff --git a/SabreTools.DatFiles.Test/DatFileTests.Splitting.cs b/SabreTools.DatFiles.Test/DatFileTests.Splitting.cs index 3de6f06b..a658c12e 100644 --- a/SabreTools.DatFiles.Test/DatFileTests.Splitting.cs +++ b/SabreTools.DatFiles.Test/DatFileTests.Splitting.cs @@ -51,7 +51,98 @@ namespace SabreTools.DatFiles.Test #region AddItemsFromCloneOfParent - // TODO: Implement AddItemsFromCloneOfParent tests + [Fact] + public void AddItemsFromCloneOfParent_Items() + { + Source source = new Source(0, source: null); + + Machine parentMachine = new Machine(); + parentMachine.SetFieldValue(Models.Metadata.Machine.NameKey, "parent"); + + Machine childMachine = new Machine(); + childMachine.SetFieldValue(Models.Metadata.Machine.NameKey, "child"); + childMachine.SetFieldValue(Models.Metadata.Machine.CloneOfKey, "parent"); + childMachine.SetFieldValue(Models.Metadata.Machine.IsBiosKey, true); + + DatItem parentItem = new Rom(); + parentItem.SetName("parent_rom"); + parentItem.SetFieldValue(Models.Metadata.Rom.SizeKey, 12345); + parentItem.SetFieldValue(Models.Metadata.Rom.CRCKey, "deadbeef"); + parentItem.SetFieldValue(DatItem.MachineKey, parentMachine); + parentItem.SetFieldValue(DatItem.SourceKey, source); + + DatItem matchChildItem = new Rom(); + matchChildItem.SetName("match_child_rom"); + matchChildItem.SetFieldValue(Models.Metadata.Rom.SizeKey, 12345); + matchChildItem.SetFieldValue(Models.Metadata.Rom.CRCKey, "deadbeef"); + matchChildItem.SetFieldValue(DatItem.MachineKey, childMachine); + matchChildItem.SetFieldValue(DatItem.SourceKey, source); + + DatItem noMatchChildItem = new Rom(); + noMatchChildItem.SetName("no_match_child_rom"); + noMatchChildItem.SetFieldValue(Models.Metadata.Rom.SizeKey, 12345); + noMatchChildItem.SetFieldValue(Models.Metadata.Rom.CRCKey, "beefdead"); + noMatchChildItem.SetFieldValue(DatItem.MachineKey, childMachine); + noMatchChildItem.SetFieldValue(DatItem.SourceKey, source); + + DatFile datFile = new Logiqx(datFile: null, deprecated: false); + datFile.AddItem(parentItem, statsOnly: false); + datFile.AddItem(matchChildItem, statsOnly: false); + datFile.AddItem(noMatchChildItem, statsOnly: false); + + datFile.BucketBy(ItemKey.Machine, DedupeType.None); + datFile.AddItemsFromCloneOfParent(); + + Assert.Equal(2, datFile.GetItemsForBucket("child").Count); + } + + [Fact] + public void AddItemsFromCloneOfParent_ItemsDB() + { + Source source = new Source(0, source: null); + + Machine parentMachine = new Machine(); + parentMachine.SetFieldValue(Models.Metadata.Machine.NameKey, "parent"); + + Machine childMachine = new Machine(); + childMachine.SetFieldValue(Models.Metadata.Machine.NameKey, "child"); + childMachine.SetFieldValue(Models.Metadata.Machine.CloneOfKey, "parent"); + childMachine.SetFieldValue(Models.Metadata.Machine.IsBiosKey, true); + + DatItem parentItem = new Rom(); + parentItem.SetName("parent_rom"); + parentItem.SetFieldValue(Models.Metadata.Rom.SizeKey, 12345); + parentItem.SetFieldValue(Models.Metadata.Rom.CRCKey, "deadbeef"); + parentItem.SetFieldValue(DatItem.MachineKey, parentMachine); + parentItem.SetFieldValue(DatItem.SourceKey, source); + + DatItem matchChildItem = new Rom(); + matchChildItem.SetName("match_child_rom"); + matchChildItem.SetFieldValue(Models.Metadata.Rom.SizeKey, 12345); + matchChildItem.SetFieldValue(Models.Metadata.Rom.CRCKey, "deadbeef"); + matchChildItem.SetFieldValue(DatItem.MachineKey, childMachine); + matchChildItem.SetFieldValue(DatItem.SourceKey, source); + + DatItem noMatchChildItem = new Rom(); + noMatchChildItem.SetName("no_match_child_rom"); + noMatchChildItem.SetFieldValue(Models.Metadata.Rom.SizeKey, 12345); + noMatchChildItem.SetFieldValue(Models.Metadata.Rom.CRCKey, "beefdead"); + noMatchChildItem.SetFieldValue(DatItem.MachineKey, childMachine); + noMatchChildItem.SetFieldValue(DatItem.SourceKey, source); + + DatFile datFile = new Logiqx(datFile: null, deprecated: false); + long biosMachineIndex = datFile.AddMachineDB(parentMachine); + long deviceMachineIndex = datFile.AddMachineDB(childMachine); + long sourceIndex = datFile.AddSourceDB(source); + _ = datFile.AddItemDB(parentItem, biosMachineIndex, sourceIndex, statsOnly: false); + _ = datFile.AddItemDB(matchChildItem, deviceMachineIndex, sourceIndex, statsOnly: false); + _ = datFile.AddItemDB(noMatchChildItem, deviceMachineIndex, sourceIndex, statsOnly: false); + + datFile.BucketBy(ItemKey.Machine, DedupeType.None); + datFile.AddItemsFromCloneOfParent(); + + Assert.Equal(2, datFile.GetItemsForBucketDB("child").Count); + } #endregion