AssemblySetUpAndCleanUp.cs :  » Bloggers » SubText » UnitTests » Subtext » C# / CSharp Open Source

using System;
using System.Configuration;
using System.Data.SqlClient;
using MbUnit.Framework;
using Subtext.Framework.Configuration;
using Subtext.Framework.Infrastructure.Installation;
using UnitTests.Subtext;

[assembly: AssemblyCleanup(typeof(AssemblySetUpAndCleanUp))]

namespace UnitTests.Subtext{
    public static class AssemblySetUpAndCleanUp
    {
        [SetUp]
        [CoverageExclude]
        public static void SetUp()
        {
            Console.WriteLine("Assembly Setup beginning...");
            if(ConfigurationManager.AppSettings["connectionStringName"] == "subtextExpress")
            {
                //For use with SQL Express. If you use "subtextData", we assume you already have the database created.
                DatabaseHelper.CreateAndInstallDatabase(Config.ConnectionString, Config.ConnectionString.Database,
                                                        "App_Data");
            }
            else
            {
                using(var connection = new SqlConnection(Config.ConnectionString))
                {
                    connection.Open();
                    using(SqlTransaction transaction = connection.BeginTransaction())
                    {
                        try
                        {
                            ScriptHelper.ExecuteScript("StoredProcedures.sql", transaction);
                            transaction.Commit();
                        }
                        catch(Exception)
                        {
                            transaction.Rollback();
                        }
                    }
                }
            }
        }

        [TearDown]
        [CoverageExclude]
        public static void TearDown()
        {
            try
            {
                if(ConfigurationManager.AppSettings["connectionStringName"] == "subtextExpress")
                {
                    try
                    {
                        DatabaseHelper.DeleteDatabase(Config.ConnectionString.Server, Config.ConnectionString.Database,
                                                      "App_Data");
                    }
                    catch(Exception e)
                    {
                        Console.WriteLine(
                            "Exception occurred while deleting the database. We'll get it the next time around.");
                        Console.WriteLine(e);
                    }
                }
            }
            catch(Exception)
            {
            }
        }
    }
}