Fix MsiAssembly table processing

2 files changed, 19 insertions, 2 deletions

diff --git a/src/WixToolset.Core.WindowsInstaller/Bind/AssemblyName.cs b/src/WixToolset.Core.WindowsInstaller/Bind/AssemblyName.cs
index 0df1a7e9..759ba303 100644
--- a/src/WixToolset.Core.WindowsInstaller/Bind/AssemblyName.cs
+++ b/src/WixToolset.Core.WindowsInstaller/Bind/AssemblyName.cs
@@ -16,7 +16,7 @@ namespace WixToolset.Core.WindowsInstaller.Bind
             this.Architecture = architecture;
 
             this.StrongNamedSigned = !String.IsNullOrEmpty(publicKeyToken);
-            this.PublicKeyToken = publicKeyToken ?? "null";
+            this.PublicKeyToken = publicKeyToken;
             this.Type = type;
         }
 
@@ -46,7 +46,7 @@ namespace WixToolset.Core.WindowsInstaller.Bind
             assemblyName.Append(", Culture=");
             assemblyName.Append(this.Culture);
             assemblyName.Append(", PublicKeyToken=");
-            assemblyName.Append(this.PublicKeyToken);
+            assemblyName.Append(this.PublicKeyToken ?? "null");
 
             if (!String.IsNullOrEmpty(this.Architecture))
             {
diff --git a/src/WixToolset.Core.WindowsInstaller/Bind/CreateOutputFromIRCommand.cs b/src/WixToolset.Core.WindowsInstaller/Bind/CreateOutputFromIRCommand.cs
index 17cac83a..081644cb 100644
--- a/src/WixToolset.Core.WindowsInstaller/Bind/CreateOutputFromIRCommand.cs
+++ b/src/WixToolset.Core.WindowsInstaller/Bind/CreateOutputFromIRCommand.cs
@@ -60,6 +60,10 @@ namespace WixToolset.Core.WindowsInstaller.Bind
                     output.EnsureTable(this.TableDefinitions["Signature"]);
                     break;
 
+                case TupleDefinitionType.Assembly:
+                    this.AddAssemblyTuple((AssemblyTuple)tuple, output);
+                    break;
+
                 case TupleDefinitionType.Binary:
                     this.AddTupleDefaultly(tuple, output, idIsPrimaryKey: true);
                     break;
@@ -239,6 +243,19 @@ namespace WixToolset.Core.WindowsInstaller.Bind
             }
         }
 
+        private void AddAssemblyTuple(AssemblyTuple tuple, Output output)
+        {
+            var attributes = tuple.Type == AssemblyType.Win32Assembly ? 1 : (int?)null;
+
+            var table = output.EnsureTable(this.TableDefinitions["MsiAssembly"]);
+            var row = table.CreateRow(tuple.SourceLineNumbers);
+            row[0] = tuple.ComponentRef;
+            row[1] = tuple.FeatureRef;
+            row[2] = tuple.ManifestFileRef;
+            row[3] = tuple.ApplicationFileRef;
+            row[4] = attributes;
+        }
+
         private void AddBBControlTuple(BBControlTuple tuple, Output output)
         {
             var attributes = tuple.Attributes;