TestDatabaseSetup.cs

using System;
using System.Collections.Generic;
using System.Data.Odbc;
using System.Data.SqlClient;
#if NETFX
using System.Data.Common;
using System.Data.SqlServerCe;
using System.Diagnostics;
#endif
using System.Data.SQLite;
using System.IO;
using FirebirdSql.Data.FirebirdClient;
using NHibernate.Test;
using Npgsql;
using NUnit.Framework;

namespace NHibernate.TestDatabaseSetup
{
	[TestFixture]
	public class DatabaseSetup
	{
		private static readonly IDictionary<string, Action<Cfg.Configuration>> SetupMethods = new Dictionary<string, Action<Cfg.Configuration>>
			{
				{"NHibernate.Driver.SqlClientDriver", SetupSqlServer},
				{"NHibernate.Driver.Sql2008ClientDriver", SetupSqlServer},
				{"NHibernate.Driver.FirebirdClientDriver", SetupFirebird},
				{"NHibernate.Driver.NpgsqlDriver", SetupNpgsql},
				{"NHibernate.Driver.OracleDataClientDriver", SetupOracle},
				{"NHibernate.Driver.MySqlDataDriver", SetupMySql},
				{"NHibernate.Driver.OracleClientDriver", SetupOracle},
				{"NHibernate.Driver.OracleManagedDataClientDriver", SetupOracle},
				{"NHibernate.Driver.OdbcDriver", SetupSqlServerOdbc},
				{"NHibernate.Driver.SQLite20Driver", SetupSQLite},
#if NETFX
				{"NHibernate.Driver.SqlServerCeDriver", SetupSqlServerCe},
				{"NHibernate.Driver.SapSQLAnywhere17Driver", SetupSqlAnywhere}
#endif
			};

		private static void SetupMySql(Cfg.Configuration obj)
		{
			//TODO: do nothing
		}

		[Test]
		public void SetupDatabase()
		{
			var cfg = TestConfigurationHelper.GetDefaultConfiguration();
			var driver = cfg.Properties[Cfg.Environment.ConnectionDriver];

			Assert.That(SetupMethods.ContainsKey(driver), "No setup method found for " + driver);

			var setupMethod = SetupMethods[driver];
			setupMethod(cfg);
		}

		private static void SetupSqlServer(Cfg.Configuration cfg)
		{
			var connStr = cfg.Properties[Cfg.Environment.ConnectionString];

			using (var conn = new SqlConnection(connStr.Replace("initial catalog=nhibernate", "initial catalog=master")))
			{
				conn.Open();

				using (var cmd = conn.CreateCommand())
				{
					cmd.CommandText = "drop database nhibernate";

					try
					{
						cmd.ExecuteNonQuery();
					}
					catch(Exception e)
					{
						Console.WriteLine(e);
					}

					cmd.CommandText = "create database nhibernate";
					cmd.ExecuteNonQuery();
				}
			}
		}

		private static void SetupSqlServerOdbc(Cfg.Configuration cfg)
		{
			var connStr = cfg.Properties[Cfg.Environment.ConnectionString];

			using (var conn = new OdbcConnection(connStr.Replace("Database=nhibernateOdbc", "Database=master")))
			{
				conn.Open();

				using (var cmd = conn.CreateCommand())
				{
					cmd.CommandText = "drop database nhibernateOdbc";

					try
					{
						cmd.ExecuteNonQuery();
					}
					catch(Exception e)
					{
						Console.WriteLine(e);
					}

					cmd.CommandText = "create database nhibernateOdbc";
					cmd.ExecuteNonQuery();
				}
			}
		}

		private static void SetupFirebird(Cfg.Configuration cfg)
		{
			var connStr = cfg.Properties[Cfg.Environment.ConnectionString];
			try
			{
				FbConnection.DropDatabase(connStr);
			}
			catch (Exception e)
			{
				Console.WriteLine(e);
			}
			// With UTF8 charset, string takes up to four times as many space, causing the
			// default page-size of 4096 to no more be enough for index key sizes. (Index key
			// size is limited to a quarter of the page size.)
			FbConnection.CreateDatabase(connStr, pageSize:16384, forcedWrites:false);
		}

#if NETFX
		private static void SetupSqlServerCe(Cfg.Configuration cfg)
		{
			var connStr = cfg.Properties[Cfg.Environment.ConnectionString];

			try
			{
				var connStrBuilder = new SqlCeConnectionStringBuilder(connStr);
				var dataSource = connStrBuilder.DataSource;
				if (File.Exists(dataSource))
					File.Delete(dataSource);
			}
			catch (Exception e)
			{
				Console.WriteLine(e);
			}

			using (var en = new SqlCeEngine(connStr))
			{
				en.CreateDatabase();
			}
		}
#endif

