Fix content-type for ilivalidator plain text log

src/ILICheck.Web/Controllers/DownloadController.cs

@@ -6,6 +6,7 @@
 using System;
 using System.Collections.Generic;
 using System.IO;
+using System.Net.Mime;
 using System.Web;
 
 namespace ILICheck.Web.Controllers
@@ -40,8 +41,14 @@ public IActionResult Download(Guid jobId, LogType logType)
             try
             {
                 logger.LogInformation("Log file (<{LogType}>) for job identifier <{JobId}> requested.", HttpUtility.HtmlEncode(logType), jobId);
-                var contentType = logType == LogType.GeoJson ? "application/geo+json" : "text/xml; charset=utf-8";
-                return File(fileProvider.OpenText(fileProvider.GetLogFile(logType)).BaseStream, contentType);
+                var fileStream = fileProvider.OpenText(fileProvider.GetLogFile(logType)).BaseStream;
+                return logType switch
+                {
+                    LogType.Log => File(fileStream, MediaTypeNames.Text.Plain),
+                    LogType.Xtf => File(fileStream, "text/xml; charset=utf-8"),
+                    LogType.GeoJson => File(fileStream, "application/geo+json"),
+                    _ => throw new NotSupportedException($"Log type <{logType}> is not supported."),
+                };
             }
             catch (Exception)
             {

src/ILICheck.Web/PhysicalFileProvider.cs

@@ -45,7 +45,7 @@ public FileStream CreateFile(string file)
         }
 
         /// <inheritdoc/>
-        public StreamReader OpenText(string file)
+        public virtual StreamReader OpenText(string file)
         {
             if (!initialized) throw new InvalidOperationException("The file provider needs to be initialized first.");
             return File.OpenText(Path.Combine(HomeDirectory.FullName, file));

tests/ILICheck.Web.Test/DownloadControllerTest.cs

@@ -0,0 +1,57 @@
+using Microsoft.AspNetCore.Mvc;
+using Microsoft.Extensions.Logging;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Moq;
+using System;
+using System.IO;
+
+namespace ILICheck.Web.Controllers
+{
+    [TestClass]
+    public sealed class DownloadControllerTest
+    {
+        private readonly Guid jobId = Guid.Parse("28e1adff-765e-4c0b-b667-90458b33e1ca");
+
+        private Mock<ILogger<DownloadController>> loggerMock;
+        private Mock<IFileProvider> fileProviderMock;
+        private DownloadController controller;
+
+        public TestContext TestContext { get; set; }
+
+        [TestInitialize]
+        public void Initialize()
+        {
+            loggerMock = new Mock<ILogger<DownloadController>>();
+            fileProviderMock = new Mock<IFileProvider>(MockBehavior.Strict);
+
+            controller = new DownloadController(loggerMock.Object, fileProviderMock.Object);
+        }
+
+        [TestCleanup]
+        public void Cleanup()
+        {
+            loggerMock.VerifyAll();
+            fileProviderMock.VerifyAll();
+
+            controller.Dispose();
+        }
+
+        [TestMethod]
+        public void Download()
+        {
+            fileProviderMock.Setup(x => x.Initialize(jobId));
+            fileProviderMock.Setup(x => x.GetFiles()).Returns(new[] { "OCEANSTEED_log.xtf", "DARKFOOT_log.log", "HAPPYPOINT_log.geojson" });
+            fileProviderMock.Setup(x => x.OpenText(It.IsAny<string>())).Returns(StreamReader.Null);
+
+            void AssertContentType(LogType logType, string contentType)
+            {
+                var result = controller.Download(jobId, logType) as FileStreamResult;
+                Assert.AreEqual(result.ContentType, contentType);
+            }
+
+            AssertContentType(LogType.Log, "text/plain");
+            AssertContentType(LogType.Xtf, "text/xml; charset=utf-8");
+            AssertContentType(LogType.GeoJson, "application/geo+json");
+        }
+    }
+}