|
|
|
@ -1,10 +1,13 @@
|
|
|
|
|
using System;
|
|
|
|
|
using System.Diagnostics;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Threading;
|
|
|
|
|
using System.Xml.Linq;
|
|
|
|
|
using NUnit.Framework;
|
|
|
|
|
using NzbDrone.Common;
|
|
|
|
|
using NzbDrone.Common.EnvironmentInfo;
|
|
|
|
|
using NzbDrone.Core.Configuration;
|
|
|
|
|
using RestSharp;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Integration.Test
|
|
|
|
@ -15,16 +18,18 @@ namespace NzbDrone.Integration.Test
|
|
|
|
|
private readonly IRestClient _restClient;
|
|
|
|
|
private Process _nzbDroneProcess;
|
|
|
|
|
|
|
|
|
|
public string AppData { get; private set; }
|
|
|
|
|
public string ApiKey { get; private set; }
|
|
|
|
|
|
|
|
|
|
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);
|
|
|
|
|
AppData = Path.Combine(Directory.GetCurrentDirectory(), "_intg_" + DateTime.Now.Ticks);
|
|
|
|
|
|
|
|
|
|
var nzbdroneConsoleExe = "NzbDrone.Console.exe";
|
|
|
|
|
|
|
|
|
@ -33,7 +38,6 @@ namespace NzbDrone.Integration.Test
|
|
|
|
|
nzbdroneConsoleExe = "NzbDrone.exe";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (BuildInfo.IsDebug)
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
@ -53,8 +57,12 @@ namespace NzbDrone.Integration.Test
|
|
|
|
|
Assert.Fail("Process has exited");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
SetApiKey();
|
|
|
|
|
|
|
|
|
|
var request = new RestRequest("system/status");
|
|
|
|
|
request.AddHeader("Authorization", ApiKey);
|
|
|
|
|
|
|
|
|
|
var statusCall = _restClient.Get(new RestRequest("system/status"));
|
|
|
|
|
var statusCall = _restClient.Get(request);
|
|
|
|
|
|
|
|
|
|
if (statusCall.ResponseStatus == ResponseStatus.Completed)
|
|
|
|
|
{
|
|
|
|
@ -76,7 +84,7 @@ namespace NzbDrone.Integration.Test
|
|
|
|
|
|
|
|
|
|
private void Start(string outputNzbdroneConsoleExe)
|
|
|
|
|
{
|
|
|
|
|
var args = "-nobrowser -data=\"" + AppDate + "\"";
|
|
|
|
|
var args = "-nobrowser -data=\"" + AppData + "\"";
|
|
|
|
|
_nzbDroneProcess = _processProvider.Start(outputNzbdroneConsoleExe, args, OnOutputDataReceived, OnOutputDataReceived);
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
@ -91,7 +99,16 @@ namespace NzbDrone.Integration.Test
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void SetApiKey()
|
|
|
|
|
{
|
|
|
|
|
var configFile = Path.Combine(AppData, "config.xml");
|
|
|
|
|
|
|
|
|
|
if (!String.IsNullOrWhiteSpace(ApiKey)) return;
|
|
|
|
|
if (!File.Exists(configFile)) return;
|
|
|
|
|
|
|
|
|
|
public string AppDate { get; private set; }
|
|
|
|
|
var xDoc = XDocument.Load(configFile);
|
|
|
|
|
var config = xDoc.Descendants(ConfigFileProvider.CONFIG_ELEMENT_NAME).Single();
|
|
|
|
|
ApiKey = config.Descendants("ApiKey").Single().Value;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|