Skip to content

Surface still-running extensions during MTP shutdown (refs #5345)#8582

Draft
Evangelink wants to merge 1 commit into
mainfrom
dev/amauryleve/mtp-shutdown-progress
Draft

Surface still-running extensions during MTP shutdown (refs #5345)#8582
Evangelink wants to merge 1 commit into
mainfrom
dev/amauryleve/mtp-shutdown-progress

Conversation

@Evangelink
Copy link
Copy Markdown
Member

Closes part of #5345 (incremental improvement, complementary to the phased-shutdown RFC in #8580).

Problem

Today when a user hits Ctrl+C on a hanging MTP test run, the cancellation token gets signalled but the user has no visibility into which extension or consumer is blocking the shutdown. The process just appears stuck.

Approach

Add a small IShutdownProgressReporter service that:

  1. Tracks units of in-flight shutdown work via using (reporter.Track(uid, displayName, phase)) { await ... }.
  2. Once the test-application cancellation token fires, waits a short quiet window (3s) to avoid noise on clean shutdowns.
  3. Then periodically (every 1s) emits a Still waiting for: <Name1> (<Phase1>, <Ns>); <Name2> ... warning through IOutputDevice.

The three known blocking await sites are now wrapped:

  • ITestSessionLifetimeHandler.OnTestSessionFinishingAsync (non-consumer pass) — CommonTestHost
  • ITestSessionLifetimeHandler.OnTestSessionFinishingAsync (consumer pass) — CommonTestHost
  • IAsyncConsumerDataProcessor.DrainDataAsync per consumer — AsynchronousMessageBus

This is intentionally complementary to #8580. The RFC there proposes a full phased-shutdown protocol with a graceful timer and forced abort. The same trackers introduced here will feed the eventual Force-killed because X didn't drain message in that bigger work — but in the meantime this PR ships value independently with zero behavior change on the happy path.

Limitations

  • Does not surface slow user test code (only platform extensions/consumers expose Uid/DisplayName).
  • Output uses WarningMessageOutputDeviceData (yellow text); coordination with the live progress line is best-effort.
  • Strings are localized via PlatformResources.resx / regenerated .xlf files.

Tests

7 new unit tests in ShutdownProgressReporterTests covering snapshot semantics, idempotent dispose, quiet-window behavior, post-cancellation emission, no-emit-if-all-disposed, and post-disposal Track. All pass on net9.0 and net462.

Out of scope

  • Server-mode / IPC progress propagation.
  • CLI options to tune the quiet window or poll interval (constants for now).
  • Acceptance-test coverage (would require a long-sleeping fake extension; deferring until the RFC lands).

When the test-application cancellation token is signalled and shutdown takes longer than expected, MTP now periodically prints a 'Still waiting for: ...' warning listing the extensions and consumers that have not yet returned. This makes a hanging Ctrl+C observable without users having to inspect the process state.

Adds a new internal IShutdownProgressReporter service plus a default ShutdownProgressReporter implementation. The reporter wraps the three known blocking await sites:

* ITestSessionLifetimeHandler.OnTestSessionFinishingAsync (both non-consumer and consumer passes in CommonTestHost)
* IAsyncConsumerDataProcessor.DrainDataAsync per consumer in AsynchronousMessageBus

The watchdog only starts after the cancellation token fires, waits a quiet window (3s) to avoid noise on clean shutdowns, then polls every second. Output goes through IOutputDevice as a WarningMessageOutputDeviceData.

Refs #5345. Complementary to #8580 - the same trackers will feed the eventual `force-killed because X didn't drain` message once the phased-shutdown RFC lands.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Copilot AI review requested due to automatic review settings May 25, 2026 21:25
Copy link
Copy Markdown
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Adds a shutdown-progress tracking/reporting service to Microsoft.Testing.Platform (MTP) so that after Ctrl+C cancellation, the host can periodically warn which extensions/consumers are still blocking shutdown.

Changes:

  • Introduces IShutdownProgressReporter + default ShutdownProgressReporter watchdog that emits “Still waiting for …” warnings after a quiet window once cancellation is requested.
  • Wraps key shutdown await sites with tracking scopes (session finishing handlers + async consumer drain).
  • Adds unit tests and localized resource strings for the warning prefix.
Show a summary per file
File Description
test/UnitTests/Microsoft.Testing.Platform.UnitTests/Services/ShutdownProgressReporterTests.cs Adds unit tests covering tracking/snapshot and watchdog emission behavior.
src/Platform/Microsoft.Testing.Platform/Services/IShutdownProgressReporter.cs Introduces internal tracking interface for shutdown in-flight work.
src/Platform/Microsoft.Testing.Platform/Services/ShutdownProgressReporter.cs Implements watchdog that reports still-running shutdown work via output device + logging.
src/Platform/Microsoft.Testing.Platform/Hosts/TestHostBuilder.CommonServices.cs Registers ShutdownProgressReporter in the common service provider (after output device).
src/Platform/Microsoft.Testing.Platform/Hosts/CommonTestHost.cs Tracks OnTestSessionFinishingAsync awaits (non-consumer + consumer passes).
src/Platform/Microsoft.Testing.Platform/Messages/AsynchronousMessageBus.cs Tracks per-consumer DrainDataAsync awaits; adds ctor overload to accept reporter.
src/Platform/Microsoft.Testing.Platform/Hosts/TestHostBuilder.Framework.cs Passes IShutdownProgressReporter into AsynchronousMessageBus construction.
src/Platform/Microsoft.Testing.Platform/Hosts/TestHostControllersTestHost.cs Passes IShutdownProgressReporter into AsynchronousMessageBus construction.
src/Platform/Microsoft.Testing.Platform/Resources/PlatformResources.resx Adds localized resource key for “Still waiting for: ” prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.cs.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.de.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.es.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.fr.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.it.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.ja.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.ko.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.pl.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.pt-BR.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.ru.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.tr.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.zh-Hans.xlf Adds new trans-unit for the shutdown progress prefix.
src/Platform/Microsoft.Testing.Platform/Resources/xlf/PlatformResources.zh-Hant.xlf Adds new trans-unit for the shutdown progress prefix.

Copilot's findings

  • Files reviewed: 22/22 changed files
  • Comments generated: 0

Comment on lines +148 to +151
catch
{
// Swallow - we are during shutdown.
}
Comment on lines +178 to +181
catch
{
// Logging during shutdown is best-effort.
}
Comment on lines +193 to +196
catch
{
// Output during shutdown is best-effort.
}
Comment on lines +100 to +102
catch (ObjectDisposedException)
{
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants