Archive-Team
/
sonarr-repo-only
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed: Rare scenario where early Radarr version messes up Sonarr database

This commit is contained in:
Taloth Saldono 2019-12-05 23:59:36 +01:00 committed by Taloth
parent c3c38880e6
commit 3cbdd6bfd3
2 changed files with 68 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/NzbDrone.Core.Test/Datastore/Migration/128_rename_quality_profiles_add_upgrade_allowedFixture.cs Normal file
View File

@ -0,0 +1,28 @@
using System.Linq;
using FluentAssertions;
using NUnit.Framework;
using NzbDrone.Core.Datastore.Migration;
using NzbDrone.Core.Datastore.Migration.Framework;
using NzbDrone.Core.Test.Framework;
namespace NzbDrone.Core.Test.Datastore.Migration
{
[TestFixture]
public class rename_quality_profiles_add_upgrade_allowedFixture : MigrationTest<rename_quality_profiles_add_upgrade_allowed>
{
[Test]
public void should_handle_injected_radarr_migration()
{
var dbBefore = WithTestDb(new MigrationContext(MigrationType, 110));
// Ensure 111 isn't applied
dbBefore.GetDirectDataMapper().Query("INSERT INTO VersionInfo (Version, AppliedOn, Description) VALUES (111, '2018-12-24T18:21:07', 'remove_bitmetv')");
var dbAfter = WithMigrationTestDb();
var result = dbAfter.QueryScalar<int>("SELECT COUNT(*) FROM VersionInfo WHERE Description = 'remove_bitmetv'");
result.Should().Be(0);
}
}
}

40
src/NzbDrone.Core/Datastore/Migration/128_rename_quality_profiles_add_upgrade_allowed.cs
View File

@ -1,13 +1,20 @@
using FluentMigrator;
using FluentMigrator.Infrastructure;
using NzbDrone.Core.Datastore.Migration.Framework;
using System.Data;
using System.Data.SQLite;
namespace NzbDrone.Core.Datastore.Migration
{
[Migration(128)]
public class rename_quality_profiles_add_upgrade_allowed : NzbDroneMigrationBase
{
private IMigrationContext _capturedContext;
protected override void MainDbUpgrade()
{
FixupMigration111();
Rename.Table("Profiles").To("QualityProfiles");
Alter.Table("QualityProfiles").AddColumn("UpgradeAllowed").AsInt32().Nullable();
@ -19,5 +26,38 @@ namespace NzbDrone.Core.Datastore.Migration
Rename.Column("ProfileId").OnTable("Series").To("QualityProfileId");
}
public override void GetUpExpressions(IMigrationContext context)
{
_capturedContext = context;
base.GetUpExpressions(context);
_capturedContext = null;
}
// Early Radarr versions can mess up Sonarr's database if they point to the same config. Fixup the damage.
private void FixupMigration111()
{
// In order to get the expressions we need to check the database directly
using (var connection = new SQLiteConnection(ConnectionString))
using (var command = connection.CreateCommand())
{
connection.Open();
command.CommandText = "SELECT Description FROM VersionInfo WHERE Version = 111";
var description = command.ExecuteScalar() as string;
connection.Close();
if (description == "remove_bitmetv")
{
// Get the migration expressions from the 111 migration
var migration111 = new create_language_profiles();
migration111.GetUpExpressions(_capturedContext);
Execute.Sql("UPDATE VersionInfo SET Description = 'create_language_profiles fixup' WHERE Version = 111");
}
}
}
}
}