dtos-service-layer/tests/ServiceLayer.Mesh.Tests/Functions/FileRetryFunctionTests.cs at a7ebf88c2750457710ca3da9aee92966c195eb4d · NHSDigital/dtos-service-layer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
using Microsoft.Azure.Functions.Worker;
using Moq;
using ServiceLayer.Data.Models;
using ServiceLayer.Mesh.Functions;
using ServiceLayer.Mesh.Messaging;
using ServiceLayer.Mesh.Configuration;

namespace ServiceLayer.Mesh.Tests.Functions;

public class FileRetryFunctionTests : FunctionTestBase<FileRetryFunction>
{
    private readonly Mock<IFileExtractQueueClient> _fileExtractQueueClientMock;
    private readonly Mock<IFileTransformQueueClient> _fileTransformQueueClientMock;
    private readonly FileRetryFunction _function;

    public FileRetryFunctionTests()
    {
        _fileExtractQueueClientMock = new Mock<IFileExtractQueueClient>();
        _fileTransformQueueClientMock = new Mock<IFileTransformQueueClient>();

        var configurationMock = new Mock<IFileRetryFunctionConfiguration>();
        configurationMock.Setup(c => c.StaleHours).Returns(12);

        _function = new FileRetryFunction(
            LoggerMock.Object,
            DbContext,
            _fileExtractQueueClientMock.Object,
            _fileTransformQueueClientMock.Object,
            configurationMock.Object
        );
    }

    [Theory]
    [InlineData(MeshFileStatus.Discovered)]
    [InlineData(MeshFileStatus.Extracting)]
    public async Task Run_EnqueuesDiscoveredOrExtractingFilesOlderThan12Hours(MeshFileStatus testStatus)
    {
        // Arrange
        var file = SaveMeshFile(testStatus, 13);

        // Act
        await _function.Run(new TimerInfo());

        // Assert
        _fileExtractQueueClientMock.Verify(q => q.EnqueueFileExtractAsync(It.Is<MeshFile>(f => f.FileId == file.FileId)), Times.Once);
        _fileTransformQueueClientMock.Verify(q => q.EnqueueFileTransformAsync(It.Is<MeshFile>(f => f.FileId == file.FileId)), Times.Never);

        AssertFileUpdated(file.FileId, testStatus);
    }

    [Theory]
    [InlineData(MeshFileStatus.Extracted)]
    [InlineData(MeshFileStatus.Transforming)]
    public async Task Run_EnqueuesExtractedOrTransformingFilesOlderThan12Hours(MeshFileStatus testStatus)
    {
        // Arrange
        var file = SaveMeshFile(testStatus, 13);

        // Act
        await _function.Run(new TimerInfo());

        // Assert
        _fileTransformQueueClientMock.Verify(q => q.EnqueueFileTransformAsync(It.Is<MeshFile>(f => f.FileId == file.FileId)), Times.Once);
        _fileExtractQueueClientMock.Verify(q => q.EnqueueFileExtractAsync(It.Is<MeshFile>(f => f.FileId == file.FileId)), Times.Never);

        AssertFileUpdated(file.FileId, testStatus);
    }

    [Theory]
    [InlineData(MeshFileStatus.Discovered)]
    [InlineData(MeshFileStatus.Extracting)]
    [InlineData(MeshFileStatus.Extracted)]
    [InlineData(MeshFileStatus.Transforming)]
    public async Task Run_SkipsFreshFiles(MeshFileStatus testStatus)
    {
        // Arrange
        var file = SaveMeshFile(testStatus);

        // Act
        await _function.Run(new TimerInfo());

        // Assert
        _fileExtractQueueClientMock.Verify(q => q.EnqueueFileExtractAsync(It.IsAny<MeshFile>()), Times.Never);
        _fileTransformQueueClientMock.Verify(q => q.EnqueueFileTransformAsync(It.IsAny<MeshFile>()), Times.Never);

        AssertFileUnchanged(file.FileId, file.Status, file.LastUpdatedUtc);
    }

    [Theory]
    [InlineData(MeshFileStatus.Transformed)]
    [InlineData(MeshFileStatus.FailedExtract)]
    [InlineData(MeshFileStatus.FailedTransform)]
    public async Task Run_IgnoresFilesInOtherStatuses(MeshFileStatus ignoredStatus)
    {
        // Arrange
        SaveMeshFile(ignoredStatus, 20);

        // Act
        await _function.Run(new TimerInfo());

        // Assert
        _fileTransformQueueClientMock.Verify(q => q.EnqueueFileTransformAsync(It.IsAny<MeshFile>()), Times.Never);
        _fileExtractQueueClientMock.Verify(q => q.EnqueueFileExtractAsync(It.IsAny<MeshFile>()), Times.Never);
    }

    [Fact]
    public async Task Run_IfNoFilesFoundDoNothing()
    {
        // Act
        await _function.Run(new TimerInfo());

        // Assert
        _fileExtractQueueClientMock.Verify(q => q.EnqueueFileExtractAsync(It.IsAny<MeshFile>()), Times.Never);
        _fileTransformQueueClientMock.Verify(q => q.EnqueueFileTransformAsync(It.IsAny<MeshFile>()), Times.Never);
    }

    [Fact]
    public async Task Run_ProcessesMultipleEligibleFiles()
    {
        // Arrange
        var file1 = SaveMeshFile(MeshFileStatus.Discovered, 13);
        var file2 = SaveMeshFile(MeshFileStatus.Extracted, 13);
        var file3 = SaveMeshFile(MeshFileStatus.Transforming, 13);

        // Act
        await _function.Run(new TimerInfo());

        // Assert
        _fileExtractQueueClientMock.Verify(q => q.EnqueueFileExtractAsync(It.Is<MeshFile>(f => f.FileId == file1.FileId)), Times.Once);
        _fileTransformQueueClientMock.Verify(q => q.EnqueueFileTransformAsync(It.Is<MeshFile>(f => f.FileId == file2.FileId)), Times.Once);
        _fileTransformQueueClientMock.Verify(q => q.EnqueueFileTransformAsync(It.Is<MeshFile>(f => f.FileId == file3.FileId)), Times.Once);

        AssertFileUpdated(file1.FileId, MeshFileStatus.Discovered);
        AssertFileUpdated(file2.FileId, MeshFileStatus.Extracted);
        AssertFileUpdated(file3.FileId, MeshFileStatus.Transforming);
    }
}