From f9f931fc2df3e7059dc4bb317cb8690a1a4b4527 Mon Sep 17 00:00:00 2001
From: Matthew Chen <matthew@signal.org>
Date: Wed, 13 Jun 2018 16:37:44 -0400
Subject: [PATCH] Fix assert around failing sync message sends.

---
 SignalServiceKit/src/Messages/OWSMessageSender.m | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/SignalServiceKit/src/Messages/OWSMessageSender.m b/SignalServiceKit/src/Messages/OWSMessageSender.m
index 68bebe27e..b7efd340f 100644
--- a/SignalServiceKit/src/Messages/OWSMessageSender.m
+++ b/SignalServiceKit/src/Messages/OWSMessageSender.m
@@ -771,11 +771,13 @@ NSString *const OWSMessageSenderRateLimitedException = @"RateLimitedException";
 
 - (void)sendMessageToService:(TSOutgoingMessage *)message
                    recipient:(SignalRecipient *)recipient
-                      thread:(TSThread *)thread
+                      thread:(nullable TSThread *)thread
                     attempts:(int)remainingAttempts
                      success:(void (^)(void))successHandler
                      failure:(RetryableFailureHandler)failureHandler
 {
+    OWSAssert(thread || [message isKindOfClass:[OWSOutgoingSyncMessage class]]);
+
     DDLogInfo(@"%@ attempting to send message: %@, timestamp: %llu, recipient: %@",
         self.logTag,
         message.class,
@@ -1065,7 +1067,7 @@ NSString *const OWSMessageSenderRateLimitedException = @"RateLimitedException";
 
 - (void)messageSendDidFail:(TSOutgoingMessage *)message
                  recipient:(SignalRecipient *)recipient
-                    thread:(TSThread *)thread
+                    thread:(nullable TSThread *)thread
              isLocalNumber:(BOOL)isLocalNumber
             deviceMessages:(NSArray<NSDictionary *> *)deviceMessages
          remainingAttempts:(int)remainingAttempts
@@ -1077,7 +1079,7 @@ NSString *const OWSMessageSenderRateLimitedException = @"RateLimitedException";
 {
     OWSAssert(message);
     OWSAssert(recipient);
-    OWSAssert(thread);
+    OWSAssert(thread || [message isKindOfClass:[OWSOutgoingSyncMessage class]]);
     OWSAssert(deviceMessages);
     OWSAssert(responseError);
     OWSAssert(successHandler);