78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni * The contents of this file are subject to the terms of the Common Development and
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni * Distribution License (the License). You may not use this file except in compliance with the
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni * You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni * specific language governing permission and limitations under the License.
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni * When distributing Covered Software, include this CDDL Header Notice in each file and include
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni * the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni * Header, with the fields enclosed by brackets [] replaced by your own identifying
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni * information: "Portions copyright [year] [name of copyright owner]".
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni * Copyright 2016 ForgeRock AS.
e89f4330a4e6e51eb675c5644317545156125894Diego Colantoni#import "FRABlockAlertView.h"
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni#import "FRAFMDatabaseConnectionHelper.h"
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni#import "FRAIdentity.h"
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni#import "FRAIdentityDatabase.h"
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni#import "FRAIdentityDatabaseSQLiteOperations.h"
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni#import "FRAMechanismReaderAction.h"
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni#import "FRAMessageUtils.h"
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni#import "FRAModelsFromDatabase.h"
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni#import "FRAPushMechanism.h"
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni#import "FRAPushMechanismFactory.h"
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni#import "FRAUriMechanismReader.h"
e89f4330a4e6e51eb675c5644317545156125894Diego Colantonistatic NSInteger const ALERT_HANDLER_CALLBACK_PARAMETER_INDEX = 7;
e89f4330a4e6e51eb675c5644317545156125894Diego Colantonistatic NSInteger const ALERT_CANCEL_SELECTION = 1;
e89f4330a4e6e51eb675c5644317545156125894Diego Colantonistatic NSInteger const ALERT_OK_SELECTION = 0;
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantonistatic NSString * const DEVICE_ID = @"device id";
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni@interface FRAMechanismReaderActionTests : XCTestCase
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni@implementation FRAMechanismReaderActionTests {
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni mockMessageUtils = OCMClassMock([FRAMessageUtils class]);
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni OCMStub([mockMessageUtils respondWithEndpoint:[OCMArg any]
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni mockModelsFromDatabase = OCMClassMock([FRAModelsFromDatabase class]);
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni OCMStub([mockModelsFromDatabase allIdentitiesWithDatabase:[OCMArg any] identityDatabase:[OCMArg any] identityModel:[OCMArg any] error:[OCMArg anyObjectRef]]).andReturn(@[]);
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni mockSQLiteOperations = OCMClassMock([FRAIdentityDatabaseSQLiteOperations class]);
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni OCMStub([mockSQLiteOperations insertIdentity:[OCMArg any] error:[OCMArg anyObjectRef]]).andReturn(YES);
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni OCMStub([mockSQLiteOperations insertMechanism:[OCMArg any] error:[OCMArg anyObjectRef]]).andReturn(YES);
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni OCMStub([mockSQLiteOperations deleteIdentity:[OCMArg any] error:[OCMArg anyObjectRef]]).andReturn(YES);
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni OCMStub([mockSQLiteOperations deleteMechanism:[OCMArg any] error:[OCMArg anyObjectRef]]).andReturn(YES);
e89f4330a4e6e51eb675c5644317545156125894Diego Colantoni mockAlertView = OCMClassMock([FRABlockAlertView class]);
e89f4330a4e6e51eb675c5644317545156125894Diego Colantoni OCMStub([mockAlertView alloc]).andReturn(mockAlertView);
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni identityDatabase = [[FRAIdentityDatabase alloc] initWithSqlOperations:mockSQLiteOperations];
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni identityModel = [[FRAIdentityModel alloc] initWithDatabase:identityDatabase sqlDatabase:nil];
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni mechanismReader = [[FRAUriMechanismReader alloc] initWithDatabase:identityDatabase identityModel:identityModel];
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni mockGateway = OCMClassMock([FRANotificationGateway class]);
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni OCMStub(((FRANotificationGateway *)mockGateway).deviceToken).andReturn(DEVICE_ID);
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni [mechanismReader addMechanismFactory:[[FRAPushMechanismFactory alloc] initWithGateway:mockGateway]];
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni action = [[FRAMechanismReaderAction alloc] initWithMechanismReader:mechanismReader];
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni- (void)testReadAddsIdentityAndMechanismToIdentityModel {
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni NSString *qrCode = @"pushauth://push/forgerock:demo3?a=aHR0cDovL2FtcWEtY2xvbmU2OS50ZXN0LmZvcmdlcm9jay5jb206ODA4MC9vcGVuYW0vanNvbi9wdXNoL3Nucy9tZXNzYWdlP19hY3Rpb249YXV0aGVudGljYXRl&image=aHR0cDovL3NlYXR0bGV3cml0ZXIuY29tL3dwLWNvbnRlbnQvdXBsb2Fkcy8yMDEzLzAxL3dlaWdodC13YXRjaGVycy1zbWFsbC5naWY&b=ff00ff&r=aHR0cDovL2FtcWEtY2xvbmU2OS50ZXN0LmZvcmdlcm9jay5jb206ODA4MC9vcGVuYW0vanNvbi9wdXNoL3Nucy9tZXNzYWdlP19hY3Rpb249cmVnaXN0ZXI&s=qrcodesecret&c=Yf66ojm3Pm80PVvNpljTB6X9CUhgSJ0WZUzB4su3vCY=&l=YW1sYmNvb2tpZT0wMT1hbWxiY29va2ll&m=9326d19c-4d08-4538-8151-f8558e71475f1464361288472&issuer=Rm9yZ2Vyb2Nr";
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni [self assertOnlyOneMechanismRegisteredWithSharedSecret:@"qrcodesecret"];
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni- (void)testReadDoesntAddMechanismIfDuplicateAndUserCancels {
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni [self whenAttemptingToRegisterDuplicateMechanismUserChoosesToCancel:YES];
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni [self assertOnlyOneMechanismRegisteredWithSharedSecret:@"firstcodesecret"];
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni- (void)testReadDoesntAddMechanismIfDuplicateAndUserAccepts {
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni [self whenAttemptingToRegisterDuplicateMechanismUserChoosesToCancel:NO];
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni [self assertOnlyOneMechanismRegisteredWithSharedSecret:@"secondcodesecret"];
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni- (void)whenAttemptingToRegisterDuplicateMechanismUserChoosesToCancel:(BOOL)cancel {
e89f4330a4e6e51eb675c5644317545156125894Diego Colantoni OCMStub([mockAlertView initWithTitle:[OCMArg any] message:[OCMArg any] delegate:[OCMArg any] cancelButtonTitle:[OCMArg any] otherButtonTitle:[OCMArg any] handler:[OCMArg any]])
e89f4330a4e6e51eb675c5644317545156125894Diego Colantoni [invocation getArgument:&callback atIndex:ALERT_HANDLER_CALLBACK_PARAMETER_INDEX];
e89f4330a4e6e51eb675c5644317545156125894Diego Colantoni callback(cancel ? ALERT_CANCEL_SELECTION : ALERT_OK_SELECTION);
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni- (void)attemptToRegisterDuplicateMechanism {
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni NSString *firstCode = @"pushauth://push/forgerock:demo3?a=aHR0cDovL2FtcWEtY2xvbmU2OS50ZXN0LmZvcmdlcm9jay5jb206ODA4MC9vcGVuYW0vanNvbi9wdXNoL3Nucy9tZXNzYWdlP19hY3Rpb249YXV0aGVudGljYXRl&image=aHR0cDovL3NlYXR0bGV3cml0ZXIuY29tL3dwLWNvbnRlbnQvdXBsb2Fkcy8yMDEzLzAxL3dlaWdodC13YXRjaGVycy1zbWFsbC5naWY&b=ff00ff&r=aHR0cDovL2FtcWEtY2xvbmU2OS50ZXN0LmZvcmdlcm9jay5jb206ODA4MC9vcGVuYW0vanNvbi9wdXNoL3Nucy9tZXNzYWdlP19hY3Rpb249cmVnaXN0ZXI&s=firstcodesecret&c=Yf66ojm3Pm80PVvNpljTB6X9CUhgSJ0WZUzB4su3vCY=&l=YW1sYmNvb2tpZT0wMT1hbWxiY29va2ll&m=9326d19c-4d08-4538-8151-f8558e71475f1464361288472&issuer=Rm9yZ2Vyb2Nr";
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni NSString *secondCode = @"pushauth://push/forgerock:demo3?a=aHR0cDovL2FtcWEtY2xvbmU2OS50ZXN0LmZvcmdlcm9jay5jb206ODA4MC9vcGVuYW0vanNvbi9wdXNoL3Nucy9tZXNzYWdlP19hY3Rpb249YXV0aGVudGljYXRl&image=aHR0cDovL3NlYXR0bGV3cml0ZXIuY29tL3dwLWNvbnRlbnQvdXBsb2Fkcy8yMDEzLzAxL3dlaWdodC13YXRjaGVycy1zbWFsbC5naWY&b=ff00ff&r=aHR0cDovL2FtcWEtY2xvbmU2OS50ZXN0LmZvcmdlcm9jay5jb206ODA4MC9vcGVuYW0vanNvbi9wdXNoL3Nucy9tZXNzYWdlP19hY3Rpb249cmVnaXN0ZXI&s=secondcodesecret&c=Yf66ojm3Pm80PVvNpljTB6X9CUhgSJ0WZUzB4su3vCY=&l=YW1sYmNvb2tpZT0wMT1hbWxiY29va2ll&m=9326d19c-4d08-4538-8151-f8558e71475f1464361288472&issuer=Rm9yZ2Vyb2Nr";
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni- (void)assertOnlyOneMechanismRegisteredWithSharedSecret:(NSString *)sharedSecret {
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni FRAIdentity *identity = [[identityModel identities] objectAtIndex:0];
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni FRAPushMechanism *mechanism = (FRAPushMechanism *)[[identity mechanisms] objectAtIndex:0];
78c07714ec1113f7f21c75b818f2bf6a7021618aDiego Colantoni XCTAssertEqual([identity mechanisms].count, 1);