using System ;
using System.Diagnostics ;
using System.IO ;
using System.Linq ;
using System.Threading ;
using System.Xml.Linq ;
using NLog ;
using NUnit.Framework ;
using NzbDrone.Common.EnvironmentInfo ;
using NzbDrone.Common.Extensions ;
using NzbDrone.Common.Processes ;
using NzbDrone.Core.Configuration ;
using RestSharp ;
namespace NzbDrone.Test.Common
{
public class NzbDroneRunner
{
private readonly IProcessProvider _processProvider ;
private readonly IRestClient _restClient ;
private Process _nzbDroneProcess ;
public string AppData { get ; private set ; }
public string ApiKey { get ; private set ; }
public int Port { get ; private set ; }
public NzbDroneRunner ( Logger logger , int port = 9696 )
{
_processProvider = new ProcessProvider ( logger ) ;
_restClient = new RestClient ( $"http://localhost:{port}/api/v1" ) ;
Port = port ;
}
public void Start ( )
{
AppData = Path . Combine ( TestContext . CurrentContext . TestDirectory , "_intg_" + TestBase . GetUID ( ) ) ;
Directory . CreateDirectory ( AppData ) ;
GenerateConfigFile ( ) ;
string consoleExe ;
if ( OsInfo . IsWindows )
{
consoleExe = "Prowlarr.Console.exe" ;
}
else
{
consoleExe = "Prowlarr" ;
}
if ( BuildInfo . IsDebug )
{
var frameworkFolder = "net6.0" ;
Start ( Path . Combine ( TestContext . CurrentContext . TestDirectory , ".." , ".." , "_output" , frameworkFolder , consoleExe ) ) ;
}
else
{
Start ( Path . Combine ( TestContext . CurrentContext . TestDirectory , ".." , "bin" , consoleExe ) ) ;
}
while ( true )
{
_nzbDroneProcess . Refresh ( ) ;
if ( _nzbDroneProcess . HasExited )
{
Assert . Fail ( "Process has exited" ) ;
}
var request = new RestRequest ( "system/status" ) ;
request . AddHeader ( "Authorization" , ApiKey ) ;
request . AddHeader ( "X-Api-Key" , ApiKey ) ;
var statusCall = _restClient . Get ( request ) ;
if ( statusCall . ResponseStatus = = ResponseStatus . Completed )
{
TestContext . Progress . WriteLine ( $"Prowlarr {Port} is started. Running Tests" ) ;
return ;
}
TestContext . Progress . WriteLine ( "Waiting for Prowlarr to start. Response Status : {0} [{1}] {2}" , statusCall . ResponseStatus , statusCall . StatusDescription , statusCall . ErrorException . Message ) ;
Thread . Sleep ( 500 ) ;
}
}
public void Kill ( )
{
try
{
if ( _nzbDroneProcess ! = null )
{
_nzbDroneProcess . Refresh ( ) ;
if ( _nzbDroneProcess . HasExited )
{
var log = File . ReadAllLines ( Path . Combine ( AppData , "logs" , "Prowlarr.trace.txt" ) ) ;
var output = log . Join ( Environment . NewLine ) ;
TestContext . Progress . WriteLine ( "Process has exited prematurely: ExitCode={0} Output:\n{1}" , _nzbDroneProcess . ExitCode , output ) ;
}
_processProvider . Kill ( _nzbDroneProcess . Id ) ;
}
}
catch ( InvalidOperationException )
{
// May happen if the process closes while being closed
}
TestBase . DeleteTempFolder ( AppData ) ;
}
public void KillAll ( )
{
try
{
if ( _nzbDroneProcess ! = null )
{
_processProvider . Kill ( _nzbDroneProcess . Id ) ;
}
_processProvider . KillAll ( ProcessProvider . PROWLARR_CONSOLE_PROCESS_NAME ) ;
_processProvider . KillAll ( ProcessProvider . PROWLARR_PROCESS_NAME ) ;
}
catch ( InvalidOperationException )
{
// May happen if the process closes while being closed
}
TestBase . DeleteTempFolder ( AppData ) ;
}
private void Start ( string outputProwlarrConsoleExe )
{
TestContext . Progress . WriteLine ( "Starting instance from {0} on port {1}" , outputProwlarrConsoleExe , Port ) ;
var args = "-nobrowser -nosingleinstancecheck -data=\"" + AppData + "\"" ;
_nzbDroneProcess = _processProvider . Start ( outputProwlarrConsoleExe , args , null , OnOutputDataReceived , OnOutputDataReceived ) ;
}
private void OnOutputDataReceived ( string data )
{
TestContext . Progress . WriteLine ( $" [{Port}] > " + data ) ;
if ( data . Contains ( "Press enter to exit" ) )
{
_nzbDroneProcess . StandardInput . WriteLine ( " " ) ;
}
}
private void GenerateConfigFile ( )
{
var configFile = Path . Combine ( AppData , "config.xml" ) ;
// Generate and set the api key so we don't have to poll the config file
var apiKey = Guid . NewGuid ( ) . ToString ( ) . Replace ( "-" , "" ) ;
var xDoc = new XDocument (
new XDeclaration ( "1.0" , "utf-8" , "yes" ) ,
new XElement ( ConfigFileProvider . CONFIG_ELEMENT_NAME ,
new XElement ( nameof ( ConfigFileProvider . ApiKey ) , apiKey ) ,
new XElement ( nameof ( ConfigFileProvider . LogLevel ) , "trace" ) ,
new XElement ( nameof ( ConfigFileProvider . AnalyticsEnabled ) , false ) ,
new XElement ( nameof ( ConfigFileProvider . Port ) , Port ) ) ) ;
var data = xDoc . ToString ( ) ;
File . WriteAllText ( configFile , data ) ;
ApiKey = apiKey ;
}
}
}