caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni/*
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni * The contents of this file are subject to the terms of the Common Development and
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni * Distribution License (the License). You may not use this file except in compliance with the
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni * License.
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni *
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni * You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni * specific language governing permission and limitations under the License.
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni *
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni * When distributing Covered Software, include this CDDL Header Notice in each file and include
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni * the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni * Header, with the fields enclosed by brackets [] replaced by your own identifying
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni * information: "Portions copyright [year] [name of copyright owner]".
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni *
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni * Copyright 2016 ForgeRock AS.
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni */
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni#import <XCTest/XCTest.h>
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni#import <OCMock/OCMock.h>
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni#import "FRAHotpOathMechanism.h"
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni#import "FRAIdentityDatabase.h"
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni#import "FRAIdentityDatabaseSQLiteOperations.h"
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni#import "FRAIdentityModel.h"
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni#import "FRAModelsFromDatabase.h"
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni#import "FRAOathMechanismFactory.h"
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni#import "FRAPushMechanism.h"
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni#import "FRAPushMechanismFactory.h"
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni#import "FRATotpOathMechanism.h"
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni#import "FRAUriMechanismReader.h"
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantonistatic NSString * const DEVICE_ID = @"device id";
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni@interface FRAUriMechanismReaderTests : XCTestCase
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni@end
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni@implementation FRAUriMechanismReaderTests {
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni FRAIdentityDatabase *identityDatabase;
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni FRAIdentityModel *identityModel;
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni FRAUriMechanismReader *reader;
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni id mockGateway;
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni id mockDatabaseOperations;
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni id mockModelsFromDatabase;
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni}
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni- (void)setUp {
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni [super setUp];
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni mockGateway = OCMClassMock([FRANotificationGateway class]);
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni OCMStub(((FRANotificationGateway *)mockGateway).deviceToken).andReturn(DEVICE_ID);
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni mockModelsFromDatabase = OCMClassMock([FRAModelsFromDatabase class]);
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni OCMStub([mockModelsFromDatabase allIdentitiesWithDatabase:[OCMArg any] identityDatabase:[OCMArg any] identityModel:[OCMArg any] error:[OCMArg anyObjectRef]]).andReturn(@[]);
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni mockDatabaseOperations = OCMClassMock([FRAIdentityDatabaseSQLiteOperations class]);
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni OCMStub([mockDatabaseOperations insertIdentity:[OCMArg any] error:[OCMArg anyObjectRef]]).andReturn(YES);
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni OCMStub([mockDatabaseOperations insertMechanism:[OCMArg any] error:[OCMArg anyObjectRef]]).andReturn(YES);
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni identityDatabase = [[FRAIdentityDatabase alloc] initWithSqlOperations:mockDatabaseOperations];
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni identityModel = [[FRAIdentityModel alloc] initWithDatabase:mockDatabaseOperations sqlDatabase:nil];
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni reader = [[FRAUriMechanismReader alloc] initWithDatabase:identityDatabase identityModel:identityModel];
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni FRAPushMechanismFactory *pushMechanismFactory = [[FRAPushMechanismFactory alloc] initWithGateway:mockGateway];
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni FRAOathMechanismFactory *oathMechanismFactory = [[FRAOathMechanismFactory alloc] init];
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni [reader addMechanismFactory:pushMechanismFactory];
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni [reader addMechanismFactory:oathMechanismFactory];
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni}
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni- (void)tearDown {
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni [mockGateway stopMocking];
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni [mockDatabaseOperations stopMocking];
6a2ae9c7fb4d2c40d75cab0edaf940f22c18224fDiego Colantoni [mockModelsFromDatabase stopMocking];
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni [super tearDown];
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni}
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni- (void)testReaderCanParsePushMechanism {
33f578498e5bdaf445a4f509ac419f0a23148f61Diego Colantoni NSURL *qrUrl = [NSURL URLWithString:@"pushauth://push/forgerock:demo3?a=aHR0cDovL2FtcWEtY2xvbmU2OS50ZXN0LmZvcmdlcm9jay5jb206ODA4MC9vcGVuYW0vanNvbi9wdXNoL3Nucy9tZXNzYWdlP19hY3Rpb249YXV0aGVudGljYXRl&image=aHR0cDovL3NlYXR0bGV3cml0ZXIuY29tL3dwLWNvbnRlbnQvdXBsb2Fkcy8yMDEzLzAxL3dlaWdodC13YXRjaGVycy1zbWFsbC5naWY&b=ff00ff&r=aHR0cDovL2FtcWEtY2xvbmU2OS50ZXN0LmZvcmdlcm9jay5jb206ODA4MC9vcGVuYW0vanNvbi9wdXNoL3Nucy9tZXNzYWdlP19hY3Rpb249cmVnaXN0ZXI&s=dA18Iph3slIUDVuRc5+3y7nv9NLGnPksH66d3jIF6uE=&c=Yf66ojm3Pm80PVvNpljTB6X9CUhgSJ0WZUzB4su3vCY=&l=YW1sYmNvb2tpZT0wMT1hbWxiY29va2ll&m=9326d19c-4d08-4538-8151-f8558e71475f1464361288472&issuer=Rm9yZ2Vyb2Nr"];
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
7c51d54f23c5581d2cf894f9eafb9798e3febd22Diego Colantoni FRAPushMechanism *mechanism = (FRAPushMechanism *)[reader parseFromURL:qrUrl handler:nil error:nil];
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni XCTAssertNotNil(mechanism);
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni XCTAssertEqual([mechanism class], [FRAPushMechanism class]);
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni}
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni- (void)testReaderCanParseHotpOathMechanism {
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni NSURL *qrUrl = [NSURL URLWithString:@"otpauth://hotp/Forgerock:demo?secret=IJQWIZ3FOIQUEYLE&issuer=Forgerock&counter=0"];
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
7c51d54f23c5581d2cf894f9eafb9798e3febd22Diego Colantoni FRAHotpOathMechanism *mechanism = (FRAHotpOathMechanism *)[reader parseFromURL:qrUrl handler:nil error:nil];
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni XCTAssertNotNil(mechanism);
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni XCTAssertEqual([mechanism class], [FRAHotpOathMechanism class]);
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni}
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni- (void)testReaderCanParseTotpOathMechanism {
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni NSURL *qrUrl = [NSURL URLWithString:@"otpauth://totp/ForgeRock:demo?secret=EE3PFF5BM6GHVRNZIBBQWBNRLQ======&issuer=ForgeRock&digits=8&period=30"];
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
7c51d54f23c5581d2cf894f9eafb9798e3febd22Diego Colantoni FRATotpOathMechanism *mechanism = (FRATotpOathMechanism *)[reader parseFromURL:qrUrl handler:nil error:nil];
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni XCTAssertNotNil(mechanism);
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni XCTAssertEqual([mechanism class], [FRATotpOathMechanism class]);
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni}
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni
caa9e77dc369fea8df9ae2c598d3c83b7214c1cfDiego Colantoni@end