Sonarr/NzbDrone.Integration.Test/NzbDroneRunner.cs

83 lines
2.2 KiB
C#
Raw Normal View History

2013-08-14 03:22:28 +00:00
using System;
using System.Diagnostics;
using System.IO;
using System.Threading;
using NUnit.Framework;
using NzbDrone.Common;
using NzbDrone.Common.EnvironmentInfo;
using RestSharp;
namespace NzbDrone.Integration.Test
{
public class NzbDroneRunner
{
private readonly IProcessProvider _processProvider;
private readonly IRestClient _restClient;
private Process _nzbDroneProcess;
public NzbDroneRunner(int port = 8989)
{
_processProvider = new ProcessProvider();
_restClient = new RestClient("http://localhost:8989/api");
}
public void Start()
{
AppDate = Path.Combine(Directory.GetCurrentDirectory(), "_intg_" + DateTime.Now.Ticks);
if (BuildInfo.IsDebug)
{
Start("..\\..\\..\\..\\_output\\NzbDrone.Console.exe");
}
else
{
2013-08-14 04:09:34 +00:00
Start(Path.Combine("bin","NzbDrone.Console.exe"));
2013-08-14 03:22:28 +00:00
}
while (true)
{
_nzbDroneProcess.Refresh();
if (_nzbDroneProcess.HasExited)
{
Assert.Fail("Process has exited");
}
if (_restClient.Get(new RestRequest("system/status")).ResponseStatus == ResponseStatus.Completed)
{
return;
}
Thread.Sleep(500);
}
}
public void KillAll()
{
2013-08-14 05:20:24 +00:00
_processProvider.KillAll(ProcessProvider.NZB_DRONE_CONSOLE_PROCESS_NAME);
_processProvider.KillAll(ProcessProvider.NZB_DRONE_PROCESS_NAME);
2013-08-14 03:22:28 +00:00
}
private void Start(string outputNzbdroneConsoleExe)
{
var args = "-nobrowser -data=\"" + AppDate + "\"";
2013-08-14 05:20:24 +00:00
_nzbDroneProcess = _processProvider.Start(outputNzbdroneConsoleExe, args, OnOutputDataReceived, OnOutputDataReceived);
2013-08-14 03:22:28 +00:00
}
private void OnOutputDataReceived(string data)
{
2013-08-14 03:28:13 +00:00
Console.WriteLine(data);
2013-08-14 03:22:28 +00:00
if (data.Contains("Press enter to exit"))
{
_nzbDroneProcess.StandardInput.WriteLine(" ");
}
}
public string AppDate { get; private set; }
}
}