From 0feb4130fea4b544ed16bab3caadcc37abb3745f Mon Sep 17 00:00:00 2001
From: Matt Nadareski <mnadareski@outlook.com>
Date: Thu, 16 Apr 2026 16:56:02 -0400
Subject: [PATCH] Fix some DB-specific tests

---
 .../DatFileTests.cs                           | 31 +++++++++++++------
 1 file changed, 21 insertions(+), 10 deletions(-)

diff --git a/SabreTools.Metadata.DatFiles.Test/DatFileTests.cs b/SabreTools.Metadata.DatFiles.Test/DatFileTests.cs
index 2dce442f..e9bedc5e 100644
--- a/SabreTools.Metadata.DatFiles.Test/DatFileTests.cs
+++ b/SabreTools.Metadata.DatFiles.Test/DatFileTests.cs
@@ -2162,8 +2162,6 @@ namespace SabreTools.Metadata.DatFiles.Test
                 Name = "name",
                 Size = 12345,
                 CRC32 = "crc",
-                Machine = (Machine)machine.Clone(),
-                Source = (Source)source.Clone()
             };
 
             List<KeyValuePair<long, DatItem>> mappings =
@@ -2172,6 +2170,11 @@ namespace SabreTools.Metadata.DatFiles.Test
             ];
             DatFile datFile = new Logiqx(null, useGame: false);
 
+            long sourceIndex = datFile.AddSourceDB(source);
+            romA.SourceIndex = sourceIndex;
+            long machineIndex = datFile.AddMachineDB(machine);
+            romA.MachineIndex = machineIndex;
+
             List<KeyValuePair<long, DatItem>> actual = datFile.ResolveNamesDB(mappings);
             KeyValuePair<long, DatItem> actualItemA = Assert.Single(actual);
             Rom? actualRomA = actualItemA.Value as Rom;
@@ -2193,8 +2196,6 @@ namespace SabreTools.Metadata.DatFiles.Test
                 Name = "romA",
                 Size = 12345,
                 CRC32 = "crc",
-                Machine = (Machine)machine.Clone(),
-                Source = (Source)source.Clone()
             };
 
             Rom romB = new Rom
@@ -2202,8 +2203,6 @@ namespace SabreTools.Metadata.DatFiles.Test
                 Name = "romB",
                 Size = 23456,
                 CRC32 = "crc2",
-                Machine = (Machine)machine.Clone(),
-                Source = (Source)source.Clone()
             };
 
             List<KeyValuePair<long, DatItem>> mappings =
@@ -2213,6 +2212,14 @@ namespace SabreTools.Metadata.DatFiles.Test
             ];
             DatFile datFile = new Logiqx(null, useGame: false);
 
+            long sourceIndex = datFile.AddSourceDB(source);
+            romA.SourceIndex = sourceIndex;
+            romB.SourceIndex = sourceIndex;
+
+            long machineIndex = datFile.AddMachineDB(machine);
+            romA.MachineIndex = machineIndex;
+            romB.MachineIndex = machineIndex;
+
             List<KeyValuePair<long, DatItem>> actual = datFile.ResolveNamesDB(mappings);
             Assert.Equal(2, actual.Count);
 
@@ -2287,8 +2294,6 @@ namespace SabreTools.Metadata.DatFiles.Test
                 Name = "rom",
                 Size = 12345,
                 CRC32 = "crc",
-                Machine = (Machine)machine.Clone(),
-                Source = (Source)source.Clone()
             };
 
             Rom romB = new Rom
@@ -2296,8 +2301,6 @@ namespace SabreTools.Metadata.DatFiles.Test
                 Name = "rom",
                 Size = 23456,
                 CRC32 = "crc2",
-                Machine = (Machine)machine.Clone(),
-                Source = (Source)source.Clone()
             };
 
             List<KeyValuePair<long, DatItem>> mappings =
@@ -2307,6 +2310,14 @@ namespace SabreTools.Metadata.DatFiles.Test
             ];
             DatFile datFile = new Logiqx(null, useGame: false);
 
+            long sourceIndex = datFile.AddSourceDB(source);
+            romA.SourceIndex = sourceIndex;
+            romB.SourceIndex = sourceIndex;
+
+            long machineIndex = datFile.AddMachineDB(machine);
+            romA.MachineIndex = machineIndex;
+            romB.MachineIndex = machineIndex;
+
             List<KeyValuePair<long, DatItem>> actual = datFile.ResolveNamesDB(mappings);
             Assert.Equal(2, actual.Count);