2 files changed, 53 insertions, 12 deletions

diff --git a/src/test/WixToolsetTest.WixCop/ConverterFixture.cs b/src/test/WixToolsetTest.WixCop/ConverterFixture.cs
index 863a781a..ceac07d6 100644
--- a/src/test/WixToolsetTest.WixCop/ConverterFixture.cs
+++ b/src/test/WixToolsetTest.WixCop/ConverterFixture.cs
@@ -421,6 +421,34 @@ namespace WixToolsetTest.WixCop
         }
 
         [Fact]
+        public void CanConvertShortNameDirectoryWithoutName()
+        {
+            var parse = String.Join(Environment.NewLine,
+                "<?xml version='1.0' encoding='utf-8'?>",
+                "<Wix xmlns='http://wixtoolset.org/schemas/v4/wxs'>",
+                "  <Directory ShortName='iamshort' />",
+                "</Wix>");
+
+            var expected = String.Join(Environment.NewLine,
+                "<?xml version=\"1.0\" encoding=\"utf-16\"?>",
+                "<Wix xmlns=\"http://wixtoolset.org/schemas/v4/wxs\">",
+                "  <Directory Name=\"iamshort\" />",
+                "</Wix>");
+
+            var document = XDocument.Parse(parse, LoadOptions.PreserveWhitespace | LoadOptions.SetLineInfo);
+
+            var messaging = new DummyMessaging();
+            var converter = new Converter(messaging, 2, null, null);
+
+            var errors = converter.ConvertDocument(document);
+
+            var actual = UnformattedDocumentString(document);
+
+            Assert.Equal(1, errors);
+            Assert.Equal(expected, actual);
+        }
+
+        [Fact]
         public void CanConvertSuppressSignatureValidationNo()
         {
             var parse = String.Join(Environment.NewLine,
diff --git a/src/wixcop/Converter.cs b/src/wixcop/Converter.cs
index 5408d370..37016c19 100644
--- a/src/wixcop/Converter.cs
+++ b/src/wixcop/Converter.cs
@@ -22,6 +22,7 @@ namespace WixToolset.Tools.WixCop
         private const char XDocumentNewLine = '\n'; // XDocument normalizes "\r\n" to just "\n".
         private static readonly XNamespace WixNamespace = "http://wixtoolset.org/schemas/v4/wxs";
 
+        private static readonly XName DirectoryElementName = WixNamespace + "Directory";
         private static readonly XName FileElementName = WixNamespace + "File";
         private static readonly XName ExePackageElementName = WixNamespace + "ExePackage";
         private static readonly XName MsiPackageElementName = WixNamespace + "MsiPackage";
@@ -71,19 +72,9 @@ namespace WixToolset.Tools.WixCop
         /// <param name="ignoreErrors">Test errors to ignore.</param>
         public Converter(IMessaging messaging, int indentationAmount, IEnumerable<string> errorsAsWarnings = null, IEnumerable<string> ignoreErrors = null)
         {
-            // workaround IDE0009 bug
-            /*this.ConvertElementMapping = new Dictionary<XName, Action<XElement>>()
-            {
-                { Converter.FileElementName, this.ConvertFileElement },
-                { Converter.ExePackageElementName, this.ConvertSuppressSignatureValidation },
-                { Converter.MsiPackageElementName, this.ConvertSuppressSignatureValidation },
-                { Converter.MspPackageElementName, this.ConvertSuppressSignatureValidation },
-                { Converter.MsuPackageElementName, this.ConvertSuppressSignatureValidation },
-                { Converter.PayloadElementName, this.ConvertSuppressSignatureValidation },
-                { Converter.WixElementWithoutNamespaceName, this.ConvertWixElementWithoutNamespace },
-            };*/
             this.ConvertElementMapping = new Dictionary<XName, Action<XElement>>
             {
+                { Converter.DirectoryElementName, this.ConvertDirectoryElement },
                 { Converter.FileElementName, this.ConvertFileElement },
                 { Converter.ExePackageElementName, this.ConvertSuppressSignatureValidation },
                 { Converter.MsiPackageElementName, this.ConvertSuppressSignatureValidation },
@@ -92,7 +83,7 @@ namespace WixToolset.Tools.WixCop
                 { Converter.PayloadElementName, this.ConvertSuppressSignatureValidation },
                 { Converter.CustomActionElementName, this.ConvertCustomActionElement },
                 { Converter.PropertyElementName, this.ConvertPropertyElement },
-                { Converter.WixElementWithoutNamespaceName, this.ConvertWixElementWithoutNamespace }
+                { Converter.WixElementWithoutNamespaceName, this.ConvertWixElementWithoutNamespace },
             };
 
             this.Messaging = messaging;
@@ -296,6 +287,23 @@ namespace WixToolset.Tools.WixCop
             }
         }
 
+        private void ConvertDirectoryElement(XElement element)
+        {
+            if (null == element.Attribute("Name"))
+            {
+                var attribute = element.Attribute("ShortName");
+                if (null != attribute)
+                {
+                    var shortName = attribute.Value;
+                    if (this.OnError(ConverterTestType.AssignDirectoryNameFromShortName, element, "The directory ShortName attribute is being renamed to Name since Name wasn't specified for value '{0}'", shortName))
+                    {
+                        element.Add(new XAttribute("Name", shortName));
+                        attribute.Remove();
+                    }
+                }
+            }
+        }
+
         private void ConvertFileElement(XElement element)
         {
             if (null == element.Attribute("Id"))
@@ -611,6 +619,11 @@ namespace WixToolset.Tools.WixCop
             /// QtExecCmdTimeout was previously used for both CAQuietExec and CAQuietExec64. For WixQuietExec, use WixQuietExecCmdTimeout. For WixQuietExec64, use WixQuietExec64CmdTimeout.
             /// </summary>
             QtExecCmdTimeoutAmbiguous,
+
+            /// <summary>
+            /// Directory/@ShortName may only be specified with Directory/@Name.
+            /// </summary>
+            AssignDirectoryNameFromShortName,
         }
     }
 }