Add cancellation token handling

OddsCollector.Functions.Tests/Infrastructure/CancellationToken/CancellationTokenGenerator.cs

@@ -0,0 +1,13 @@
+namespace OddsCollector.Functions.Tests.Infrastructure.CancellationToken;
+
+internal static class CancellationTokenGenerator
+{
+    public static async Task<System.Threading.CancellationToken> GetRequestedForCancellationToken()
+    {
+        var cancellationTokenSource = new CancellationTokenSource();
+
+        await cancellationTokenSource.CancelAsync();
+
+        return cancellationTokenSource.Token;
+    }
+}

OddsCollector.Functions.Tests/Infrastructure/Logger/LoggerFactory.cs

@@ -0,0 +1,11 @@
+using Microsoft.Extensions.Logging;
+
+namespace OddsCollector.Functions.Tests.Infrastructure.Logger;
+
+internal static class LoggerFactory
+{
+    public static ILogger<T> GetLoggerMock<T>() where T : class
+    {
+        return Substitute.For<ILogger<T>>();
+    }
+}

OddsCollector.Functions.Tests/Infrastructure/ServiceBus/ServiceBusMessageActionFactory.cs

@@ -0,0 +1,11 @@
+using Microsoft.Azure.Functions.Worker;
+
+namespace OddsCollector.Functions.Tests.Infrastructure.ServiceBus;
+
+internal static class ServiceBusMessageActionsFactory
+{
+    public static ServiceBusMessageActions GetServiceBusMessageActionsMock()
+    {
+        return Substitute.For<ServiceBusMessageActions>();
+    }
+}

OddsCollector.Functions.Tests/Infrastructure/ServiceBus/ServiceBusReceivedMessageFactory.cs

