From 9c1b2f045e55e52dcef4dc3018b3d0b41b667a19 Mon Sep 17 00:00:00 2001
From: Matt Nadareski <mnadareski@outlook.com>
Date: Thu, 16 Jan 2025 14:07:32 -0500
Subject: [PATCH] Add quick AddMachine and AddSource DB tests

---
 .../ItemDictionaryDBTests.cs                  | 22 +++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/SabreTools.DatFiles.Test/ItemDictionaryDBTests.cs b/SabreTools.DatFiles.Test/ItemDictionaryDBTests.cs
index 4dfcb10e..f4550587 100644
--- a/SabreTools.DatFiles.Test/ItemDictionaryDBTests.cs
+++ b/SabreTools.DatFiles.Test/ItemDictionaryDBTests.cs
@@ -20,13 +20,31 @@ namespace SabreTools.DatFiles.Test
 
         #region AddMachine
 
-        // TODO: Add AddMachine tests
+        [Fact]
+        public void AddMachineTest()
+        {
+            Machine machine = new Machine();
+            var dict = new ItemDictionaryDB();
+            long machineIndex = dict.AddMachine(machine);
+
+            Assert.Equal(0, machineIndex);
+            Assert.Single(dict.GetMachines());
+        }
 
         #endregion
 
         #region AddSource
 
-        // TODO: Add AddSource tests
+        [Fact]
+        public void AddSourceTest()
+        {
+            Source source = new Source(0, source: null);
+            var dict = new ItemDictionaryDB();
+            long sourceIndex = dict.AddSource(source);
+
+            Assert.Equal(0, sourceIndex);
+            Assert.Single(dict.GetSources());
+        }
 
         #endregion