2011-12-15 04:15:53 +00:00
using System ;
using System.Linq ;
2011-11-13 07:27:16 +00:00
using System.IO ;
using Moq ;
using NUnit.Framework ;
using NzbDrone.Common ;
2011-11-14 00:22:18 +00:00
using NzbDrone.Test.Common.AutoMoq ;
2011-11-13 07:27:16 +00:00
namespace NzbDrone.Test.Common
{
public class TestBase : LoggingTest
{
2011-11-14 00:22:18 +00:00
2011-12-02 01:33:17 +00:00
protected const string INTEGRATION_TEST = "Integration Test" ;
2011-11-14 00:22:18 +00:00
2012-02-05 06:34:36 +00:00
2011-12-15 04:15:53 +00:00
private AutoMoqer _mocker ;
protected AutoMoqer Mocker
{
get
{
if ( _mocker = = null )
{
_mocker = new AutoMoqer ( ) ;
}
return _mocker ;
}
}
2011-11-13 07:27:16 +00:00
2012-02-05 06:34:36 +00:00
protected Mock < RestProvider > MockedRestProvider { get ; private set ; }
2013-01-19 20:59:36 +00:00
private string VirtualPath
2011-11-13 07:27:16 +00:00
{
get
{
var virtualPath = Path . Combine ( TempFolder , "VirtualNzbDrone" ) ;
if ( ! Directory . Exists ( virtualPath ) ) Directory . CreateDirectory ( virtualPath ) ;
return virtualPath ;
}
}
2013-01-19 20:59:36 +00:00
protected string TempFolder { get ; private set ; }
2011-11-13 07:27:16 +00:00
[SetUp]
2011-11-13 19:15:40 +00:00
public void TestBaseSetup ( )
2011-11-13 07:27:16 +00:00
{
2013-01-19 20:59:36 +00:00
TempFolder = Path . Combine ( Directory . GetCurrentDirectory ( ) , "_temp_" + DateTime . Now . Ticks ) ;
2012-02-05 06:34:36 +00:00
MockedRestProvider = new Mock < RestProvider > ( ) ;
ReportingService . RestProvider = MockedRestProvider . Object ;
2012-04-30 01:24:24 +00:00
ReportingService . SetupExceptronDriver ( ) ;
2012-04-22 23:14:02 +00:00
2011-11-13 07:27:16 +00:00
Directory . CreateDirectory ( TempFolder ) ;
}
2011-11-13 19:15:40 +00:00
[TearDown]
public void TestBaseTearDown ( )
{
2011-12-15 04:15:53 +00:00
_mocker = null ;
2013-01-19 20:59:36 +00:00
try
{
if ( Directory . Exists ( TempFolder ) )
{
Directory . Delete ( TempFolder , true ) ;
}
}
catch ( Exception )
{
}
2011-11-13 19:15:40 +00:00
}
2011-12-02 01:37:38 +00:00
protected void WithStrictMocker ( )
2011-11-13 07:27:16 +00:00
{
2011-12-15 04:15:53 +00: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 07:27:16 +00:00
}
protected void WithTempAsAppPath ( )
{
2012-03-07 02:59:43 +00:00
Mocker . GetMock < EnvironmentProvider > ( )
2011-11-13 07:27:16 +00:00
. SetupGet ( c = > c . ApplicationPath )
. Returns ( VirtualPath ) ;
}
protected string GetTestFilePath ( string fileName )
{
return Path . Combine ( @".\Files\" , fileName ) ;
}
protected string ReadTestFile ( string fileName )
{
return File . ReadAllText ( GetTestFilePath ( fileName ) ) ;
}
}
}