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/ScanResultServiceTests.cs
+++ b/MalwareMultiScan.Tests/Api/ScanResultServiceTests.cs
@@ -0,0 +1,121 @@
+using System.IO;
+using System.Threading.Tasks;
+using MalwareMultiScan.Api.Data.Configuration;
+using MalwareMultiScan.Api.Data.Models;
+using MalwareMultiScan.Api.Services.Implementations;
+using MalwareMultiScan.Api.Services.Interfaces;
+using Mongo2Go;
+using MongoDB.Bson;
+using MongoDB.Driver;
+using MongoDB.Driver.GridFS;
+using Moq;
+using NUnit.Framework;
+
+namespace MalwareMultiScan.Tests.Api
+{
+    public class ScanResultServiceTest
+    {
+        private MongoDbRunner _mongoDbRunner;
+        private IScanResultService _resultService;
+        private Mock<IScanBackendService> _scanBackendService;
+
+        [SetUp]
+        public void SetUp()
+        {
+            _mongoDbRunner = MongoDbRunner.Start();
+
+            var connection = new MongoClient(_mongoDbRunner.ConnectionString);
+            var database = connection.GetDatabase("Test");
+            var gridFsBucket = new GridFSBucket(database);
+
+            _scanBackendService = new Mock<IScanBackendService>();
+
+            _scanBackendService
+                .SetupGet(x => x.List)
+                .Returns(() => new[]
+                {
+                    new ScanBackend {Id = "dummy", Enabled = true},
+                    new ScanBackend {Id = "clamav", Enabled = true},
+                    new ScanBackend {Id = "disabled", Enabled = false}
+                });
+
+            _resultService = new ScanResultService(
+                database, gridFsBucket, _scanBackendService.Object);
+        }
+
+        [TearDown]
+        public void TearDown()
+        {
+            _mongoDbRunner.Dispose();
+        }
+
+        [Test]
+        public async Task TestFileStorageStoreObtain()
+        {
+            var originalData = new byte[] {0xDE, 0xAD, 0xBE, 0xEF};
+
+            string fileId;
+
+            await using (var dataStream = new MemoryStream(originalData))
+            {
+                fileId = await _resultService.StoreFile("test.txt", dataStream);
+            }
+
+            Assert.NotNull(fileId);
+
+            Assert.IsNull(await _resultService.ObtainFile("wrong-id"));
+            Assert.IsNull(await _resultService.ObtainFile(ObjectId.GenerateNewId().ToString()));
+
+            await using var fileSteam = await _resultService.ObtainFile(fileId);
+
+            var readData = new[]
+            {
+                (byte) fileSteam.ReadByte(),
+                (byte) fileSteam.ReadByte(),
+                (byte) fileSteam.ReadByte(),
+                (byte) fileSteam.ReadByte()
+            };
+
+            Assert.That(readData, Is.EquivalentTo(originalData));
+        }
+
+        [Test]
+        public async Task TestScanResultCreateGet()
+        {
+            var result = await _resultService.CreateScanResult();
+
+            Assert.NotNull(result.Id);
+            Assert.That(result.Results, Contains.Key("dummy"));
+
+            await _resultService.UpdateScanResultForBackend(
+                result.Id, "dummy", 100, true, true, new[] {"Test"});
+
+            result = await _resultService.GetScanResult(result.Id);
+
+            Assert.NotNull(result.Id);
+            Assert.That(result.Results, Contains.Key("dummy"));
+
+            var dummyResult = result.Results["dummy"];
+
+            Assert.IsTrue(dummyResult.Completed);
+            Assert.IsTrue(dummyResult.Succeeded);
+            Assert.AreEqual(dummyResult.Duration, 100);
+            Assert.IsNotEmpty(dummyResult.Threats);
+            Assert.That(dummyResult.Threats, Is.EquivalentTo(new[] {"Test"}));
+        }
+
+        [Test]
+        public async Task TestQueueUrlScan()
+        {
+            var result = await _resultService.CreateScanResult();
+
+            await _resultService.QueueUrlScan(
+                result, "http://url.com");
+
+            _scanBackendService.Verify(x => x.QueueUrlScan(
+                It.IsAny<ScanResult>(),
+                It.IsAny<ScanBackend>(),
+                "http://url.com"), Times.Exactly(2));
+        }
+    }
+}