Mvc2VS2010Helper.cs :  » Testing » xUnit.net » Xunit » Installer » C# / CSharp Open Source

using System;
using System.IO;
using System.Reflection;
using System.Windows.Forms;
using Microsoft.Win32;

namespace Xunit.Installer{
    public static class Mvc2VS2010Helper
    {
        const string TEMPLATE_NAME_CSHARP = "XunitMvc2TestProjectTemplate.cs.zip";
        const string TEMPLATE_NAME_VB = "XunitMvc2TestProjectTemplate.vb.zip";

        public static bool CanBeEnabled
        {
            get
            {
                using (RegistryKey key = OpenMvc2Key())
                    return key != null;
            }
        }

        public static bool IsEnabled
        {
            get
            {
                using (RegistryKey key = OpenMvc2Key())
                {
                    if (key == null)
                        return false;

                    using (RegistryKey templateKey = key.OpenSubKey("xUnit.net"))
                        return templateKey != null;
                }
            }
        }

        public static bool Disable(Form owner)
        {
            using (RegistryKey key = OpenMvc2Key())
                key.DeleteSubKeyTree("xUnit.net");

            UninstallZipFiles(VS2010Helper.GetTestProjectTemplatePath("CSharp"));
            UninstallZipFiles(VS2010Helper.GetTestProjectTemplatePath("VisualBasic"));

            VS2010Helper.ResetVisualStudio(owner);
            return true;
        }

        public static bool Enable(Form owner)
        {
            ResourceHelper.WriteResourceToFile("Xunit.Installer.Templates.MVC2-CS-VS2010.zip",
                                               Path.Combine(VS2010Helper.GetTestProjectTemplatePath("CSharp"), TEMPLATE_NAME_CSHARP));
            ResourceHelper.WriteResourceToFile("Xunit.Installer.Templates.MVC2-VB-VS2010.zip",
                                               Path.Combine(VS2010Helper.GetTestProjectTemplatePath("VisualBasic"), TEMPLATE_NAME_VB));

            using (RegistryKey key = OpenMvc2Key())
            using (RegistryKey templateKey = key.CreateSubKey("xUnit.net"))
            {
                using (RegistryKey csharpKey = templateKey.CreateSubKey("C#"))
                {
                    csharpKey.SetValue("AdditionalInfo", "http://xunit.codeplex.com/");
                    csharpKey.SetValue("Package", "");
                    csharpKey.SetValue("Path", @"CSharp\Test");
                    csharpKey.SetValue("Template", TEMPLATE_NAME_CSHARP);
                    csharpKey.SetValue("TestFrameworkName", "xUnit.net build " + Assembly.GetExecutingAssembly().GetName().Version);
                }
                using (RegistryKey vbKey = templateKey.CreateSubKey("VB"))
                {
                    vbKey.SetValue("AdditionalInfo", "http://xunit.codeplex.com/");
                    vbKey.SetValue("Package", "");
                    vbKey.SetValue("Path", @"VisualBasic\Test");
                    vbKey.SetValue("Template", TEMPLATE_NAME_VB);
                    vbKey.SetValue("TestFrameworkName", "xUnit.net build " + Assembly.GetExecutingAssembly().GetName().Version);
                }
            }

            VS2010Helper.ResetVisualStudio(owner);
            return true;
        }

        public static string InstalledXunitVersion()
        {
            using (RegistryKey key = OpenMvc2Key())
            using (RegistryKey csharpKey = key.OpenSubKey(@"xUnit.net\C#"))
                return String.Format("(With templates for {0})", csharpKey.GetValue("TestFrameworkName"));
        }

        static RegistryKey OpenMvc2Key()
        {
            return Registry.LocalMachine.OpenSubKey(@"SOFTWARE\Microsoft\VisualStudio\10.0\MVC2\TestProjectTemplates", true);
        }

        static void UninstallZipFiles(string templatePath)
        {
            foreach (string file in Directory.GetFiles(templatePath, "XunitMvc2TestProjectTemplate.*"))
                File.Delete(Path.Combine(templatePath, file));

            foreach (string directory in Directory.GetDirectories(templatePath))
                UninstallZipFiles(Path.Combine(templatePath, directory));
        }
    }
}