@@ -7,7 +7,12 @@ namespace OddsCollector.Functions.Tests.Infrastructure.ServiceBus;
 
 internal static class ServiceBusReceivedMessageFactory
 {
-    public static ServiceBusReceivedMessage CreateFromObject(object obj)
+    public static IEnumerable<ServiceBusReceivedMessage> CreateFromObjects(IEnumerable<object> objects)
+    {
+        return objects.Select(CreateFromObject);
+    }
+
+    private static ServiceBusReceivedMessage CreateFromObject(object obj)
     {
         var serialized = Encoding.ASCII.GetBytes(JsonSerializer.Serialize(obj))
             .Select(x => new ReadOnlyMemory<byte>([x]));

OddsCollector.Functions.Tests/Tests/Functions/EventResultsFunction.cs

@@ -3,6 +3,7 @@
 using NSubstitute.ExceptionExtensions;
 using OddsCollector.Functions.Models;
 using OddsCollector.Functions.OddsApi;
+using LoggerFactory = OddsCollector.Functions.Tests.Infrastructure.Logger.LoggerFactory;
 
 namespace OddsCollector.Functions.Tests.Tests.Functions;
 
@@ -15,18 +16,16 @@ public async Task Run_WithValidParameters_ReturnsEventResults()
         // Arrange
         IEnumerable<EventResult> expectedEventResults = new List<EventResult> { new() };
 
-        var loggerMock = Substitute.For<ILogger<OddsCollector.Functions.Functions.EventResultsFunction>>();
+        var loggerMock = LoggerFactory.GetLoggerMock<OddsCollector.Functions.Functions.EventResultsFunction>();
 
-        var clientStub = Substitute.For<IOddsApiClient>();
-        clientStub.GetEventResultsAsync(Arg.Any<Guid>(), Arg.Any<DateTime>(), Arg.Any<CancellationToken>())
-            .Returns(Task.FromResult(expectedEventResults));
+        var clientStub = GetOddsApiClientStub(expectedEventResults);
 
         var function = new OddsCollector.Functions.Functions.EventResultsFunction(loggerMock, clientStub);
 
-        var token = new CancellationToken();
+        var cancellationToken = new CancellationToken();
 
         // Act
-        var eventResults = await function.Run(token);
+        var eventResults = await function.Run(cancellationToken);
 
         // Assert
         eventResults.Should().NotBeNull().And.BeEquivalentTo(expectedEventResults);
@@ -40,18 +39,18 @@ public async Task Run_WithException_ReturnsEmptyEventResults()
         // Arrange
         var exception = new Exception();
 
-        var loggerMock = Substitute.For<ILogger<OddsCollector.Functions.Functions.EventResultsFunction>>();
+        var loggerMock = LoggerFactory.GetLoggerMock<OddsCollector.Functions.Functions.EventResultsFunction>();
 
         var clientStub = Substitute.For<IOddsApiClient>();
         clientStub.GetEventResultsAsync(Arg.Any<Guid>(), Arg.Any<DateTime>(), Arg.Any<CancellationToken>())
             .Throws(exception);
 
         var function = new OddsCollector.Functions.Functions.EventResultsFunction(loggerMock, clientStub);
 
-        var token = new CancellationToken();
+        var cancellationToken = new CancellationToken();
 
         // Act
-        var eventResults = await function.Run(token);
+        var eventResults = await function.Run(cancellationToken);
 
         // Assert
         eventResults.Should().NotBeNull().And.BeEmpty();
@@ -66,22 +65,30 @@ public async Task Run_WithValidParameters_ReturnsNoEventResults()
         // Arrange
         IEnumerable<EventResult> expectedEventResults = new List<EventResult>();
 
-        var loggerMock = Substitute.For<ILogger<OddsCollector.Functions.Functions.EventResultsFunction>>();
+        var loggerMock = LoggerFactory.GetLoggerMock<OddsCollector.Functions.Functions.EventResultsFunction>();
 
-        var clientStub = Substitute.For<IOddsApiClient>();
-        clientStub.GetEventResultsAsync(Arg.Any<Guid>(), Arg.Any<DateTime>(), Arg.Any<CancellationToken>())
-            .Returns(Task.FromResult(expectedEventResults));
+        var clientStub = GetOddsApiClientStub(expectedEventResults);
 
         var function = new OddsCollector.Functions.Functions.EventResultsFunction(loggerMock, clientStub);
 
-        var token = new CancellationToken();
+        var cancellationToken = new CancellationToken();
 
         // Act
-        var eventResults = await function.Run(token);
+        var eventResults = await function.Run(cancellationToken);
 
         // Assert
         eventResults.Should().NotBeNull().And.BeEquivalentTo(expectedEventResults);
 
         loggerMock.ReceivedWithAnyArgs().LogWarning(string.Empty, 1);
     }
+
+    private static IOddsApiClient GetOddsApiClientStub(IEnumerable<EventResult> expectedEventResults)
+    {
+        var clientStub = Substitute.For<IOddsApiClient>();
+
+        clientStub.GetEventResultsAsync(Arg.Any<Guid>(), Arg.Any<DateTime>(), Arg.Any<CancellationToken>())
+            .Returns(Task.FromResult(expectedEventResults));
+
+        return clientStub;
+    }
 }

OddsCollector.Functions.Tests/Tests/Functions/PredictionFunction.cs

@@ -1,10 +1,9 @@
-using Azure.Messaging.ServiceBus;
-using Microsoft.Azure.Functions.Worker;
-using Microsoft.Extensions.Logging;
-using NSubstitute.ReceivedExtensions;
+using NSubstitute.ReceivedExtensions;
 using OddsCollector.Functions.Models;
 using OddsCollector.Functions.Strategies;
+using OddsCollector.Functions.Tests.Infrastructure.CancellationToken;
 using OddsCollector.Functions.Tests.Infrastructure.Data;
+using OddsCollector.Functions.Tests.Infrastructure.Logger;
 using OddsCollector.Functions.Tests.Infrastructure.ServiceBus;
 
 namespace OddsCollector.Functions.Tests.Tests.Functions;
