public async Task InvokeMethodOnModuleTest(ITransportSettings[] transportSettings)
{
// Arrange
string iotHubConnectionString = await SecretsHelper.GetSecretFromConfigKey("iotHubConnStrKey");
string edgeDeviceConnectionString = await SecretsHelper.GetSecretFromConfigKey("edgeCapableDeviceConnStrKey");
IotHubConnectionStringBuilder connectionStringBuilder = IotHubConnectionStringBuilder.Create(edgeDeviceConnectionString);
RegistryManager rm = RegistryManager.CreateFromConnectionString(edgeDeviceConnectionString);
ModuleClient receiver = null;
var request = new TestMethodRequest("Prop1", 10);
var response = new TestMethodResponse("RespProp1", 20);
TestMethodRequest receivedRequest = null;
Task <MethodResponse> MethodHandler(MethodRequest methodRequest, object context)
{
receivedRequest = JsonConvert.DeserializeObject <TestMethodRequest>(methodRequest.DataAsJson);
return(Task.FromResult(new MethodResponse(
Encoding.UTF8.GetBytes(JsonConvert.SerializeObject(response)), 200)));
}
try
{
string receiverModuleName = "receiver1";
ServiceClient sender = ServiceClient.CreateFromConnectionString(iotHubConnectionString);
string receiverModuleConnectionString = await RegistryManagerHelper.CreateModuleIfNotExists(rm, connectionStringBuilder.HostName, connectionStringBuilder.DeviceId, receiverModuleName);
receiver = ModuleClient.CreateFromConnectionString(receiverModuleConnectionString, transportSettings);
await receiver.OpenAsync();
await receiver.SetMethodHandlerAsync("poke", MethodHandler, null);
// Act
CloudToDeviceMethodResult cloudToDeviceMethodResult = await sender.InvokeDeviceMethodAsync(
connectionStringBuilder.DeviceId,
receiverModuleName,
new CloudToDeviceMethod("poke").SetPayloadJson(JsonConvert.SerializeObject(request)));
// Assert
Assert.NotNull(cloudToDeviceMethodResult);
Assert.NotNull(receivedRequest);
Assert.Equal(receivedRequest.RequestProp1, request.RequestProp1);
Assert.Equal(receivedRequest.RequestProp2, request.RequestProp2);
Assert.Equal(200, cloudToDeviceMethodResult.Status);
var receivedResponse = JsonConvert.DeserializeObject <TestMethodResponse>(cloudToDeviceMethodResult.GetPayloadAsJson());
Assert.NotNull(receivedResponse);
Assert.Equal(receivedResponse.ResponseProp1, response.ResponseProp1);
Assert.Equal(receivedResponse.ResponseProp2, response.ResponseProp2);
}
finally
{
if (rm != null)
{
await rm.CloseAsync();
}
if (receiver != null)
{
await receiver.CloseAsync();
}
}
// wait for the connection to be closed on the Edge side
await Task.Delay(TimeSpan.FromSeconds(10));
}
public async Task InvokeMethodOnModuleTest(ITransportSettings[] transportSettings)
{
// Arrange
string iotHubConnectionString = await SecretsHelper.GetSecretFromConfigKey("iotHubConnStrKey");
IotHubConnectionStringBuilder connectionStringBuilder = IotHubConnectionStringBuilder.Create(iotHubConnectionString);
RegistryManager rm = RegistryManager.CreateFromConnectionString(iotHubConnectionString);
ModuleClient receiver = null;
string edgeDeviceConnectionString = ConfigHelper.TestConfig[EdgeHubConstants.ConfigKey.IotHubConnectionString];
Client.IotHubConnectionStringBuilder edgeHubConnectionStringBuilder = Client.IotHubConnectionStringBuilder.Create(edgeDeviceConnectionString);
string edgeDeviceId = edgeHubConnectionStringBuilder.DeviceId;
var request = new TestMethodRequest("Prop1", 10);
var response = new TestMethodResponse("RespProp1", 20);
TestMethodRequest receivedRequest = null;
Task <MethodResponse> MethodHandler(MethodRequest methodRequest, object context)
{
receivedRequest = JsonConvert.DeserializeObject <TestMethodRequest>(methodRequest.DataAsJson);
return(Task.FromResult(
new MethodResponse(
Encoding.UTF8.GetBytes(JsonConvert.SerializeObject(response)),
200)));
}
string receiverModuleName = "method-module";
try
{
ServiceClient sender = ServiceClient.CreateFromConnectionString(iotHubConnectionString);
string receiverModuleConnectionString = await RegistryManagerHelper.CreateModuleIfNotExists(rm, connectionStringBuilder.HostName, edgeDeviceId, receiverModuleName);
receiver = ModuleClient.CreateFromConnectionString(receiverModuleConnectionString, transportSettings);
await receiver.OpenAsync();
await receiver.SetMethodHandlerAsync("poke", MethodHandler, null);
var waitStart = DateTime.Now;
var isConnected = false;
while (!isConnected && (DateTime.Now - waitStart) < TimeSpan.FromSeconds(30))
{
var connectedDevice = await rm.GetModuleAsync(edgeDeviceId, receiverModuleName);
isConnected = connectedDevice.ConnectionState == DeviceConnectionState.Connected;
}
Assert.True(isConnected);
// Need longer sleep to ensure receiver is completely initialized
await Task.Delay(TimeSpan.FromSeconds(10));
// Act
CloudToDeviceMethodResult cloudToDeviceMethodResult = await sender.InvokeDeviceMethodAsync(
edgeDeviceId,
receiverModuleName,
new CloudToDeviceMethod("poke").SetPayloadJson(JsonConvert.SerializeObject(request)));
// Assert
Assert.NotNull(cloudToDeviceMethodResult);
Assert.NotNull(receivedRequest);
Assert.Equal(receivedRequest.RequestProp1, request.RequestProp1);
Assert.Equal(receivedRequest.RequestProp2, request.RequestProp2);
Assert.Equal(200, cloudToDeviceMethodResult.Status);
var receivedResponse = JsonConvert.DeserializeObject <TestMethodResponse>(cloudToDeviceMethodResult.GetPayloadAsJson());
Assert.NotNull(receivedResponse);
Assert.Equal(receivedResponse.ResponseProp1, response.ResponseProp1);
Assert.Equal(receivedResponse.ResponseProp2, response.ResponseProp2);
}
finally
{
await rm.CloseAsync();
if (receiver != null)
{
await receiver.CloseAsync();
}
try
{
await RegistryManagerHelper.RemoveModule(edgeDeviceId, receiverModuleName, rm);
}
catch (Exception)
{
// ignored
}
}
// wait for the connection to be closed on the Edge side
await Task.Delay(TimeSpan.FromSeconds(10));
}
public async Task InvokeMethodOnModuleTest(ITransportSettings[] transportSettings)
{
// Arrange
string deviceName = string.Format("moduleMethodTest-{0}", transportSettings.First().GetTransportType().ToString("g"));
string iotHubConnectionString = await SecretsHelper.GetSecretFromConfigKey("iotHubConnStrKey");
IotHubConnectionStringBuilder connectionStringBuilder = IotHubConnectionStringBuilder.Create(iotHubConnectionString);
RegistryManager rm = RegistryManager.CreateFromConnectionString(iotHubConnectionString);
ModuleClient receiver = null;
var request = new TestMethodRequest("Prop1", 10);
var response = new TestMethodResponse("RespProp1", 20);
TestMethodRequest receivedRequest = null;
Task <MethodResponse> MethodHandler(MethodRequest methodRequest, object context)
{
receivedRequest = JsonConvert.DeserializeObject <TestMethodRequest>(methodRequest.DataAsJson);
return(Task.FromResult(
new MethodResponse(
Encoding.UTF8.GetBytes(JsonConvert.SerializeObject(response)),
200)));
}
string receiverModuleName = "method-module";
(string edgeDeviceId, string deviceConnStr) = await RegistryManagerHelper.CreateDevice(deviceName, iotHubConnectionString, rm, true, false);
try
{
ServiceClient sender = ServiceClient.CreateFromConnectionString(iotHubConnectionString);
string receiverModuleConnectionString = await RegistryManagerHelper.CreateModuleIfNotExists(rm, connectionStringBuilder.HostName, edgeDeviceId, receiverModuleName);
receiver = ModuleClient.CreateFromConnectionString(receiverModuleConnectionString, transportSettings);
await receiver.OpenAsync();
await receiver.SetMethodHandlerAsync("poke", MethodHandler, null);
// Need longer sleep to ensure receiver is completely initialized
await Task.Delay(TimeSpan.FromSeconds(10));
// Act
CloudToDeviceMethodResult cloudToDeviceMethodResult = await sender.InvokeDeviceMethodAsync(
edgeDeviceId,
receiverModuleName,
new CloudToDeviceMethod("poke").SetPayloadJson(JsonConvert.SerializeObject(request)));
// Assert
Assert.NotNull(cloudToDeviceMethodResult);
Assert.NotNull(receivedRequest);
Assert.Equal(receivedRequest.RequestProp1, request.RequestProp1);
Assert.Equal(receivedRequest.RequestProp2, request.RequestProp2);
Assert.Equal(200, cloudToDeviceMethodResult.Status);
var receivedResponse = JsonConvert.DeserializeObject <TestMethodResponse>(cloudToDeviceMethodResult.GetPayloadAsJson());
Assert.NotNull(receivedResponse);
Assert.Equal(receivedResponse.ResponseProp1, response.ResponseProp1);
Assert.Equal(receivedResponse.ResponseProp2, response.ResponseProp2);
}
finally
{
if (rm != null)
{
await rm.CloseAsync();
}
if (receiver != null)
{
await receiver.CloseAsync();
}
try
{
await RegistryManagerHelper.RemoveDevice(edgeDeviceId, rm);
}
catch (Exception)
{
// ignored
}
}
// wait for the connection to be closed on the Edge side
await Task.Delay(TimeSpan.FromSeconds(10));
}