DeviceBridgeTests/Controllers/MethodsControllerTests.cs

// Copyright (c) Microsoft Corporation. All rights reserved.

using System.Threading;
using System.Threading.Tasks;
using DeviceBridge.Models;
using DeviceBridge.Services;
using Moq;
using NLog;
using NUnit.Framework;

namespace DeviceBridge.Controllers.Tests
{
    [TestFixture]
    public class MethodsControllerTests
    {
        private const string MockDeviceId = "test-device";
        private const string MockCallbackUrl = "mock-callback-url";
        private Mock<IDataSubscriptionService> _dataSubscriptionServiceMock;
        private MethodsController _methodsController;
        private DeviceSubscriptionWithStatus _deviceSubscriptionWithStatus;

        [SetUp]
        public void Setup()
        {
            _dataSubscriptionServiceMock = new Mock<IDataSubscriptionService>();
            _methodsController = new MethodsController(LogManager.GetCurrentClassLogger(), _dataSubscriptionServiceMock.Object);

            var deviceSubscription = new DeviceSubscription();
            _deviceSubscriptionWithStatus = new DeviceSubscriptionWithStatus(deviceSubscription);
            _dataSubscriptionServiceMock.Setup(s => s.GetDataSubscription(It.IsAny<Logger>(), MockDeviceId, DeviceSubscriptionType.Methods, It.IsAny<CancellationToken>())).Returns(Task.FromResult(_deviceSubscriptionWithStatus));
            _dataSubscriptionServiceMock.Setup(s => s.CreateOrUpdateDataSubscription(It.IsAny<Logger>(), MockDeviceId, DeviceSubscriptionType.Methods, MockCallbackUrl, It.IsAny<CancellationToken>())).Returns(Task.FromResult(_deviceSubscriptionWithStatus));
        }

        [Test]
        [Description("Test to ensure GetMethodsSubscription calls DataSubscriptionService.GetDataSubscription and the value is returned.")]
        public async Task TestGetMethodsSubscription()
        {
            var methodSubscription = await _methodsController.GetMethodsSubscription(MockDeviceId);
            _dataSubscriptionServiceMock.Verify(p => p.GetDataSubscription(It.IsAny<Logger>(), MockDeviceId, DeviceSubscriptionType.Methods, It.IsAny<CancellationToken>()));
            Assert.AreEqual(_deviceSubscriptionWithStatus, methodSubscription.Value);
        }

        [Test]
        [Description("Test to ensure that CreateOrUpdateMethodsSubscription calls DataSubscriptionService.CreateOrUpdateSubscription and the value is returned.")]
        public async Task TestCreateOrUpdateMethodsSubscription()
        {
            var body = new SubscriptionCreateOrUpdateBody()
            {
                CallbackUrl = MockCallbackUrl,
            };

            var methodSubscription = await _methodsController.CreateOrUpdateMethodsSubscription(MockDeviceId, body);
            _dataSubscriptionServiceMock.Verify(p => p.CreateOrUpdateDataSubscription(It.IsAny<Logger>(), MockDeviceId, DeviceSubscriptionType.Methods, body.CallbackUrl, It.IsAny<CancellationToken>()));
            Assert.AreEqual(_deviceSubscriptionWithStatus, methodSubscription.Value);
        }

        [Test]
        [Description("Test to ensure that CreateOrUpdateMethodsSubscription calls DataSubscriptionService.DeleteDataSubscription.")]
        public async Task TestDeleteMethodsSubscription()
        {
            await _methodsController.DeleteMethodsSubscription(MockDeviceId);
            _dataSubscriptionServiceMock.Verify(p => p.DeleteDataSubscription(It.IsAny<Logger>(), MockDeviceId, DeviceSubscriptionType.Methods, It.IsAny<CancellationToken>()));
        }
    }
}