examples/objc/AppRTCMobile/tests/ARDAppClient_xctest.mm - src/webrtc - Git at Google

 /*
  *  Copyright 2014 The WebRTC Project Authors. All rights reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */

 #import <Foundation/Foundation.h>
 #import <OCMock/OCMock.h>
 #import <QuartzCore/CoreAnimation.h>
 #import <XCTest/XCTest.h>

 #include "webrtc/rtc_base/ssladapter.h"

 #import "WebRTC/RTCMediaConstraints.h"
 #import "WebRTC/RTCPeerConnectionFactory.h"

 #import "ARDAppClient+Internal.h"
 #import "ARDJoinResponse+Internal.h"
 #import "ARDMessageResponse+Internal.h"
 #import "ARDSDPUtils.h"
 #import "ARDSettingsModel.h"

 @interface ARDAppClientTest : XCTestCase
 @end

 @implementation ARDAppClientTest

 #pragma mark - Mock helpers

 - (id)mockRoomServerClientForRoomId:(NSString *)roomId
                            clientId:(NSString *)clientId
                         isInitiator:(BOOL)isInitiator
                            messages:(NSArray *)messages
                      messageHandler:
     (void (^)(ARDSignalingMessage *))messageHandler {
   id mockRoomServerClient =
       [OCMockObject mockForProtocol:@protocol(ARDRoomServerClient)];

   // Successful join response.
   ARDJoinResponse *joinResponse = [[ARDJoinResponse alloc] init];
   joinResponse.result = kARDJoinResultTypeSuccess;
   joinResponse.roomId = roomId;
   joinResponse.clientId = clientId;
   joinResponse.isInitiator = isInitiator;
   joinResponse.messages = messages;

   // Successful message response.
   ARDMessageResponse *messageResponse = [[ARDMessageResponse alloc] init];
   messageResponse.result = kARDMessageResultTypeSuccess;

   // Return join response from above on join.
   [[[mockRoomServerClient stub] andDo:^(NSInvocation *invocation) {
     __unsafe_unretained void (^completionHandler)(ARDJoinResponse *response,
                                                   NSError *error);
     [invocation getArgument:&completionHandler atIndex:4];
     completionHandler(joinResponse, nil);
   }] joinRoomWithRoomId:roomId isLoopback:NO completionHandler:[OCMArg any]];

   // Return message response from above on join.
   [[[mockRoomServerClient stub] andDo:^(NSInvocation *invocation) {
     __unsafe_unretained ARDSignalingMessage *message;
     __unsafe_unretained void (^completionHandler)(ARDMessageResponse *response,
                                                   NSError *error);
     [invocation getArgument:&message atIndex:2];
     [invocation getArgument:&completionHandler atIndex:5];
     messageHandler(message);
     completionHandler(messageResponse, nil);
   }] sendMessage:[OCMArg any]
             forRoomId:roomId
              clientId:clientId
     completionHandler:[OCMArg any]];

   // Do nothing on leave.
   [[[mockRoomServerClient stub] andDo:^(NSInvocation *invocation) {
     __unsafe_unretained void (^completionHandler)(NSError *error);
     [invocation getArgument:&completionHandler atIndex:4];
     if (completionHandler) {
       completionHandler(nil);
     }
   }] leaveRoomWithRoomId:roomId
                 clientId:clientId
        completionHandler:[OCMArg any]];

   return mockRoomServerClient;
 }

 - (id)mockSignalingChannelForRoomId:(NSString *)roomId
                            clientId:(NSString *)clientId
                      messageHandler:
     (void (^)(ARDSignalingMessage *message))messageHandler {
   id mockSignalingChannel =
       [OCMockObject niceMockForProtocol:@protocol(ARDSignalingChannel)];
   [[mockSignalingChannel stub] registerForRoomId:roomId clientId:clientId];
   [[[mockSignalingChannel stub] andDo:^(NSInvocation *invocation) {
     __unsafe_unretained ARDSignalingMessage *message;
     [invocation getArgument:&message atIndex:2];
     messageHandler(message);
   }] sendMessage:[OCMArg any]];
   return mockSignalingChannel;
 }

 - (id)mockTURNClient {
   id mockTURNClient =
       [OCMockObject mockForProtocol:@protocol(ARDTURNClient)];
   [[[mockTURNClient stub] andDo:^(NSInvocation *invocation) {
     // Don't return anything in TURN response.
     __unsafe_unretained void (^completionHandler)(NSArray *turnServers,
                                                   NSError *error);
     [invocation getArgument:&completionHandler atIndex:2];
     completionHandler([NSArray array], nil);
   }] requestServersWithCompletionHandler:[OCMArg any]];
   return mockTURNClient;
 }

 - (id)mockSettingsModel {
   ARDSettingsModel *model = [[ARDSettingsModel alloc] init];
   id partialMock = [OCMockObject partialMockForObject:model];
   [[[partialMock stub] andReturn:@[ @"640x480", @"960x540", @"1280x720" ]]
       availableVideoResolutions];

   return model;
 }

 - (ARDAppClient *)createAppClientForRoomId:(NSString *)roomId
                                   clientId:(NSString *)clientId
                                isInitiator:(BOOL)isInitiator
                                   messages:(NSArray *)messages
                             messageHandler:
     (void (^)(ARDSignalingMessage *message))messageHandler
                           connectedHandler:(void (^)(void))connectedHandler
                     localVideoTrackHandler:(void (^)(void))localVideoTrackHandler {
   id turnClient = [self mockTURNClient];
   id signalingChannel = [self mockSignalingChannelForRoomId:roomId
                                                    clientId:clientId
                                              messageHandler:messageHandler];
   id roomServerClient =
       [self mockRoomServerClientForRoomId:roomId
                                  clientId:clientId
                               isInitiator:isInitiator
                                  messages:messages
                            messageHandler:messageHandler];
   id delegate =
       [OCMockObject niceMockForProtocol:@protocol(ARDAppClientDelegate)];
   [[[delegate stub] andDo:^(NSInvocation *invocation) {
     connectedHandler();
   }] appClient:[OCMArg any]
       didChangeConnectionState:RTCIceConnectionStateConnected];
   [[[delegate stub] andDo:^(NSInvocation *invocation) {
     localVideoTrackHandler();
   }] appClient:[OCMArg any]
       didReceiveLocalVideoTrack:[OCMArg any]];

   return [[ARDAppClient alloc] initWithRoomServerClient:roomServerClient
                                        signalingChannel:signalingChannel
                                              turnClient:turnClient
                                                delegate:delegate];
 }

 // Tests that an ICE connection is established between two ARDAppClient objects
 // where one is set up as a caller and the other the answerer. Network
 // components are mocked out and messages are relayed directly from object to
 // object. It's expected that both clients reach the
 // RTCIceConnectionStateConnected state within a reasonable amount of time.
 - (void)testSession {
   // Need block arguments here because we're setting up a callbacks before we
   // create the clients.
   ARDAppClient *caller = nil;
   ARDAppClient *answerer = nil;
   __block __weak ARDAppClient *weakCaller = nil;
   __block __weak ARDAppClient *weakAnswerer = nil;
   NSString *roomId = @"testRoom";
   NSString *callerId = @"testCallerId";
   NSString *answererId = @"testAnswererId";

   XCTestExpectation *callerConnectionExpectation =
       [self expectationWithDescription:@"Caller PC connected."];
   XCTestExpectation *answererConnectionExpectation =
       [self expectationWithDescription:@"Answerer PC connected."];

   caller = [self createAppClientForRoomId:roomId
                                  clientId:callerId
                               isInitiator:YES
                                  messages:[NSArray array]
                            messageHandler:^(ARDSignalingMessage *message) {
     ARDAppClient *strongAnswerer = weakAnswerer;
     [strongAnswerer channel:strongAnswerer.channel didReceiveMessage:message];
   }                      connectedHandler:^{
     [callerConnectionExpectation fulfill];
   }                localVideoTrackHandler:^{
   }];
   // TODO(tkchin): Figure out why DTLS-SRTP constraint causes thread assertion
   // crash in Debug.
   caller.defaultPeerConnectionConstraints =
       [[RTCMediaConstraints alloc] initWithMandatoryConstraints:nil
                                             optionalConstraints:nil];
   weakCaller = caller;

   answerer = [self createAppClientForRoomId:roomId
                                    clientId:answererId
                                 isInitiator:NO
                                    messages:[NSArray array]
                              messageHandler:^(ARDSignalingMessage *message) {
     ARDAppClient *strongCaller = weakCaller;
     [strongCaller channel:strongCaller.channel didReceiveMessage:message];
   }                        connectedHandler:^{
     [answererConnectionExpectation fulfill];
   }                  localVideoTrackHandler:^{
   }];
   // TODO(tkchin): Figure out why DTLS-SRTP constraint causes thread assertion
   // crash in Debug.
   answerer.defaultPeerConnectionConstraints =
       [[RTCMediaConstraints alloc] initWithMandatoryConstraints:nil
                                             optionalConstraints:nil];
   weakAnswerer = answerer;

   // Kick off connection.
   [caller connectToRoomWithId:roomId settings:[self mockSettingsModel] isLoopback:NO];
   [answerer connectToRoomWithId:roomId settings:[self mockSettingsModel] isLoopback:NO];
   [self waitForExpectationsWithTimeout:20 handler:^(NSError *error) {
     if (error) {
       XCTFail(@"Expectation failed with error %@.", error);
     }
   }];
 }

 // Test to see that we get a local video connection
 // Note this will currently pass even when no camera is connected as a local
 // video track is created regardless (Perhaps there should be a test for that...)
 #if !TARGET_IPHONE_SIMULATOR // Expect to fail on simulator due to no camera support
 - (void)testSessionShouldGetLocalVideoTrackCallback {
   ARDAppClient *caller = nil;
   NSString *roomId = @"testRoom";
   NSString *callerId = @"testCallerId";

   XCTestExpectation *localVideoTrackExpectation =
       [self expectationWithDescription:@"Caller got local video."];

   caller = [self createAppClientForRoomId:roomId
                                  clientId:callerId
                               isInitiator:YES
                                  messages:[NSArray array]
                            messageHandler:^(ARDSignalingMessage *message) {}
                          connectedHandler:^{}
                    localVideoTrackHandler:^{ [localVideoTrackExpectation fulfill]; }];
   caller.defaultPeerConnectionConstraints =
       [[RTCMediaConstraints alloc] initWithMandatoryConstraints:nil
                                           optionalConstraints:nil];

   // Kick off connection.
   [caller connectToRoomWithId:roomId
                      settings:[self mockSettingsModel]
                    isLoopback:NO];
   [self waitForExpectationsWithTimeout:20 handler:^(NSError *error) {
     if (error) {
       XCTFail("Expectation timed out with error: %@.", error);
     }
   }];
 }
 #endif

 @end
	/*
	* Copyright 2014 The WebRTC Project Authors. All rights reserved.
	*
	* Use of this source code is governed by a BSD-style license
	* that can be found in the LICENSE file in the root of the source
	* tree. An additional intellectual property rights grant can be found
	* in the file PATENTS. All contributing project authors may
	* be found in the AUTHORS file in the root of the source tree.
	*/

	#import <Foundation/Foundation.h>
	#import <OCMock/OCMock.h>
	#import <QuartzCore/CoreAnimation.h>
	#import <XCTest/XCTest.h>

	#include "webrtc/rtc_base/ssladapter.h"

	#import "WebRTC/RTCMediaConstraints.h"
	#import "WebRTC/RTCPeerConnectionFactory.h"

	#import "ARDAppClient+Internal.h"
	#import "ARDJoinResponse+Internal.h"
	#import "ARDMessageResponse+Internal.h"
	#import "ARDSDPUtils.h"
	#import "ARDSettingsModel.h"

	@interface ARDAppClientTest : XCTestCase
	@end

	@implementation ARDAppClientTest

	#pragma mark - Mock helpers

	- (id)mockRoomServerClientForRoomId:(NSString *)roomId
	clientId:(NSString *)clientId
	isInitiator:(BOOL)isInitiator
	messages:(NSArray *)messages
	messageHandler:
	(void (^)(ARDSignalingMessage *))messageHandler {
	id mockRoomServerClient =
	[OCMockObject mockForProtocol:@protocol(ARDRoomServerClient)];

	// Successful join response.
	ARDJoinResponse *joinResponse = [[ARDJoinResponse alloc] init];
	joinResponse.result = kARDJoinResultTypeSuccess;
	joinResponse.roomId = roomId;
	joinResponse.clientId = clientId;
	joinResponse.isInitiator = isInitiator;
	joinResponse.messages = messages;

	// Successful message response.
	ARDMessageResponse *messageResponse = [[ARDMessageResponse alloc] init];
	messageResponse.result = kARDMessageResultTypeSuccess;

	// Return join response from above on join.
	[[[mockRoomServerClient stub] andDo:^(NSInvocation *invocation) {
	__unsafe_unretained void (^completionHandler)(ARDJoinResponse *response,
	NSError *error);
	[invocation getArgument:&completionHandler atIndex:4];
	completionHandler(joinResponse, nil);
	}] joinRoomWithRoomId:roomId isLoopback:NO completionHandler:[OCMArg any]];

	// Return message response from above on join.
	[[[mockRoomServerClient stub] andDo:^(NSInvocation *invocation) {
	__unsafe_unretained ARDSignalingMessage *message;
	__unsafe_unretained void (^completionHandler)(ARDMessageResponse *response,
	NSError *error);
	[invocation getArgument:&message atIndex:2];
	[invocation getArgument:&completionHandler atIndex:5];
	messageHandler(message);
	completionHandler(messageResponse, nil);
	}] sendMessage:[OCMArg any]
	forRoomId:roomId
	clientId:clientId
	completionHandler:[OCMArg any]];

	// Do nothing on leave.
	[[[mockRoomServerClient stub] andDo:^(NSInvocation *invocation) {
	__unsafe_unretained void (^completionHandler)(NSError *error);
	[invocation getArgument:&completionHandler atIndex:4];
	if (completionHandler) {
	completionHandler(nil);
	}
	}] leaveRoomWithRoomId:roomId
	clientId:clientId
	completionHandler:[OCMArg any]];

	return mockRoomServerClient;
	}

	- (id)mockSignalingChannelForRoomId:(NSString *)roomId
	clientId:(NSString *)clientId
	messageHandler:
	(void (^)(ARDSignalingMessage *message))messageHandler {
	id mockSignalingChannel =
	[OCMockObject niceMockForProtocol:@protocol(ARDSignalingChannel)];
	[[mockSignalingChannel stub] registerForRoomId:roomId clientId:clientId];
	[[[mockSignalingChannel stub] andDo:^(NSInvocation *invocation) {
	__unsafe_unretained ARDSignalingMessage *message;
	[invocation getArgument:&message atIndex:2];
	messageHandler(message);
	}] sendMessage:[OCMArg any]];
	return mockSignalingChannel;
	}

	- (id)mockTURNClient {
	id mockTURNClient =
	[OCMockObject mockForProtocol:@protocol(ARDTURNClient)];
	[[[mockTURNClient stub] andDo:^(NSInvocation *invocation) {
	// Don't return anything in TURN response.
	__unsafe_unretained void (^completionHandler)(NSArray *turnServers,
	NSError *error);
	[invocation getArgument:&completionHandler atIndex:2];
	completionHandler([NSArray array], nil);
	}] requestServersWithCompletionHandler:[OCMArg any]];
	return mockTURNClient;
	}

	- (id)mockSettingsModel {
	ARDSettingsModel *model = [[ARDSettingsModel alloc] init];
	id partialMock = [OCMockObject partialMockForObject:model];
	[[[partialMock stub] andReturn:@[ @"640x480", @"960x540", @"1280x720" ]]
	availableVideoResolutions];

	return model;
	}

	- (ARDAppClient )createAppClientForRoomId:(NSString )roomId
	clientId:(NSString *)clientId
	isInitiator:(BOOL)isInitiator
	messages:(NSArray *)messages
	messageHandler:
	(void (^)(ARDSignalingMessage *message))messageHandler
	connectedHandler:(void (^)(void))connectedHandler
	localVideoTrackHandler:(void (^)(void))localVideoTrackHandler {
	id turnClient = [self mockTURNClient];
	id signalingChannel = [self mockSignalingChannelForRoomId:roomId
	clientId:clientId
	messageHandler:messageHandler];
	id roomServerClient =
	[self mockRoomServerClientForRoomId:roomId
	clientId:clientId
	isInitiator:isInitiator
	messages:messages
	messageHandler:messageHandler];
	id delegate =
	[OCMockObject niceMockForProtocol:@protocol(ARDAppClientDelegate)];
	[[[delegate stub] andDo:^(NSInvocation *invocation) {
	connectedHandler();
	}] appClient:[OCMArg any]
	didChangeConnectionState:RTCIceConnectionStateConnected];
	[[[delegate stub] andDo:^(NSInvocation *invocation) {
	localVideoTrackHandler();
	}] appClient:[OCMArg any]
	didReceiveLocalVideoTrack:[OCMArg any]];

	return [[ARDAppClient alloc] initWithRoomServerClient:roomServerClient
	signalingChannel:signalingChannel
	turnClient:turnClient
	delegate:delegate];
	}

	// Tests that an ICE connection is established between two ARDAppClient objects
	// where one is set up as a caller and the other the answerer. Network
	// components are mocked out and messages are relayed directly from object to
	// object. It's expected that both clients reach the
	// RTCIceConnectionStateConnected state within a reasonable amount of time.
	- (void)testSession {
	// Need block arguments here because we're setting up a callbacks before we
	// create the clients.
	ARDAppClient *caller = nil;
	ARDAppClient *answerer = nil;
	__block __weak ARDAppClient *weakCaller = nil;
	__block __weak ARDAppClient *weakAnswerer = nil;
	NSString *roomId = @"testRoom";
	NSString *callerId = @"testCallerId";
	NSString *answererId = @"testAnswererId";

	XCTestExpectation *callerConnectionExpectation =
	[self expectationWithDescription:@"Caller PC connected."];
	XCTestExpectation *answererConnectionExpectation =
	[self expectationWithDescription:@"Answerer PC connected."];

	caller = [self createAppClientForRoomId:roomId
	clientId:callerId
	isInitiator:YES
	messages:[NSArray array]
	messageHandler:^(ARDSignalingMessage *message) {
	ARDAppClient *strongAnswerer = weakAnswerer;
	[strongAnswerer channel:strongAnswerer.channel didReceiveMessage:message];
	} connectedHandler:^{
	[callerConnectionExpectation fulfill];
	} localVideoTrackHandler:^{
	}];
	// TODO(tkchin): Figure out why DTLS-SRTP constraint causes thread assertion
	// crash in Debug.
	caller.defaultPeerConnectionConstraints =
	[[RTCMediaConstraints alloc] initWithMandatoryConstraints:nil
	optionalConstraints:nil];
	weakCaller = caller;

	answerer = [self createAppClientForRoomId:roomId
	clientId:answererId
	isInitiator:NO
	messages:[NSArray array]
	messageHandler:^(ARDSignalingMessage *message) {
	ARDAppClient *strongCaller = weakCaller;
	[strongCaller channel:strongCaller.channel didReceiveMessage:message];
	} connectedHandler:^{
	[answererConnectionExpectation fulfill];
	} localVideoTrackHandler:^{
	}];
	// TODO(tkchin): Figure out why DTLS-SRTP constraint causes thread assertion
	// crash in Debug.
	answerer.defaultPeerConnectionConstraints =
	[[RTCMediaConstraints alloc] initWithMandatoryConstraints:nil
	optionalConstraints:nil];
	weakAnswerer = answerer;

	// Kick off connection.
	[caller connectToRoomWithId:roomId settings:[self mockSettingsModel] isLoopback:NO];
	[answerer connectToRoomWithId:roomId settings:[self mockSettingsModel] isLoopback:NO];
	[self waitForExpectationsWithTimeout:20 handler:^(NSError *error) {
	if (error) {
	XCTFail(@"Expectation failed with error %@.", error);
	}
	}];
	}

	// Test to see that we get a local video connection
	// Note this will currently pass even when no camera is connected as a local
	// video track is created regardless (Perhaps there should be a test for that...)
	#if !TARGET_IPHONE_SIMULATOR // Expect to fail on simulator due to no camera support
	- (void)testSessionShouldGetLocalVideoTrackCallback {
	ARDAppClient *caller = nil;
	NSString *roomId = @"testRoom";
	NSString *callerId = @"testCallerId";

	XCTestExpectation *localVideoTrackExpectation =
	[self expectationWithDescription:@"Caller got local video."];

	caller = [self createAppClientForRoomId:roomId
	clientId:callerId
	isInitiator:YES
	messages:[NSArray array]
	messageHandler:^(ARDSignalingMessage *message) {}
	connectedHandler:^{}
	localVideoTrackHandler:^{ [localVideoTrackExpectation fulfill]; }];
	caller.defaultPeerConnectionConstraints =
	[[RTCMediaConstraints alloc] initWithMandatoryConstraints:nil
	optionalConstraints:nil];

	// Kick off connection.
	[caller connectToRoomWithId:roomId
	settings:[self mockSettingsModel]
	isLoopback:NO];
	[self waitForExpectationsWithTimeout:20 handler:^(NSError *error) {
	if (error) {
	XCTFail("Expectation timed out with error: %@.", error);
	}
	}];
	}
	#endif

	@end