Fixup tests

// FREEBIE
pull/1/head
Michael Kirk 8 years ago
parent 01e808febe
commit 935b51aa19

@ -44,6 +44,7 @@ NS_ASSUME_NONNULL_BEGIN
return recipient; return recipient;
} }
// TODO This method should probably live on the TSAccountManager rather than grabbing a global singleton.
+ (instancetype)selfRecipient + (instancetype)selfRecipient
{ {
SignalRecipient *myself = [self recipientWithTextSecureIdentifier:[TSAccountManager localNumber]]; SignalRecipient *myself = [self recipientWithTextSecureIdentifier:[TSAccountManager localNumber]];

@ -3,10 +3,16 @@
// //
#import "SignalRecipient.h" #import "SignalRecipient.h"
#import "TSAccountManager.h"
#import "TSStorageManager+keyingMaterial.h" #import "TSStorageManager+keyingMaterial.h"
#import "TSStorageManager.h"
#import <XCTest/XCTest.h> #import <XCTest/XCTest.h>
@interface TSAccountManager (Testing)
- (void)storeLocalNumber:(NSString *)localNumber;
@end
@interface SignalRecipientTest : XCTestCase @interface SignalRecipientTest : XCTestCase
@property (nonatomic) NSString *localNumber; @property (nonatomic) NSString *localNumber;
@ -19,7 +25,7 @@
{ {
[super setUp]; [super setUp];
self.localNumber = @"+13231231234"; self.localNumber = @"+13231231234";
[[TSStorageManager sharedManager] storePhoneNumber:self.localNumber]; [[TSAccountManager sharedInstance] storeLocalNumber:self.localNumber];
} }
- (void)testSelfRecipientWithExistingRecord - (void)testSelfRecipientWithExistingRecord

@ -10,13 +10,13 @@
#import "OWSFakeNetworkManager.h" #import "OWSFakeNetworkManager.h"
#import "OWSMessageSender.h" #import "OWSMessageSender.h"
#import "OWSUploadingService.h" #import "OWSUploadingService.h"
#import "TSAccountManager.h"
#import "TSContactThread.h" #import "TSContactThread.h"
#import "TSGroupModel.h" #import "TSGroupModel.h"
#import "TSGroupThread.h" #import "TSGroupThread.h"
#import "TSMessagesManager.h" #import "TSMessagesManager.h"
#import "TSNetworkManager.h" #import "TSNetworkManager.h"
#import "TSOutgoingMessage.h" #import "TSOutgoingMessage.h"
#import "TSStorageManager+keyingMaterial.h"
#import "TSStorageManager.h" #import "TSStorageManager.h"
#import <AxolotlKit/AxolotlExceptions.h> #import <AxolotlKit/AxolotlExceptions.h>
#import <AxolotlKit/SessionBuilder.h> #import <AxolotlKit/SessionBuilder.h>
@ -174,6 +174,12 @@ NS_ASSUME_NONNULL_BEGIN
@end @end
@interface TSAccountManager (Testing)
- (void)storeLocalNumber:(NSString *)localNumber;
@end
@interface OWSMessageSenderTest : XCTestCase @interface OWSMessageSenderTest : XCTestCase
@property (nonatomic) TSThread *thread; @property (nonatomic) TSThread *thread;
@ -192,7 +198,7 @@ NS_ASSUME_NONNULL_BEGIN
[super setUp]; [super setUp];
// Hack to make sure we don't explode when sending sync message. // Hack to make sure we don't explode when sending sync message.
[[TSStorageManager sharedManager] storePhoneNumber:@"+13231231234"]; [[TSAccountManager sharedInstance] storeLocalNumber:@"+13231231234"];
self.thread = [[TSContactThread alloc] initWithUniqueId:@"fake-thread-id"]; self.thread = [[TSContactThread alloc] initWithUniqueId:@"fake-thread-id"];
[self.thread save]; [self.thread save];

Loading…
Cancel
Save