Add overloads to WindowsInstallerData.Load for table definitions.

1 files changed, 40 insertions, 0 deletions

diff --git a/src/test/WixToolsetTest.Data/SerializeFixture.cs b/src/test/WixToolsetTest.Data/SerializeFixture.cs
index 6e224438..198b2571 100644
--- a/src/test/WixToolsetTest.Data/SerializeFixture.cs
+++ b/src/test/WixToolsetTest.Data/SerializeFixture.cs
@@ -8,8 +8,11 @@ namespace WixToolsetTest.Data
     using WixToolset.Data;
     using WixToolset.Data.Bind;
     using WixToolset.Data.Tuples;
+    using WixToolset.Data.WindowsInstaller.Rows;
     using Xunit;
 
+    using Wid = WixToolset.Data.WindowsInstaller;
+
     public class SerializeFixture
     {
         [Fact]
@@ -383,5 +386,42 @@ namespace WixToolsetTest.Data
                 File.Delete(path);
             }
         }
+
+        [Fact]
+        public void CanSaveAndLoadWindowsInstallerData()
+        {
+            var sln = new SourceLineNumber("test.wxs", 1);
+            var windowsInstallerData = new Wid.WindowsInstallerData(sln)
+            {
+                Type = OutputType.Product,
+            };
+
+            var fileTable = windowsInstallerData.EnsureTable(Wid.WindowsInstallerTableDefinitions.File);
+            var fileRow = (FileRow)fileTable.CreateRow(sln);
+            fileRow.File = "TestFile";
+
+            var path = Path.GetTempFileName();
+            try
+            {
+                using (var wixout = WixOutput.Create(path))
+                {
+                    windowsInstallerData.Save(wixout);
+                }
+
+                var loaded = Wid.WindowsInstallerData.Load(path);
+
+                var loadedTable = Assert.Single(loaded.Tables);
+                Assert.Equal(Wid.WindowsInstallerTableDefinitions.File.Name, loadedTable.Name);
+
+                var loadedRow = Assert.Single(loadedTable.Rows);
+                var loadedFileRow = Assert.IsType<FileRow>(loadedRow);
+
+                Assert.Equal("TestFile", loadedFileRow.File);
+            }
+            finally
+            {
+                File.Delete(path);
+            }
+        }
     }
 }