Adds unit tests for the /version flag

Author: Kevin Johnson

src/GitVersionExe.Tests/GitVersionExe.Tests.csproj

@@ -128,6 +128,7 @@
     <Compile Include="MsBuildProjectArgTest.cs" />
     <Compile Include="PullRequestInTeamCityTest.cs" />
     <Compile Include="AssemblyInfoFileUpdateTests.cs" />
+    <Compile Include="VersionWriterTests.cs" />
   </ItemGroup>
   <ItemGroup>
     <Service Include="{82A7F48D-3B50-4B1E-B82E-3ADA8210C358}" />

src/GitVersionExe.Tests/VersionWriterTests.cs

@@ -0,0 +1,63 @@
+namespace GitVersionExe.Tests
+{
+    using System;
+    using System.CodeDom.Compiler;
+    using System.Reflection;
+    using GitVersion;
+    using NUnit.Framework;
+
+    public class VersionWriterTests
+    {
+        [Test]
+        public void WriteVersion_ShouldWriteFileVersion_WithNoPrereleaseTag()
+        {
+            var asm = GenerateAssembly(new Version(1, 0, 0), "");
+
+            string version = null;
+            VersionWriter.WriteTo(asm, v => version = v);
+
+            Assert.IsNotNull(asm);
+            Assert.AreEqual("1.0.0", version);
+        }
+
+        [Test]
+        public void WriteVersion_ShouldWriteFileVersion_WithPrereleaseTag()
+        {
+            var asm = GenerateAssembly(new Version(1, 0, 0), "-beta0004");
+
+            string version = null;
+            VersionWriter.WriteTo(asm, v => version = v);
+
+            Assert.IsNotNull(asm);
+            Assert.AreEqual("1.0.0-beta0004", version);
+        }
+
+        private Assembly GenerateAssembly(Version fileVersion, string prereleaseInfo)
+        {
+            var source = string.Format(@"
+using System.Reflection;
+using System.Runtime.CompilerServices;
+[assembly: AssemblyTitle(""GitVersion.DynamicUnitTests"")]
+[assembly: AssemblyProduct(""GitVersion"")]
+[assembly: AssemblyVersion(""{0}"")]
+[assembly: AssemblyFileVersion(""{0}"")]
+[assembly: AssemblyInformationalVersion(""{0}{1}"")]
+
+public class B
+{{
+    public static int k=7;
+}}
+", fileVersion, prereleaseInfo);
+
+            CompilerParameters parameters = new CompilerParameters
+            {
+                GenerateInMemory = true,
+                GenerateExecutable = false,
+                OutputAssembly = "GitVersion.DynamicUnitTests.dll"
+            };
+
+            var r = CodeDomProvider.CreateProvider("CSharp").CompileAssemblyFromSource(parameters, source);
+            return r.CompiledAssembly;
+        }
+    }
+}

src/GitVersionExe/HelpWriter.cs

@@ -1,6 +1,7 @@
 namespace GitVersion
 {
     using System;
+    using System.Reflection;
 
     class HelpWriter
     {
@@ -12,7 +13,8 @@ public static void Write()
         public static void WriteTo(Action<string> writeAction)
         {
             string version = string.Empty;
-            VersionWriter.WriteTo(v => version = v);
+            Assembly assembly = Assembly.GetExecutingAssembly();
+            VersionWriter.WriteTo(assembly, v => version = v);
 
             string message = "GitVersion " + version + @"
 Use convention to derive a SemVer product version from a GitFlow or GitHub based repository.

src/GitVersionExe/Program.cs

@@ -7,6 +7,7 @@ namespace GitVersion
     using System.Diagnostics;
     using System.IO;
     using System.Linq;
+    using System.Reflection;
     using System.Text;
 
     class Program
@@ -59,7 +60,8 @@ static int VerifyArgumentsAndRun()
 
                 if (arguments.IsVersion)
                 {
-                    VersionWriter.Write();
+                    var assembly = Assembly.GetExecutingAssembly();
+                    VersionWriter.Write(assembly);
                     return 0;
                 }
 

src/GitVersionExe/VersionWriter.cs

@@ -6,42 +6,29 @@
 
     class VersionWriter
     {
-        /// <summary>
-        /// Gets the xss version
-        /// </summary>
-        private static Version Version
+        public static void Write(Assembly assembly)
         {
-            get { return Assembly.GetExecutingAssembly().GetName().Version; }
+            WriteTo(assembly, Console.WriteLine);
         }
 
-        /// <summary>
-        /// Gets the AssemblyInformationalVersion
-        /// </summary>
-        private static string InformationalVersion
+        public static void WriteTo(Assembly assembly, Action<string> writeAction)
         {
-            get
-            {
-                var attribute = Assembly.GetExecutingAssembly()
+            var version = GetAssemblyVersion(assembly);
+            writeAction(version);
+        }
+
+        private static string GetAssemblyVersion(Assembly assembly)
+        {
+            var attribute = assembly
                     .GetCustomAttributes(typeof(AssemblyInformationalVersionAttribute), false)
                     .FirstOrDefault() as AssemblyInformationalVersionAttribute;
 
-                if (attribute != null)
-                {
-                    return attribute.InformationalVersion;
-                }
-
-                return Version.ToString();
+            if (attribute != null)
+            {
+                return attribute.InformationalVersion;
             }
-        }
 
-        public static void Write()
-        {
-            WriteTo(Console.WriteLine);
-        }
-
-        public static void WriteTo(Action<string> writeAction)
-        {
-            writeAction(InformationalVersion);
+            return assembly.GetName().Version.ToString();
         }
     }
 }