public async Task Should_delete_staging_queue_when_completing()
{
var endpointName = "FakeEndpoint";
var sut = new MsmqTarget(new TestLoggingAdapter(), connectionString, schema);
await using var endpointTarget = await sut.PrepareTargetEndpointBatchMigrator(endpointName);
await sut.Complete(endpointName);
await using var connection = new SqlConnection(connectionString);
await connection.OpenAsync();
await using var command = connection.CreateCommand();
command.CommandText = string.Format(@"
SELECT COUNT(*)
FROM INFORMATION_SCHEMA.TABLES
WHERE TABLE_SCHEMA = '{1}' AND TABLE_NAME = '{0}' AND TABLE_CATALOG = '{2}'
", MsmqSqlConstants.TimeoutMigrationStagingTable, schema, databaseName);
var result = await command.ExecuteScalarAsync() as int?;
Assert.That(Convert.ToBoolean(result), Is.False);
}
public async Task Should_migrate_into_delayed_table()
{
var endpointDelayedTableName = MsmqSqlConstants.DelayedTableName(ExistingEndpointName);
var sut = new MsmqTarget(new TestLoggingAdapter(), connectionString, schema);
await using var connection = new SqlConnection(connectionString);
await connection.OpenAsync();
await using var command = connection.CreateCommand();
command.CommandText = string.Format(@"
CREATE TABLE [{1}].[{0}] (
Id nvarchar(250) not null primary key,
Destination nvarchar(200),
State varbinary(max),
Time datetime,
Headers varbinary(max) not null,
RetryCount INT NOT NULL default(0)
);
", endpointDelayedTableName, schema);
await command.ExecuteNonQueryAsync();
const int batchNumber = 2;
await using var endpointTarget = await sut.PrepareTargetEndpointBatchMigrator(ExistingEndpointName);
await endpointTarget.StageBatch(new List <TimeoutData>
{
new TimeoutData
{
Id = "SomeID",
Headers = new Dictionary <string, string>
{
{ "NServiceBus.MessageId", "SomeMessageId" }
},
Destination = "SomeDestination",
State = new byte[2],
Time = new DateTime(2021, 12, 12, 12, 12, 12, DateTimeKind.Utc)
},
new TimeoutData
{
Id = "SomeOtherId",
Headers = new Dictionary <string, string>
{
{ "NServiceBus.MessageId", "SomeOtherMessageId" }
},
Destination = "SomeOtherDestination",
State = new byte[2],
Time = new DateTime(2021, 12, 12, 12, 13, 13, DateTimeKind.Utc)
},
}, batchNumber);
await endpointTarget.CompleteBatch(batchNumber);
var endpointDelayedTableDataTable = new DataTable();
using var endpointDelayedTableDataAdapter = new SqlDataAdapter(string.Format("SELECT * FROM [{1}].[{0}]", endpointDelayedTableName, schema), connection);
endpointDelayedTableDataAdapter.Fill(endpointDelayedTableDataTable);
var migrationTableDataTable = new DataTable();
using var migrationTableDataAdapter = new SqlDataAdapter(string.Format("SELECT * FROM [{1}].[{0}]", MsmqSqlConstants.TimeoutMigrationStagingTable, schema), connection);
migrationTableDataAdapter.Fill(migrationTableDataTable);
Approver.Verify(endpointDelayedTableDataTable.Rows.OfType <DataRow>().SelectMany(r => r.ItemArray.Take(3)));
Assert.IsEmpty(migrationTableDataTable.Rows);
}
