From 4eec1d31e43cb10b4fdc0b72b79f89c6543cc5e9 Mon Sep 17 00:00:00 2001
From: Graham Watts <grahamcharleswatts@gmail.com>
Date: Wed, 12 Jan 2022 19:49:52 +0200
Subject: [PATCH] Add pre/post test hooks

---
 .../AdapterTestsBase.cs                       | 43 +++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/src/DataCore.Adapter.Tests.Helpers/AdapterTestsBase.cs b/src/DataCore.Adapter.Tests.Helpers/AdapterTestsBase.cs
index de3947db..50ae7303 100644
--- a/src/DataCore.Adapter.Tests.Helpers/AdapterTestsBase.cs
+++ b/src/DataCore.Adapter.Tests.Helpers/AdapterTestsBase.cs
@@ -74,6 +74,47 @@ protected virtual IAdapterCallContext CreateCallContext(TestContext context) {
         }
 
 
+        /// <summary>
+        /// Called immediately prior to the callback in <see cref="RunAdapterTest"/>, after the 
+        /// <paramref name="adapter"/> has been started.
+        /// </summary>
+        /// <param name="adapter">
+        ///   The adapter for the test.
+        /// </param>
+        /// <param name="context">
+        ///   The call context for the test.
+        /// </param>
+        /// <param name="cancellationToken">
+        ///   The cancellation token for the test.
+        /// </param>
+        /// <returns>
+        ///   A <see cref="Task"/> that will perform any pre-test actions.
+        /// </returns>
+        protected virtual Task BeforeAdapterTestAsync(TAdapter adapter, IAdapterCallContext context, CancellationToken cancellationToken) {
+            return Task.CompletedTask;
+        }
+
+
+        /// <summary>
+        /// Called immediately after the callback in <see cref="RunAdapterTest"/>.
+        /// </summary>
+        /// <param name="adapter">
+        ///   The adapter for the test.
+        /// </param>
+        /// <param name="context">
+        ///   The call context for the test.
+        /// </param>
+        /// <param name="cancellationToken">
+        ///   The cancellation token for the test.
+        /// </param>
+        /// <returns>
+        ///   A <see cref="Task"/> that will perform any post-test actions.
+        /// </returns>
+        protected virtual Task AfterAdapterTestAsync(TAdapter adapter, IAdapterCallContext context, CancellationToken cancellationToken) {
+            return Task.CompletedTask;
+        }
+
+
         /// <summary>
         /// Creates and initialises a <typeparamref name="TAdapter"/> instance and runs an adapter 
         /// test.
@@ -104,7 +145,9 @@ protected async Task RunAdapterTest(Func<TAdapter, IAdapterCallContext, Cancella
                         await adapter.StartAsync(CancellationToken).ConfigureAwait(false);
                     }
                     var context = CreateCallContext(TestContext);
+                    await BeforeAdapterTestAsync(adapter, context, CancellationToken).ConfigureAwait(false);
                     await callback(adapter, context, CancellationToken).ConfigureAwait(false);
+                    await AfterAdapterTestAsync(adapter, context, CancellationToken).ConfigureAwait(false);
                 }
                 finally {
                     if (adapter is IAsyncDisposable iad) {