110 lines
3.4 KiB
C#
110 lines
3.4 KiB
C#
using System.Collections.Generic;
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.Datastore.Migration.Framework;
|
|
using NzbDrone.Core.Test.Framework;
|
|
using NzbDrone.Core.Tv;
|
|
using System.Linq;
|
|
|
|
namespace NzbDrone.Core.Test.Datastore
|
|
{
|
|
[TestFixture]
|
|
public class SQLiteMigrationHelperFixture : DbTest
|
|
{
|
|
private SQLiteMigrationHelper _subject;
|
|
|
|
[SetUp]
|
|
public void SetUp()
|
|
{
|
|
_subject = Mocker.Resolve<SQLiteMigrationHelper>();
|
|
}
|
|
|
|
[Test]
|
|
public void should_parse_existing_columns()
|
|
{
|
|
var columns = _subject.GetColumns("Series");
|
|
|
|
columns.Should().NotBeEmpty();
|
|
|
|
columns.Values.Should().NotContain(c => string.IsNullOrWhiteSpace(c.Name));
|
|
columns.Values.Should().NotContain(c => string.IsNullOrWhiteSpace(c.Schema));
|
|
}
|
|
|
|
[Test]
|
|
public void should_create_table_from_column_list()
|
|
{
|
|
var columns = _subject.GetColumns("Series");
|
|
columns.Remove("Title");
|
|
|
|
_subject.CreateTable("Series_New", columns.Values, new List<SQLiteMigrationHelper.SQLiteIndex>());
|
|
|
|
var newColumns = _subject.GetColumns("Series_New");
|
|
|
|
newColumns.Values.Should().HaveSameCount(columns.Values);
|
|
newColumns.Should().NotContainKey("Title");
|
|
}
|
|
|
|
|
|
[Test]
|
|
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, indexes);
|
|
|
|
|
|
_subject.CopyData("Series", "Series_New", columns.Values);
|
|
}
|
|
|
|
[Test]
|
|
public void should_transfer_table_with_data()
|
|
{
|
|
var originalEpisodes = Builder<Episode>.CreateListOfSize(10).BuildListOfNew();
|
|
|
|
Mocker.Resolve<EpisodeRepository>().InsertMany(originalEpisodes);
|
|
|
|
var columns = _subject.GetColumns("Episodes");
|
|
var indexes = _subject.GetIndexes("Episodes");
|
|
|
|
columns.Remove("Title");
|
|
|
|
_subject.CreateTable("Episodes_New", columns.Values, indexes);
|
|
|
|
_subject.CopyData("Episodes", "Episodes_New", columns.Values);
|
|
|
|
_subject.GetRowCount("Episodes_New").Should().Be(originalEpisodes.Count);
|
|
}
|
|
|
|
[Test]
|
|
public void should_read_existing_indexes()
|
|
{
|
|
var indexes = _subject.GetIndexes("QualitySizes");
|
|
|
|
indexes.Should().NotBeEmpty();
|
|
|
|
indexes.Should().OnlyContain(c => c != null);
|
|
indexes.Should().OnlyContain(c => !string.IsNullOrWhiteSpace(c.Column));
|
|
indexes.Should().OnlyContain(c => c.Table == "QualitySizes");
|
|
indexes.Should().OnlyContain(c => c.Unique);
|
|
}
|
|
|
|
[Test]
|
|
public void should_add_indexes_when_creating_new_table()
|
|
{
|
|
var columns = _subject.GetColumns("QualitySizes");
|
|
var indexes = _subject.GetIndexes("QualitySizes");
|
|
|
|
|
|
_subject.CreateTable("QualityB", columns.Values, indexes);
|
|
|
|
|
|
var newIndexes = _subject.GetIndexes("QualityB");
|
|
|
|
newIndexes.Should().HaveSameCount(indexes);
|
|
newIndexes.Select(c=>c.Column).Should().BeEquivalentTo(indexes.Select(c=>c.Column));
|
|
}
|
|
}
|
|
} |