mirror of https://github.com/Radarr/Radarr
164 lines
4.3 KiB
C#
164 lines
4.3 KiB
C#
![]() |
using System;
|
|||
|
using System.Diagnostics;
|
|||
|
using System.Drawing.Imaging;
|
|||
|
using System.IO;
|
|||
|
using System.Linq;
|
|||
|
using NUnit.Framework;
|
|||
|
using NzbDrone.Common;
|
|||
|
using OpenQA.Selenium;
|
|||
|
using OpenQA.Selenium.Firefox;
|
|||
|
using OpenQA.Selenium.Remote;
|
|||
|
|
|||
|
namespace NzbDrone.Web.UI.Automation
|
|||
|
{
|
|||
|
public abstract class AutomationTestBase
|
|||
|
{
|
|||
|
static readonly EnviromentProvider enviromentProvider = new EnviromentProvider();
|
|||
|
private static readonly string testFolder;
|
|||
|
|
|||
|
public string AppUrl
|
|||
|
{
|
|||
|
get
|
|||
|
{
|
|||
|
return "http://localhost:8989";
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
|
|||
|
public RemoteWebDriver Driver { get; private set; }
|
|||
|
|
|||
|
static AutomationTestBase()
|
|||
|
{
|
|||
|
CleanBinFolder();
|
|||
|
testFolder = CreatePackage();
|
|||
|
StartNzbDrone();
|
|||
|
}
|
|||
|
|
|||
|
[SetUp]
|
|||
|
public void AutomationSetup()
|
|||
|
{
|
|||
|
Driver = new FirefoxDriver();
|
|||
|
}
|
|||
|
|
|||
|
[TearDown]
|
|||
|
public void AutomationTearDown()
|
|||
|
{
|
|||
|
Driver.Close();
|
|||
|
|
|||
|
if (!Directory.Exists(Directory.GetCurrentDirectory() + "\\Screenshots"))
|
|||
|
{
|
|||
|
Directory.CreateDirectory(Directory.GetCurrentDirectory() + "\\Screenshots");
|
|||
|
}
|
|||
|
|
|||
|
foreach (var file in Directory.GetFiles(Directory.GetCurrentDirectory(), "*__*.png").Select(c => new FileInfo(c)))
|
|||
|
{
|
|||
|
File.Copy(file.FullName, Directory.GetCurrentDirectory() + "\\Screenshots\\" + file.Name, true);
|
|||
|
file.Delete();
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
|
|||
|
[TestFixtureSetUp]
|
|||
|
public void AutomationTestFixtureSetup()
|
|||
|
{
|
|||
|
StopNzbDrone();
|
|||
|
ResetUserData();
|
|||
|
StartNzbDrone();
|
|||
|
}
|
|||
|
|
|||
|
[TestFixtureTearDown]
|
|||
|
public void AutomationTestFixtureTearDown()
|
|||
|
{
|
|||
|
StopNzbDrone();
|
|||
|
}
|
|||
|
|
|||
|
|
|||
|
protected void CaptureScreen()
|
|||
|
{
|
|||
|
var method = new StackFrame(1).GetMethod().Name;
|
|||
|
|
|||
|
var fileName = String.Format("{0}__{1}.png", this.GetType().Name, method);
|
|||
|
|
|||
|
((ITakesScreenshot)Driver).GetScreenshot().SaveAsFile(fileName, ImageFormat.Png);
|
|||
|
}
|
|||
|
|
|||
|
private void ResetUserData()
|
|||
|
{
|
|||
|
var appDataPath = Path.Combine(testFolder, "NzbDrone.Web", "app_data");
|
|||
|
|
|||
|
if (Directory.Exists(appDataPath))
|
|||
|
Directory.Delete(appDataPath, true);
|
|||
|
|
|||
|
}
|
|||
|
|
|||
|
|
|||
|
private static void CleanBinFolder()
|
|||
|
{
|
|||
|
var folderName = "Debug";
|
|||
|
|
|||
|
if (EnviromentProvider.IsDebug)
|
|||
|
{
|
|||
|
folderName = "Release";
|
|||
|
}
|
|||
|
|
|||
|
var dirs = Directory.GetDirectories(enviromentProvider.ApplicationPath, folderName, SearchOption.AllDirectories);
|
|||
|
|
|||
|
|
|||
|
foreach (var dir in dirs)
|
|||
|
{
|
|||
|
Directory.Delete(dir, true);
|
|||
|
}
|
|||
|
|
|||
|
}
|
|||
|
|
|||
|
static void StartNzbDrone()
|
|||
|
{
|
|||
|
Process.Start(Path.Combine(testFolder, "nzbdrone.exe"));
|
|||
|
}
|
|||
|
|
|||
|
public static void StopNzbDrone()
|
|||
|
{
|
|||
|
foreach (var process in Process.GetProcessesByName("nzbdrone"))
|
|||
|
{
|
|||
|
process.Kill();
|
|||
|
process.WaitForExit();
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
private static string CreatePackage()
|
|||
|
{
|
|||
|
Console.WriteLine("Creating NzbDrone Package");
|
|||
|
|
|||
|
StopNzbDrone();
|
|||
|
|
|||
|
var rootDirectory = new DirectoryInfo(enviromentProvider.ApplicationPath);
|
|||
|
|
|||
|
if (rootDirectory.GetDirectories("_rawPackage").Any())
|
|||
|
{
|
|||
|
rootDirectory.GetDirectories("_rawPackage").ToList().ForEach(c => c.Delete(true));
|
|||
|
}
|
|||
|
|
|||
|
var startInfo = new ProcessStartInfo
|
|||
|
{
|
|||
|
FileName = Path.Combine(rootDirectory.FullName, "package.bat"),
|
|||
|
WorkingDirectory = rootDirectory.FullName
|
|||
|
};
|
|||
|
|
|||
|
Process.Start(startInfo).WaitForExit();
|
|||
|
|
|||
|
var testFolder = Path.Combine(enviromentProvider.SystemTemp, "NzbDroneAutomation");
|
|||
|
|
|||
|
if (Directory.Exists(testFolder))
|
|||
|
{
|
|||
|
Directory.Delete(testFolder, true);
|
|||
|
}
|
|||
|
|
|||
|
Directory.Move(Path.Combine(rootDirectory.FullName, "_rawPackage", "nzbdrone"), testFolder);
|
|||
|
|
|||
|
|
|||
|
|
|||
|
return testFolder;
|
|||
|
}
|
|||
|
}
|
|||
|
}
|