@@ -15,16 +14,18 @@ internal class PredictionFunction
     public async Task Run_WithValidServiceBusMessage_ReturnsEventPrediction()
     {
         // Arrange
-        var loggerStub = Substitute.For<ILogger<OddsCollector.Functions.Functions.PredictionFunction>>();
+        var loggerStub = LoggerFactory.GetLoggerMock<OddsCollector.Functions.Functions.PredictionFunction>();
 
         var upcomingEvent = new UpcomingEventBuilder().SetSampleData().Instance;
 
         var expectedPrediction = new EventPredictionBuilder().SetSampleData().Instance;
 
-        ServiceBusReceivedMessage[] receivedMessages =
-            [ServiceBusReceivedMessageFactory.CreateFromObject(upcomingEvent)];
+        var receivedMessages =
+            ServiceBusReceivedMessageFactory.CreateFromObjects([
+                upcomingEvent
+            ]).ToArray();
 
-        var actionsMock = Substitute.For<ServiceBusMessageActions>();
+        var actionsMock = ServiceBusMessageActionsFactory.GetServiceBusMessageActionsMock();
 
         var strategyStub = Substitute.For<IPredictionStrategy>();
         strategyStub.GetPrediction(Arg.Any<UpcomingEvent>(), Arg.Any<DateTime>()).Returns(expectedPrediction);
@@ -43,11 +44,43 @@ public async Task Run_WithValidServiceBusMessage_ReturnsEventPrediction()
         await actionsMock.Received(Quantity.Exactly(1)).CompleteMessageAsync(receivedMessages[0], token);
     }
 
