Make sure our code works

test/Squirrel.Tests.csproj

@@ -84,6 +84,7 @@
     <Compile Include="Properties\AssemblyInfo.cs" />
     <Compile Include="ReleaseEntryTests.cs" />
     <Compile Include="ReleasePackageTests.cs" />
+    <Compile Include="SquirrelAwareExecutableDetectorTests.cs" />
     <Compile Include="TestHelpers\AssertExtensions.cs" />
     <Compile Include="TestHelpers\ExposedClass.cs" />
     <Compile Include="TestHelpers\ExposedObject.cs" />
@@ -98,6 +99,10 @@
       <Project>{1436e22a-fe3c-4d68-9a85-9e74df2e6a92}</Project>
       <Name>Squirrel</Name>
     </ProjectReference>
+    <ProjectReference Include="..\src\Update\Update.csproj">
+      <Project>{1eebacbc-6982-4696-bd4e-899ed0ac6cd2}</Project>
+      <Name>Update</Name>
+    </ProjectReference>
   </ItemGroup>
   <ItemGroup>
     <None Include="packages.config" />

test/SquirrelAwareExecutableDetectorTests.cs

@@ -0,0 +1,53 @@
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.IO;
+using System.Linq;
+using System.Reflection;
+using System.Runtime.InteropServices;
+using System.Text;
+using System.Threading.Tasks;
+using Squirrel.Tests.TestHelpers;
+using Xunit;
+
+namespace Squirrel.Tests
+{
+    public class SquirrelAwareExecutableDetectorTests
+    {
+        [Fact]
+        public void SquirrelAwareViaVersionBlock()
+        {
+            var target = Path.Combine(
+                IntegrationTestHelper.GetIntegrationTestRootDirectory(),
+                "..", "src", "Setup", "bin", "Release", "Setup.exe");
+
+            Assert.True(File.Exists(target));
+
+            var ret = SquirrelAwareExecutableDetector.GetPESquirrelAwareVersion(target);
+            Assert.Equal(1, ret.Value);
+        }
+
+        [Fact]
+        public void SquirrelAwareViaAssemblyAttribute()
+        {
+            var target = Path.Combine(
+                IntegrationTestHelper.GetIntegrationTestRootDirectory(),
+                "..", "src", "Update", "bin", "Release", "Update.exe");
+
+            Assert.True(File.Exists(target));
+
+            var ret = SquirrelAwareExecutableDetector.GetPESquirrelAwareVersion(target);
+            Assert.Equal(1, ret.Value);
+        }
+
+        [Fact]
+        public void NotSquirrelAware()
+        {
+            var target = Assembly.GetExecutingAssembly().Location;
+            Assert.True(File.Exists(target));
+
+            var ret = SquirrelAwareExecutableDetector.GetPESquirrelAwareVersion(target);
+            Assert.Null(ret);
+        }
+    }
+}