diff --git a/SabreTools.DatTools/Remover.cs b/SabreTools.DatTools/Remover.cs
index 715e3401..3c202c9f 100644
--- a/SabreTools.DatTools/Remover.cs
+++ b/SabreTools.DatTools/Remover.cs
@@ -211,7 +211,7 @@ namespace SabreTools.DatTools
///
/// Remove machine fields with given values
///
- private void RemoveFields(Machine? machine)
+ internal void RemoveFields(Machine? machine)
{
// If we have an invalid input, return
if (machine == null || MachineFieldNames.Count == 0)
@@ -227,7 +227,7 @@ namespace SabreTools.DatTools
/// Remove fields with given values
///
/// DatItem to remove fields from
- private void RemoveFields(DatItem? datItem)
+ internal void RemoveFields(DatItem? datItem)
{
if (datItem == null)
return;
diff --git a/SabreTools.Test/RemoverTests.cs b/SabreTools.Test/RemoverTests.cs
index c01fae45..a2e19fb0 100644
--- a/SabreTools.Test/RemoverTests.cs
+++ b/SabreTools.Test/RemoverTests.cs
@@ -1,45 +1,37 @@
using SabreTools.DatItems;
using SabreTools.DatItems.Formats;
+using SabreTools.DatTools;
+using Xunit;
namespace SabreTools.Test
{
- // TODO: Reenable tests when there's a reasonable way of doing so
public class RemoverTests
{
- //[Fact]
- //public void RemoveFieldsDatItemTest()
- //{
- // var datItem = CreateDatItem();
- // var remover = new Remover();
- // remover.PopulateExclusions("DatItem.Name");
- // remover.RemoveFields(datItem);
- // Assert.Null(datItem.GetName());
- //}
-
- //[Fact]
- //public void RemoveFieldsMachineTest()
- //{
- // var datItem = CreateDatItem();
- // var remover = new Remover();
- // remover.PopulateExclusions("Machine.Name");
- // remover.RemoveFields(datItem);
- // Assert.Null(datItem.GetFieldValue(DatItem.MachineKey)!.GetStringFieldValue(Models.Metadata.Machine.NameKey));
- //}
-
- ///
- /// Generate a consistent DatItem for testing
- ///
- private static DatItem CreateDatItem()
+ [Fact]
+ public void RemoveFieldsDatItemTest()
{
- var machine = new Machine();
+ var datItem = new Rom();
+ datItem.SetName("foo");
+
+ var remover = new Remover();
+ remover.PopulateExclusions("DatItem.Name");
+ remover.RemoveFields(datItem);
+
+ Assert.Null(datItem.GetName());
+ }
+
+ [Fact]
+ public void RemoveFieldsMachineTest()
+ {
+ var machine = new Machine();
machine.SetFieldValue(Models.Metadata.Machine.NameKey, "bar");
machine.SetFieldValue(Models.Metadata.Machine.DescriptionKey, "bar");
- var rom = new Rom();
- rom.SetName("foo");
- rom.SetFieldValue(DatItem.MachineKey, machine);
+ var remover = new Remover();
+ remover.PopulateExclusions("Machine.Name");
+ remover.RemoveFields(machine);
- return rom;
+ Assert.Null(machine.GetStringFieldValue(Models.Metadata.Machine.NameKey));
}
}
}
\ No newline at end of file