Add canary assert

// FREEBIE
pull/1/head
Michael Kirk 8 years ago
parent ecf767ea53
commit a01164e8b1

@ -7,6 +7,7 @@
#import "OWSPrimaryStorage.h" #import "OWSPrimaryStorage.h"
#import "TSIncomingMessage.h" #import "TSIncomingMessage.h"
#import "TSMessage.h" #import "TSMessage.h"
#import "TSOutgoingMessage.h"
#import "TSThread.h" #import "TSThread.h"
#import <YapDatabase/YapDatabase.h> #import <YapDatabase/YapDatabase.h>
#import <YapDatabase/YapDatabaseQuery.h> #import <YapDatabase/YapDatabaseQuery.h>
@ -60,6 +61,10 @@ static NSString *const OWSDisappearingMessageFinderExpiresAtIndex = @"index_mess
object); object);
return; return;
} }
// We'll need to update if we ever support expiring other message types
OWSAssert([object isKindOfClass:[TSOutgoingMessage class]] || [object isKindOfClass:[TSIncomingMessage class]]);
TSMessage *message = (TSMessage *)object; TSMessage *message = (TSMessage *)object;
if ([message shouldStartExpireTimerWithTransaction:transaction]) { if ([message shouldStartExpireTimerWithTransaction:transaction]) {
if ([message isKindOfClass:[TSIncomingMessage class]]) { if ([message isKindOfClass:[TSIncomingMessage class]]) {
@ -67,9 +72,6 @@ static NSString *const OWSDisappearingMessageFinderExpiresAtIndex = @"index_mess
if (!incomingMessage.wasRead) { if (!incomingMessage.wasRead) {
return; return;
} }
} else {
// update if we ever support other disappearing message types
OWSProdLogAndFail([message isKindOfClass:[TSOutgoingMessage class]]);
} }
[messageIds addObject:key]; [messageIds addObject:key];
} }

Loading…
Cancel
Save