mirror of
https://github.com/oxen-io/session-ios.git
synced 2023-12-13 21:30:14 +01:00
59 lines
1.5 KiB
Objective-C
59 lines
1.5 KiB
Objective-C
//
|
|
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "SignalRecipient.h"
|
|
#import "MockSSKEnvironment.h"
|
|
#import "OWSPrimaryStorage.h"
|
|
#import "SSKBaseTest.h"
|
|
#import "TSAccountManager.h"
|
|
#import "TestAppContext.h"
|
|
#import <SignalServiceKit/SignalServiceKit-Swift.h>
|
|
|
|
@interface TSAccountManager (Testing)
|
|
|
|
- (void)storeLocalNumber:(NSString *)localNumber;
|
|
|
|
@end
|
|
|
|
@interface SignalRecipientTest : SSKBaseTest
|
|
|
|
@property (nonatomic) NSString *localNumber;
|
|
|
|
@end
|
|
|
|
@implementation SignalRecipientTest
|
|
|
|
- (void)setUp
|
|
{
|
|
[super setUp];
|
|
self.localNumber = @"+13231231234";
|
|
[[TSAccountManager sharedInstance] storeLocalNumber:self.localNumber];
|
|
}
|
|
|
|
- (void)testSelfRecipientWithExistingRecord
|
|
{
|
|
// Sanity Check
|
|
XCTAssertNotNil(self.localNumber);
|
|
[[[SignalRecipient alloc] initWithTextSecureIdentifier:self.localNumber relay:nil] save];
|
|
XCTAssertNotNil([SignalRecipient recipientWithTextSecureIdentifier:self.localNumber]);
|
|
|
|
SignalRecipient *me = [SignalRecipient selfRecipient];
|
|
XCTAssert(me);
|
|
XCTAssertEqualObjects(self.localNumber, me.uniqueId);
|
|
}
|
|
|
|
- (void)testSelfRecipientWithoutExistingRecord
|
|
{
|
|
XCTAssertNotNil(self.localNumber);
|
|
[[SignalRecipient fetchObjectWithUniqueID:self.localNumber] remove];
|
|
// Sanity Check that there's no existing user.
|
|
XCTAssertNil([SignalRecipient recipientWithTextSecureIdentifier:self.localNumber]);
|
|
|
|
SignalRecipient *me = [SignalRecipient selfRecipient];
|
|
XCTAssert(me);
|
|
XCTAssertEqualObjects(self.localNumber, me.uniqueId);
|
|
}
|
|
|
|
@end
|