		private static void SetupNpgsql(Cfg.Configuration cfg)
		{
			var connStr = cfg.Properties[Cfg.Environment.ConnectionString];

			using (var conn = new NpgsqlConnection(connStr.Replace("Database=nhibernate", "Database=postgres")))
			{
				conn.Open();

				using (var cmd = conn.CreateCommand())
				{
					cmd.CommandText = "drop database nhibernate";

					try
					{
						cmd.ExecuteNonQuery();
					}
					catch (Exception e)
					{
						Console.WriteLine(e);
					}

					cmd.CommandText = "create database nhibernate";
					cmd.ExecuteNonQuery();
				}
			}

			// Install the GUID generator function that uses the most common "random" algorithm.
			using (var conn = new NpgsqlConnection(connStr))
			{
				conn.Open();

				using (var cmd = conn.CreateCommand())
				{
					cmd.CommandText = "CREATE EXTENSION IF NOT EXISTS \"uuid-ossp\";";

					cmd.ExecuteNonQuery();
				}
			}
		}

		private static void SetupSQLite(Cfg.Configuration cfg)
		{
			var connStr = cfg.Properties[Cfg.Environment.ConnectionString];

			try
			{
				var connStrBuilder = new SQLiteConnectionStringBuilder(connStr);
				var dataSource = connStrBuilder.DataSource;
				if (File.Exists(dataSource))
					File.Delete(dataSource);
			}
			catch (Exception e)
			{
				Console.WriteLine(e);
			}
		}

		private static void SetupOracle(Cfg.Configuration cfg)
		{
			// disabled until system password is set on TeamCity

			//var connStr =
			//    cfg.Properties[Cfg.Environment.ConnectionString]
			//        .Replace("User ID=nhibernate", "User ID=SYSTEM")
			//        .Replace("Password=nhibernate", "Password=password");

			//cfg.DataBaseIntegration(db =>
			//    {
			//        db.ConnectionString = connStr;
			//        db.Dialect<NHibernate.Dialect.Oracle10gDialect>();
			//        db.KeywordsAutoImport = Hbm2DDLKeyWords.None;
			//    });

			//using (var sf = cfg.BuildSessionFactory())
			//{
			//    try
			//    {
			//        using(var s = sf.OpenSession())
			//            s.CreateSQLQuery("drop user nhibernate cascade").ExecuteUpdate();
			//    }
			//    catch {}

			//    using (var s = sf.OpenSession())
			//    {
			//        s.CreateSQLQuery("create user nhibernate identified by nhibernate").ExecuteUpdate();
			//        s.CreateSQLQuery("grant dba to nhibernate with admin option").ExecuteUpdate();
			//    }
			//}
		}

#if NETFX
		private static void SetupSqlAnywhere(Cfg.Configuration cfg)
		{
			var connStr = cfg.Properties[Cfg.Environment.ConnectionString];

			var factory = DbProviderFactories.GetFactory("Sap.Data.SQLAnywhere");
			var connBuilder = factory.CreateConnectionStringBuilder();
			connBuilder.ConnectionString = connStr;
			var filename = (string) connBuilder["DBF"];

			RunProcess("dbstop", $"-c \"UID=nhibernate;PWD=nhibernate;DBN=nhibernate\" -d", false);
			RunProcess("dberase", $"-y {filename}", false);
			// -dba: login,pwd
			RunProcess("dbinit", $"-dba nhibernate,nhibernate {filename}", true);

			using (var conn = factory.CreateConnection())
			{
				conn.ConnectionString = connStr;
				conn.Open();
				using (var cmd = conn.CreateCommand())
				{
					cmd.CommandText = "set option ansi_update_constraints = 'Off'";
					cmd.ExecuteNonQuery();
				}
			}
		}

		private static void RunProcess(string processName, string arguments, bool checkSuccess)
		{
			using (var process = new Process())
			{
				process.StartInfo.FileName = processName;
				process.StartInfo.Arguments = arguments;
				process.StartInfo.CreateNoWindow = true;
				process.StartInfo.UseShellExecute = false;
				process.StartInfo.RedirectStandardOutput = true;
				process.StartInfo.RedirectStandardError = true;
				process.Start();
				Console.WriteLine($"{processName} output:");
				Console.Write(process.StandardOutput.ReadToEnd());
				Console.WriteLine();
				Console.WriteLine($"{processName} error output:");
				Console.Write(process.StandardError.ReadToEnd());
				Console.WriteLine();
				process.WaitForExit();
				if (checkSuccess && process.ExitCode != 0)
					throw new InvalidOperationException($"{processName} has failed");
			}
		}
#endif
	}
}