self-hosted/Sonarr
1
0
Fork 0
mirror of https://github.com/Sonarr/Sonarr.git synced 2024-12-16 11:37:58 +02:00
Code Issues Releases Activity

Added before migration hook, this can be used to insert "old" data into test Db

Browse Source 
so data migration could be tested.
This commit is contained in:
Keivan Beigi 2014-12-02 11:21:43 -08:00 committed by Mark McDowall
parent 37a1398338
commit c7ed76f6d3
5 changed files with 51 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/NzbDrone.Core.Test/Framework/DbTest.cs
View File

@ -89,7 +89,6 @@ private void WithTestDb()
{
WithTempAsAppPath();
Mocker.SetConstant<IAnnouncer>(Mocker.Resolve<MigrationLogger>());
Mocker.SetConstant<IConnectionStringFactory>(Mocker.Resolve<ConnectionStringFactory>());
Mocker.SetConstant<IMigrationController>(Mocker.Resolve<MigrationController>());
@ -97,9 +96,9 @@ private void WithTestDb()
MapRepository.Instance.EnableTraceLogging = true;
var factory = Mocker.Resolve<DbFactory>();
var _database = factory.Create(MigrationType);
_db = new TestDatabase(_database);
Mocker.SetConstant(_database);
var database = factory.Create(MigrationType);
_db = new TestDatabase(database);
Mocker.SetConstant(database);
}
[SetUp]

6
src/NzbDrone.Core/Datastore/DbFactory.cs
View File

@ -12,7 +12,7 @@ namespace NzbDrone.Core.Datastore
{
public interface IDbFactory
{
IDatabase Create(MigrationType migrationType = MigrationType.Main);
IDatabase Create(MigrationType migrationType = MigrationType.Main, Action<NzbDroneMigrationBase> beforeMigration = null);
}
public class DbFactory : IDbFactory
@ -43,7 +43,7 @@ public DbFactory(IMigrationController migrationController, IConnectionStringFact
_connectionStringFactory = connectionStringFactory;
}
public IDatabase Create(MigrationType migrationType = MigrationType.Main)
public IDatabase Create(MigrationType migrationType = MigrationType.Main, Action<NzbDroneMigrationBase> beforeMigration = null)
{
string connectionString;
@ -66,7 +66,7 @@ public IDatabase Create(MigrationType migrationType = MigrationType.Main)
}
}
_migrationController.MigrateToLatest(connectionString, migrationType);
_migrationController.MigrateToLatest(connectionString, migrationType, beforeMigration);
var db = new Database(() =>
{

15
src/NzbDrone.Core/Datastore/Migration/Framework/MigrationContext.cs
View File

@ -1,7 +1,18 @@
namespace NzbDrone.Core.Datastore.Migration.Framework
using System;
namespace NzbDrone.Core.Datastore.Migration.Framework
{
public class MigrationContext
{
public MigrationType MigrationType { get; set; }
public MigrationType MigrationType { get; private set; }
public Action<NzbDroneMigrationBase> BeforeMigration { get; private set; }
public MigrationContext(MigrationType migrationType, Action<NzbDroneMigrationBase> beforeAction)
{
MigrationType = migrationType;
BeforeMigration = beforeAction;
}
}
}

13
src/NzbDrone.Core/Datastore/Migration/Framework/MigrationController.cs
View File

@ -1,14 +1,14 @@
using System.Diagnostics;
using System;
using System.Diagnostics;
using System.Reflection;
using FluentMigrator.Runner;
using FluentMigrator.Runner.Initialization;
using FluentMigrator.Runner.Processors.SQLite;
namespace NzbDrone.Core.Datastore.Migration.Framework
{
public interface IMigrationController
{
void MigrateToLatest(string connectionString, MigrationType migrationType);
void MigrateToLatest(string connectionString, MigrationType migrationType, Action<NzbDroneMigrationBase> beforeMigration);
}
public class MigrationController : IMigrationController
@ -20,7 +20,7 @@ public MigrationController(IAnnouncer announcer)
_announcer = announcer;
}
public void MigrateToLatest(string connectionString, MigrationType migrationType)
public void MigrateToLatest(string connectionString, MigrationType migrationType, Action<NzbDroneMigrationBase> beforeMigration)
{
var sw = Stopwatch.StartNew();
@ -31,10 +31,7 @@ public void MigrateToLatest(string connectionString, MigrationType migrationType
var migrationContext = new RunnerContext(_announcer)
{
Namespace = "NzbDrone.Core.Datastore.Migration",
ApplicationContext = new MigrationContext
{
MigrationType = migrationType
}
ApplicationContext = new MigrationContext(migrationType, beforeMigration)
};
var options = new MigrationOptions { PreviewOnly = false, Timeout = 60 };

28
src/NzbDrone.Core/Datastore/Migration/Framework/NzbDroneMigrationBase.cs
View File

@ -1,4 +1,5 @@
using System;
using FluentMigrator;
using NLog;
using NzbDrone.Common.Instrumentation;
@ -7,10 +8,12 @@ namespace NzbDrone.Core.Datastore.Migration.Framework
public abstract class NzbDroneMigrationBase : FluentMigrator.Migration
{
protected readonly Logger _logger;
private readonly MigrationContext _migrationContext;
protected NzbDroneMigrationBase()
{
_logger = NzbDroneLogger.GetLogger(this);
_migrationContext = (MigrationContext)ApplicationContext;
}
protected virtual void MainDbUpgrade()
@ -21,11 +24,32 @@ protected virtual void LogDbUpgrade()
{
}
public int Version
{
get
{
var migrationAttribute = (MigrationAttribute)Attribute.GetCustomAttribute(GetType(), typeof(MigrationAttribute));
return (int)migrationAttribute.Version;
}
}
public MigrationContext Context
{
get
{
return _migrationContext;
}
}
public override void Up()
{
var context = (MigrationContext)ApplicationContext;
switch (context.MigrationType)
if (Context.BeforeMigration != null)
{
Context.BeforeMigration(this);
}
switch (Context.MigrationType)
{
case MigrationType.Main:
MainDbUpgrade();