public async Task ConfirmsMessageAfterSent()
{
var capturedLockId = default(string);
var capturedStatus = (FeedbackStatus)(-1);
var connectionRegistry = GetConnectionRegistry(
confirmAction: (lockId, status) =>
{
capturedLockId = lockId;
capturedStatus = status;
});
var connector = GetConnector();
var identity = new DeviceIdentity("hub", "device_id");
var message = new EdgeMessage
.Builder(new byte[] { 0x01, 0x02, 0x03 })
.SetSystemProperties(new Dictionary <string, string>()
{
[SystemProperties.LockToken] = "12345"
})
.Build();
var sut = new Cloud2DeviceMessageHandler(connectionRegistry);
sut.SetConnector(connector);
await sut.SendC2DMessageAsync(message, identity);
Assert.Equal("12345", capturedLockId);
Assert.Equal(FeedbackStatus.Complete, capturedStatus);
}
public async Task EncodesPropertiesInTopic()
{
var capture = new SendCapture();
var connector = GetConnector(capture);
var connectionRegistry = GetConnectionRegistry();
var identity = new DeviceIdentity("hub", "device_id");
var message = new EdgeMessage
.Builder(new byte[] { 0x01, 0x02, 0x03 })
.SetProperties(new Dictionary <string, string>()
{
["prop1"] = "value1", ["prop2"] = "value2"
})
.SetSystemProperties(new Dictionary <string, string>()
{
["userId"] = "userid", ["cid"] = "corrid"
})
.Build();
var sut = new Cloud2DeviceMessageHandler(connectionRegistry);
sut.SetConnector(connector);
await sut.SendC2DMessageAsync(message, identity);
Assert.StartsWith("$edgehub/device_id/messages/c2d/post/", capture.Topic);
Assert.Contains("prop1=value1", capture.Topic);
Assert.Contains("prop2=value2", capture.Topic);
Assert.Contains("%24.uid=userid", capture.Topic);
Assert.Contains("%24.cid=corrid", capture.Topic);
}
public async Task SendsMessageDataAsPayload()
{
var capture = new SendCapture();
var connector = GetConnector(capture);
var connectionRegistry = GetConnectionRegistry();
var identity = new DeviceIdentity("hub", "device_id");
var message = new EdgeMessage
.Builder(new byte[] { 0x01, 0x02, 0x03 })
.Build();
var sut = new Cloud2DeviceMessageHandler(connectionRegistry);
sut.SetConnector(connector);
await sut.SendC2DMessageAsync(message, identity);
Assert.Equal(new byte[] { 0x01, 0x02, 0x03 }, capture.Content);
}
public async Task EncodesDeviceNameInTopic()
{
var capture = new SendCapture();
var connector = GetConnector(capture);
var connectionRegistry = GetConnectionRegistry();
var identity = new DeviceIdentity("hub", "device_id");
var message = new EdgeMessage
.Builder(new byte[] { 0x01, 0x02, 0x03 })
.Build();
var sut = new Cloud2DeviceMessageHandler(connectionRegistry);
sut.SetConnector(connector);
await sut.SendC2DMessageAsync(message, identity);
Assert.Equal("$edgehub/device_id/messages/c2d/post/", capture.Topic);
}
public async Task DoesNotSendToModule()
{
var connector = GetConnector();
var connectionRegistry = GetConnectionRegistry();
var identity = new ModuleIdentity("hub", "device_id", "module_id");
var message = new EdgeMessage
.Builder(new byte[] { 0x01, 0x02, 0x03 })
.Build();
var sut = new Cloud2DeviceMessageHandler(connectionRegistry);
sut.SetConnector(connector);
await sut.SendC2DMessageAsync(message, identity);
Mock.Get(connector)
.Verify(c => c.SendAsync(It.IsAny <string>(), It.IsAny <byte[]>()), Times.Never());
}
public async Task SendsMessageDataAsPayload()
{
var capture = new SendCapture();
var connector = GetConnector(capture);
var connectionRegistry = GetConnectionRegistry();
var identity = new DeviceIdentity("hub", "device_id");
var message = new EdgeMessage
.Builder(new byte[] { 0x01, 0x02, 0x03 })
.SetSystemProperties(new Dictionary <string, string>()
{
[SystemProperties.LockToken] = "12345"
})
.Build();
var sut = new Cloud2DeviceMessageHandler(connectionRegistry);
sut.SetConnector(connector);
await sut.SendC2DMessageAsync(message, identity, true);
Assert.Equal(new byte[] { 0x01, 0x02, 0x03 }, capture.Content);
}
async Task <(IMessageConsumer, IMessageConsumer, NullBrokerConnector)> SetupEnvironment()
{
Routing.UserMetricLogger = NullRoutingUserMetricLogger.Instance;
Routing.PerfCounter = NullRoutingPerfCounter.Instance;
Routing.UserAnalyticsLogger = NullUserAnalyticsLogger.Instance;
var defaultRetryStrategy = new FixedInterval(5, TimeSpan.FromSeconds(5));
var defaultRevivePeriod = TimeSpan.FromHours(1);
var defaultTimeout = TimeSpan.FromSeconds(60);
var endpointExecutorConfig = new EndpointExecutorConfig(defaultTimeout, defaultRetryStrategy, defaultRevivePeriod, true);
var cloudProxyDispatcher = new BrokeredCloudProxyDispatcher();
var cloudConnectionProvider = new BrokeredCloudConnectionProvider(cloudProxyDispatcher);
var identityProvider = new IdentityProvider(iotHubName);
var deviceConnectivityManager = new BrokeredDeviceConnectivityManager(cloudProxyDispatcher);
var connectionManager = new ConnectionManager(cloudConnectionProvider, Mock.Of <ICredentialsCache>(), new IdentityProvider(iotHubName), deviceConnectivityManager);
var routingMessageConverter = new RoutingMessageConverter();
var routeFactory = new EdgeRouteFactory(new EndpointFactory(connectionManager, routingMessageConverter, edgeDeviceId, 10, 10));
var routesList = new[] { routeFactory.Create("FROM /messages INTO $upstream") };
var endpoints = routesList.Select(r => r.Endpoint);
var routerConfig = new RouterConfig(endpoints, routesList);
var dbStoreProvider = new InMemoryDbStoreProvider();
var storeProvider = new StoreProvider(dbStoreProvider);
var messageStore = new MessageStore(storeProvider, CheckpointStore.Create(storeProvider), TimeSpan.MaxValue, false, 1800);
var endpointExecutorFactory = new StoringAsyncEndpointExecutorFactory(endpointExecutorConfig, new AsyncEndpointExecutorOptions(1, TimeSpan.FromMilliseconds(10)), messageStore);
var router = await Router.CreateAsync(Guid.NewGuid().ToString(), iotHubName, routerConfig, endpointExecutorFactory);
var messageConverterProvider = new MessageConverterProvider(
new Dictionary <Type, IMessageConverter>()
{
{ typeof(Twin), new TwinMessageConverter() },
{ typeof(TwinCollection), new TwinCollectionMessageConverter() }
});
var twinManager = TwinManager.CreateTwinManager(connectionManager, messageConverterProvider, Option.None <IStoreProvider>());
var invokeMethodHandler = Mock.Of <IInvokeMethodHandler>();
var subscriptionProcessor = new SubscriptionProcessor(connectionManager, invokeMethodHandler, deviceConnectivityManager);
var edgeHub = new RoutingEdgeHub(router, routingMessageConverter, connectionManager, twinManager, edgeDeviceId, edgeModuleName, invokeMethodHandler, subscriptionProcessor, Mock.Of <IDeviceScopeIdentitiesCache>());
var brokerConnector = new NullBrokerConnector(cloudProxyDispatcher);
cloudProxyDispatcher.SetConnector(brokerConnector);
cloudProxyDispatcher.BindEdgeHub(edgeHub);
var connectionProvider = new ConnectionProvider(connectionManager, edgeHub, TimeSpan.FromSeconds(30));
var authenticator = new NullAuthenticator();
var edgeHubIdentity = new ModuleIdentity(iotHubName, edgeDeviceId, edgeModuleName);
var tokenCredentials = new TokenCredentials(edgeHubIdentity, "qwerty", "test-product", Option.Some("test-model"), Option.None <string>(), false);
var systemComponentProvider = new SystemComponentIdProvider(tokenCredentials);
var connectionHandler = default(ConnectionHandler);
connectionHandler = new ConnectionHandler(
Task.FromResult <IConnectionProvider>(connectionProvider),
Task.FromResult <IAuthenticator>(authenticator),
identityProvider,
systemComponentProvider,
DeviceProxyFactory);
DeviceProxy DeviceProxyFactory(IIdentity identity, bool isDirectClient)
{
return(new DeviceProxy(identity, isDirectClient, connectionHandler, Mock.Of <ITwinHandler>(), Mock.Of <IModuleToModuleMessageHandler>(), Mock.Of <ICloud2DeviceMessageHandler>(), Mock.Of <IDirectMethodHandler>()));
}
var cloud2DeviceMessageHandler = new Cloud2DeviceMessageHandler(connectionHandler);
var moduleToModuleMessageHandler = new ModuleToModuleMessageHandler(connectionHandler, identityProvider, new ModuleToModuleResponseTimeout(TimeSpan.FromSeconds(10)));
var directMethodHandler = new DirectMethodHandler(connectionHandler, identityProvider);
var twinHandler = new TwinHandler(connectionHandler, identityProvider);
var subscriptionChangeHandler = new SubscriptionChangeHandler(
cloud2DeviceMessageHandler,
moduleToModuleMessageHandler,
directMethodHandler,
twinHandler,
connectionHandler,
identityProvider);
return(subscriptionChangeHandler, cloudProxyDispatcher, brokerConnector);
}
