diff --git a/NzbDrone.Core.Test/Datastore/SQLiteMigrationHelperFixture.cs b/NzbDrone.Core.Test/Datastore/SQLiteMigrationHelperFixture.cs index 6f6dca730..386f63bf0 100644 --- a/NzbDrone.Core.Test/Datastore/SQLiteMigrationHelperFixture.cs +++ b/NzbDrone.Core.Test/Datastore/SQLiteMigrationHelperFixture.cs @@ -5,6 +5,7 @@ using NzbDrone.Core.Datastore.Migration.Framework; using NzbDrone.Core.Test.Framework; using NzbDrone.Core.Tv; +using System.Linq; namespace NzbDrone.Core.Test.Datastore { @@ -49,9 +50,10 @@ public void should_create_table_from_column_list() public void should_be_able_to_transfer_empty_tables() { var columns = _subject.GetColumns("Series"); + var indexes = _subject.GetIndexes("Series"); columns.Remove("Title"); - _subject.CreateTable("Series_New", columns.Values, new List()); + _subject.CreateTable("Series_New", columns.Values, indexes); _subject.CopyData("Series", "Series_New", columns.Values); @@ -65,9 +67,11 @@ public void should_transfer_table_with_data() Mocker.Resolve().InsertMany(originalEpisodes); var columns = _subject.GetColumns("Episodes"); + var indexes = _subject.GetIndexes("Episodes"); + columns.Remove("Title"); - _subject.CreateTable("Episodes_New", columns.Values, new List()); + _subject.CreateTable("Episodes_New", columns.Values, indexes); _subject.CopyData("Episodes", "Episodes_New", columns.Values); @@ -100,7 +104,7 @@ public void should_add_indexes_when_creating_new_table() var newIndexes = _subject.GetIndexes("QualityB"); newIndexes.Should().HaveSameCount(indexes); - newIndexes.Should().BeEquivalentTo(columns); + newIndexes.Select(c=>c.Column).Should().BeEquivalentTo(indexes.Select(c=>c.Column)); } } } \ No newline at end of file