updated eloquera's test setup.

This commit is contained in:
kay.one 2013-02-03 16:10:15 -08:00
parent 23b90ae1c4
commit d13f977bb5
8 changed files with 92 additions and 147 deletions

View file

@ -0,0 +1,45 @@
using System;
using System.Linq;
using NUnit.Framework;
using NzbDrone.Core.Datastore;
namespace NzbDrone.Core.Test.Framework
{
public abstract class ObjectDbTest : CoreTest
{
private EloqueraDb _db;
protected EloqueraDb Db
{
get
{
if (_db == null)
throw new InvalidOperationException("Test object database doesn't exists. Make sure you call WithRealDb() if you intend to use an actual database.");
return _db;
}
}
protected void WithObjectDb(bool memory = true)
{
if (memory)
{
_db = new EloqueraDbFactory().CreateMemoryDb();
}
else
{
_db = new EloqueraDbFactory().Create(Guid.NewGuid().ToString());
}
Mocker.SetConstant(Db);
}
[TearDown]
public void ObjectDbTearDown()
{
if (_db != null)
{
_db.Dispose();
}
}
}
}