PowerToys/src/modules/Hosts/Hosts.Tests/HostsServiceTest.cs

// Copyright (c) Microsoft Corporation
// The Microsoft Corporation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using System.Collections.Generic;
using System.IO.Abstractions.TestingHelpers;
using System.Linq;
using System.Threading.Tasks;
using Hosts.Helpers;
using Hosts.Models;
using Hosts.Settings;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using Moq;
using Settings.UI.Library.Enumerations;

namespace Hosts.Tests
{
    [TestClass]
    public class HostsServiceTest
    {
        private static Mock<IElevationHelper> _elevationHelper;

        [ClassInitialize]
        public static void ClassInitialize(TestContext context)
        {
            _elevationHelper = new Mock<IElevationHelper>();
            _elevationHelper.Setup(m => m.IsElevated).Returns(true);
        }

        [TestMethod]
        public void Hosts_Exists()
        {
            var fileSystem = new MockFileSystem
            {
                FileSystemWatcher = new TestFileSystemWatcherFactory(),
            };

            var userSettings = new Mock<IUserSettings>();

            var service = new HostsService(fileSystem, userSettings.Object, _elevationHelper.Object);
            fileSystem.AddFile(service.HostsFilePath, new MockFileData(string.Empty));
            var result = service.Exists();

            Assert.IsTrue(result);
        }

        [TestMethod]
        public void Hosts_Not_Exists()
        {
            var fileSystem = new MockFileSystem(new Dictionary<string, MockFileData>
            {
            })
            {
                FileSystemWatcher = new TestFileSystemWatcherFactory(),
            };

            var userSettings = new Mock<IUserSettings>();

            var service = new HostsService(fileSystem, userSettings.Object, _elevationHelper.Object);
            var result = service.Exists();

            Assert.IsFalse(result);
        }

        [TestMethod]
        public async Task Host_Added()
        {
            var content =
@"10.1.1.1 host host.local # comment
10.1.1.2 host2 host2.local # another comment
";

            var contentResult =
@"  10.1.1.1  host host.local     # comment
  10.1.1.2  host2 host2.local   # another comment
# 10.1.1.30 host30 host30.local # new entry
";

            var fileSystem = new MockFileSystem
            {
                FileSystemWatcher = new TestFileSystemWatcherFactory(),
            };

            var userSettings = new Mock<IUserSettings>();

            var service = new HostsService(fileSystem, userSettings.Object, _elevationHelper.Object);
            fileSystem.AddFile(service.HostsFilePath, new MockFileData(content));

            var (_, entries) = await service.ReadAsync();
            entries.Add(new Entry("10.1.1.30", "host30 host30.local", "new entry", false));
            await service.WriteAsync(string.Empty, entries);

            var result = fileSystem.GetFile(service.HostsFilePath);
            Assert.AreEqual(result.TextContents, contentResult);
        }

        [TestMethod]
        public async Task Host_Deleted()
        {
            var content =
@"10.1.1.1 host host.local # comment
10.1.1.2 host2 host2.local # another comment
";

            var contentResult =
@"10.1.1.2 host2 host2.local # another comment
";

            var fileSystem = new MockFileSystem
            {
                FileSystemWatcher = new TestFileSystemWatcherFactory(),
            };

            var userSettings = new Mock<IUserSettings>();

            var service = new HostsService(fileSystem, userSettings.Object, _elevationHelper.Object);
            fileSystem.AddFile(service.HostsFilePath, new MockFileData(content));

            var (_, entries) = await service.ReadAsync();
            entries.RemoveAt(0);
            await service.WriteAsync(string.Empty, entries);

            var result = fileSystem.GetFile(service.HostsFilePath);
            Assert.AreEqual(result.TextContents, contentResult);
        }

        [TestMethod]
        public async Task Host_Updated()
        {
            var content =
@"10.1.1.1 host host.local # comment
10.1.1.2 host2 host2.local # another comment
";

            var contentResult =
@"# 10.1.1.10 host host.local host1.local # updated comment
  10.1.1.2  host2 host2.local           # another comment
";

            var fileSystem = new MockFileSystem
            {
                FileSystemWatcher = new TestFileSystemWatcherFactory(),
            };

            var userSettings = new Mock<IUserSettings>();

            var service = new HostsService(fileSystem, userSettings.Object, _elevationHelper.Object);
            fileSystem.AddFile(service.HostsFilePath, new MockFileData(content));

            var (_, entries) = await service.ReadAsync();
            var entry = entries[0];
            entry.Address = "10.1.1.10";
            entry.Hosts = "host host.local host1.local";
            entry.Comment = "updated comment";
            entry.Active = false;
            await service.WriteAsync(string.Empty, entries);

            var result = fileSystem.GetFile(service.HostsFilePath);
            Assert.AreEqual(result.TextContents, contentResult);
        }

        [TestMethod]
        public async Task Empty_Hosts()
        {
            var fileSystem = new MockFileSystem
            {
                FileSystemWatcher = new TestFileSystemWatcherFactory(),
            };

            var userSettings = new Mock<IUserSettings>();

            var service = new HostsService(fileSystem, userSettings.Object, _elevationHelper.Object);
            fileSystem.AddFile(service.HostsFilePath, new MockFileData(string.Empty));

            await service.WriteAsync(string.Empty, Enumerable.Empty<Entry>());

            var result = fileSystem.GetFile(service.HostsFilePath);
            Assert.AreEqual(result.TextContents, string.Empty);
        }

        [TestMethod]
        public async Task AdditionalLines_Top()
        {
            var content =
@"# header
10.1.1.1 host host.local   # comment
# comment
10.1.1.2 host2 host2.local # another comment
# footer
";

            var contentResult =
@"# header
# comment
# footer
10.1.1.1 host host.local   # comment
10.1.1.2 host2 host2.local # another comment
";

            var fileSystem = new MockFileSystem
            {
                FileSystemWatcher = new TestFileSystemWatcherFactory(),
            };

            var userSettings = new Mock<IUserSettings>();
            userSettings.Setup(m => m.AdditionalLinesPosition).Returns(AdditionalLinesPosition.Top);

            var service = new HostsService(fileSystem, userSettings.Object, _elevationHelper.Object);
            fileSystem.AddFile(service.HostsFilePath, new MockFileData(content));

            var (additionalLines, entries) = await service.ReadAsync();
            await service.WriteAsync(additionalLines, entries);

            var result = fileSystem.GetFile(service.HostsFilePath);
            Assert.AreEqual(result.TextContents, contentResult);
        }

        [TestMethod]
        public async Task AdditionalLines_Bottom()
        {
            var content =
@"# header
10.1.1.1 host host.local   # comment
# comment
10.1.1.2 host2 host2.local # another comment
# footer
";

            var contentResult =
@"10.1.1.1 host host.local   # comment
10.1.1.2 host2 host2.local # another comment
# header
# comment
# footer
";

            var fileSystem = new MockFileSystem
            {
                FileSystemWatcher = new TestFileSystemWatcherFactory(),
            };

            var userSettings = new Mock<IUserSettings>();
            userSettings.Setup(m => m.AdditionalLinesPosition).Returns(AdditionalLinesPosition.Bottom);

            var service = new HostsService(fileSystem, userSettings.Object, _elevationHelper.Object);
            fileSystem.AddFile(service.HostsFilePath, new MockFileData(content));

            var (additionalLines, entries) = await service.ReadAsync();
            await service.WriteAsync(additionalLines, entries);

            var result = fileSystem.GetFile(service.HostsFilePath);
            Assert.AreEqual(result.TextContents, contentResult);
        }
    }
}