Sonarr/NzbDrone.Core.Test/MockLib.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using Moq;
using NzbDrone.Core.Providers;
using SubSonic.DataProviders;
using SubSonic.Repository;

namespace NzbDrone.Core.Test
{
    /// <summary>
    /// Provides the standard Mocks needed for a typical test
    /// </summary>
    static class MockLib
    {
        public static string[] StandardSeries
        {
            get { return new string[] { "C:\\TV\\The Simpsons", "C:\\TV\\Family Guy" }; }
        }

        public const string MemoryConnection = "Data Source=:memory:;Version=3;New=True";


        public static IRepository MemoryRepository
        {
            get
            {
                var provider = ProviderFactory.GetProvider(MemoryConnection, "System.Data.SQLite");
                return new SimpleRepository(provider, SimpleRepositoryOptions.RunMigrations);
            }
        }

        public static IConfigProvider StandardConfig
        {
            get
            {
                var mock = new Mock<IConfigProvider>();
                mock.SetupGet(c => c.SeriesRoot).Returns("C:\\");
                return mock.Object;
            }
        }

        public static IDiskProvider StandardDisk
        {
            get
            {
                var mock = new Mock<IDiskProvider>();
                mock.Setup(c => c.GetDirectories(It.IsAny<String>())).Returns(StandardSeries);
                mock.Setup(c => c.Exists(It.Is<String>(d => StandardSeries.Contains(d)))).Returns(true);
                return mock.Object;
            }
        }
    }
}