Lidarr/src/NzbDrone.Test.Common/NzbDroneRunner.cs

137 lines
4.1 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;
2014-02-27 17:04:09 +00:00
using System.Xml;
2013-09-21 02:07:42 +00:00
using System.Xml.Linq;
using System.Xml.XPath;
2014-12-17 07:12:26 +00:00
using NLog;
2013-08-14 03:22:28 +00:00
using NUnit.Framework;
using NzbDrone.Common.EnvironmentInfo;
using NzbDrone.Common.Processes;
2013-08-14 03:22:28 +00:00
using RestSharp;
2013-11-12 03:25:54 +00:00
namespace NzbDrone.Test.Common
2013-08-14 03:22:28 +00:00
{
public class NzbDroneRunner
{
private readonly IProcessProvider _processProvider;
private readonly IRestClient _restClient;
private Process _nzbDroneProcess;
2013-09-21 02:07:42 +00:00
public string AppData { get; private set; }
public string ApiKey { get; private set; }
2017-03-31 18:55:07 +00:00
public NzbDroneRunner(Logger logger, int port = 8686)
2013-08-14 03:22:28 +00:00
{
2014-12-17 07:12:26 +00:00
_processProvider = new ProcessProvider(logger);
2017-03-31 18:55:07 +00:00
_restClient = new RestClient("http://localhost:8686/api");
2013-08-14 03:22:28 +00:00
}
public void Start()
{
AppData = Path.Combine(TestContext.CurrentContext.TestDirectory, "_intg_" + DateTime.Now.Ticks);
2013-08-14 03:22:28 +00:00
var nzbdroneConsoleExe = "Lidarr.Console.exe";
2013-08-26 19:35:53 +00:00
2014-12-07 20:54:07 +00:00
if (OsInfo.IsNotWindows)
2013-08-26 19:35:53 +00:00
{
nzbdroneConsoleExe = "Lidarr.exe";
2013-08-26 19:35:53 +00:00
}
2013-08-14 03:22:28 +00:00
if (BuildInfo.IsDebug)
{
Start(Path.Combine(TestContext.CurrentContext.TestDirectory, "..\\..\\..\\..\\..\\_output\\Lidarr.Console.exe"));
2013-08-14 03:22:28 +00:00
}
else
{
2013-08-26 19:35:53 +00:00
Start(Path.Combine("bin", nzbdroneConsoleExe));
2013-08-14 03:22:28 +00:00
}
while (true)
{
_nzbDroneProcess.Refresh();
if (_nzbDroneProcess.HasExited)
{
Assert.Fail("Process has exited");
}
SetApiKey();
2013-09-21 02:07:42 +00:00
var request = new RestRequest("system/status");
request.AddHeader("Authorization", ApiKey);
request.AddHeader("X-Api-Key", ApiKey);
2013-09-21 02:07:42 +00:00
var statusCall = _restClient.Get(request);
if (statusCall.ResponseStatus == ResponseStatus.Completed)
2013-08-14 03:22:28 +00:00
{
2013-09-01 00:30:35 +00:00
Console.WriteLine("NzbDrone is started. Running Tests");
2013-08-14 03:22:28 +00:00
return;
}
Console.WriteLine("Waiting for NzbDrone to start. Response Status : {0} [{1}] {2}", statusCall.ResponseStatus, statusCall.StatusDescription, statusCall.ErrorException);
2013-08-14 03:22:28 +00:00
Thread.Sleep(500);
}
}
public void KillAll()
{
if (_nzbDroneProcess != null)
{
_processProvider.Kill(_nzbDroneProcess.Id);
}
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)
{
2013-09-21 02:07:42 +00:00
var args = "-nobrowser -data=\"" + AppData + "\"";
_nzbDroneProcess = _processProvider.Start(outputNzbdroneConsoleExe, args, null, 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(" ");
}
}
private void SetApiKey()
2013-09-21 02:07:42 +00:00
{
var configFile = Path.Combine(AppData, "config.xml");
var attempts = 0;
2013-09-21 02:07:42 +00:00
while (ApiKey == null && attempts < 50)
{
try
{
if (File.Exists(configFile))
{
var apiKeyElement = XDocument.Load(configFile)
.XPathSelectElement("Config/ApiKey");
if (apiKeyElement != null)
{
ApiKey = apiKeyElement.Value;
}
}
}
2014-02-27 17:04:09 +00:00
catch (XmlException ex)
{
Console.WriteLine("Error getting API Key from XML file: " + ex.Message, ex);
}
attempts++;
Thread.Sleep(1000);
}
2013-09-21 02:07:42 +00:00
}
2013-08-14 03:22:28 +00:00
}
}