mirror of https://github.com/oxen-io/session-ios
Christine Corbett
10 years ago
committed by
Frederic Jacobs
parent
71320a690c
commit
b954ff2445
@ -1 +1 @@
|
|||||||
Subproject commit dd570a162f44f9cc88575962b2b08fda698d9f8e
|
Subproject commit 135d1331112829ed1b2842d61e5ae0036aa8f78e
|
||||||
@ -0,0 +1,134 @@
|
|||||||
|
//
|
||||||
|
// VersionMigrationsTests.m
|
||||||
|
// TextSecureKit
|
||||||
|
//
|
||||||
|
// Created by Frederic Jacobs on 06/11/14.
|
||||||
|
// Copyright (c) 2014 Open Whisper Systems. All rights reserved.
|
||||||
|
//
|
||||||
|
|
||||||
|
#import <XCTest/XCTest.h>
|
||||||
|
#import "VersionMigrations.h"
|
||||||
|
|
||||||
|
#import "CategorizingLogger.h"
|
||||||
|
#import "Cryptography.h"
|
||||||
|
#import "Environment.h"
|
||||||
|
#import "TSStorageManager.h"
|
||||||
|
#import "TSStorageManager+IdentityKeyStore.h"
|
||||||
|
#import "RecentCall.h"
|
||||||
|
#import "RecentCallManager.h"
|
||||||
|
#import "Release.h"
|
||||||
|
#import "SecurityUtils.h"
|
||||||
|
#import "TestUtil.h"
|
||||||
|
#import "TSCall.h"
|
||||||
|
#import "TSContactThread.h"
|
||||||
|
#import "TSDatabaseView.h"
|
||||||
|
#import "SignalKeyingStorage.h"
|
||||||
|
#import "UICKeyChainStore.h"
|
||||||
|
#import "YapDatabaseConnection.h"
|
||||||
|
|
||||||
|
|
||||||
|
@interface VersionMigrations(Testing)
|
||||||
|
+(void) migrateRecentCallsToVersion2Dot0;
|
||||||
|
+(void) migrateKeyingStorageToVersion2Dot0;
|
||||||
|
@end
|
||||||
|
|
||||||
|
|
||||||
|
@interface SignalKeyingStorage(Testing)
|
||||||
|
+(void)storeData:(NSData*)data forKey:(NSString*)key;
|
||||||
|
+(NSData*)dataForKey:(NSString*)key;
|
||||||
|
+(void)storeString:(NSString*)string forKey:(NSString*)key;
|
||||||
|
+(NSString*)stringForKey:(NSString*)key;
|
||||||
|
@end
|
||||||
|
|
||||||
|
@interface TSDatabaseView(Testing)
|
||||||
|
+ (BOOL)threadShouldBeInInbox:(TSThread*)thread;
|
||||||
|
@end
|
||||||
|
|
||||||
|
|
||||||
|
@interface VersionMigrationsTests : XCTestCase
|
||||||
|
@property (nonatomic,strong) NSString* localNumber;
|
||||||
|
@property (nonatomic,strong) NSString* passwordCounter;
|
||||||
|
@property (nonatomic,strong) NSString* savedPassword;
|
||||||
|
|
||||||
|
@property (nonatomic,strong) NSData* signalingMacKey;
|
||||||
|
@property (nonatomic,strong) NSData* signalingCipherKey;
|
||||||
|
@property (nonatomic,strong) NSData* zidKey;
|
||||||
|
@property (nonatomic,strong) NSData* signalingExtraKey;
|
||||||
|
@property (nonatomic,strong) NSMutableArray* recentCalls;
|
||||||
|
@end
|
||||||
|
|
||||||
|
@implementation VersionMigrationsTests
|
||||||
|
|
||||||
|
- (void)setUp {
|
||||||
|
[super setUp];
|
||||||
|
[Environment setCurrent:[Release unitTestEnvironment:@[]]];
|
||||||
|
|
||||||
|
|
||||||
|
_localNumber = @"+123456789";
|
||||||
|
_passwordCounter = @"20";
|
||||||
|
_savedPassword = @"muchlettersverysecure";
|
||||||
|
_signalingMacKey = [Cryptography generateRandomBytes:8];
|
||||||
|
_signalingCipherKey = [Cryptography generateRandomBytes:8];
|
||||||
|
_zidKey = [Cryptography generateRandomBytes:8];
|
||||||
|
_signalingExtraKey = [Cryptography generateRandomBytes:8];
|
||||||
|
|
||||||
|
// setup the keys
|
||||||
|
[UICKeyChainStore setString:_localNumber forKey:LOCAL_NUMBER_KEY];
|
||||||
|
[UICKeyChainStore setString:_passwordCounter forKey:PASSWORD_COUNTER_KEY];
|
||||||
|
[UICKeyChainStore setString:_savedPassword forKey:SAVED_PASSWORD_KEY];
|
||||||
|
[UICKeyChainStore setData:_signalingMacKey forKey:SIGNALING_MAC_KEY];
|
||||||
|
[UICKeyChainStore setData:_signalingCipherKey forKey:SIGNALING_CIPHER_KEY];
|
||||||
|
[UICKeyChainStore setData:_zidKey forKey:ZID_KEY];
|
||||||
|
[UICKeyChainStore setData:_signalingExtraKey forKey:SIGNALING_EXTRA_KEY];
|
||||||
|
|
||||||
|
// setup the recent calls
|
||||||
|
RecentCall* r1 = [RecentCall recentCallWithContactID:123
|
||||||
|
andNumber:testPhoneNumber1
|
||||||
|
andCallType:RPRecentCallTypeIncoming];
|
||||||
|
RecentCall* r2 = [RecentCall recentCallWithContactID:456
|
||||||
|
andNumber:testPhoneNumber2
|
||||||
|
andCallType:RPRecentCallTypeMissed];
|
||||||
|
|
||||||
|
r2.isArchived = YES;
|
||||||
|
|
||||||
|
_recentCalls = [[NSMutableArray alloc] initWithObjects:r1,r2,nil];
|
||||||
|
|
||||||
|
NSUserDefaults *defaults = NSUserDefaults.standardUserDefaults;
|
||||||
|
NSData *saveData = [NSKeyedArchiver archivedDataWithRootObject:_recentCalls.copy];
|
||||||
|
[defaults setObject:saveData forKey:RECENT_CALLS_DEFAULT_KEY];
|
||||||
|
[defaults synchronize];
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
- (void)tearDown {
|
||||||
|
// Put teardown code here. This method is called after the invocation of each test method in the class.
|
||||||
|
[super tearDown];
|
||||||
|
}
|
||||||
|
|
||||||
|
- (void)testMigrateKeyingStorageToVersion2Dot0 {
|
||||||
|
// migrate keying storage
|
||||||
|
[VersionMigrations migrateKeyingStorageToVersion2Dot0];
|
||||||
|
|
||||||
|
// checking that everything is migrated correctly
|
||||||
|
XCTAssert([[SignalKeyingStorage stringForKey:LOCAL_NUMBER_KEY] isEqualToString:_localNumber]);
|
||||||
|
XCTAssert([[SignalKeyingStorage stringForKey:PASSWORD_COUNTER_KEY] isEqualToString:_passwordCounter]);
|
||||||
|
XCTAssert([[SignalKeyingStorage stringForKey:SAVED_PASSWORD_KEY] isEqualToString:_savedPassword]);
|
||||||
|
|
||||||
|
XCTAssert([[SignalKeyingStorage dataForKey:SIGNALING_MAC_KEY] isEqualToData:_signalingMacKey]);
|
||||||
|
XCTAssert([[SignalKeyingStorage dataForKey:SIGNALING_CIPHER_KEY] isEqualToData:_signalingCipherKey]);
|
||||||
|
XCTAssert([[SignalKeyingStorage dataForKey:ZID_KEY] isEqualToData:_zidKey]);
|
||||||
|
XCTAssert([[SignalKeyingStorage dataForKey:SIGNALING_EXTRA_KEY] isEqualToData:_signalingExtraKey]);
|
||||||
|
|
||||||
|
// checking that the old storage is empty
|
||||||
|
XCTAssert([UICKeyChainStore stringForKey:LOCAL_NUMBER_KEY] == nil);
|
||||||
|
XCTAssert([UICKeyChainStore stringForKey:PASSWORD_COUNTER_KEY] == nil);
|
||||||
|
XCTAssert([UICKeyChainStore stringForKey:SAVED_PASSWORD_KEY] == nil);
|
||||||
|
|
||||||
|
XCTAssert([UICKeyChainStore dataForKey:SIGNALING_MAC_KEY] == nil);
|
||||||
|
XCTAssert([UICKeyChainStore dataForKey:SIGNALING_CIPHER_KEY] == nil);
|
||||||
|
XCTAssert([UICKeyChainStore dataForKey:ZID_KEY] == nil);
|
||||||
|
XCTAssert([UICKeyChainStore dataForKey:SIGNALING_EXTRA_KEY] == nil);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@end
|
||||||
Loading…
Reference in New Issue