Joseph Doherty
49 lines
1.9 KiB
C#
49 lines
1.9 KiB
C#
using Microsoft.Extensions.DependencyInjection;
|
|
using ZB.MOM.WW.ScadaBridge.SiteEventLogging;
|
|
|
|
namespace ZB.MOM.WW.ScadaBridge.SiteRuntime.Tests.TestSupport;
|
|
|
|
/// <summary>
|
|
/// WP3.1 parity pin 3: an <see cref="IServiceProvider"/> that counts
|
|
/// <see cref="IServiceScopeFactory.CreateScope"/> calls and each scope's
|
|
/// <see cref="IDisposable.Dispose"/>, so a test can assert that one script run creates
|
|
/// exactly one DI scope and disposes it exactly once — on the success, failure, AND timeout
|
|
/// paths alike. Removing the per-run execution actor moved the scope's <c>finally</c> into
|
|
/// <see cref="ZB.MOM.WW.ScadaBridge.SiteRuntime.Scripts.ScriptRunLauncher"/>; a scope leaked
|
|
/// there would silently leak every scoped service a script touches.
|
|
/// </summary>
|
|
public sealed class ScopeSpyServiceProvider(ISiteEventLogger? logger = null)
|
|
: IServiceProvider, IServiceScopeFactory
|
|
{
|
|
private int _scopesCreated;
|
|
private int _scopesDisposed;
|
|
|
|
/// <summary>Scopes created so far.</summary>
|
|
public int ScopesCreated => Volatile.Read(ref _scopesCreated);
|
|
|
|
/// <summary>Scope disposals observed so far (double-disposal is counted twice, deliberately).</summary>
|
|
public int ScopesDisposed => Volatile.Read(ref _scopesDisposed);
|
|
|
|
/// <inheritdoc />
|
|
public object? GetService(Type serviceType)
|
|
{
|
|
if (serviceType == typeof(ISiteEventLogger)) return logger;
|
|
if (serviceType == typeof(IServiceScopeFactory)) return this;
|
|
return null;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public IServiceScope CreateScope()
|
|
{
|
|
Interlocked.Increment(ref _scopesCreated);
|
|
return new SpyScope(this);
|
|
}
|
|
|
|
private sealed class SpyScope(ScopeSpyServiceProvider owner) : IServiceScope
|
|
{
|
|
public IServiceProvider ServiceProvider => owner;
|
|
|
|
public void Dispose() => Interlocked.Increment(ref owner._scopesDisposed);
|
|
}
|
|
}
|