Sonarr/NzbDrone.Core.Test/Framework/TestDbHelper.cs

61 lines
1.9 KiB
C#
Raw Normal View History

2011-05-22 16:53:21 +00:00
// ReSharper disable RedundantUsingDirective
using System;
using System.Collections.Generic;
using System.IO;
2011-04-22 19:16:52 +00:00
using FizzWare.NBuilder;
using Moq;
2011-06-05 06:02:31 +00:00
using NzbDrone.Core.Datastore;
2011-04-04 03:50:12 +00:00
using NzbDrone.Core.Providers.Core;
2011-04-22 19:16:52 +00:00
using NzbDrone.Core.Repository;
using NzbDrone.Core.Repository.Quality;
2011-06-15 02:31:41 +00:00
using PetaPoco;
namespace NzbDrone.Core.Test.Framework
{
internal static class TestDbHelper
{
private const string DB_TEMPLATE_NAME = "_dbtemplate.sdf";
public static string ConnectionString { get; private set; }
2011-04-10 02:44:01 +00:00
2011-06-15 02:31:41 +00:00
public static IDatabase GetEmptyDatabase(bool enableLogging = false, string fileName = "")
{
Console.WriteLine("====================DataBase====================");
Console.WriteLine("Cloning database from template.");
2011-06-15 02:31:41 +00:00
if (String.IsNullOrWhiteSpace(fileName))
{
2011-06-23 06:56:17 +00:00
fileName = Guid.NewGuid() + ".sdf";
2011-06-15 02:31:41 +00:00
}
2011-06-18 01:46:22 +00:00
File.Copy(DB_TEMPLATE_NAME, fileName);
ConnectionString = Connection.GetConnectionString(fileName);
2011-06-18 01:46:22 +00:00
var database = Connection.GetPetaPocoDb(ConnectionString);
2011-06-15 02:31:41 +00:00
Console.WriteLine("====================DataBase====================");
Console.WriteLine();
Console.WriteLine();
2011-06-15 02:31:41 +00:00
return database;
}
public static void CreateDataBaseTemplate()
{
Console.WriteLine("Creating an empty PetaPoco database");
var connectionString = Connection.GetConnectionString(DB_TEMPLATE_NAME);
var database = Connection.GetPetaPocoDb(connectionString);
database.Dispose();
}
public static Series GetFakeSeries(int id, string title)
2011-04-22 19:16:52 +00:00
{
return Builder<Series>.CreateNew()
.With(c => c.SeriesId = id)
.With(c => c.Title = title)
.With(c => c.CleanTitle = Parser.NormalizeTitle(title))
.Build();
}
}
2011-04-10 02:44:01 +00:00
}