mirror of
https://github.com/ckaczor/sqltoolsservice.git
synced 2026-01-14 01:25:40 -05:00
a053457ba1
* initial commit * adding prototypefile and server conditions * Saving query store options completed, todo:tests * adding LOC stings according to the LOC version of the ms doc * removing common constants * merge conflict fix * fixing null reference exception * Adjusting the null reference exception property value in prototype.cs * removing unused directive * test fix that checks wrong value * Purge query store data changes * adding comment and uncommented the line
873 lines
51 KiB
C#
873 lines
51 KiB
C#
//
|
|
// Copyright (c) Microsoft. All rights reserved.
|
|
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
|
//
|
|
|
|
#nullable disable
|
|
|
|
using System.Collections.Generic;
|
|
using System.Collections.Specialized;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.Data.SqlClient;
|
|
using Microsoft.SqlServer.Management.Common;
|
|
using Microsoft.SqlServer.Management.Smo;
|
|
using Microsoft.SqlTools.ServiceLayer.Admin;
|
|
using Microsoft.SqlTools.ServiceLayer.Connection;
|
|
using Microsoft.SqlTools.ServiceLayer.IntegrationTests.Utility;
|
|
using Microsoft.SqlTools.ServiceLayer.ObjectManagement;
|
|
using Microsoft.SqlTools.ServiceLayer.ObjectManagement.Contracts;
|
|
using Microsoft.SqlTools.ServiceLayer.Test.Common;
|
|
using NUnit.Framework;
|
|
using static Microsoft.SqlTools.ServiceLayer.Admin.AzureSqlDbHelper;
|
|
using DatabaseFile = Microsoft.SqlTools.ServiceLayer.ObjectManagement.DatabaseFile;
|
|
using FileGroup = Microsoft.SqlServer.Management.Smo.FileGroup;
|
|
|
|
namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.ObjectManagement
|
|
{
|
|
/// <summary>
|
|
/// Tests for the Database management component
|
|
/// </summary>
|
|
public class DatabaseHandlerTests
|
|
{
|
|
/// <summary>
|
|
/// Test the basic Create Database method handler by creating, updating, and then deleting a database.
|
|
/// </summary>
|
|
[Test]
|
|
public async Task DatabaseCreateAndUpdateTest_OnPrem()
|
|
{
|
|
await RunDatabaseCreateAndUpdateTest(TestServerType.OnPrem);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Test the Create Database method handler functionality against an Azure SQL database.
|
|
/// </summary>
|
|
[Test]
|
|
[Ignore("Test is not supported in the integration test pipeline.")]
|
|
public async Task DatabaseCreateAndUpdateTest_Azure()
|
|
{
|
|
await RunDatabaseCreateAndUpdateTest(TestServerType.Azure);
|
|
}
|
|
|
|
private async Task RunDatabaseCreateAndUpdateTest(TestServerType serverType)
|
|
{
|
|
// setup, drop database if exists.
|
|
var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master", serverType: serverType);
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var server = new Server(new ServerConnection(sqlConn));
|
|
|
|
var testDatabase = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.Name);
|
|
await ObjectManagementTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, objUrn);
|
|
|
|
try
|
|
{
|
|
// create and update
|
|
var parametersForCreation = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, "master", true, SqlObjectType.Database, "", "");
|
|
await ObjectManagementTestUtils.SaveObject(parametersForCreation, testDatabase);
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), $"Expected database '{testDatabase.Name}' was not created succesfully");
|
|
|
|
var parametersForUpdate = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, "master", false, SqlObjectType.Database, "", objUrn);
|
|
await ObjectManagementTestUtils.SaveObject(parametersForUpdate, testDatabase);
|
|
|
|
// cleanup
|
|
await ObjectManagementTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, objUrn, throwIfNotExist: true);
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), Is.False, $"Database '{testDatabase.Name}' was not dropped succesfully");
|
|
}
|
|
finally
|
|
{
|
|
// Cleanup using SMO if Drop didn't work
|
|
DropDatabase(server, testDatabase.Name!);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Test that the handler can export the Create Database operation to a SQL script.
|
|
/// </summary>
|
|
[Test]
|
|
public async Task DatabaseScriptTest()
|
|
{
|
|
var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master");
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var server = new Server(new ServerConnection(sqlConn));
|
|
|
|
var testDatabase = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.Name);
|
|
await ObjectManagementTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, objUrn);
|
|
|
|
try
|
|
{
|
|
var parametersForCreation = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, "master", true, SqlObjectType.Database, "", "");
|
|
var script = await ObjectManagementTestUtils.ScriptObject(parametersForCreation, testDatabase);
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), Is.False, $"Database should not have been created for scripting operation");
|
|
Assert.That(script.ToLowerInvariant(), Does.Contain($"create database [{testDatabase.Name!.ToLowerInvariant()}]"));
|
|
}
|
|
finally
|
|
{
|
|
// Cleanup database on the off-chance that scripting somehow created the database
|
|
DropDatabase(server, testDatabase.Name!);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Test that the handler correctly throws an error when trying to drop a database that doesn't exist.
|
|
/// </summary>
|
|
[Test]
|
|
public async Task DatabaseNotExistsErrorTest()
|
|
{
|
|
var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master");
|
|
var testDatabase = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.Name);
|
|
try
|
|
{
|
|
await ObjectManagementTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, objUrn, throwIfNotExist: true);
|
|
Assert.Fail("Did not throw an exception when trying to drop non-existent database.");
|
|
}
|
|
catch (FailedOperationException ex)
|
|
{
|
|
Assert.That(ex.InnerException, Is.Not.Null, "Expected inner exception was null.");
|
|
Assert.That(ex.InnerException, Is.InstanceOf<MissingObjectException>(), $"Received unexpected inner exception type: {ex.InnerException!.GetType()}");
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Test that the handler correctly throws an error when trying to create a database with the same name as an existing database.
|
|
/// </summary>
|
|
[Test]
|
|
public async Task DatabaseAlreadyExistsErrorTest()
|
|
{
|
|
var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master");
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var server = new Server(new ServerConnection(sqlConn));
|
|
|
|
var testDatabase = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.Name);
|
|
await ObjectManagementTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, objUrn);
|
|
|
|
try
|
|
{
|
|
var parametersForCreation = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, "master", true, SqlObjectType.Database, "", "");
|
|
await ObjectManagementTestUtils.SaveObject(parametersForCreation, testDatabase);
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), $"Expected database '{testDatabase.Name}' was not created succesfully");
|
|
|
|
await ObjectManagementTestUtils.SaveObject(parametersForCreation, testDatabase);
|
|
Assert.Fail("Did not throw an exception when trying to create database with same name.");
|
|
}
|
|
catch (FailedOperationException ex)
|
|
{
|
|
Assert.That(ex.InnerException, Is.Not.Null, "Expected inner exception was null.");
|
|
Assert.That(ex.InnerException, Is.InstanceOf<ExecutionFailureException>(), $"Received unexpected inner exception type: {ex.InnerException!.GetType()}");
|
|
Assert.That(ex.InnerException.InnerException, Is.Not.Null, "Expected inner-inner exception was null.");
|
|
Assert.That(ex.InnerException.InnerException, Is.InstanceOf<SqlException>(), $"Received unexpected inner-inner exception type: {ex.InnerException.InnerException!.GetType()}");
|
|
}
|
|
finally
|
|
{
|
|
DropDatabase(server, testDatabase.Name!);
|
|
}
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
public void GetAzureEditionsTest()
|
|
{
|
|
var actualEditionNames = DatabaseHandler.AzureEditionNames;
|
|
var expectedEditionNames = AzureSqlDbHelper.GetValidAzureEditionOptions().Select(edition => edition.DisplayName);
|
|
Assert.That(actualEditionNames, Is.EquivalentTo(expectedEditionNames));
|
|
}
|
|
|
|
[Test]
|
|
public void GetAzureBackupRedundancyLevelsTest()
|
|
{
|
|
var actualLevels = DatabaseHandler.AzureBackupLevels;
|
|
var expectedLevels = new string[] { "Geo", "Local", "Zone" };
|
|
Assert.That(actualLevels, Is.EquivalentTo(expectedLevels));
|
|
}
|
|
|
|
[Test]
|
|
public void GetAzureServiceLevelObjectivesTest()
|
|
{
|
|
var actualLevelsMap = new Dictionary<string, OptionsCollection>();
|
|
foreach (AzureEditionDetails serviceDetails in DatabaseHandler.AzureServiceLevels)
|
|
{
|
|
actualLevelsMap.Add(serviceDetails.EditionDisplayName, serviceDetails.EditionOptions);
|
|
}
|
|
|
|
var expectedDefaults = new Dictionary<AzureEdition, string>()
|
|
{
|
|
{ AzureEdition.Basic, "Basic" },
|
|
{ AzureEdition.Standard, "S0" },
|
|
{ AzureEdition.Premium, "P1" },
|
|
{ AzureEdition.BusinessCritical, "BC_Gen5_2" },
|
|
{ AzureEdition.GeneralPurpose, "GP_Gen5_2" },
|
|
{ AzureEdition.Hyperscale, "HS_Gen5_2" }
|
|
};
|
|
Assert.That(actualLevelsMap.Count, Is.EqualTo(expectedDefaults.Count), "Did not get expected number of editions for DatabaseHandler's service levels");
|
|
foreach (AzureEdition edition in expectedDefaults.Keys)
|
|
{
|
|
if (AzureSqlDbHelper.TryGetServiceObjectiveInfo(edition, out var expectedLevelInfo))
|
|
{
|
|
var expectedServiceLevels = expectedLevelInfo.Value;
|
|
var actualServiceLevels = actualLevelsMap[edition.DisplayName];
|
|
Assert.That(actualServiceLevels.Options, Is.EquivalentTo(expectedServiceLevels), "Did not get expected SLO list for edition '{0}'", edition.DisplayName);
|
|
|
|
var expectedDefaultIndex = expectedLevelInfo.Key;
|
|
var expectedDefault = expectedServiceLevels[expectedDefaultIndex];
|
|
var actualDefault = actualServiceLevels.Options[actualServiceLevels.DefaultValueIndex];
|
|
Assert.That(actualDefault, Is.EqualTo(expectedDefault), "Did not get expected default SLO for edition '{0}'", edition.DisplayName);
|
|
}
|
|
else
|
|
{
|
|
Assert.Fail("Could not retrieve SLO info for Azure edition '{0}'", edition.DisplayName);
|
|
}
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
public void GetAzureMaxSizesTest()
|
|
{
|
|
var actualSizesMap = new Dictionary<string, OptionsCollection>();
|
|
foreach (AzureEditionDetails sizeDetails in DatabaseHandler.AzureMaxSizes)
|
|
{
|
|
actualSizesMap.Add(sizeDetails.EditionDisplayName, sizeDetails.EditionOptions);
|
|
}
|
|
|
|
var expectedDefaults = new Dictionary<AzureEdition, string>()
|
|
{
|
|
{ AzureEdition.Basic, "2GB" },
|
|
{ AzureEdition.Standard, "250GB" },
|
|
{ AzureEdition.Premium, "500GB" },
|
|
{ AzureEdition.BusinessCritical, "32GB" },
|
|
{ AzureEdition.GeneralPurpose, "32GB" },
|
|
{ AzureEdition.Hyperscale, "0MB" }
|
|
};
|
|
Assert.That(actualSizesMap.Count, Is.EqualTo(expectedDefaults.Count), "Did not get expected number of editions for DatabaseHandler's max sizes");
|
|
foreach (AzureEdition edition in expectedDefaults.Keys)
|
|
{
|
|
if (AzureSqlDbHelper.TryGetDatabaseSizeInfo(edition, out var expectedSizeInfo))
|
|
{
|
|
var expectedSizes = expectedSizeInfo.Value.Select(size => size.ToString()).ToArray();
|
|
var actualSizes = actualSizesMap[edition.DisplayName];
|
|
Assert.That(actualSizes.Options, Is.EquivalentTo(expectedSizes), "Did not get expected size list for edition '{0}'", edition.DisplayName);
|
|
|
|
var expectedDefaultIndex = expectedSizeInfo.Key;
|
|
var expectedDefault = expectedSizes[expectedDefaultIndex];
|
|
var actualDefault = actualSizes.Options[actualSizes.DefaultValueIndex];
|
|
Assert.That(actualDefault, Is.EqualTo(expectedDefault.ToString()), "Did not get expected default size for edition '{0}'", edition.DisplayName);
|
|
}
|
|
else
|
|
{
|
|
Assert.Fail("Could not retrieve max size info for Azure edition '{0}'", edition.DisplayName);
|
|
}
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
/// This test validates the newly created database properties and verifies with some default values
|
|
public async Task VerifyDatabasePropertiesTest()
|
|
{
|
|
// setup, drop database if exists.
|
|
var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master", serverType: TestServerType.OnPrem);
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var server = new Server(new ServerConnection(sqlConn));
|
|
|
|
var testDatabase = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.Name);
|
|
await ObjectManagementTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, objUrn);
|
|
|
|
try
|
|
{
|
|
// create database
|
|
var parametersForCreation = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, "master", true, SqlObjectType.Database, "", "");
|
|
await ObjectManagementTestUtils.SaveObject(parametersForCreation, testDatabase);
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), $"Expected database '{testDatabase.Name}' was not created succesfully");
|
|
|
|
// Get database properties and verify
|
|
var parametersForUpdate = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, testDatabase.Name, false, SqlObjectType.Database, "", objUrn);
|
|
DatabaseViewInfo databaseViewInfo = await ObjectManagementTestUtils.GetDatabaseObject(parametersForUpdate, testDatabase);
|
|
Assert.That(databaseViewInfo.ObjectInfo, Is.Not.Null, $"Expected result should not be empty");
|
|
Assert.That(databaseViewInfo.ObjectInfo.Name, Is.EqualTo(testDatabase.Name), $"database name should be matched");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).DateCreated, Is.Not.Null, $"database name should be matched");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).NumberOfUsers, Is.GreaterThan(0), $"Default database users count should not be zero");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).LastDatabaseBackup, Is.EqualTo(testDatabase.LastDatabaseBackup), $"Should have no database last backup date");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).LastDatabaseLogBackup, Is.EqualTo(testDatabase.LastDatabaseLogBackup), $"Should have no database backup log date");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).SizeInMb, Is.GreaterThan(0), $"Should have default database size when created");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).AutoCreateIncrementalStatistics, Is.True, $"AutoCreateIncrementalStatistics match with testdata");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).AutoCreateStatistics, Is.True, $"AutoCreateStatistics should match with testdata");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).AutoShrink, Is.False, $"AutoShrink should match with testdata");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).AutoUpdateStatistics, Is.True, $"AutoUpdateStatistics should match with testdata");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).AutoUpdateStatisticsAsynchronously, Is.False, $"AutoUpdateStatisticsAsynchronously should match with testdata");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).PageVerify, Is.EqualTo(testDatabase.PageVerify), $"PageVerify should match with testdata");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).RestrictAccess, Is.EqualTo(testDatabase.RestrictAccess), $"RestrictAccess should match with testdata");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).DatabaseScopedConfigurations, Is.Not.Null, $"DatabaseScopedConfigurations is not null");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).DatabaseScopedConfigurations?.Length, Is.GreaterThan(0), $"DatabaseScopedConfigurations should have at least a+ few properties");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).Files.Count, Is.EqualTo(2), $"Create database should create two database files");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).Files[0].Type, Is.EqualTo("ROWS Data"), $"Database files first file should be Row type database files");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).Files[1].Type, Is.EqualTo("LOG"), $"Database files first file should be Log type database files");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).Filegroups?.Length, Is.GreaterThan(0), $"Database file groups should exists");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).QueryStoreOptions, Is.Not.Null, $"Database Query store options are not null");
|
|
|
|
// cleanup
|
|
await ObjectManagementTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, objUrn, throwIfNotExist: true);
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), Is.False, $"Database '{testDatabase.Name}' was not dropped succesfully");
|
|
}
|
|
finally
|
|
{
|
|
// Cleanup using SMO if Drop didn't work
|
|
DropDatabase(server, testDatabase.Name!);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Updating and validating database scoped configurations property values
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
[Test]
|
|
public async Task VerifyDatabaseScopedConfigurationsTest()
|
|
{
|
|
using (var testDatabase = await SqlTestDb.CreateNewAsync(serverType: TestServerType.OnPrem, dbNamePrefix: "VerifyDatabaseFilesTest"))
|
|
{
|
|
var connectionResult = LiveConnectionHelper.InitLiveConnectionInfo(testDatabase.DatabaseName);
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var server = new Server(new ServerConnection(sqlConn));
|
|
|
|
var testDatabaseInfo = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
testDatabaseInfo.Name = testDatabase.DatabaseName;
|
|
testDatabaseInfo.CollationName = "";
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.DatabaseName);
|
|
|
|
// Get database properties and verify
|
|
var parametersForUpdate = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, testDatabase.DatabaseName, false, SqlObjectType.Database, "", objUrn);
|
|
DatabaseViewInfo databaseViewInfo = await ObjectManagementTestUtils.GetDatabaseObject(parametersForUpdate, testDatabaseInfo);
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).DatabaseScopedConfigurations, Is.Not.Null, $"DatabaseScopedConfigurations is not null");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).DatabaseScopedConfigurations.Count, Is.GreaterThan(0), $"DatabaseScopedConfigurations should have at least a+ few properties");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).DatabaseScopedConfigurations[0].ValueForPrimary, Is.EqualTo("ON"), $"DatabaseScopedConfigurations primary value should match");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).DatabaseScopedConfigurations[0].ValueForSecondary, Is.EqualTo("ON"), $"DatabaseScopedConfigurations secondary value should match");
|
|
|
|
// Update few database scoped configurations
|
|
testDatabaseInfo.DatabaseScopedConfigurations = ((DatabaseInfo)databaseViewInfo.ObjectInfo).DatabaseScopedConfigurations;
|
|
if (testDatabaseInfo.DatabaseScopedConfigurations.Length > 0)
|
|
{
|
|
// ACCELERATED_PLAN_FORCING
|
|
testDatabaseInfo.DatabaseScopedConfigurations[0].ValueForPrimary = "OFF";
|
|
testDatabaseInfo.DatabaseScopedConfigurations[0].ValueForSecondary = "OFF";
|
|
}
|
|
await ObjectManagementTestUtils.SaveObject(parametersForUpdate, testDatabaseInfo);
|
|
DatabaseViewInfo updatedDatabaseViewInfo = await ObjectManagementTestUtils.GetDatabaseObject(parametersForUpdate, testDatabaseInfo);
|
|
|
|
// verify the modified properties
|
|
Assert.That(((DatabaseInfo)updatedDatabaseViewInfo.ObjectInfo).DatabaseScopedConfigurations[0].ValueForPrimary, Is.EqualTo(testDatabaseInfo.DatabaseScopedConfigurations[0].ValueForPrimary), $"DSC updated primary value should match");
|
|
Assert.That(((DatabaseInfo)updatedDatabaseViewInfo.ObjectInfo).DatabaseScopedConfigurations[0].ValueForSecondary, Is.EqualTo(testDatabaseInfo.DatabaseScopedConfigurations[0].ValueForSecondary), $"DSC updated Secondary value should match");
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Adding, modifying and deleting database files
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
[Test]
|
|
public async Task VerifyDatabaseFilesTest()
|
|
{
|
|
using (var testDatabase = await SqlTestDb.CreateNewAsync(serverType: TestServerType.OnPrem, dbNamePrefix: "VerifyDatabaseFilesTest"))
|
|
{
|
|
var connectionResult = LiveConnectionHelper.InitLiveConnectionInfo(testDatabase.DatabaseName);
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var testDatabaseInfo = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
testDatabaseInfo.Name = testDatabase.DatabaseName;
|
|
testDatabaseInfo.CollationName = "";
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.DatabaseName);
|
|
|
|
// Get database properties and verify
|
|
var parametersForUpdate = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, testDatabase.DatabaseName, false, SqlObjectType.Database, "", objUrn);
|
|
DatabaseViewInfo databaseViewInfo = await ObjectManagementTestUtils.GetDatabaseObject(parametersForUpdate, testDatabaseInfo);
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).Files.Count, Is.EqualTo(2), $"Create database should create two database files");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).Files[0].Type, Is.EqualTo("ROWS Data"), $"Database files first file should be Row type database files");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).Files[1].Type, Is.EqualTo("LOG"), $"Database files first file should be Log type database files");
|
|
|
|
List<DatabaseFile> databaseFile = new List<DatabaseFile>();
|
|
|
|
// copy exisitng Row data files to the list
|
|
databaseFile.Add(((DatabaseInfo)databaseViewInfo.ObjectInfo).Files[0]);
|
|
databaseFile.Add(((DatabaseInfo)databaseViewInfo.ObjectInfo).Files[1]);
|
|
|
|
// Update existing file
|
|
databaseFile[0].SizeInMb = 15; // size modified from 8 to 15
|
|
databaseFile[0].AutoFileGrowth = 74; // size modified from 64 to 74
|
|
|
|
// Add new Files of Row and Log types
|
|
var testDatabaseFiles = ObjectManagementTestUtils.GetTestDatabaseFiles();
|
|
databaseFile.AddRange(testDatabaseFiles);
|
|
|
|
// Attaching the files to the testdatabase
|
|
testDatabaseInfo.Files = databaseFile.ToArray();
|
|
|
|
// Validate the result files
|
|
await ObjectManagementTestUtils.SaveObject(parametersForUpdate, testDatabaseInfo);
|
|
DatabaseViewInfo updatedDatabaseViewInfo = await ObjectManagementTestUtils.GetDatabaseObject(parametersForUpdate, testDatabaseInfo);
|
|
|
|
// verify the modified properties
|
|
Assert.That(((DatabaseInfo)updatedDatabaseViewInfo.ObjectInfo).Files.Count, Is.EqualTo(3), $"Three files should exists, as we modified one and added one new files");
|
|
var file = ((DatabaseInfo)updatedDatabaseViewInfo.ObjectInfo).Files.FirstOrDefault(x => x.Name == databaseFile[0].Name);
|
|
Assert.That(file, Is.Not.Null, $"Primary file should exists");
|
|
Assert.That(file.SizeInMb, Is.EqualTo(15), $"Files updated value should match");
|
|
Assert.That(file.AutoFileGrowth, Is.EqualTo(74), $"Files updated value should match");
|
|
|
|
// Validate newly created files
|
|
file = ((DatabaseInfo)updatedDatabaseViewInfo.ObjectInfo).Files.FirstOrDefault(x => x.Name == testDatabaseFiles[0].Name);
|
|
Assert.That(file, Is.Not.Null, $"New file should be created");
|
|
Assert.That(file, Is.Not.EqualTo(0), $"Newly created file should have an Id");
|
|
|
|
// Deleting newly created file
|
|
List<DatabaseFile> newfiles = ((DatabaseInfo)updatedDatabaseViewInfo.ObjectInfo).Files.ToList();
|
|
var fileIndexTobeRemoved = newfiles.FindIndex(x => x.Name == testDatabaseFiles[0].Name);
|
|
newfiles.RemoveAt(fileIndexTobeRemoved);
|
|
testDatabaseInfo.Files = newfiles.ToArray();
|
|
|
|
// Validate the result files
|
|
await ObjectManagementTestUtils.SaveObject(parametersForUpdate, testDatabaseInfo);
|
|
DatabaseViewInfo databaseViewInfoAfterFileDelete = await ObjectManagementTestUtils.GetDatabaseObject(parametersForUpdate, testDatabaseInfo);
|
|
Assert.That(((DatabaseInfo)databaseViewInfoAfterFileDelete.ObjectInfo).Files.Count, Is.EqualTo(2), $"Should have only 2 files as we removed one");
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Adding, modifying and deleting database filegroups
|
|
/// /// </summary>
|
|
/// <returns></returns>
|
|
[Test]
|
|
public async Task VerifyDatabaseFileGroupsTest()
|
|
{
|
|
using (var testDatabase = await SqlTestDb.CreateNewAsync(serverType: TestServerType.OnPrem, dbNamePrefix: "VerifyDatabaseFilegeoupsTest"))
|
|
{
|
|
var connectionResult = LiveConnectionHelper.InitLiveConnectionInfo(testDatabase.DatabaseName);
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var testDatabaseInfo = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
testDatabaseInfo.Name = testDatabase.DatabaseName;
|
|
testDatabaseInfo.CollationName = "";
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.DatabaseName);
|
|
|
|
// Get database properties and verify
|
|
var parametersForUpdate = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, testDatabase.DatabaseName, false, SqlObjectType.Database, "", objUrn);
|
|
DatabaseViewInfo databaseViewInfo = await ObjectManagementTestUtils.GetDatabaseObject(parametersForUpdate, testDatabaseInfo);
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).Filegroups?.Length, Is.EqualTo(1), $"Create database should create one default database filegroup");
|
|
Assert.That(((DatabaseInfo)databaseViewInfo.ObjectInfo).Filegroups[0].Name, Is.EqualTo("PRIMARY"), $"Database default filegroup name should be PRIMARY");
|
|
|
|
List<FileGroupSummary> databaseFilegroup = new List<FileGroupSummary>();
|
|
|
|
// copy exisitng Row data files to the list
|
|
databaseFilegroup.Add(((DatabaseInfo)databaseViewInfo.ObjectInfo).Filegroups[0]);
|
|
|
|
// Add new Filegroups
|
|
var testDatabaseFilegroups = ObjectManagementTestUtils.GetTestDatabaseFilegroups();
|
|
databaseFilegroup.AddRange(testDatabaseFilegroups);
|
|
|
|
// Attaching the filegroups to the testdatabase
|
|
testDatabaseInfo.Filegroups = databaseFilegroup.ToArray();
|
|
|
|
// Validate the result
|
|
await ObjectManagementTestUtils.SaveObject(parametersForUpdate, testDatabaseInfo);
|
|
DatabaseViewInfo updatedDatabaseViewInfo = await ObjectManagementTestUtils.GetDatabaseObject(parametersForUpdate, testDatabaseInfo);
|
|
|
|
// verify the modified properties
|
|
Assert.That(((DatabaseInfo)updatedDatabaseViewInfo.ObjectInfo).Filegroups?.Length, Is.EqualTo(3), $"Four filegroups should exists, as we created three more");
|
|
var filegroup = ((DatabaseInfo)updatedDatabaseViewInfo.ObjectInfo).Filegroups.FirstOrDefault(x => x.Name == databaseFilegroup[1].Name);
|
|
Assert.That(filegroup, Is.Not.Null, $"filegroup should exists");
|
|
Assert.That(filegroup.Type, Is.EqualTo(FileGroupType.RowsFileGroup), $"Filegroup type should be matched");
|
|
|
|
filegroup = ((DatabaseInfo)updatedDatabaseViewInfo.ObjectInfo).Filegroups.FirstOrDefault(x => x.Name == databaseFilegroup[2].Name);
|
|
Assert.That(filegroup, Is.Not.Null, $"filegroup should exists");
|
|
Assert.That(filegroup.Type, Is.EqualTo(FileGroupType.MemoryOptimizedDataFileGroup), $"Filegroup type should be matched");
|
|
|
|
// Deleting newly created file
|
|
List<FileGroupSummary> newfilegroups = ((DatabaseInfo)updatedDatabaseViewInfo.ObjectInfo).Filegroups.ToList();
|
|
var fileIndexTobeRemoved = newfilegroups.FindIndex(x => x.Name == databaseFilegroup[1].Name);
|
|
newfilegroups.RemoveAt(fileIndexTobeRemoved);
|
|
testDatabaseInfo.Filegroups = newfilegroups.ToArray();
|
|
|
|
// Validate the result files
|
|
await ObjectManagementTestUtils.SaveObject(parametersForUpdate, testDatabaseInfo);
|
|
DatabaseViewInfo databaseViewInfoAfterFileDelete = await ObjectManagementTestUtils.GetDatabaseObject(parametersForUpdate, testDatabaseInfo);
|
|
Assert.That(((DatabaseInfo)databaseViewInfoAfterFileDelete.ObjectInfo).Filegroups.Count, Is.EqualTo(2), $"Should have only 3 filegroups as we removed one");
|
|
filegroup = ((DatabaseInfo)databaseViewInfoAfterFileDelete.ObjectInfo).Filegroups.FirstOrDefault(x => x.Name == databaseFilegroup[1].Name);
|
|
Assert.That(filegroup, Is.Null, $"filegroup should exists");
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
[Test]
|
|
public async Task DetachDatabaseTest()
|
|
{
|
|
var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master");
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var server = new Server(new ServerConnection(sqlConn));
|
|
|
|
var testDatabase = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.Name);
|
|
await ObjectManagementTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, objUrn);
|
|
|
|
try
|
|
{
|
|
// Create database to test with
|
|
var parametersForCreation = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, "master", true, SqlObjectType.Database, "", "");
|
|
await ObjectManagementTestUtils.SaveObject(parametersForCreation, testDatabase);
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), $"Expected database '{testDatabase.Name}' was not created succesfully");
|
|
|
|
var handler = new DatabaseHandler(ConnectionService.Instance);
|
|
var connectionUri = connectionResult.ConnectionInfo.OwnerUri;
|
|
|
|
var detachParams = new DetachDatabaseRequestParams()
|
|
{
|
|
ConnectionUri = connectionUri,
|
|
ObjectUrn = objUrn,
|
|
DropConnections = true,
|
|
UpdateStatistics = true,
|
|
GenerateScript = false
|
|
};
|
|
|
|
// Get databases's files so we can reattach it later before dropping it
|
|
var fileCollection = new StringCollection();
|
|
var smoDatabase = server.GetSmoObject(objUrn) as Database;
|
|
foreach (FileGroup fileGroup in smoDatabase!.FileGroups)
|
|
{
|
|
foreach (DataFile file in fileGroup.Files)
|
|
{
|
|
fileCollection.Add(file.FileName);
|
|
}
|
|
}
|
|
foreach (LogFile file in smoDatabase.LogFiles)
|
|
{
|
|
fileCollection.Add(file.FileName);
|
|
}
|
|
|
|
var script = handler.Detach(detachParams);
|
|
Assert.That(script, Is.Empty, "Should only return an empty string if GenerateScript is false");
|
|
|
|
server.Databases.Refresh();
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), Is.False, $"Expected database '{testDatabase.Name}' was not detached succesfully");
|
|
|
|
server.AttachDatabase(testDatabase.Name, fileCollection);
|
|
|
|
server.Databases.Refresh();
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), $"Expected database '{testDatabase.Name}' was not re-attached succesfully");
|
|
}
|
|
finally
|
|
{
|
|
DropDatabase(server, testDatabase.Name!);
|
|
}
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
public async Task DetachDatabaseScriptTest()
|
|
{
|
|
var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master");
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var server = new Server(new ServerConnection(sqlConn));
|
|
|
|
var testDatabase = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.Name);
|
|
await ObjectManagementTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, objUrn);
|
|
|
|
try
|
|
{
|
|
// Create database to test with
|
|
var parametersForCreation = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, "master", true, SqlObjectType.Database, "", "");
|
|
await ObjectManagementTestUtils.SaveObject(parametersForCreation, testDatabase);
|
|
|
|
var handler = new DatabaseHandler(ConnectionService.Instance);
|
|
var connectionUri = connectionResult.ConnectionInfo.OwnerUri;
|
|
|
|
// Default use case
|
|
var detachParams = new DetachDatabaseRequestParams()
|
|
{
|
|
ConnectionUri = connectionUri,
|
|
ObjectUrn = objUrn,
|
|
DropConnections = false,
|
|
UpdateStatistics = false,
|
|
GenerateScript = true
|
|
};
|
|
var expectedDetachScript = $"EXEC master.dbo.sp_detach_db @dbname = N'{testDatabase.Name}'";
|
|
var expectedAlterScript = $"ALTER DATABASE [{testDatabase.Name}] SET SINGLE_USER WITH ROLLBACK IMMEDIATE";
|
|
var expectedStatsScript = $"EXEC master.dbo.sp_detach_db @dbname = N'{testDatabase.Name}', @skipchecks = 'false'";
|
|
|
|
var actualScript = handler.Detach(detachParams);
|
|
Assert.That(actualScript, Does.Contain(expectedDetachScript).IgnoreCase);
|
|
|
|
// Drop connections only
|
|
detachParams.DropConnections = true;
|
|
actualScript = handler.Detach(detachParams);
|
|
Assert.That(actualScript, Does.Contain(expectedDetachScript).IgnoreCase);
|
|
Assert.That(actualScript, Does.Contain(expectedAlterScript).IgnoreCase);
|
|
|
|
// Update statistics only
|
|
detachParams.DropConnections = false;
|
|
detachParams.UpdateStatistics = true;
|
|
actualScript = handler.Detach(detachParams);
|
|
Assert.That(actualScript, Does.Contain(expectedStatsScript).IgnoreCase);
|
|
|
|
// Both drop and update
|
|
detachParams.DropConnections = true;
|
|
actualScript = handler.Detach(detachParams);
|
|
Assert.That(actualScript, Does.Contain(expectedAlterScript).IgnoreCase);
|
|
Assert.That(actualScript, Does.Contain(expectedStatsScript).IgnoreCase);
|
|
}
|
|
finally
|
|
{
|
|
DropDatabase(server, testDatabase.Name!);
|
|
}
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
[TestCase(true)]
|
|
[TestCase(false)]
|
|
public async Task AttachDatabaseTest(bool generateScript)
|
|
{
|
|
using (SqlTestDb testDb = await SqlTestDb.CreateNewAsync(TestServerType.OnPrem, false, null, null, nameof(AttachDatabaseTest)))
|
|
{
|
|
var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master", serverType: TestServerType.OnPrem);
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var serverConn = new ServerConnection(sqlConn);
|
|
var server = new Server(serverConn);
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDb.DatabaseName);
|
|
var database = server.GetSmoObject(objUrn) as Database;
|
|
|
|
var originalOwner = database!.Owner;
|
|
var originalFilePaths = new List<string>();
|
|
foreach (FileGroup group in database.FileGroups)
|
|
{
|
|
foreach (DataFile file in group.Files)
|
|
{
|
|
originalFilePaths.Add(file.FileName);
|
|
}
|
|
}
|
|
foreach (LogFile file in database.LogFiles)
|
|
{
|
|
originalFilePaths.Add(file.FileName);
|
|
}
|
|
|
|
// Detach database so that we can re-attach it with the database handler method.
|
|
// Have to set database to single user mode to close active connections before detaching it.
|
|
database.DatabaseOptions.UserAccess = SqlServer.Management.Smo.DatabaseUserAccess.Single;
|
|
database.Alter(TerminationClause.RollbackTransactionsImmediately);
|
|
server.DetachDatabase(testDb.DatabaseName, false);
|
|
var dbExists = this.DatabaseExists(testDb.DatabaseName, server);
|
|
Assert.That(dbExists, Is.False, "Database was not correctly detached before doing attach test.");
|
|
|
|
try
|
|
{
|
|
var handler = new DatabaseHandler(ConnectionService.Instance);
|
|
var attachParams = new AttachDatabaseRequestParams()
|
|
{
|
|
ConnectionUri = connectionResult.ConnectionInfo.OwnerUri,
|
|
Databases = new DatabaseFileData[]
|
|
{
|
|
new DatabaseFileData()
|
|
{
|
|
Owner = originalOwner,
|
|
DatabaseName = testDb.DatabaseName,
|
|
DatabaseFilePaths = originalFilePaths.ToArray()
|
|
}
|
|
},
|
|
GenerateScript = generateScript
|
|
};
|
|
var script = handler.Attach(attachParams);
|
|
|
|
if (generateScript)
|
|
{
|
|
dbExists = this.DatabaseExists(testDb.DatabaseName, server);
|
|
Assert.That(dbExists, Is.False, "Should not have attached DB when only generating a script.");
|
|
|
|
var queryBuilder = new StringBuilder();
|
|
queryBuilder.AppendLine("USE [master]");
|
|
queryBuilder.AppendLine($"CREATE DATABASE [{testDb.DatabaseName}] ON ");
|
|
|
|
for (int i = 0; i < originalFilePaths.Count - 1; i++)
|
|
{
|
|
var file = originalFilePaths[i];
|
|
queryBuilder.AppendLine($"( FILENAME = N'{file}' ),");
|
|
}
|
|
queryBuilder.AppendLine($"( FILENAME = N'{originalFilePaths[originalFilePaths.Count - 1]}' )");
|
|
|
|
queryBuilder.AppendLine(" FOR ATTACH");
|
|
queryBuilder.AppendLine($"if exists (select name from master.sys.databases sd where name = N'{testDb.DatabaseName}' and SUSER_SNAME(sd.owner_sid) = SUSER_SNAME() ) EXEC [{testDb.DatabaseName}].dbo.sp_changedbowner @loginame=N'{originalOwner}', @map=false");
|
|
|
|
Assert.That(script, Is.EqualTo(queryBuilder.ToString()), "Did not get expected attach database script");
|
|
}
|
|
else
|
|
{
|
|
Assert.That(script, Is.Empty, "Should not have generated a script for this Attach operation.");
|
|
|
|
server.Databases.Refresh();
|
|
dbExists = this.DatabaseExists(testDb.DatabaseName, server);
|
|
Assert.That(dbExists, "Database was not attached successfully");
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
dbExists = this.DatabaseExists(testDb.DatabaseName, server);
|
|
if (!dbExists)
|
|
{
|
|
// Reattach database so it can get dropped during cleanup
|
|
var fileCollection = new StringCollection();
|
|
originalFilePaths.ForEach(file => fileCollection.Add(file));
|
|
server.AttachDatabase(testDb.DatabaseName, fileCollection);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
public async Task DeleteDatabaseTest()
|
|
{
|
|
var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master");
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var server = new Server(new ServerConnection(sqlConn));
|
|
|
|
var testDatabase = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.Name);
|
|
await ObjectManagementTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, objUrn);
|
|
|
|
try
|
|
{
|
|
// Create database to test with
|
|
var parametersForCreation = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, "master", true, SqlObjectType.Database, "", "");
|
|
await ObjectManagementTestUtils.SaveObject(parametersForCreation, testDatabase);
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), $"Expected database '{testDatabase.Name}' was not created succesfully");
|
|
|
|
var handler = new DatabaseHandler(ConnectionService.Instance);
|
|
var connectionUri = connectionResult.ConnectionInfo.OwnerUri;
|
|
|
|
var deleteParams = new DropDatabaseRequestParams()
|
|
{
|
|
ConnectionUri = connectionUri,
|
|
ObjectUrn = objUrn,
|
|
DropConnections = false,
|
|
DeleteBackupHistory = false,
|
|
GenerateScript = false
|
|
};
|
|
var script = handler.Drop(deleteParams);
|
|
|
|
Assert.That(script, Is.Empty, "Should only return an empty string if GenerateScript is false");
|
|
|
|
server.Databases.Refresh();
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), Is.False, $"Database '{testDatabase.Name}' was not deleted succesfully");
|
|
}
|
|
finally
|
|
{
|
|
DropDatabase(server, testDatabase.Name!);
|
|
}
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
public async Task DeleteDatabaseScriptTest()
|
|
{
|
|
var connectionResult = await LiveConnectionHelper.InitLiveConnectionInfoAsync("master");
|
|
using (SqlConnection sqlConn = ConnectionService.OpenSqlConnection(connectionResult.ConnectionInfo))
|
|
{
|
|
var server = new Server(new ServerConnection(sqlConn));
|
|
|
|
var testDatabase = ObjectManagementTestUtils.GetTestDatabaseInfo();
|
|
var objUrn = ObjectManagementTestUtils.GetDatabaseURN(testDatabase.Name);
|
|
await ObjectManagementTestUtils.DropObject(connectionResult.ConnectionInfo.OwnerUri, objUrn);
|
|
|
|
try
|
|
{
|
|
// Create database to test with
|
|
var parametersForCreation = ObjectManagementTestUtils.GetInitializeViewRequestParams(connectionResult.ConnectionInfo.OwnerUri, "master", true, SqlObjectType.Database, "", "");
|
|
await ObjectManagementTestUtils.SaveObject(parametersForCreation, testDatabase);
|
|
|
|
var handler = new DatabaseHandler(ConnectionService.Instance);
|
|
var connectionUri = connectionResult.ConnectionInfo.OwnerUri;
|
|
|
|
var deleteParams = new DropDatabaseRequestParams()
|
|
{
|
|
ConnectionUri = connectionUri,
|
|
ObjectUrn = objUrn,
|
|
DropConnections = false,
|
|
DeleteBackupHistory = false,
|
|
GenerateScript = true
|
|
};
|
|
|
|
var expectedDeleteScript = $"DROP DATABASE [{testDatabase.Name}]";
|
|
var expectedAlterScript = $"ALTER DATABASE [{testDatabase.Name}] SET SINGLE_USER WITH ROLLBACK IMMEDIATE";
|
|
var expectedBackupScript = $"EXEC msdb.dbo.sp_delete_database_backuphistory @database_name = N'{testDatabase.Name}'";
|
|
|
|
var actualScript = handler.Drop(deleteParams);
|
|
Assert.That(DatabaseExists(testDatabase.Name!, server), "Database should not have been deleted when just generating a script.");
|
|
Assert.That(actualScript, Does.Contain(expectedDeleteScript).IgnoreCase);
|
|
|
|
// Drop connections only
|
|
deleteParams.DropConnections = true;
|
|
actualScript = handler.Drop(deleteParams);
|
|
Assert.That(actualScript, Does.Contain(expectedDeleteScript).IgnoreCase);
|
|
Assert.That(actualScript, Does.Contain(expectedAlterScript).IgnoreCase);
|
|
|
|
// Delete backup/restore history only
|
|
deleteParams.DropConnections = false;
|
|
deleteParams.DeleteBackupHistory = true;
|
|
actualScript = handler.Drop(deleteParams);
|
|
Assert.That(actualScript, Does.Contain(expectedBackupScript).IgnoreCase);
|
|
|
|
// Both drop and update
|
|
deleteParams.DropConnections = true;
|
|
actualScript = handler.Drop(deleteParams);
|
|
Assert.That(actualScript, Does.Contain(expectedAlterScript).IgnoreCase);
|
|
Assert.That(actualScript, Does.Contain(expectedBackupScript).IgnoreCase);
|
|
}
|
|
finally
|
|
{
|
|
DropDatabase(server, testDatabase.Name!);
|
|
}
|
|
}
|
|
}
|
|
|
|
private bool DatabaseExists(string dbName, Server server)
|
|
{
|
|
server.Databases.Refresh();
|
|
bool dbFound = false;
|
|
foreach (Database db in server.Databases)
|
|
{
|
|
if (db.Name == dbName)
|
|
{
|
|
dbFound = true;
|
|
break;
|
|
}
|
|
}
|
|
return dbFound;
|
|
}
|
|
|
|
private void DropDatabase(Server server, string databaseName)
|
|
{
|
|
server.Databases.Refresh();
|
|
foreach (Database db in server.Databases)
|
|
{
|
|
if (db.Name == databaseName)
|
|
{
|
|
db.DropIfExists();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|