fix: mark timed out migrations as failed

Author: jorgeparavicini

src/Rapidata.MongoDB.Migrations/Services/MigrationService.cs

@@ -68,6 +68,15 @@ public async Task ExecuteMigration(string databaseName, IMigration migration, Ca
 
             var client = _mongoClientProvider.GetClient();
             var database = client.GetDatabase(databaseName);
+            
+            // Merge cancellation token with timeout from config
+            if (_config.MigrationTimeout is TimeSpan timeout)
+            {
+                var newCancellationTokenSource = new CancellationTokenSource(timeout);
+                cancellationToken = CancellationTokenSource
+                    .CreateLinkedTokenSource(cancellationToken, newCancellationTokenSource.Token)
+                    .Token;
+            }
 
             await migration.Migrate(database, cancellationToken)
                 .ConfigureAwait(false);

tests/Rapidata.MongoDB.Migrations.Tests.Integration/Core/MigrationEngineTests.cs

@@ -339,4 +339,24 @@ public async Task Migrate_WhenMigrationWithSameVersionButDifferentDateAlreadyApp
         var count = await DefaultCollection.CountDocumentsAsync(FilterDefinition<Migration>.Empty);
         count.Should().Be(2);
     }
+    
+    [Test]
+    public async Task Migrate_WhenMigrationRunsIntoTimeout_MarksMigrationAsFailed()
+    {
+        // Arrange
+        var migration = new MigrationMockBuilder()
+            .WithAction(async (_, cancellationToken) =>
+            {
+                await Task.Delay(1000, cancellationToken);
+            })
+            .Build();
+        Setup(configure: builder => builder.WithMigrationTimeout(TimeSpan.FromMilliseconds(200)), migration.Object);
+
+        // Act
+        await Subject.Migrate(CancellationToken.None);
+
+        // Assert
+        var appliedMigration = await DefaultCollection.Find(FilterDefinition<Migration>.Empty).FirstAsync();
+        appliedMigration.State.Should().Be(MigrationState.Failed);
+    }
 }