From 205598bd1340a17345cee0b733ef284b3d14fe7b Mon Sep 17 00:00:00 2001
From: Matt Nadareski <mnadareski@outlook.com>
Date: Thu, 16 Jan 2025 15:12:28 -0500
Subject: [PATCH] Add RemoveMachine test

---
 .../ItemDictionaryDBTests.cs                     | 16 ++++++++++------
 SabreTools.DatFiles.Test/ItemDictionaryTests.cs  |  6 ++++++
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/SabreTools.DatFiles.Test/ItemDictionaryDBTests.cs b/SabreTools.DatFiles.Test/ItemDictionaryDBTests.cs
index c92b1f17..71940945 100644
--- a/SabreTools.DatFiles.Test/ItemDictionaryDBTests.cs
+++ b/SabreTools.DatFiles.Test/ItemDictionaryDBTests.cs
@@ -213,13 +213,17 @@ namespace SabreTools.DatFiles.Test
 
         #region RemoveMachine
 
-        // TODO: Add RemoveMachine tests
+        [Fact]
+        public void RemoveMachineTest()
+        {
+            Machine machine = new Machine();
+            var dict = new ItemDictionaryDB();
+            long machineIndex = dict.AddMachine(machine);
 
-        #endregion
-
-        #region SetBucketedBy
-
-        // TODO: Add SetBucketedBy tests
+            bool actual = dict.RemoveMachine(machineIndex);
+            Assert.True(actual);
+            Assert.Empty(dict.GetMachines());
+        }
 
         #endregion
 
diff --git a/SabreTools.DatFiles.Test/ItemDictionaryTests.cs b/SabreTools.DatFiles.Test/ItemDictionaryTests.cs
index 5b8a443c..cabb2809 100644
--- a/SabreTools.DatFiles.Test/ItemDictionaryTests.cs
+++ b/SabreTools.DatFiles.Test/ItemDictionaryTests.cs
@@ -73,6 +73,12 @@ namespace SabreTools.DatFiles.Test
 
         #endregion
 
+        #region SetBucketedBy
+
+        // TODO: Add SetBucketedBy tests
+
+        #endregion
+
         #region BucketBy
 
         [Theory]