session-ios/SignalServiceKit/tests/Contacts/TSContactThreadTest.m
2018-10-15 12:50:07 -04:00

43 lines
1,020 B
Objective-C

//
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
//
#import "MockSSKEnvironment.h"
#import "OWSIdentityManager.h"
#import "SSKBaseTestObjC.h"
#import "TSContactThread.h"
NS_ASSUME_NONNULL_BEGIN
@interface TSContactThreadTest : SSKBaseTestObjC
@property (nonatomic) TSContactThread *contactThread;
@end
@implementation TSContactThreadTest
- (void)setUp
{
[super setUp];
self.contactThread = [TSContactThread getOrCreateThreadWithContactId:@"fake-contact-id"];
[OWSRecipientIdentity removeAllObjectsInCollection];
}
- (void)testHasSafetyNumbersWithoutRemoteIdentity
{
XCTAssertFalse(self.contactThread.hasSafetyNumbers);
}
- (void)testHasSafetyNumbersWithRemoteIdentity
{
[[OWSIdentityManager sharedManager] saveRemoteIdentity:[[NSMutableData alloc] initWithLength:kStoredIdentityKeyLength]
recipientId:self.contactThread.contactIdentifier];
XCTAssert(self.contactThread.hasSafetyNumbers);
}
@end
NS_ASSUME_NONNULL_END