bitwarden-server/test/Core.Test/Services/AmazonSesMailDeliveryServiceTests.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Amazon.SimpleEmail;
using Amazon.SimpleEmail.Model;
using Bit.Core.Models.Mail;
using Bit.Core.Services;
using Microsoft.AspNetCore.Hosting;
using Microsoft.Extensions.Logging;
using NSubstitute;
using Xunit;

namespace Bit.Core.Test.Services
{
    public class AmazonSesMailDeliveryServiceTests : IDisposable
    {
        private readonly AmazonSesMailDeliveryService _sut;

        private readonly GlobalSettings _globalSettings;
        private readonly IWebHostEnvironment _hostingEnvironment;
        private readonly ILogger<AmazonSesMailDeliveryService> _logger;
        private readonly IAmazonSimpleEmailService _amazonSimpleEmailService;

        public AmazonSesMailDeliveryServiceTests()
        {
            _globalSettings = new GlobalSettings
            {
                Amazon =
                        {
                            AccessKeyId = "AccessKeyId-AmazonSesMailDeliveryServiceTests",
                            AccessKeySecret = "AccessKeySecret-AmazonSesMailDeliveryServiceTests",
                            Region = "Region-AmazonSesMailDeliveryServiceTests"
                        }
            };

            _hostingEnvironment = Substitute.For<IWebHostEnvironment>();
            _logger = Substitute.For<ILogger<AmazonSesMailDeliveryService>>();
            _amazonSimpleEmailService = Substitute.For<IAmazonSimpleEmailService>();

            _sut = new AmazonSesMailDeliveryService(
                _globalSettings,
                _hostingEnvironment,
                _logger,
                _amazonSimpleEmailService
            );
        }

        public void Dispose()
        {
            _sut?.Dispose();
        }

        [Fact]
        public async Task SendEmailAsync_CallsSendEmailAsync_WhenMessageIsValid()
        {
            var mailMessage = new MailMessage
            {
                ToEmails = new List<string> { "ToEmails" },
                BccEmails = new List<string> { "BccEmails" },
                Subject = "Subject",
                HtmlContent = "HtmlContent",
                TextContent = "TextContent",
                Category = "Category"
            };

            await _sut.SendEmailAsync(mailMessage);

            await _amazonSimpleEmailService.Received(1).SendEmailAsync(
                Arg.Do<SendEmailRequest>(request =>
                {
                    Assert.False(string.IsNullOrEmpty(request.Source));

                    Assert.Single(request.Destination.ToAddresses);
                    Assert.Equal(mailMessage.ToEmails.First(), request.Destination.ToAddresses.First());

                    Assert.Equal(mailMessage.Subject, request.Message.Subject.Data);
                    Assert.Equal(mailMessage.HtmlContent, request.Message.Body.Html.Data);
                    Assert.Equal(mailMessage.TextContent, request.Message.Body.Text.Data);

                    Assert.Single(request.Destination.BccAddresses);
                    Assert.Equal(mailMessage.BccEmails.First(), request.Destination.BccAddresses.First());

                    Assert.Contains(request.Tags, x => x.Name == "Environment");
                    Assert.Contains(request.Tags, x => x.Name == "Sender");
                    Assert.Contains(request.Tags, x => x.Name == "Category");
                }));
        }
    }
}