mirror of
https://github.com/Radarr/Radarr.git
synced 2024-09-11 12:02:35 +02:00
188 lines
5.1 KiB
C#
188 lines
5.1 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using FluentMigrator.Runner;
|
|
using Marr.Data;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Common.Messaging;
|
|
using NzbDrone.Core.Datastore;
|
|
using NzbDrone.Core.Datastore.Migration.Framework;
|
|
|
|
namespace NzbDrone.Core.Test.Framework
|
|
{
|
|
[Category("DbTest")]
|
|
public abstract class DbTest<TSubject, TModel> : DbTest
|
|
where TSubject : class
|
|
where TModel : ModelBase, new()
|
|
{
|
|
private TSubject _subject;
|
|
|
|
protected BasicRepository<TModel> Storage { get; private set; }
|
|
|
|
protected IList<TModel> AllStoredModels
|
|
{
|
|
get
|
|
{
|
|
return Storage.All().ToList();
|
|
}
|
|
}
|
|
|
|
protected TModel StoredModel
|
|
{
|
|
get
|
|
{
|
|
return Storage.All().Single();
|
|
}
|
|
}
|
|
|
|
[SetUp]
|
|
public void CoreTestSetup()
|
|
{
|
|
_subject = null;
|
|
Storage = Mocker.Resolve<BasicRepository<TModel>>();
|
|
}
|
|
|
|
protected TSubject Subject
|
|
{
|
|
get
|
|
{
|
|
if (_subject == null)
|
|
{
|
|
_subject = Mocker.Resolve<TSubject>();
|
|
}
|
|
|
|
return _subject;
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
public abstract class DbTest : CoreTest
|
|
{
|
|
private ITestDatabase _db;
|
|
|
|
protected virtual MigrationType MigrationType
|
|
{
|
|
get
|
|
{
|
|
return MigrationType.Main;
|
|
|
|
}
|
|
}
|
|
|
|
protected ITestDatabase 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;
|
|
}
|
|
}
|
|
|
|
private void WithTestDb()
|
|
{
|
|
WithTempAsAppPath();
|
|
|
|
|
|
Mocker.SetConstant<IAnnouncer>(Mocker.Resolve<MigrationLogger>());
|
|
Mocker.SetConstant<IConnectionStringFactory>(Mocker.Resolve<ConnectionStringFactory>());
|
|
Mocker.SetConstant<ISQLiteMigrationHelper>(Mocker.Resolve<SQLiteMigrationHelper>());
|
|
Mocker.SetConstant<ISQLiteAlter>(Mocker.Resolve<SQLiteAlter>());
|
|
Mocker.SetConstant<IMigrationController>(Mocker.Resolve<MigrationController>());
|
|
|
|
MapRepository.Instance.EnableTraceLogging = true;
|
|
|
|
var factory = Mocker.Resolve<DbFactory>();
|
|
var _database = factory.Create(MigrationType);
|
|
_db = new TestDatabase(_database);
|
|
Mocker.SetConstant(_database);
|
|
}
|
|
|
|
[SetUp]
|
|
public void SetupReadDb()
|
|
{
|
|
WithTestDb();
|
|
}
|
|
|
|
[TearDown]
|
|
public void TearDown()
|
|
{
|
|
if (TestFolderInfo != null && Directory.Exists(TestFolderInfo.AppDataFolder))
|
|
{
|
|
var files = Directory.GetFiles(TestFolderInfo.AppDataFolder);
|
|
|
|
foreach (var file in files)
|
|
{
|
|
try
|
|
{
|
|
File.Delete(file);
|
|
}
|
|
catch (Exception)
|
|
{
|
|
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
public interface ITestDatabase
|
|
{
|
|
void InsertMany<T>(IEnumerable<T> items) where T : ModelBase, new();
|
|
T Insert<T>(T item) where T : ModelBase, new();
|
|
List<T> All<T>() where T : ModelBase, new();
|
|
T Single<T>() where T : ModelBase, new();
|
|
void Update<T>(T childModel) where T : ModelBase, new();
|
|
void Delete<T>(T childModel) where T : ModelBase, new();
|
|
}
|
|
|
|
public class TestDatabase : ITestDatabase
|
|
{
|
|
private readonly IDatabase _dbConnection;
|
|
private IMessageAggregator _messageAggregator;
|
|
|
|
public TestDatabase(IDatabase dbConnection)
|
|
{
|
|
_messageAggregator = new Mock<IMessageAggregator>().Object;
|
|
_dbConnection = dbConnection;
|
|
}
|
|
|
|
public void InsertMany<T>(IEnumerable<T> items) where T : ModelBase, new()
|
|
{
|
|
new BasicRepository<T>(_dbConnection, _messageAggregator).InsertMany(items.ToList());
|
|
}
|
|
|
|
public T Insert<T>(T item) where T : ModelBase, new()
|
|
{
|
|
return new BasicRepository<T>(_dbConnection, _messageAggregator).Insert(item);
|
|
}
|
|
|
|
public List<T> All<T>() where T : ModelBase, new()
|
|
{
|
|
return new BasicRepository<T>(_dbConnection, _messageAggregator).All().ToList();
|
|
}
|
|
|
|
public T Single<T>() where T : ModelBase, new()
|
|
{
|
|
return All<T>().SingleOrDefault();
|
|
}
|
|
|
|
public void Update<T>(T childModel) where T : ModelBase, new()
|
|
{
|
|
new BasicRepository<T>(_dbConnection, _messageAggregator).Update(childModel);
|
|
}
|
|
|
|
public void Delete<T>(T childModel) where T : ModelBase, new()
|
|
{
|
|
new BasicRepository<T>(_dbConnection, _messageAggregator).Delete(childModel);
|
|
}
|
|
}
|
|
} |