mirror of https://github.com/Sonarr/Sonarr
111 lines
4.1 KiB
C#
111 lines
4.1 KiB
C#
using System.ServiceProcess;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Common;
|
|
using NzbDrone.Common.EnvironmentInfo;
|
|
using NzbDrone.Common.Processes;
|
|
using NzbDrone.Host;
|
|
using NzbDrone.Test.Common;
|
|
|
|
namespace NzbDrone.App.Test
|
|
{
|
|
[TestFixture]
|
|
public class RouterTest : TestBase<Router>
|
|
{
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
WindowsOnly();
|
|
}
|
|
|
|
|
|
[Test]
|
|
public void Route_should_call_install_service_when_application_mode_is_install()
|
|
{
|
|
var serviceProviderMock = Mocker.GetMock<IServiceProvider>(MockBehavior.Strict);
|
|
serviceProviderMock.Setup(c => c.ServiceExist(ServiceProvider.SERVICE_NAME)).Returns(false);
|
|
serviceProviderMock.Setup(c => c.Install(ServiceProvider.SERVICE_NAME));
|
|
serviceProviderMock.Setup(c => c.SetPermissions(ServiceProvider.SERVICE_NAME));
|
|
|
|
Mocker.GetMock<IProcessProvider>()
|
|
.Setup(c => c.SpawnNewProcess("sc.exe", It.IsAny<string>(), null, true));
|
|
|
|
Mocker.GetMock<IRuntimeInfo>().SetupGet(c => c.IsUserInteractive).Returns(true);
|
|
|
|
Subject.Route(ApplicationModes.InstallService);
|
|
|
|
serviceProviderMock.Verify(c => c.Install(ServiceProvider.SERVICE_NAME), Times.Once());
|
|
}
|
|
|
|
|
|
[Test]
|
|
public void Route_should_call_uninstall_service_when_application_mode_is_uninstall()
|
|
{
|
|
var serviceProviderMock = Mocker.GetMock<IServiceProvider>();
|
|
serviceProviderMock.Setup(c => c.Uninstall(ServiceProvider.SERVICE_NAME));
|
|
Mocker.GetMock<IRuntimeInfo>().SetupGet(c => c.IsUserInteractive).Returns(true);
|
|
serviceProviderMock.Setup(c => c.ServiceExist(ServiceProvider.SERVICE_NAME)).Returns(true);
|
|
|
|
Subject.Route(ApplicationModes.UninstallService);
|
|
|
|
serviceProviderMock.Verify(c => c.Uninstall(ServiceProvider.SERVICE_NAME), Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void Route_should_call_console_service_when_application_mode_is_console()
|
|
{
|
|
Mocker.GetMock<IRuntimeInfo>().SetupGet(c => c.IsUserInteractive).Returns(true);
|
|
|
|
Subject.Route(ApplicationModes.Interactive);
|
|
|
|
Mocker.GetMock<INzbDroneServiceFactory>().Verify(c => c.Start(), Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void Route_should_call_service_start_when_run_in_service_mode()
|
|
{
|
|
var envMock = Mocker.GetMock<IRuntimeInfo>();
|
|
var serviceProvider = Mocker.GetMock<IServiceProvider>();
|
|
|
|
envMock.SetupGet(c => c.IsUserInteractive).Returns(false);
|
|
|
|
serviceProvider.Setup(c => c.Run(It.IsAny<ServiceBase>()));
|
|
serviceProvider.Setup(c => c.ServiceExist(It.IsAny<string>())).Returns(true);
|
|
serviceProvider.Setup(c => c.GetStatus(It.IsAny<string>())).Returns(ServiceControllerStatus.StartPending);
|
|
|
|
Subject.Route(ApplicationModes.Service);
|
|
|
|
serviceProvider.Verify(c => c.Run(It.IsAny<ServiceBase>()), Times.Once());
|
|
}
|
|
|
|
|
|
[Test]
|
|
public void show_error_on_install_if_service_already_exist()
|
|
{
|
|
var consoleMock = Mocker.GetMock<IConsoleService>();
|
|
var serviceMock = Mocker.GetMock<IServiceProvider>();
|
|
Mocker.GetMock<IRuntimeInfo>().SetupGet(c => c.IsUserInteractive).Returns(true);
|
|
|
|
consoleMock.Setup(c => c.PrintServiceAlreadyExist());
|
|
serviceMock.Setup(c => c.ServiceExist(ServiceProvider.SERVICE_NAME)).Returns(true);
|
|
|
|
Subject.Route(ApplicationModes.InstallService);
|
|
|
|
}
|
|
|
|
[Test]
|
|
public void show_error_on_uninstall_if_service_doesnt_exist()
|
|
{
|
|
var consoleMock = Mocker.GetMock<IConsoleService>();
|
|
var serviceMock = Mocker.GetMock<IServiceProvider>();
|
|
Mocker.GetMock<IRuntimeInfo>().SetupGet(c => c.IsUserInteractive).Returns(true);
|
|
|
|
consoleMock.Setup(c => c.PrintServiceDoesNotExist());
|
|
serviceMock.Setup(c => c.ServiceExist(ServiceProvider.SERVICE_NAME)).Returns(false);
|
|
|
|
Subject.Route(ApplicationModes.UninstallService);
|
|
|
|
}
|
|
}
|
|
}
|