Archive-Team
/
sonarr-repo-only
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

better routing.

This commit is contained in:
kay.one 2013-08-13 21:02:56 -07:00
parent 3b4eacdd31
commit ddff598ba4
4 changed files with 22 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
NzbDrone.App.Test/RouterTest.cs
View File

@ -58,11 +58,8 @@ namespace NzbDrone.App.Test
Mocker.GetMock<INzbDroneServiceFactory>().Verify(c => c.Start(), Times.Once());
}
[TestCase(ApplicationModes.Interactive)]
[TestCase(ApplicationModes.InstallService)]
[TestCase(ApplicationModes.UninstallService)]
[TestCase(ApplicationModes.Help)]
public void Route_should_call_service_start_when_run_in_service_mode(ApplicationModes applicationModes)
[Test]
public void Route_should_call_service_start_when_run_in_service_mode()
{
var envMock = Mocker.GetMock<IRuntimeInfo>();
var serviceProvider = Mocker.GetMock<IServiceProvider>();
@ -71,8 +68,9 @@ namespace NzbDrone.App.Test
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);
Subject.Route();
serviceProvider.Verify(c => c.Run(It.IsAny<ServiceBase>()), Times.Once());
}
@ -90,7 +88,6 @@ namespace NzbDrone.App.Test
Subject.Route(ApplicationModes.InstallService);
Mocker.VerifyAllMocks();
}
[Test]
@ -105,7 +102,6 @@ namespace NzbDrone.App.Test
Subject.Route(ApplicationModes.UninstallService);
Mocker.VerifyAllMocks();
}
}
}

9
NzbDrone.Common/ServiceProvider.cs
View File

@ -18,6 +18,7 @@ namespace NzbDrone.Common
ServiceController GetService(string serviceName);
void Stop(string serviceName);
void Start(string serviceName);
ServiceControllerStatus GetStatus(string serviceName);
}
public class ServiceProvider : IServiceProvider
@ -136,7 +137,13 @@ namespace NzbDrone.Common
}
}
public virtual void Start(string serviceName)
public ServiceControllerStatus GetStatus(string serviceName)
{
return GetService(serviceName).Status;
}
public void Start(string serviceName)
{
Logger.Info("Starting {0} Service...", serviceName);
var service = GetService(serviceName);

16
NzbDrone.Host/Router.cs
View File

@ -93,6 +93,14 @@ namespace NzbDrone.Host
private ApplicationModes GetApplicationMode()
{
if (!_runtimeInfo.IsUserInteractive &&
OsInfo.IsWindows &&
_serviceProvider.ServiceExist(ServiceProvider.NZBDRONE_SERVICE_NAME) &&
_serviceProvider.GetStatus(ServiceProvider.NZBDRONE_SERVICE_NAME) == ServiceControllerStatus.StartPending)
{
return ApplicationModes.Service;
}
if (_startupArguments.Flags.Contains(StartupArguments.HELP))
{
return ApplicationModes.Help;
@ -108,14 +116,6 @@ namespace NzbDrone.Host
return ApplicationModes.UninstallService;
}
if (!_runtimeInfo.IsUserInteractive &&
OsInfo.IsWindows &&
_serviceProvider.ServiceExist(ServiceProvider.NZBDRONE_SERVICE_NAME) &&
_serviceProvider.GetService(ServiceProvider.NZBDRONE_SERVICE_NAME).Status == ServiceControllerStatus.StartPending)
{
return ApplicationModes.Service;
}
return ApplicationModes.Interactive;
}
}