2011-12-15 05:15:53 +01:00
using System ;
using System.Linq ;
2011-11-13 08:27:16 +01:00
using System.IO ;
using Moq ;
2013-02-18 08:59:43 +01:00
using NLog ;
2011-11-13 08:27:16 +01:00
using NUnit.Framework ;
using NzbDrone.Common ;
2013-02-24 20:18:48 +01:00
using NzbDrone.Common.Eventing ;
2011-11-14 01:22:18 +01:00
using NzbDrone.Test.Common.AutoMoq ;
2011-11-13 08:27:16 +01:00
namespace NzbDrone.Test.Common
{
2013-02-02 20:39:27 +01:00
public abstract class TestBase : LoggingTest
2011-11-13 08:27:16 +01:00
{
2011-12-02 02:33:17 +01:00
protected const string INTEGRATION_TEST = "Integration Test" ;
2011-11-14 01:22:18 +01:00
2011-12-15 05:15:53 +01:00
private AutoMoqer _mocker ;
protected AutoMoqer Mocker
{
get
{
if ( _mocker = = null )
{
_mocker = new AutoMoqer ( ) ;
}
return _mocker ;
}
}
2011-11-13 08:27:16 +01:00
2012-02-05 07:34:36 +01:00
protected Mock < RestProvider > MockedRestProvider { get ; private set ; }
2013-01-19 21:59:36 +01:00
private string VirtualPath
2011-11-13 08:27:16 +01:00
{
get
{
var virtualPath = Path . Combine ( TempFolder , "VirtualNzbDrone" ) ;
if ( ! Directory . Exists ( virtualPath ) ) Directory . CreateDirectory ( virtualPath ) ;
return virtualPath ;
}
}
2013-01-19 21:59:36 +01:00
protected string TempFolder { get ; private set ; }
2011-11-13 08:27:16 +01:00
[SetUp]
2011-11-13 20:15:40 +01:00
public void TestBaseSetup ( )
2011-11-13 08:27:16 +01:00
{
2013-01-19 21:59:36 +01:00
2013-02-18 08:59:43 +01:00
Mocker . SetConstant ( LogManager . GetLogger ( "TestLogger" ) ) ;
2013-03-04 06:53:02 +01:00
LogManager . ReconfigExistingLoggers ( ) ;
2013-01-19 21:59:36 +01:00
TempFolder = Path . Combine ( Directory . GetCurrentDirectory ( ) , "_temp_" + DateTime . Now . Ticks ) ;
2012-02-05 07:34:36 +01:00
MockedRestProvider = new Mock < RestProvider > ( ) ;
ReportingService . RestProvider = MockedRestProvider . Object ;
2012-04-23 01:14:02 +02:00
2011-11-13 08:27:16 +01:00
Directory . CreateDirectory ( TempFolder ) ;
}
2011-11-13 20:15:40 +01:00
[TearDown]
public void TestBaseTearDown ( )
{
2011-12-15 05:15:53 +01:00
_mocker = null ;
2013-01-19 21:59:36 +01:00
try
{
if ( Directory . Exists ( TempFolder ) )
{
Directory . Delete ( TempFolder , true ) ;
}
}
catch ( Exception )
{
}
2011-11-13 20:15:40 +01:00
}
2013-02-19 02:57:08 +01:00
[Obsolete("Use Mock.Verify() instead")]
2011-12-02 02:37:38 +01:00
protected void WithStrictMocker ( )
2011-11-13 08:27:16 +01:00
{
2013-02-19 02:57:08 +01:00
return ;
//TODO: Remove dependency on restrict mocks!
2011-12-15 05:15:53 +01:00
if ( _mocker ! = null )
throw new InvalidOperationException ( "Can not switch to a strict container after container has been used. make sure this is the first call in your test." ) ;
_mocker = new AutoMoqer ( MockBehavior . Strict ) ;
2011-11-13 08:27:16 +01:00
}
protected void WithTempAsAppPath ( )
{
2012-03-07 03:59:43 +01:00
Mocker . GetMock < EnvironmentProvider > ( )
2011-11-13 08:27:16 +01:00
. SetupGet ( c = > c . ApplicationPath )
. Returns ( VirtualPath ) ;
}
protected string GetTestFilePath ( string fileName )
{
2013-02-17 20:19:38 +01:00
return Path . Combine ( Directory . GetCurrentDirectory ( ) , "Files" , fileName ) ;
2011-11-13 08:27:16 +01:00
}
2013-02-24 20:18:48 +01:00
2013-03-07 05:34:56 +01:00
2013-02-24 20:39:31 +01:00
protected void VerifyEventPublished < TEvent > ( ) where TEvent : IEvent
2013-02-24 20:18:48 +01:00
{
2013-03-07 05:34:56 +01:00
VerifyEventPublished < TEvent > ( Times . Once ( ) ) ;
}
protected void VerifyEventPublished < TEvent > ( Times times ) where TEvent : IEvent
{
Mocker . GetMock < IEventAggregator > ( ) . Verify ( c = > c . Publish ( It . IsAny < TEvent > ( ) ) , times ) ;
2013-02-24 20:18:48 +01:00
}
protected void VerifyEventNotPublished < TEvent > ( ) where TEvent : IEvent
{
Mocker . GetMock < IEventAggregator > ( ) . Verify ( c = > c . Publish ( It . IsAny < TEvent > ( ) ) , Times . Never ( ) ) ;
}
2011-11-13 08:27:16 +01:00
}
}