#import <LocalAuthentication/LocalAuthentication.h>

#import <OCMock/OCMock.h>

#import <UIKit/UIKit.h>

#import <XCTest/XCTest.h>

#import "FRAIdentity.h"

#import "FRAPushMechanism.h"

#import "FRANotification.h"

#import "FRANotificationViewController.h"

@interface FRANotificationViewControllerTests : XCTestCase

@end

@implementation FRANotificationViewControllerTests {

FRANotificationViewController *viewController;

FRAIdentity *identity;

FRAPushMechanism *mechanism;

FRANotification *notification;

LAContext *authContext;

}

- (void)setUp {

[super setUp];

identity = OCMClassMock([FRAIdentity class]);

mechanism = OCMClassMock([FRAPushMechanism class]);

notification = OCMClassMock([FRANotification class]);

authContext = OCMClassMock([LAContext class]);

OCMStub([notification parent]).andReturn(mechanism);

OCMStub([mechanism parent]).andReturn(identity);

// load notification controller from storyboard

UIStoryboard *storyboard = [UIStoryboard storyboardWithName:@"Main" bundle:[NSBundle mainBundle]];

viewController = [storyboard instantiateViewControllerWithIdentifier:FRANotificationViewControllerStoryboardIdentifer];

viewController.notification = notification;

viewController.authContext = authContext;

}

- (void)tearDown {

[self simulateUnloadingOfView];

[super tearDown];

}

- (void)testIfTouchIDUnavailableThenUserShouldBeShownSliderToApproveAndButtonToDenyAuthentication {

// Given

// When

[self simulateLoadingOfView];

// Then

OCMVerifyAll((id)authContext);

XCTAssertEqual(viewController.authorizeSlider.hidden, NO, "slider should be shown if Touch ID unavailable");

XCTAssertEqual(viewController.denyButton.hidden, NO, "deny button should be shown if Touch ID unavailable");

}

- (void)testTouchIDShouldBeUsedIfAvailable {

// Given

OCMExpect([authContext canEvaluatePolicy:LAPolicyDeviceOwnerAuthenticationWithBiometrics

error:[OCMArg anyObjectRef]]).andReturn(YES);

OCMStub([identity issuer]).andReturn(@"Umbrella Corp");

OCMStub([identity accountName]).andReturn(@"alice");

OCMExpect([authContext evaluatePolicy:LAPolicyDeviceOwnerAuthenticationWithBiometrics

localizedReason:@"Log in to Umbrella Corp as alice using Touch ID" reply:[OCMArg any]]);

// When

[self simulateLoadingOfView];

// Then

OCMVerifyAll((id)authContext);

XCTAssertEqual(viewController.authorizeSlider.hidden, YES, "slider should not be shown if Touch ID is available");

XCTAssertEqual(viewController.denyButton.hidden, YES, "deny button should not be shown if Touch ID is available");

}

- (void)simulateLoadingOfView {

[viewController loadViewIfNeeded]; // force IBOutlets etc to be initialized

XCTAssertNotNil(viewController.view);

[viewController viewWillAppear:YES];

}

- (void)simulateUnloadingOfView {

[viewController viewWillDisappear:YES];

}

@end