public void SetUp()
        {
            _loggerFactory  = new LoggerFactory();
            _loggerProvider = new TestLoggerProvider();
            _loggerFactory.AddProvider(_loggerProvider);

            _consumerClientMock = new Mock <IEventHubConsumerClient>(MockBehavior.Strict);
            _consumerClientMock.Setup(c => c.ConsumerGroup).Returns(_consumerGroup);
            _consumerClientMock.Setup(c => c.EventHubName).Returns(_eventHubName);
            _consumerClientMock.Setup(c => c.FullyQualifiedNamespace).Returns(_namespace);
            _consumerClientMock.Setup(client => client.GetPartitionsAsync())
            .Returns(() => Task.FromResult(_partitions.Select(p => p.Id).ToArray()));
            _consumerClientMock.Setup(client => client.GetPartitionPropertiesAsync(IsAny <string>()))
            .Returns((string id) => Task.FromResult(_partitions.SingleOrDefault(p => p.Id == id)));

            this._mockCheckpointStore = new Mock <BlobCheckpointStoreInternal>(MockBehavior.Strict);

            _mockCheckpointStore.Setup(s => s.GetCheckpointAsync(_namespace, _eventHubName, _consumerGroup, It.IsAny <string>(), default))
            .Returns <string, string, string, string, CancellationToken>((ns, hub, cg, partitionId, ct) => Task.FromResult(_checkpoints.SingleOrDefault(cp => cp.PartitionId == partitionId)));

            _scaleMonitor = new EventHubsScaleMonitor(
                _functionId,
                _consumerClientMock.Object,
                _mockCheckpointStore.Object,
                _loggerFactory.CreateLogger(LogCategories.CreateTriggerCategory("EventHub")));
        }
Beispiel #2
0
        public void SetUp()
        {
            _mockBlobContainer = new Mock <BlobContainerClient>(MockBehavior.Strict);
            _loggerFactory     = new LoggerFactory();
            _loggerProvider    = new TestLoggerProvider();
            _loggerFactory.AddProvider(_loggerProvider);

            _scaleMonitor = new EventHubsScaleMonitor(
                _functionId,
                _eventHubName,
                _consumerGroup,
                _eventHubConnectionString,
                _storageConnectionString,
                _loggerFactory.CreateLogger(LogCategories.CreateTriggerCategory("EventHub")),
                _mockBlobContainer.Object);
        }