tests/MassTransit.Tests/Encryption/EncryptedFallbackDeserializerTests.cs - develop - vshmidt/masstransit

tests/MassTransit.Tests/Encryption/EncryptedFallbackDeserializerTests.cs
147 строк6 KB
denys.kozhevnikov
Add NUnit.Analyzer, rework test assertion recommendations
19 апр 2024, 23:19
19 апр 2024, 23:194603f90
#nullable enable
namespace MassTransit.Tests.Encryption
{
    using System;
    using System.Runtime.Serialization;
    using System.Security.Cryptography;
    using InMemoryTransport;
    using MassTransit.Serialization;
    using NUnit.Framework;


    [TestFixture]
    public class EncryptedFallbackDeserializerTests
    {
        [Test]
        public void MessageEncryptionTest_HavingMessageIsEncrypted_WhenMessageIsReceived_MessageCannotBeDecryptedWithDifferentKeySet()
        {
            // Arrange
            Tuple<string, string> firstServiceKeySet = GenerateEncryptionKeys();

            // second factory, representing another service.
            Tuple<string, string> secondServiceKeySet = GenerateEncryptionKeys();

            var firstFactory = CreateSerializerFactory(firstServiceKeySet);
            var firstSerializer = firstFactory.CreateSerializer();

            var secondFactory = CreateSerializerFactory(secondServiceKeySet);
            var secondDeserializer = secondFactory.CreateDeserializer();

            var inputMessage = CreateTestMessage(out var inputTestEvent);

            // Act
            var inMemorySendContext = new InMemorySendContext<TestMessage>(inputTestEvent);
            var messageBody = firstSerializer.GetMessageBody(inMemorySendContext);
            var encryptedMessage = messageBody.GetString();

            void DeserializeAction()
            {
                secondDeserializer.Deserialize(messageBody, EmptyHeaders.Instance);
            }

            // Assert

            Assert.That(encryptedMessage, Is.Not.EqualTo(inputMessage));
            Assert.Throws<SerializationException>(DeserializeAction);
        }

        [Test]
        public void MessageEncryptionTest_HavingMessageIsEncrypted_WhenMessageIsReceived_MessageIsDecryptedWithPrimaryKey()
        {
            // Arrange
            Tuple<string, string> keys = GenerateEncryptionKeys();

            var factory = CreateSerializerFactory(keys);

            var serializer = factory.CreateSerializer();
            var deserializer = factory.CreateDeserializer();

            var inputMessage = CreateTestMessage(out var inputTestEvent);

            // Act
            var inMemorySendContext = new InMemorySendContext<TestMessage>(inputTestEvent);
            var messageBody = serializer.GetMessageBody(inMemorySendContext);
            var encryptedMessage = messageBody.GetString();

            var deserializerContext = deserializer.Deserialize(messageBody, EmptyHeaders.Instance);
            var canDecryptMessage = deserializerContext.TryGetMessage(out TestMessage? decryptedReceivedTestEvent);

            Assert.Multiple(() =>
            {
                // Assert
                Assert.That(encryptedMessage, Is.Not.EqualTo(inputMessage));
                Assert.That(canDecryptMessage, Is.True);
                Assert.That(decryptedReceivedTestEvent, Is.EqualTo(inputTestEvent));
            });
        }

        [Test]
        public void MessageEncryptionTest_HavingMessageIsEncrypted_WhenMessageIsReceived_MessageIsDecryptedWithSecondaryKey()
        {
            // Arrange
            Tuple<string, string> firstServiceKeySet = GenerateEncryptionKeys();

            // second factory, representing another service.
            var keyNotInUse = GenerateEncryptionKeys().Item1;
            var secondServiceKeySet = new Tuple<string, string>(keyNotInUse, firstServiceKeySet.Item1);

            var firstFactory = CreateSerializerFactory(firstServiceKeySet);
            var firstSerializer = firstFactory.CreateSerializer();

            var secondFactory = CreateSerializerFactory(secondServiceKeySet);
            var secondDeserializer = secondFactory.CreateDeserializer();

            var inputMessage = CreateTestMessage(out var inputTestEvent);

            // Act
            var inMemorySendContext = new InMemorySendContext<TestMessage>(inputTestEvent);
            var messageBody = firstSerializer.GetMessageBody(inMemorySendContext);
            var encryptedMessage = messageBody.GetString();

            var deserializerContext = secondDeserializer.Deserialize(messageBody, EmptyHeaders.Instance);
            var canDecryptMessage = deserializerContext.TryGetMessage(out TestMessage? decryptedReceivedTestEvent);

            Assert.Multiple(() =>
            {
                // Assert
                Assert.That(encryptedMessage, Is.Not.EqualTo(inputMessage));
                Assert.That(canDecryptMessage, Is.True);
                Assert.That(decryptedReceivedTestEvent, Is.EqualTo(inputTestEvent));
            });
        }

        static string CreateTestMessage(out TestMessage inputTestEvent)
        {
            var inputMessage = "test message";
            inputTestEvent = new TestMessage(Guid.NewGuid(), inputMessage);

            return inputMessage;
        }

        static EncryptedFallbackSerializerFactoryV2 CreateSerializerFactory(Tuple<string, string> keys)
        {
            var primaryKey = Convert.FromBase64String(keys.Item1);
            var keyProvider = new ConstantSecureKeyProvider(primaryKey);
            var primaryCrypto = new AesCryptoStreamProviderV2(keyProvider);

            var secondaryKey = Convert.FromBase64String(keys.Item2);
            var keyProvider2 = new ConstantSecureKeyProvider(secondaryKey);
            var secondaryCrypto = new AesCryptoStreamProviderV2(keyProvider2);

            return new EncryptedFallbackSerializerFactoryV2(primaryCrypto, secondaryCrypto);
        }

        static Tuple<string, string> GenerateEncryptionKeys()
        {
            var aes = Aes.Create();
            aes.GenerateKey();

            var key1 = Convert.ToBase64String(aes.Key);

            aes.GenerateKey();
            var key2 = Convert.ToBase64String(aes.Key);

            return new Tuple<string, string>(key1, key2);
        }
    }
}