protected async Task ReceiveMessageAsync(CancellationToken ct)
{
ExceptionDispatchInfo exInfo = null;
_mRecv.ScheduleTime = null;
_swRecv.Restart();
try
{
Task <Client.Message> t = _dc.ReceiveAsync(ct);
_mRecv.ScheduleTime = _swRecv.ElapsedMilliseconds;
_swRecv.Restart();
Client.Message msg = await t.ConfigureAwait(false);
await _dc.CompleteAsync(msg).ConfigureAwait(false);
int deviceIdFromMessage = BitConverter.ToInt32(msg.GetBytes());
if (_id != deviceIdFromMessage)
{
throw new InvalidOperationException($"DeviceId mismatch: Expected {_id} actual {deviceIdFromMessage}.");
}
}
catch (Exception ex)
{
_mRecv.ErrorMessage = $"{ex.GetType().Name} - {ex.Message}";
exInfo = ExceptionDispatchInfo.Capture(ex);
}
_mRecv.ExecuteTime = _swRecv.ElapsedMilliseconds;
await _writer.WriteAsync(_mRecv).ConfigureAwait(false);
exInfo?.Throw();
}
public static async Task VerifyReceivedC2DMessageAsync(Client.TransportType transport, DeviceClient dc, string deviceId, string payload, string p1Value)
{
var wait = true;
Stopwatch sw = new Stopwatch();
sw.Start();
while (wait)
{
Client.Message receivedMessage = null;
_log.WriteLine($"Receiving messages for device {deviceId}.");
if (transport == Client.TransportType.Http1)
{
// timeout on HTTP is not supported
receivedMessage = await dc.ReceiveAsync().ConfigureAwait(false);
}
else
{
receivedMessage = await dc.ReceiveAsync(TimeSpan.FromSeconds(1)).ConfigureAwait(false);
}
if (receivedMessage != null)
{
string messageData = Encoding.ASCII.GetString(receivedMessage.GetBytes());
_log.WriteLine($"{nameof(VerifyReceivedC2DMessageAsync)}: Received message: for {deviceId}: {messageData}");
Assert.AreEqual(payload, messageData, $"The payload did not match for device {deviceId}");
Assert.AreEqual(1, receivedMessage.Properties.Count, $"The count of received properties did not match for device {deviceId}");
var prop = receivedMessage.Properties.Single();
Assert.AreEqual("property1", prop.Key, $"The key \"property1\" did not match for device {deviceId}");
Assert.AreEqual(p1Value, prop.Value, $"The value of \"property1\" did not match for device {deviceId}");
await dc.CompleteAsync(receivedMessage).ConfigureAwait(false);
break;
}
if (sw.Elapsed.TotalMilliseconds > FaultInjection.RecoveryTimeMilliseconds)
{
throw new TimeoutException("Test is running longer than expected.");
}
}
sw.Stop();
}
public DeserializableCommand(Client.Message message)
{
if (message == null)
{
throw new ArgumentNullException("message");
}
Debug.Assert(
!string.IsNullOrEmpty(message.LockToken),
"message.LockToken is a null reference or empty string.");
_lockToken = message.LockToken;
byte[] messageBytes = message.GetBytes(); // this needs to be saved if needed later, because it can only be read once from the original Message
string jsonData = Encoding.UTF8.GetString(messageBytes);
_commandHistory = JsonConvert.DeserializeObject<CommandHistory>(jsonData);
}
public static async Task VerifyReceivedC2dMessageWithCancellationTokenAsync(Client.TransportType transport, DeviceClient dc, string deviceId, string payload, string p1Value, MsTestLogger logger)
{
var sw = new Stopwatch();
bool received = false;
sw.Start();
while (!received && sw.ElapsedMilliseconds < FaultInjection.RecoveryTimeMilliseconds)
{
logger.Trace($"Receiving messages for device {deviceId}.");
using var cts = new CancellationTokenSource(s_oneMinute);
using Client.Message receivedMessage = await dc.ReceiveAsync(cts.Token).ConfigureAwait(false);
if (receivedMessage == null)
{
Assert.Fail($"No message is received for device {deviceId} in {s_oneMinute}.");
}
try
{
// always complete message
await dc.CompleteAsync(receivedMessage).ConfigureAwait(false);
}
catch (Exception)
{
// ignore exception from CompleteAsync
}
string messageData = Encoding.ASCII.GetString(receivedMessage.GetBytes());
logger.Trace($"{nameof(VerifyReceivedC2dMessageWithCancellationTokenAsync)}: Received message: for {deviceId}: {messageData}");
if (Equals(payload, messageData))
{
Assert.AreEqual(1, receivedMessage.Properties.Count, $"The count of received properties did not match for device {deviceId}");
System.Collections.Generic.KeyValuePair <string, string> prop = receivedMessage.Properties.Single();
Assert.AreEqual("property1", prop.Key, $"The key \"property1\" did not match for device {deviceId}");
Assert.AreEqual(p1Value, prop.Value, $"The value of \"property1\" did not match for device {deviceId}");
received = true;
}
}
sw.Stop();
Assert.IsTrue(received, $"No message received for device {deviceId} with payload={payload} in {FaultInjection.RecoveryTimeMilliseconds}.");
}
public DeserializableCommand(Client.Message message, ISerialize serializer)
{
if (message == null)
{
throw new ArgumentNullException("message");
}
if (serializer == null)
{
throw new ArgumentNullException("serializer");
}
Debug.Assert(
!string.IsNullOrEmpty(message.LockToken),
"message.LockToken is a null reference or empty string.");
_lockToken = message.LockToken;
byte[] messageBytes = message.GetBytes(); // this needs to be saved if needed later, because it can only be read once from the original Message
_command = serializer.DeserializeObject <dynamic>(messageBytes);
}
public static async Task VerifyReceivedC2DMessageAsync(Client.TransportType transport, DeviceClient dc, string payload, string p1Value)
{
var wait = true;
Stopwatch sw = new Stopwatch();
sw.Start();
while (wait)
{
Client.Message receivedMessage = null;
receivedMessage = await dc.ReceiveAsync(TimeSpan.FromSeconds(1)).ConfigureAwait(false);
if (receivedMessage != null)
{
string messageData = Encoding.ASCII.GetString(receivedMessage.GetBytes());
_log.WriteLine($"{nameof(VerifyReceivedC2DMessageAsync)}: Received message: {receivedMessage}");
Assert.AreEqual(payload, messageData);
Assert.AreEqual(1, receivedMessage.Properties.Count);
var prop = receivedMessage.Properties.Single();
Assert.AreEqual("property1", prop.Key);
Assert.AreEqual(p1Value, prop.Value);
await dc.CompleteAsync(receivedMessage).ConfigureAwait(false);
break;
}
if (sw.Elapsed.TotalMilliseconds > FaultInjection.RecoveryTimeMilliseconds)
{
throw new TimeoutException("Test is running longer than expected.");
}
}
sw.Stop();
}
public static async Task VerifyReceivedC2DMessageAsync(Client.TransportType transport, DeviceClient dc, string deviceId, Message message, string payload, MsTestLogger logger)
{
string receivedMessageDestination = $"/devices/{deviceId}/messages/deviceBound";
var sw = new Stopwatch();
bool received = false;
sw.Start();
while (!received && sw.ElapsedMilliseconds < FaultInjection.RecoveryTimeMilliseconds)
{
Client.Message receivedMessage = null;
try
{
logger.Trace($"Receiving messages for device {deviceId}.");
if (transport == Client.TransportType.Http1)
{
// timeout on HTTP is not supported
receivedMessage = await dc.ReceiveAsync().ConfigureAwait(false);
}
else
{
receivedMessage = await dc.ReceiveAsync(s_oneMinute).ConfigureAwait(false);
}
if (receivedMessage == null)
{
Assert.Fail($"No message is received for device {deviceId} in {s_oneMinute}.");
}
try
{
// always complete message
await dc.CompleteAsync(receivedMessage).ConfigureAwait(false);
}
catch (Exception)
{
// ignore exception from CompleteAsync
}
Assert.AreEqual(receivedMessage.MessageId, message.MessageId, "Recieved message Id is not what was sent by service");
Assert.AreEqual(receivedMessage.UserId, message.UserId, "Recieved user Id is not what was sent by service");
Assert.AreEqual(receivedMessage.To, receivedMessageDestination, "Recieved message destination is not what was sent by service");
string messageData = Encoding.ASCII.GetString(receivedMessage.GetBytes());
logger.Trace($"{nameof(VerifyReceivedC2DMessageAsync)}: Received message: for {deviceId}: {messageData}");
if (Equals(payload, messageData))
{
Assert.AreEqual(1, receivedMessage.Properties.Count, $"The count of received properties did not match for device {deviceId}");
System.Collections.Generic.KeyValuePair <string, string> prop = receivedMessage.Properties.Single();
string propertyKey = "property1";
Assert.AreEqual(propertyKey, prop.Key, $"The key \"property1\" did not match for device {deviceId}");
Assert.AreEqual(message.Properties[propertyKey], prop.Value, $"The value of \"property1\" did not match for device {deviceId}");
received = true;
}
}
finally
{
receivedMessage?.Dispose();
}
}
sw.Stop();
Assert.IsTrue(received, $"No message received for device {deviceId} with payload={payload} in {FaultInjection.RecoveryTimeMilliseconds}.");
}