tgstation-server/tests/Tgstation.Server.Host.Tests/Database/TestDatabaseConnectionFactory.cs

using System;

using Microsoft.Data.SqlClient;
using Microsoft.VisualStudio.TestTools.UnitTesting;

using MySqlConnector;

using Tgstation.Server.Host.Configuration;

namespace Tgstation.Server.Host.Database.Tests
{
	[TestClass]
	public sealed class TestDatabaseConnectionFactory
	{
		[TestMethod]
		public void TestBadParameters()
		{
			var factory = new DatabaseConnectionFactory();
			Assert.ThrowsExactly<ArgumentNullException>(() => factory.CreateConnection(null, default));
			Assert.ThrowsExactly<ArgumentOutOfRangeException>(() => factory.CreateConnection(String.Empty, (DatabaseType)42));
		}

		[TestMethod]
		public void TestWorks()
		{
			var factory = new DatabaseConnectionFactory();
			Assert.IsInstanceOfType(factory.CreateConnection(String.Empty, DatabaseType.MariaDB), typeof(MySqlConnection));
			Assert.IsInstanceOfType(factory.CreateConnection(String.Empty, DatabaseType.MySql), typeof(MySqlConnection));
			Assert.IsInstanceOfType(factory.CreateConnection(String.Empty, DatabaseType.SqlServer), typeof(SqlConnection));
		}
	}
}