212 lines
8.3 KiB
Mathematica
212 lines
8.3 KiB
Mathematica
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||
|
|
|
||
|
|
#import <XCTest/XCTest.h>
|
||
|
|
#import <OCMock/OCMock.h>
|
||
|
|
|
||
|
|
#import "OCTSubmanagerBootstrapImpl.h"
|
||
|
|
#import "OCTSubmanagerDataSource.h"
|
||
|
|
#import "OCTTox.h"
|
||
|
|
#import "OCTRealmManager.h"
|
||
|
|
#import "OCTSettingsStorageObject.h"
|
||
|
|
#import "OCTNode.h"
|
||
|
|
|
||
|
|
@interface OCTSubmanagerBootstrapImpl (Tests)
|
||
|
|
|
||
|
|
@property (strong, nonatomic) NSMutableSet *addedNodes;
|
||
|
|
|
||
|
|
@property (assign, nonatomic) NSTimeInterval didConnectDelay;
|
||
|
|
@property (assign, nonatomic) NSTimeInterval iterationTime;
|
||
|
|
|
||
|
|
@end
|
||
|
|
|
||
|
|
@interface OCTSubmanagerBootstrapImplTests : XCTestCase
|
||
|
|
|
||
|
|
@property (strong, nonatomic) id dataSource;
|
||
|
|
@property (strong, nonatomic) id tox;
|
||
|
|
@property (strong, nonatomic) id realmManager;
|
||
|
|
@property (strong, nonatomic) id settingsStorage;
|
||
|
|
@property (strong, nonatomic) OCTSubmanagerBootstrapImpl *submanager;
|
||
|
|
|
||
|
|
@end
|
||
|
|
|
||
|
|
@implementation OCTSubmanagerBootstrapImplTests
|
||
|
|
|
||
|
|
- (void)setUp
|
||
|
|
{
|
||
|
|
[super setUp];
|
||
|
|
// Put setup code here. This method is called before the invocation of each test method in the class.
|
||
|
|
|
||
|
|
self.dataSource = OCMProtocolMock(@protocol(OCTSubmanagerDataSource));
|
||
|
|
// self.tox = OCMClassMock([OCTTox class]);
|
||
|
|
self.tox = OCMStrictClassMock([OCTTox class]);
|
||
|
|
self.realmManager = OCMClassMock([OCTRealmManager class]);
|
||
|
|
self.settingsStorage = OCMClassMock([OCTSettingsStorageObject class]);
|
||
|
|
|
||
|
|
OCMStub([self.dataSource managerGetTox]).andReturn(self.tox);
|
||
|
|
OCMStub([self.dataSource managerGetRealmManager]).andReturn(self.realmManager);
|
||
|
|
OCMStub([self.realmManager settingsStorage]).andReturn(self.settingsStorage);
|
||
|
|
|
||
|
|
self.submanager = [OCTSubmanagerBootstrapImpl new];
|
||
|
|
self.submanager.dataSource = self.dataSource;
|
||
|
|
}
|
||
|
|
|
||
|
|
- (void)tearDown
|
||
|
|
{
|
||
|
|
self.dataSource = nil;
|
||
|
|
self.tox = nil;
|
||
|
|
self.realmManager = nil;
|
||
|
|
self.settingsStorage = nil;
|
||
|
|
self.submanager = nil;
|
||
|
|
// Put teardown code here. This method is called after the invocation of each test method in the class.
|
||
|
|
[super tearDown];
|
||
|
|
}
|
||
|
|
|
||
|
|
- (void)testAddPredefinedNodes
|
||
|
|
{
|
||
|
|
[self.submanager addPredefinedNodes];
|
||
|
|
|
||
|
|
XCTAssertTrue(self.submanager.addedNodes.count > 0);
|
||
|
|
|
||
|
|
for (OCTNode *node in self.submanager.addedNodes) {
|
||
|
|
XCTAssertTrue(node.ipv4Host.length > 0);
|
||
|
|
XCTAssertTrue(node.udpPort > 0);
|
||
|
|
XCTAssertNotNil(node.tcpPorts);
|
||
|
|
XCTAssertEqual(node.publicKey.length, kOCTToxPublicKeyLength);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
- (void)testBootstrapCustomNodes
|
||
|
|
{
|
||
|
|
XCTestExpectation *expectation = [self expectationWithDescription:@"bootstrap"];
|
||
|
|
|
||
|
|
self.submanager.didConnectDelay = 0.0;
|
||
|
|
self.submanager.iterationTime = 0.05;
|
||
|
|
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"one" ipv6Host:@"one6" udpPort:1 tcpPorts:@[@1, @11] publicKey:@"1"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"two" ipv6Host:nil udpPort:2 tcpPorts:@[@2] publicKey:@"2"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:nil ipv6Host:@"three6" udpPort:3 tcpPorts:@[@3] publicKey:@"3"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"four" ipv6Host:@"four6" udpPort:4 tcpPorts:@[] publicKey:@"4"];
|
||
|
|
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"one" port:1 publicKey:@"1" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"one6" port:1 publicKey:@"1" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox addTCPRelayWithHost:@"one" port:1 publicKey:@"1" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox addTCPRelayWithHost:@"one6" port:1 publicKey:@"1" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox addTCPRelayWithHost:@"one" port:11 publicKey:@"1" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox addTCPRelayWithHost:@"one6" port:11 publicKey:@"1" error:[OCMArg anyObjectRef]]);
|
||
|
|
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"two" port:2 publicKey:@"2" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox addTCPRelayWithHost:@"two" port:2 publicKey:@"2" error:[OCMArg anyObjectRef]]);
|
||
|
|
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"three6" port:3 publicKey:@"3" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox addTCPRelayWithHost:@"three6" port:3 publicKey:@"3" error:[OCMArg anyObjectRef]]);
|
||
|
|
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"four" port:4 publicKey:@"4" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"four6" port:4 publicKey:@"4" error:[OCMArg anyObjectRef]]);
|
||
|
|
|
||
|
|
[self.submanager bootstrap];
|
||
|
|
|
||
|
|
[self performBlock:^{
|
||
|
|
[expectation fulfill];
|
||
|
|
} afterDelay:0.1];
|
||
|
|
[self waitForExpectationsWithTimeout:0.5 handler:nil];
|
||
|
|
|
||
|
|
OCMVerifyAll(self.tox);
|
||
|
|
}
|
||
|
|
|
||
|
|
- (void)testBootstrapSeveralPortions
|
||
|
|
{
|
||
|
|
XCTestExpectation *expectation = [self expectationWithDescription:@"bootstrap"];
|
||
|
|
|
||
|
|
self.submanager.didConnectDelay = 0.0;
|
||
|
|
self.submanager.iterationTime = 0.1;
|
||
|
|
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"h1" ipv6Host:nil udpPort:1 tcpPorts:@[] publicKey:@"1"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"h2" ipv6Host:nil udpPort:2 tcpPorts:@[] publicKey:@"2"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"h3" ipv6Host:nil udpPort:3 tcpPorts:@[] publicKey:@"3"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"h4" ipv6Host:nil udpPort:4 tcpPorts:@[] publicKey:@"4"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"h5" ipv6Host:nil udpPort:5 tcpPorts:@[] publicKey:@"5"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"h6" ipv6Host:nil udpPort:6 tcpPorts:@[] publicKey:@"6"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"h7" ipv6Host:nil udpPort:7 tcpPorts:@[] publicKey:@"7"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"h8" ipv6Host:nil udpPort:8 tcpPorts:@[] publicKey:@"8"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"h9" ipv6Host:nil udpPort:9 tcpPorts:@[] publicKey:@"9"];
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"h10" ipv6Host:nil udpPort:10 tcpPorts:@[] publicKey:@"10"];
|
||
|
|
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"h1" port:1 publicKey:@"1" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"h2" port:2 publicKey:@"2" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"h3" port:3 publicKey:@"3" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"h4" port:4 publicKey:@"4" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"h5" port:5 publicKey:@"5" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"h6" port:6 publicKey:@"6" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"h7" port:7 publicKey:@"7" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"h8" port:8 publicKey:@"8" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"h9" port:9 publicKey:@"9" error:[OCMArg anyObjectRef]]);
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"h10" port:10 publicKey:@"10" error:[OCMArg anyObjectRef]]);
|
||
|
|
|
||
|
|
[self.submanager bootstrap];
|
||
|
|
|
||
|
|
[self performBlock:^{
|
||
|
|
[expectation fulfill];
|
||
|
|
} afterDelay:0.6];
|
||
|
|
[self waitForExpectationsWithTimeout:0.8 handler:nil];
|
||
|
|
|
||
|
|
OCMVerifyAll(self.tox);
|
||
|
|
}
|
||
|
|
|
||
|
|
- (void)testBootstrapDidConnectVerify
|
||
|
|
{
|
||
|
|
XCTestExpectation *expectation = [self expectationWithDescription:@"bootstrap"];
|
||
|
|
|
||
|
|
OCMStub([self.settingsStorage bootstrapDidConnect]).andReturn(YES);
|
||
|
|
|
||
|
|
self.submanager.didConnectDelay = 0.1;
|
||
|
|
self.submanager.iterationTime = 0.05;
|
||
|
|
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"one" ipv6Host:nil udpPort:1 tcpPorts:@[] publicKey:@"1"];
|
||
|
|
|
||
|
|
OCMExpect([self.tox bootstrapFromHost:@"one" port:1 publicKey:@"1" error:[OCMArg anyObjectRef]]);
|
||
|
|
|
||
|
|
[self.submanager bootstrap];
|
||
|
|
|
||
|
|
[self performBlock:^{
|
||
|
|
[expectation fulfill];
|
||
|
|
} afterDelay:0.3];
|
||
|
|
[self waitForExpectationsWithTimeout:0.5 handler:nil];
|
||
|
|
|
||
|
|
OCMVerifyAll(self.tox);
|
||
|
|
}
|
||
|
|
|
||
|
|
- (void)testBootstrapDidConnectReject
|
||
|
|
{
|
||
|
|
XCTestExpectation *expectation = [self expectationWithDescription:@"bootstrap"];
|
||
|
|
|
||
|
|
OCMStub([self.settingsStorage bootstrapDidConnect]).andReturn(YES);
|
||
|
|
|
||
|
|
self.submanager.didConnectDelay = 0.2;
|
||
|
|
self.submanager.iterationTime = 0.05;
|
||
|
|
|
||
|
|
[self.submanager addNodeWithIpv4Host:@"one" ipv6Host:nil udpPort:1 tcpPorts:@[] publicKey:@"1"];
|
||
|
|
|
||
|
|
[self.submanager bootstrap];
|
||
|
|
|
||
|
|
[self performBlock:^{
|
||
|
|
[expectation fulfill];
|
||
|
|
} afterDelay:0.1];
|
||
|
|
[self waitForExpectationsWithTimeout:0.15 handler:nil];
|
||
|
|
|
||
|
|
OCMVerifyAll(self.tox);
|
||
|
|
}
|
||
|
|
|
||
|
|
- (void)performBlock:(void (^)())block afterDelay:(NSTimeInterval)delay
|
||
|
|
{
|
||
|
|
[self performSelector:@selector(runBlock:) withObject:block afterDelay:delay];
|
||
|
|
}
|
||
|
|
|
||
|
|
- (void)runBlock:(void (^)())block
|
||
|
|
{
|
||
|
|
block();
|
||
|
|
}
|
||
|
|
|
||
|
|
@end
|