using System; using System.Collections.Generic; using System.Linq; using System.Threading.Tasks; using Bit.Core.Models.Mail; using Bit.Core.Services; using Bit.Core.Settings; using Microsoft.AspNetCore.Hosting; using Microsoft.Extensions.Logging; using NSubstitute; using SendGrid; using SendGrid.Helpers.Mail; using Xunit; namespace Bit.Core.Test.Services { public class SendGridMailDeliveryServiceTests : IDisposable { private readonly SendGridMailDeliveryService _sut; private readonly GlobalSettings _globalSettings; private readonly IWebHostEnvironment _hostingEnvironment; private readonly ILogger _logger; private readonly ISendGridClient _sendGridClient; public SendGridMailDeliveryServiceTests() { _globalSettings = new GlobalSettings { Mail = { SendGridApiKey = "SendGridApiKey" } }; _hostingEnvironment = Substitute.For(); _logger = Substitute.For>(); _sendGridClient = Substitute.For(); _sut = new SendGridMailDeliveryService( _sendGridClient, _globalSettings, _hostingEnvironment, _logger ); } public void Dispose() { _sut?.Dispose(); } [Fact] public async Task SendEmailAsync_CallsSendEmailAsync_WhenMessageIsValid() { var mailMessage = new MailMessage { ToEmails = new List { "ToEmails" }, BccEmails = new List { "BccEmails" }, Subject = "Subject", HtmlContent = "HtmlContent", TextContent = "TextContent", Category = "Category" }; _sendGridClient.SendEmailAsync(Arg.Any()).Returns( new Response(System.Net.HttpStatusCode.OK, null, null)); await _sut.SendEmailAsync(mailMessage); await _sendGridClient.Received(1).SendEmailAsync( Arg.Do(msg => { msg.Received(1).AddTos(new List { new EmailAddress(mailMessage.ToEmails.First()) }); msg.Received(1).AddBccs(new List { new EmailAddress(mailMessage.ToEmails.First()) }); Assert.Equal(mailMessage.Subject, msg.Subject); Assert.Equal(mailMessage.HtmlContent, msg.HtmlContent); Assert.Equal(mailMessage.TextContent, msg.PlainTextContent); Assert.Contains("type:Cateogry", msg.Categories); Assert.Contains(msg.Categories, x => x.StartsWith("env:")); Assert.Contains(msg.Categories, x => x.StartsWith("sender:")); msg.Received(1).SetClickTracking(false, false); msg.Received(1).SetOpenTracking(false); })); } } }