using System; using System.Diagnostics; using System.IO; using System.Threading; using FluentAssertions; using Moq; using NLog; using NUnit.Framework; using NzbDrone.Common.Cache; using NzbDrone.Common.EnvironmentInfo; using NzbDrone.Common.Messaging; using NzbDrone.Core.Messaging.Events; using NzbDrone.Test.Common.AutoMoq; namespace NzbDrone.Test.Common { public abstract class TestBase : TestBase where TSubject : class { private TSubject _subject; [SetUp] public void CoreTestSetup() { _subject = null; } protected TSubject Subject { get { if (_subject == null) { _subject = Mocker.Resolve(); } return _subject; } } } public abstract class TestBase : LoggingTest { private static readonly Random _random = new Random(); private static int _nextUid; private AutoMoqer _mocker; protected AutoMoqer Mocker { get { if (_mocker == null) { _mocker = new AutoMoqer(); _mocker.SetConstant(new CacheManager()); _mocker.SetConstant(new StartupContext(Array.Empty())); _mocker.SetConstant(TestLogger); } return _mocker; } } protected int RandomNumber { get { Thread.Sleep(1); return _random.Next(0, int.MaxValue); } } private string VirtualPath { get { var virtualPath = Path.Combine(TempFolder, "VirtualNzbDrone"); if (!Directory.Exists(virtualPath)) { Directory.CreateDirectory(virtualPath); } return virtualPath; } } private string _tempFolder; protected string TempFolder { get { if (_tempFolder == null) { _tempFolder = Path.Combine(TestContext.CurrentContext.TestDirectory, "_temp_" + GetUID()); Directory.CreateDirectory(_tempFolder); } return _tempFolder; } } [SetUp] public void TestBaseSetup() { GetType().IsPublic.Should().BeTrue("All Test fixtures should be public to work in mono."); LogManager.ReconfigExistingLoggers(); _tempFolder = null; } [TearDown] public void TestBaseTearDown() { _mocker = null; DeleteTempFolder(_tempFolder); } public static string GetUID() { return Process.GetCurrentProcess().Id + "_" + DateTime.Now.Ticks + "_" + Interlocked.Increment(ref _nextUid); } public static void DeleteTempFolder(string folder) { if (folder == null) { return; } try { var tempFolder = new DirectoryInfo(folder); if (tempFolder.Exists) { foreach (var file in tempFolder.GetFiles("*", SearchOption.AllDirectories)) { file.IsReadOnly = false; } tempFolder.Delete(true); } } catch (Exception) { } } protected IAppFolderInfo TestFolderInfo { get; private set; } protected void WindowsOnly() { if (OsInfo.IsNotWindows) { throw new IgnoreException("windows specific test"); } } protected void PosixOnly() { if (OsInfo.IsWindows) { throw new IgnoreException("non windows specific test"); } } protected void MonoOnly() { if (!PlatformInfo.IsMono) { throw new IgnoreException("mono specific test"); } } protected void WithTempAsAppPath() { Mocker.GetMock() .SetupGet(c => c.AppDataFolder) .Returns(VirtualPath); TestFolderInfo = Mocker.GetMock().Object; } protected string GetTestPath(string path) { return Path.Combine(TestContext.CurrentContext.TestDirectory, Path.Combine(path.Split('/'))); } protected string ReadAllText(string path) { return File.ReadAllText(GetTestPath(path)); } protected string GetTempFilePath() { return Path.Combine(TempFolder, Path.GetRandomFileName()); } protected void VerifyEventPublished() where TEvent : class, IEvent { VerifyEventPublished(Times.Once()); } protected void VerifyEventPublished(Times times) where TEvent : class, IEvent { Mocker.GetMock().Verify(c => c.PublishEvent(It.IsAny()), times); } protected void VerifyEventNotPublished() where TEvent : class, IEvent { Mocker.GetMock().Verify(c => c.PublishEvent(It.IsAny()), Times.Never()); } } }