Automation ;)

NzbDrone.Web.UI.Test/AutomationTestBase.cs

@@ -0,0 +1,163 @@
+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;
+        }
+    }
+}