sonarr-repo-only/src/NzbDrone.App.Test/NzbDroneProcessServiceFixture.cs

using System.Collections.Generic;
using Moq;
using NUnit.Framework;
using NzbDrone.Common.Model;
using NzbDrone.Common.Processes;
using NzbDrone.Host;
using NzbDrone.Test.Common;

namespace NzbDrone.App.Test
{
    [TestFixture]
    public class NzbDroneProcessServiceFixture : TestBase<SingleInstancePolicy>
    {
        private const int CURRENT_PROCESS_ID = 5;

        [SetUp]
        public void Setup()
        {
            Mocker.GetMock<IProcessProvider>().Setup(c => c.GetCurrentProcess())
                .Returns(new ProcessInfo() { Id = CURRENT_PROCESS_ID });
        }

        [Test]
        public void should_continue_if_only_instance()
        {
            Mocker.GetMock<IProcessProvider>()
                .Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_CONSOLE_PROCESS_NAME))
                .Returns(new List<ProcessInfo>());

            Mocker.GetMock<IProcessProvider>().Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_PROCESS_NAME))
                .Returns(new List<ProcessInfo>
                {
                    new ProcessInfo{Id = CURRENT_PROCESS_ID}
                });


            Subject.EnforceSingleInstance();


            Mocker.GetMock<IBrowserService>().Verify(c => c.LaunchWebUI(), Times.Never());

        }

        [Test]
        public void should_enforce_if_another_console_is_running()
        {
            Mocker.GetMock<IProcessProvider>()
                .Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_CONSOLE_PROCESS_NAME))
                    .Returns(new List<ProcessInfo>
                {
                    new ProcessInfo{Id = 10}
                });

            Mocker.GetMock<IProcessProvider>().Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_PROCESS_NAME))
                .Returns(new List<ProcessInfo>
                {
                    new ProcessInfo{Id = CURRENT_PROCESS_ID}
                });



            Assert.Throws<TerminateApplicationException>(() => Subject.EnforceSingleInstance());
            Mocker.GetMock<IBrowserService>().Verify(c => c.LaunchWebUI(), Times.Once());
            ExceptionVerification.ExpectedWarns(1);
        }

        [Test]
        public void should_return_false_if_another_gui_is_running()
        {
            Mocker.GetMock<IProcessProvider>()
                .Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_CONSOLE_PROCESS_NAME))
                .Returns(new List<ProcessInfo>
                {
                      new ProcessInfo{Id = CURRENT_PROCESS_ID}
                
                });

            Mocker.GetMock<IProcessProvider>().Setup(c => c.FindProcessByName(ProcessProvider.NZB_DRONE_PROCESS_NAME))
                .Returns(new List<ProcessInfo>
                {
                      new ProcessInfo{Id = 10}
                });



            Assert.Throws<TerminateApplicationException>(() => Subject.EnforceSingleInstance());
            Mocker.GetMock<IBrowserService>().Verify(c => c.LaunchWebUI(), Times.Once());
            ExceptionVerification.ExpectedWarns(1);
        }
    }
}