macSVG-mirror/CocoaAsyncSocket-master/Tests/Shared/GCDAsyncSocketConnectionTests.m

//
//  GCDAsyncSocketConnectionTests.m
//  GCDAsyncSocketConnectionTests
//
//  Created by Christopher Ballinger on 10/31/14.
//
//

#import <Foundation/Foundation.h>
#import <XCTest/XCTest.h>
@import CocoaAsyncSocket;

@interface GCDAsyncSocketConnectionTests : XCTestCase <GCDAsyncSocketDelegate>
@property (nonatomic) uint16_t portNumber;
@property (nonatomic, strong) GCDAsyncSocket *clientSocket;
@property (nonatomic, strong) GCDAsyncSocket *serverSocket;
@property (nonatomic, strong) GCDAsyncSocket *acceptedServerSocket;

@property (nonatomic, strong) XCTestExpectation *expectation;
@end

@implementation GCDAsyncSocketConnectionTests

- (void)setUp {
    [super setUp];
    // Put setup code here. This method is called before the invocation of each test method in the class.
    self.portNumber = [self randomValidPort];
    self.clientSocket = [[GCDAsyncSocket alloc] initWithDelegate:self delegateQueue:dispatch_get_main_queue()];
    self.serverSocket = [[GCDAsyncSocket alloc] initWithDelegate:self delegateQueue:dispatch_get_main_queue()];
}

- (void)tearDown {
    // Put teardown code here. This method is called after the invocation of each test method in the class.
    [super tearDown];
    [self.clientSocket disconnect];
    [self.serverSocket disconnect];
    [self.acceptedServerSocket disconnect];
    self.clientSocket = nil;
    self.serverSocket = nil;
    self.acceptedServerSocket = nil;
}

- (uint16_t) randomValidPort {
    uint16_t minPort = 1024;
    uint16_t maxPort = UINT16_MAX;
    return minPort + arc4random_uniform(maxPort - minPort + 1);
}

- (void)testFullConnection {
    NSError *error = nil;
    BOOL success = NO;
    success = [self.serverSocket acceptOnPort:self.portNumber error:&error];
    XCTAssertTrue(success, @"Server failed setting up socket on port %d %@", self.portNumber, error);
    success = [self.clientSocket connectToHost:@"127.0.0.1" onPort:self.portNumber error:&error];
    XCTAssertTrue(success, @"Client failed connecting to up server socket on port %d %@", self.portNumber, error);
    
    self.expectation = [self expectationWithDescription:@"Test Full Connection"];
    [self waitForExpectationsWithTimeout:30 handler:^(NSError *error) {
        if (error) {
            NSLog(@"Error establishing test connection");
        }
    }];
}

- (void)testConnectionWithAnIPv4OnlyServer {
    self.serverSocket.IPv6Enabled = NO;
    
    NSError *error = nil;
    BOOL success = NO;
    success = [self.serverSocket acceptOnPort:self.portNumber error:&error];
    XCTAssertTrue(success, @"Server failed setting up socket on port %d %@", self.portNumber, error);
    success = [self.clientSocket connectToHost:@"127.0.0.1" onPort:self.portNumber error:&error];
    XCTAssertTrue(success, @"Client failed connecting to up server socket on port %d %@", self.portNumber, error);
    
    self.expectation = [self expectationWithDescription:@"Test Full Connection"];
    [self waitForExpectationsWithTimeout:30 handler:^(NSError *error) {
        if (error) {
            NSLog(@"Error establishing test connection");
        }
        else {
            XCTAssertTrue(self.acceptedServerSocket.isIPv4, @"Established connection is not IPv4");
        }
    }];
}

- (void)testConnectionWithAnIPv6OnlyServer {
    self.serverSocket.IPv4Enabled = NO;
    
    NSError *error = nil;
    BOOL success = NO;
    success = [self.serverSocket acceptOnPort:self.portNumber error:&error];
    XCTAssertTrue(success, @"Server failed setting up socket on port %d %@", self.portNumber, error);
    success = [self.clientSocket connectToHost:@"::1" onPort:self.portNumber error:&error];
    XCTAssertTrue(success, @"Client failed connecting to up server socket on port %d %@", self.portNumber, error);
    
    self.expectation = [self expectationWithDescription:@"Test Full Connection"];
    [self waitForExpectationsWithTimeout:30 handler:^(NSError *error) {
        if (error) {
            NSLog(@"Error establishing test connection");
        }
        else {
            XCTAssertTrue(self.acceptedServerSocket.isIPv6, @"Established connection is not IPv6");
        }
    }];
}

- (void)testConnectionWithLocalhostWithClientPreferringIPv4 {
    [self.clientSocket setIPv4PreferredOverIPv6:YES];
    
    NSError *error = nil;
    BOOL success = NO;
    success = [self.serverSocket acceptOnPort:self.portNumber error:&error];
    XCTAssertTrue(success, @"Server failed setting up socket on port %d %@", self.portNumber, error);
    success = [self.clientSocket connectToHost:@"localhost" onPort:self.portNumber error:&error];
    XCTAssertTrue(success, @"Client failed connecting to up server socket on port %d %@", self.portNumber, error);
    
    self.expectation = [self expectationWithDescription:@"Test Full Connection"];
    [self waitForExpectationsWithTimeout:30 handler:^(NSError *error) {
        if (error) {
            NSLog(@"Error establishing test connection");
        }
    }];
}

- (void)testConnectionWithLocalhostWithClientPreferringIPv6 {
    [self.clientSocket setIPv4PreferredOverIPv6:NO];

    NSError *error = nil;
    BOOL success = NO;
    success = [self.serverSocket acceptOnPort:self.portNumber error:&error];
    XCTAssertTrue(success, @"Server failed setting up socket on port %d %@", self.portNumber, error);
    success = [self.clientSocket connectToHost:@"localhost" onPort:self.portNumber error:&error];
    XCTAssertTrue(success, @"Client failed connecting to up server socket on port %d %@", self.portNumber, error);
    
    self.expectation = [self expectationWithDescription:@"Test Full Connection"];
    [self waitForExpectationsWithTimeout:30 handler:^(NSError *error) {
        if (error) {
            NSLog(@"Error establishing test connection");
        }
    }];
}

#pragma mark GCDAsyncSocketDelegate methods

/**
 * Called when a socket accepts a connection.
 * Another socket is automatically spawned to handle it.
 *
 * You must retain the newSocket if you wish to handle the connection.
 * Otherwise the newSocket instance will be released and the spawned connection will be closed.
 *
 * By default the new socket will have the same delegate and delegateQueue.
 * You may, of course, change this at any time.
 **/
- (void)socket:(GCDAsyncSocket *)sock didAcceptNewSocket:(GCDAsyncSocket *)newSocket {
    NSLog(@"didAcceptNewSocket %@ %@", sock, newSocket);
    self.acceptedServerSocket = newSocket;
}

/**
 * Called when a socket connects and is ready for reading and writing.
 * The host parameter will be an IP address, not a DNS name.
 **/
- (void)socket:(GCDAsyncSocket *)sock didConnectToHost:(NSString *)host port:(uint16_t)port {
    NSLog(@"didConnectToHost %@ %@ %d", sock, host, port);
    [self.expectation fulfill];
}


@end