finished unit tests and docstrings

2025-08-25 14:01:13 +00:00 · 2020-10-29 16:09:56 +02:00
parent b2902c128a
commit b68c285ce5
48 changed files with 910 additions and 194 deletions
--- a/MalwareMultiScan.Tests/Api/ControllersTests.cs
+++ b/MalwareMultiScan.Tests/Api/ControllersTests.cs
@@ -0,0 +1,103 @@
+using System.IO;
+using System.Threading.Tasks;
+using MalwareMultiScan.Api.Controllers;
+using MalwareMultiScan.Api.Data.Models;
+using MalwareMultiScan.Api.Services.Interfaces;
+using Microsoft.AspNetCore.Http;
+using Microsoft.AspNetCore.Mvc;
+using Moq;
+using NUnit.Framework;
+
+namespace MalwareMultiScan.Tests.Api
+{
+    public class ControllersTests
+    {
+        private DownloadController _downloadController;
+        private QueueController _queueController;
+        private ScanResultsController _scanResultsController;
+        private Mock<IScanResultService> _scanResultServiceMock;
+
+        [SetUp]
+        public void SetUp()
+        {
+            _scanResultServiceMock = new Mock<IScanResultService>();
+
+            _scanResultServiceMock
+                .Setup(x => x.ObtainFile("existing"))
+                .Returns(Task.FromResult(new MemoryStream() as Stream));
+
+            _scanResultServiceMock
+                .Setup(x => x.ObtainFile("non-existing"))
+                .Returns(Task.FromResult((Stream) null));
+
+            _scanResultServiceMock
+                .Setup(x => x.GetScanResult("existing"))
+                .Returns(Task.FromResult(new ScanResult
+                {
+                    Id = "existing"
+                }));
+
+            _scanResultServiceMock
+                .Setup(x => x.GetScanResult("non-existing"))
+                .Returns(Task.FromResult((ScanResult) null));
+
+            _scanResultServiceMock
+                .Setup(x => x.ObtainFile("non-existing"))
+                .Returns(Task.FromResult((Stream) null));
+
+            _scanResultServiceMock
+                .Setup(x => x.CreateScanResult())
+                .Returns(Task.FromResult(new ScanResult()));
+
+            _downloadController = new DownloadController(_scanResultServiceMock.Object);
+            _scanResultsController = new ScanResultsController(_scanResultServiceMock.Object);
+
+            _queueController = new QueueController(_scanResultServiceMock.Object)
+            {
+                Url = Mock.Of<IUrlHelper>()
+            };
+        }
+
+        [Test]
+        public async Task TestFileDownload()
+        {
+            Assert.True(await _downloadController.Index("existing") is FileStreamResult);
+            Assert.True(await _downloadController.Index("non-existing") is NotFoundResult);
+        }
+
+        [Test]
+        public async Task TestScanResults()
+        {
+            Assert.True(
+                await _scanResultsController.Index("existing") is OkObjectResult okObjectResult &&
+                okObjectResult.Value is ScanResult scanResult && scanResult.Id == "existing");
+
+            Assert.True(await _scanResultsController.Index("non-existing") is NotFoundResult);
+        }
+
+        [Test]
+        public async Task TestFileQueue()
+        {
+            Assert.True(await _queueController.ScanFile(
+                Mock.Of<IFormFile>(x =>
+                    x.FileName == "test.exe" &&
+                    x.OpenReadStream() == new MemoryStream())
+            ) is CreatedAtActionResult);
+
+            _scanResultServiceMock.Verify(
+                x => x.StoreFile("test.exe", It.IsAny<Stream>()), Times.Once);
+
+            _scanResultServiceMock.Verify(
+                x => x.QueueUrlScan(It.IsAny<ScanResult>(), It.IsAny<string>()), Times.Once);
+        }
+
+        [Test]
+        public async Task TestUrlQueue()
+        {
+            Assert.True(await _queueController.ScanUrl("http://test.com") is CreatedAtActionResult);
+
+            _scanResultServiceMock.Verify(
+                x => x.QueueUrlScan(It.IsAny<ScanResult>(), "http://test.com"), Times.Once);
+        }
+    }
+}