+    [Test]
+    public async Task Run_WithValidServiceBusMessageAndRequestedCancellation_ReturnsNoPredictions()
+    {
+        // Arrange
+        var loggerStub = LoggerFactory.GetLoggerMock<OddsCollector.Functions.Functions.PredictionFunction>();
+
+        var upcomingEvent = new UpcomingEventBuilder().SetSampleData().Instance;
+
+        var expectedPrediction = new EventPredictionBuilder().SetSampleData().Instance;
+
+        var receivedMessages =
+            ServiceBusReceivedMessageFactory.CreateFromObjects([
+                upcomingEvent
+            ]).ToArray();
+
+        var actionsMock = ServiceBusMessageActionsFactory.GetServiceBusMessageActionsMock();
+
+        var strategyStub = Substitute.For<IPredictionStrategy>();
+        strategyStub.GetPrediction(Arg.Any<UpcomingEvent>(), Arg.Any<DateTime>()).Returns(expectedPrediction);
+
+        var function = new OddsCollector.Functions.Functions.PredictionFunction(loggerStub, strategyStub);
+
+        var cancellationToken = await CancellationTokenGenerator.GetRequestedForCancellationToken();
+
+        // Act
+        var prediction = await function.Run(receivedMessages, actionsMock, cancellationToken)
+            .ConfigureAwait(false);
+
+        // Assert
+        prediction.Should().NotBeNull().And.HaveCount(0);
+    }
+
     [Test]
     public async Task Run_WithInvalidItems_ReturnsSuccessfullyProcessedEventPredictions()
     {
         // Arrange
-        var loggerMock = Substitute.For<ILogger<OddsCollector.Functions.Functions.PredictionFunction>>();
+        var loggerMock = LoggerFactory.GetLoggerMock<OddsCollector.Functions.Functions.PredictionFunction>();
 
         var goodUpcomingEvent = new UpcomingEventBuilder().SetSampleData().Instance;
 
@@ -56,13 +89,13 @@ public async Task Run_WithInvalidItems_ReturnsSuccessfullyProcessedEventPredicti
 
         var expectedPrediction = new EventPredictionBuilder().SetSampleData().Instance;
 
-        ServiceBusReceivedMessage[] receivedMessages =
-        [
-            ServiceBusReceivedMessageFactory.CreateFromObject(badUpcomingEvent),
-            ServiceBusReceivedMessageFactory.CreateFromObject(goodUpcomingEvent)
-        ];
+        var receivedMessages =
+            ServiceBusReceivedMessageFactory.CreateFromObjects([
+                badUpcomingEvent,
+                goodUpcomingEvent
+            ]).ToArray();
 
-        var actionsMock = Substitute.For<ServiceBusMessageActions>();
+        var actionsMock = ServiceBusMessageActionsFactory.GetServiceBusMessageActionsMock();
 
         var exception = new Exception();
 

OddsCollector.Functions.Tests/Tests/Functions/PredictionsHttpFunction.cs

@@ -2,9 +2,9 @@
 using System.Text.Json;
 using Microsoft.Azure.Functions.Worker;
 using Microsoft.Azure.Functions.Worker.Http;
-using Microsoft.Extensions.Logging;
 using OddsCollector.Functions.Models;
 using OddsCollector.Functions.Tests.Infrastructure.Data;
+using OddsCollector.Functions.Tests.Infrastructure.Logger;
 
 namespace OddsCollector.Functions.Tests.Tests.Functions;
 
@@ -14,7 +14,7 @@ internal class PredictionsHttpFunction
     public void Run_WithValidArguments_ReturnsValidResponse()
     {
         // Arrange
-        var loggerStub = Substitute.For<ILogger<OddsCollector.Functions.Functions.PredictionsHttpFunction>>();
+        var loggerStub = LoggerFactory.GetLoggerMock<OddsCollector.Functions.Functions.PredictionsHttpFunction>();
 
         var function = new OddsCollector.Functions.Functions.PredictionsHttpFunction(loggerStub);
 

OddsCollector.Functions.Tests/Tests/Functions/UpcomingEventsFunction.cs

@@ -3,6 +3,7 @@
 using NSubstitute.ExceptionExtensions;
 using OddsCollector.Functions.Models;
 using OddsCollector.Functions.OddsApi;
+using LoggerFactory = OddsCollector.Functions.Tests.Infrastructure.Logger.LoggerFactory;
 
 namespace OddsCollector.Functions.Tests.Tests.Functions;
 
@@ -15,7 +16,7 @@ public async Task Run_WithValidParameters_ReturnsEventResults()
         // Arrange
         IEnumerable<UpcomingEvent> expectedEventResults = new List<UpcomingEvent> { new() };
 
-        var loggerMock = Substitute.For<ILogger<OddsCollector.Functions.Functions.UpcomingEventsFunction>>();
+        var loggerMock = LoggerFactory.GetLoggerMock<OddsCollector.Functions.Functions.UpcomingEventsFunction>();
 
         var clientStub = Substitute.For<IOddsApiClient>();
         clientStub.GetUpcomingEventsAsync(Arg.Any<Guid>(), Arg.Any<DateTime>(), Arg.Any<CancellationToken>())
@@ -38,7 +39,7 @@ public async Task Run_WithException_ReturnsEmptyEventResults()
         // Arrange
         var exception = new Exception();
 
-        var loggerMock = Substitute.For<ILogger<OddsCollector.Functions.Functions.UpcomingEventsFunction>>();
+        var loggerMock = LoggerFactory.GetLoggerMock<OddsCollector.Functions.Functions.UpcomingEventsFunction>();
 
         var clientStub = Substitute.For<IOddsApiClient>();
         clientStub.GetUpcomingEventsAsync(Arg.Any<Guid>(), Arg.Any<DateTime>(), Arg.Any<CancellationToken>())
@@ -62,7 +63,7 @@ public async Task Run_WithValidParameters_ReturnsNoEventResults()
         // Arrange
         IEnumerable<UpcomingEvent> expectedEventResults = new List<UpcomingEvent>();
 
-        var loggerMock = Substitute.For<ILogger<OddsCollector.Functions.Functions.UpcomingEventsFunction>>();
+        var loggerMock = LoggerFactory.GetLoggerMock<OddsCollector.Functions.Functions.UpcomingEventsFunction>();
 
         var clientStub = Substitute.For<IOddsApiClient>();
         clientStub.GetUpcomingEventsAsync(Arg.Any<Guid>(), Arg.Any<DateTime>(), Arg.Any<CancellationToken>())

OddsCollector.Functions.Tests/Tests/OddsApi/OddsApiClient.cs

@@ -3,6 +3,7 @@
 using OddsCollector.Functions.OddsApi.Configuration;
 using OddsCollector.Functions.OddsApi.Converter;
 using OddsCollector.Functions.OddsApi.WebApi;
+using OddsCollector.Functions.Tests.Infrastructure.CancellationToken;
 
 namespace OddsCollector.Functions.Tests.Tests.OddsApi;
 
@@ -118,4 +119,80 @@ public async Task GetEventResultsAsync_WithLeagues_ReturnsEventResults()
         firstReceivedArguments[1].Should().Be(traceId);
         firstReceivedArguments[2].Should().Be(timestamp);
     }
+
+    [Test]
+    public async Task GetUpcomingEventsAsync_WithLeaguesAndRequestedCancellation_ReturnsNoUpcomingEvents()
+    {
+        // Arrange
+        ICollection<Anonymous2> rawUpcomingEvents = [new Anonymous2()];
+        var webApiClientMock = Substitute.For<IClient>();
+        webApiClientMock
+            .OddsAsync(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<Regions>(), Arg.Any<Markets>(),
+                Arg.Any<DateFormat>(), Arg.Any<OddsFormat>(), Arg.Any<string>(), Arg.Any<string>(),
+                Arg.Any<CancellationToken>())
+            .Returns(Task.FromResult(rawUpcomingEvents));
+
+        // ReSharper disable once CollectionNeverUpdated.Local
+        List<UpcomingEvent> upcomingEvents = [];
+        var converterMock = Substitute.For<IOddsApiObjectConverter>();
+        converterMock.ToUpcomingEvents(Arg.Any<ICollection<Anonymous2>?>(), Arg.Any<Guid>(), Arg.Any<DateTime>())
+            .Returns(new List<UpcomingEvent>());
+
+        const string secretValue = nameof(secretValue);
+
+        const string league = nameof(league);
+        var optionsStub = Substitute.For<IOptions<OddsApiClientOptions>>();
+        optionsStub.Value.Returns(new OddsApiClientOptions { Leagues = [league], ApiKey = secretValue });
+
+        var oddsClient =
+            new OddsCollector.Functions.OddsApi.OddsApiClient(optionsStub, webApiClientMock, converterMock);
+
+        var traceId = Guid.NewGuid();
+        var timestamp = DateTime.UtcNow;
+
+        var cancellationToken = await CancellationTokenGenerator.GetRequestedForCancellationToken();
+
+        // Act
+        var results = (await oddsClient.GetUpcomingEventsAsync(traceId, timestamp, cancellationToken)).ToList();
+
+        // Assert
+        results.Should().NotBeNull().And.HaveCount(0);
+    }
+
+    [Test]
+    public async Task GetEventResultsAsync_WithLeaguesAndRequestedCancellation_ReturnsNoEventResults()
+    {
+        // Arrange
+        ICollection<Anonymous3> rawEventResults = [new Anonymous3()];
+        var webApiClientMock = Substitute.For<IClient>();
+        webApiClientMock.ScoresAsync(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<int?>(),
+                Arg.Any<CancellationToken>())
+            .Returns(Task.FromResult(rawEventResults));
+
+        // ReSharper disable once CollectionNeverUpdated.Local
+        List<EventResult> eventResults = [];
+        var converterMock = Substitute.For<IOddsApiObjectConverter>();
+        converterMock.ToEventResults(Arg.Any<ICollection<Anonymous3>?>(), Arg.Any<Guid>(), Arg.Any<DateTime>())
+            .Returns(eventResults);
+
+        const string secretValue = nameof(secretValue);
+
+        const string league = nameof(league);
+        var optionsStub = Substitute.For<IOptions<OddsApiClientOptions>>();
+        optionsStub.Value.Returns(new OddsApiClientOptions { Leagues = [league], ApiKey = secretValue });
+
+        var oddsClient =
+            new OddsCollector.Functions.OddsApi.OddsApiClient(optionsStub, webApiClientMock, converterMock);
+
+        var traceId = Guid.NewGuid();
+        var timestamp = DateTime.UtcNow;
+
+        var cancellationToken = await CancellationTokenGenerator.GetRequestedForCancellationToken();
+
+        // Act
+        var results = (await oddsClient.GetEventResultsAsync(traceId, timestamp, cancellationToken)).ToList();
+
+        // Assert
+        results.Should().NotBeNull().And.HaveCount(0);
+    }
 }

OddsCollector.Functions/Functions/PredictionFunction.cs

@@ -20,6 +20,11 @@ public async Task<EventPrediction[]> Run(
 
         foreach (var message in messages)
         {
+            if (cancellationToken.IsCancellationRequested)
+            {
+                break;
+            }
+
             try
             {
                 var upcomingEvent = message.Body.ToObjectFromJson<UpcomingEvent>();