Prowlarr/src/NzbDrone.Core.Test/Datastore/Migration/039_email_encryptionFixture.cs

using System.Collections.Generic;
using System.Linq;
using FluentAssertions;
using NUnit.Framework;
using NzbDrone.Common.Serializer;
using NzbDrone.Core.Datastore.Migration;
using NzbDrone.Core.Notifications.Email;
using NzbDrone.Core.Test.Framework;

namespace NzbDrone.Core.Test.Datastore.Migration
{
    [TestFixture]
    public class email_encryptionFixture : MigrationTest<email_encryption>
    {
        [Test]
        public void should_convert_do_not_require_encryption_to_auto()
        {
            var db = WithMigrationTestDb(c =>
            {
                c.Insert.IntoTable("Notifications").Row(new
                {
                    OnGrab = true,
                    OnHealthIssue = true,
                    IncludeHealthWarnings = true,
                    Name = "Mail Prowlarr",
                    Implementation = "Email",
                    Tags = "[]",
                    Settings = new EmailSettings38
                    {
                        Server = "smtp.gmail.com",
                        Port = 563,
                        To = new List<string> { "dont@email.me" },
                        RequireEncryption = false
                    }.ToJson(),
                    ConfigContract = "EmailSettings"
                });
            });

            var items = db.Query<NotificationDefinition39>("SELECT * FROM \"Notifications\"");

            items.Should().HaveCount(1);
            items.First().Implementation.Should().Be("Email");
            items.First().ConfigContract.Should().Be("EmailSettings");
            items.First().Settings.UseEncryption.Should().Be((int)EmailEncryptionType.Preferred);
        }

        [Test]
        public void should_convert_require_encryption_to_always()
        {
            var db = WithMigrationTestDb(c =>
            {
                c.Insert.IntoTable("Notifications").Row(new
                {
                    OnGrab = true,
                    OnHealthIssue = true,
                    IncludeHealthWarnings = true,
                    Name = "Mail Prowlarr",
                    Implementation = "Email",
                    Tags = "[]",
                    Settings = new EmailSettings38
                    {
                        Server = "smtp.gmail.com",
                        Port = 563,
                        To = new List<string> { "dont@email.me" },
                        RequireEncryption = true
                    }.ToJson(),
                    ConfigContract = "EmailSettings"
                });
            });

            var items = db.Query<NotificationDefinition39>("SELECT * FROM \"Notifications\"");

            items.Should().HaveCount(1);
            items.First().Implementation.Should().Be("Email");
            items.First().ConfigContract.Should().Be("EmailSettings");
            items.First().Settings.UseEncryption.Should().Be((int)EmailEncryptionType.Always);
        }

        [Test]
        public void should_use_defaults_when_settings_are_empty()
        {
            var db = WithMigrationTestDb(c =>
            {
                c.Insert.IntoTable("Notifications").Row(new
                {
                    OnGrab = true,
                    OnHealthIssue = true,
                    IncludeHealthWarnings = true,
                    Name = "Mail Prowlarr",
                    Implementation = "Email",
                    Tags = "[]",
                    Settings = new { }.ToJson(),
                    ConfigContract = "EmailSettings"
                });
            });

            var items = db.Query<NotificationDefinition39>("SELECT * FROM \"Notifications\"");

            items.Should().HaveCount(1);
            items.First().Implementation.Should().Be("Email");
            items.First().ConfigContract.Should().Be("EmailSettings");
            items.First().Settings.UseEncryption.Should().Be((int)EmailEncryptionType.Preferred);
        }
    }

    public class NotificationDefinition39
    {
        public int Id { get; set; }
        public string Implementation { get; set; }
        public string ConfigContract { get; set; }
        public EmailSettings39 Settings { get; set; }
        public string Name { get; set; }
        public bool OnGrab { get; set; }
        public bool OnHealthIssue { get; set; }
        public bool OnHealthRestored { get; set; }
        public bool OnApplicationUpdate { get; set; }
        public bool SupportsOnGrab { get; set; }
        public bool IncludeManualGrabs { get; set; }
        public bool SupportsOnHealthIssue { get; set; }
        public bool SupportsOnHealthRestored { get; set; }
        public bool IncludeHealthWarnings { get; set; }
        public bool SupportsOnApplicationUpdate { get; set; }
        public List<int> Tags { get; set; }
    }

    public class EmailSettings38
    {
        public string Server { get; set; }
        public int Port { get; set; }
        public bool RequireEncryption { get; set; }
        public string Username { get; set; }
        public string Password { get; set; }
        public string From { get; set; }
        public IEnumerable<string> To { get; set; }
        public IEnumerable<string> Cc { get; set; }
        public IEnumerable<string> Bcc { get; set; }
    }

    public class EmailSettings39
    {
        public string Server { get; set; }
        public int Port { get; set; }
        public int UseEncryption { get; set; }
        public string Username { get; set; }
        public string Password { get; set; }
        public string From { get; set; }
        public IEnumerable<string> To { get; set; }
        public IEnumerable<string> Cc { get; set; }
        public IEnumerable<string> Bcc { get; set; }
    }
}