ccb4a88742
git remote add ssk ../SignalServiceKit git remote update git merge -s ours --allow-unrelated-histories --no-commit ssk/master git read-tree --prefix=SignalServiceKit -u ssk/master git commit
94 lines
2.9 KiB
Objective-C
94 lines
2.9 KiB
Objective-C
// Copyright © 2016 Open Whisper Systems. All rights reserved.
|
|
|
|
#import "OWSDeviceProvisioner.h"
|
|
#import "OWSDeviceProvisioningCodeService.h"
|
|
#import "OWSDeviceProvisioningService.h"
|
|
#import "OWSFakeNetworkManager.h"
|
|
#import "TSNetworkManager.h"
|
|
|
|
#import <XCTest/XCTest.h>
|
|
|
|
@interface OWSFakeDeviceProvisioningService : OWSDeviceProvisioningService
|
|
|
|
@end
|
|
|
|
@implementation OWSFakeDeviceProvisioningService
|
|
|
|
- (void)provisionWithMessageBody:(NSData *)messageBody
|
|
ephemeralDeviceId:(NSString *)deviceId
|
|
success:(void (^)())successCallback
|
|
failure:(void (^)(NSError *))failureCallback
|
|
{
|
|
NSLog(@"faking successful provisioning");
|
|
successCallback();
|
|
}
|
|
|
|
@end
|
|
|
|
@interface OWSFakeDeviceProvisioningCodeService : OWSDeviceProvisioningCodeService
|
|
|
|
@end
|
|
|
|
@implementation OWSFakeDeviceProvisioningCodeService
|
|
|
|
- (void)requestProvisioningCodeWithSuccess:(void (^)(NSString *))successCallback
|
|
failure:(void (^)(NSError *))failureCallback
|
|
{
|
|
NSLog(@"faking successful provisioning code fetching");
|
|
successCallback(@"fake-provisioning-code");
|
|
}
|
|
|
|
@end
|
|
|
|
@interface OWSDeviceProvisioner (Testing)
|
|
|
|
@property OWSDeviceProvisioningCodeService *provisioningCodeService;
|
|
@property OWSDeviceProvisioningService *provisioningService;
|
|
|
|
@end
|
|
|
|
@interface OWSDeviceProvisionerTest : XCTestCase
|
|
|
|
@end
|
|
|
|
@implementation OWSDeviceProvisionerTest
|
|
|
|
- (void)testProvisioning
|
|
{
|
|
|
|
XCTestExpectation *expectation = [self expectationWithDescription:@"Provisioning Success"];
|
|
|
|
NSData *nullKey = [[NSMutableData dataWithLength:32] copy];
|
|
NSData *myPublicKey = [nullKey copy];
|
|
NSData *myPrivateKey = [nullKey copy];
|
|
NSData *theirPublicKey = [nullKey copy];
|
|
NSString *accountIdentifier;
|
|
NSString *theirEphemeralDeviceId;
|
|
|
|
OWSFakeNetworkManager *networkManager = [OWSFakeNetworkManager new];
|
|
OWSDeviceProvisioner *provisioner = [[OWSDeviceProvisioner alloc]
|
|
initWithMyPublicKey:myPublicKey
|
|
myPrivateKey:myPrivateKey
|
|
theirPublicKey:theirPublicKey
|
|
theirEphemeralDeviceId:theirEphemeralDeviceId
|
|
accountIdentifier:accountIdentifier
|
|
provisioningCodeService:[[OWSFakeDeviceProvisioningCodeService alloc] initWithNetworkManager:networkManager]
|
|
provisioningService:[[OWSFakeDeviceProvisioningService alloc] initWithNetworkManager:networkManager]];
|
|
|
|
[provisioner provisionWithSuccess:^{
|
|
[expectation fulfill];
|
|
}
|
|
failure:^(NSError *_Nonnull error) {
|
|
XCTAssert(NO, @"Failed to provision with error: %@", error);
|
|
}];
|
|
|
|
[self waitForExpectationsWithTimeout:5.0
|
|
handler:^(NSError *error) {
|
|
if (error) {
|
|
NSLog(@"Timeout Error: %@", error);
|
|
}
|
|
}];
|
|
}
|
|
|
|
@end
|