FRANotificationTest.m revision a1e92b2783be4bfeb0c7e267223cc7779a6f324c
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan/*
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan * The contents of this file are subject to the terms of the Common Development and
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan * Distribution License (the License). You may not use this file except in compliance with the
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan * License.
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan *
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan * You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan * specific language governing permission and limitations under the License.
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan *
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan * When distributing Covered Software, include this CDDL Header Notice in each file and include
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan * the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan * Header, with the fields enclosed by brackets [] replaced by your own identifying
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan * information: "Portions copyright [year] [name of copyright owner]".
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan *
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan * Copyright 2016 ForgeRock AS.
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan */
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan#import <OCMock/OCMock.h>
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan#import <XCTest/XCTest.h>
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan#import "FRAIdentityDatabase.h"
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan#import "FRAIdentityDatabaseSQLiteOperations.h"
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan#import "FRAMessageUtils.h"
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan#import "FRANotification.h"
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan#import "FRAPushMechanism.h"
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan@interface FRANotificationTest : XCTestCase
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan@end
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan@implementation FRANotificationTest {
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan id mockSqlOperations;
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan id databaseObserverMock;
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan FRAIdentityDatabase *database;
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan FRANotification* notification;
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan id messageUtilsMock;
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan FRAPushMechanism *mechanism;
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan}
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan- (void)setUp {
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan [super setUp];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan mockSqlOperations = OCMClassMock([FRAIdentityDatabaseSQLiteOperations class]);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan database = [[FRAIdentityDatabase alloc] initWithSqlOperations:mockSqlOperations];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan NSTimeInterval timeToLive = 120.0;
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan mechanism = [[FRAPushMechanism alloc] initWithDatabase:database authEndpoint:@"http://service.endpoint" secret:@"secret"];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan notification = [[FRANotification alloc] initWithDatabase:database messageId:@"messageId" challenge:@"challenge" timeReceived:[NSDate date] timeToLive:timeToLive];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan OCMStub([(FRAIdentityDatabaseSQLiteOperations *)mockSqlOperations insertNotification:notification error:nil]).andReturn(YES);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan OCMStub([(FRAIdentityDatabaseSQLiteOperations *)mockSqlOperations updateNotification:notification error:nil]).andReturn(YES);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan databaseObserverMock = OCMObserverMock();
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan messageUtilsMock = OCMClassMock([FRAMessageUtils class]);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan [notification setParent:mechanism];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan}
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan- (void)tearDown {
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan [mockSqlOperations stopMocking];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan [messageUtilsMock stopMocking];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan [super tearDown];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan}
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan- (void)testInitialStateOfNotification {
addc51736ee961d7b52d048021139920d4364671Neil Madden // Given
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan // When
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan // Then
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan XCTAssertEqual([notification isPending], YES);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan XCTAssertEqual([notification isApproved], NO);
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan XCTAssertEqual([notification isDenied], NO);
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan XCTAssertEqual([notification isExpired], NO);
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan}
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan- (void)testShouldSetTimeExpired {
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan // Given
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan NSDate *timeReceived = [NSDate date];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan NSTimeInterval timeToLive = 120.0;
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan FRANotification *expiringNotification = [[FRANotification alloc] initWithDatabase:database messageId:@"messageId" challenge:@"challenge" timeReceived:timeReceived timeToLive:timeToLive];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan // When
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan // Then
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan XCTAssertEqualObjects([expiringNotification timeExpired], [timeReceived dateByAddingTimeInterval:timeToLive]);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan}
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan- (void)testShouldApproveNotification {
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan // Given
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan // When
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan [notification approveWithError:nil];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan // Then
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan XCTAssertEqual([notification isPending], NO);
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan XCTAssertEqual([notification isApproved], YES);
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan XCTAssertEqual([notification isDenied], NO);
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan}
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan- (void)testApproveNotificationSendsMessageToService {
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan // Given
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan [database insertNotification:notification error:nil];
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan OCMExpect([messageUtilsMock respondWithEndpoint:@"http://service.endpoint"
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan base64Secret:@"secret"
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan messageId:@"messageId"
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan data:@{@"response":[FRAMessageUtils generateChallengeResponse:@"challenge"
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan secret:@"secret"]}
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan handler:[OCMArg any]]);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan // When
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan [notification approveWithError:nil];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan // Then
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan OCMVerifyAll(messageUtilsMock);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan}
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan- (void)testShouldDenyNotification {
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan // Given
6057736124e265febf1603affd1ec9be4428d99dDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan // When
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan [notification denyWithError:nil];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan // Then
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan XCTAssertEqual([notification isPending], NO);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan XCTAssertEqual([notification isApproved], NO);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan XCTAssertEqual([notification isDenied], YES);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan}
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan- (void)testSavedNotificationAutomaticallySavesItselfToDatabaseWhenApproved {
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan // Given
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan OCMStub([(FRAIdentityDatabaseSQLiteOperations*)mockSqlOperations insertNotification:notification error:nil]).andReturn(YES);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan [database insertNotification:notification error:nil];
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan OCMStub([(FRAIdentityDatabaseSQLiteOperations*)mockSqlOperations updateNotification:notification error:nil]).andReturn(YES);
9df1de8309feb43d162df756c9290ce3afab802aDirk Hogan OCMStub([messageUtilsMock respondWithEndpoint:@"http://service.endpoint"
base64Secret:@"secret"
messageId:@"messageId"
data:@{@"response":[FRAMessageUtils generateChallengeResponse:@"challenge"
secret:@"secret"]}
handler:[OCMArg any]]);
// When
BOOL notificationApproved = [notification approveWithError:nil];
// Then
XCTAssertTrue(notificationApproved);
}
- (void)testSavedNotificationAutomaticallySavesItselfToDatabaseWhenDenied {
// Given
OCMStub([(FRAIdentityDatabaseSQLiteOperations*)mockSqlOperations insertNotification:notification error:nil]).andReturn(YES);
[database insertNotification:notification error:nil];
OCMStub([(FRAIdentityDatabaseSQLiteOperations*)mockSqlOperations updateNotification:notification error:nil]).andReturn(YES);
// When
BOOL notificationDenied = [notification denyWithError:nil];
// Then
XCTAssertTrue(notificationDenied);
}
- (void)testBroadcastsOneChangeNotificationWhenNotificationUpdateIsAutomaticallySavedToDatabase {
// Given
OCMStub([(FRAIdentityDatabaseSQLiteOperations*)mockSqlOperations insertNotification:notification error:nil]).andReturn(YES);
[database insertNotification:notification error:nil];
OCMStub([(FRAIdentityDatabaseSQLiteOperations*)mockSqlOperations updateNotification:notification error:nil]).andReturn(YES);
[[NSNotificationCenter defaultCenter] addMockObserver:databaseObserverMock name:FRAIdentityDatabaseChangedNotification object:database];
[[databaseObserverMock expect] notificationWithName:FRAIdentityDatabaseChangedNotification object:database userInfo:[OCMArg any]];
OCMStub([messageUtilsMock respondWithEndpoint:@"http://service.endpoint"
base64Secret:@"secret"
messageId:@"messageId"
data:@{@"response":[FRAMessageUtils generateChallengeResponse:@"challenge"
secret:@"secret"]}
handler:[OCMArg any]]);
// When
BOOL notificationApproved = [notification approveWithError:nil];
// Then
XCTAssertTrue(notificationApproved);
OCMVerifyAll(databaseObserverMock);
}
- (void)testIsExpiredReturnsYesIfNotificationHasExpired {
// Given
FRANotification *expiredNotification = [[FRANotification alloc] initWithDatabase:database messageId:@"messageId" challenge:@"challenge" timeReceived:[NSDate date] timeToLive:-10.0];
// When
// Then
XCTAssertEqual([expiredNotification isExpired], YES);
XCTAssertEqual([expiredNotification isPending], NO);
XCTAssertEqual([expiredNotification isApproved], NO);
XCTAssertEqual([expiredNotification isDenied], NO);
}
- (void)testIsExpiredReturnsNoIfNotificationHasNotExpired {
// Given
FRANotification *expiredNotification = [[FRANotification alloc] initWithDatabase:database messageId:@"messageId" challenge:@"challenge" timeReceived:[NSDate date] timeToLive:120.0];
// When
// Then
XCTAssertEqual([expiredNotification isExpired], NO);
XCTAssertEqual([expiredNotification isPending], YES);
XCTAssertEqual([expiredNotification isApproved], NO);
XCTAssertEqual([expiredNotification isDenied], NO);
}
@end