1  //
  2  //  SecTapToRadarTests.m
  3  //  TrustedPeersTests
  4  //
  5  
  6  #import <TargetConditionals.h>
  7  
  8  #if !TARGET_OS_BRIDGE
  9  
 10  #import <XCTest/XCTest.h>
 11  #import <OCMock/OCMock.h>
 12  #import "utilities/SecTapToRadar.h"
 13  
 14  @interface SecTapToRadarTests : XCTestCase
 15  @property BOOL isRateLimited;
 16  @property BOOL userSay;
 17  
 18  @property bool ttrDidAppear;
 19  @property id mockTTR;
 20  
 21  @end
 22  
 23  @implementation SecTapToRadarTests
 24  
 25  - (void)triggerTapToRadar:(SecTapToRadar *)ttrRequest
 26  {
 27      self.ttrDidAppear = true;
 28  }
 29  
 30  - (BOOL)isRateLimited:(SecTapToRadar *)ttrRequest
 31  {
 32      return self.isRateLimited;
 33  }
 34  
 35  - (BOOL)askUserIfTTR:(SecTapToRadar *)ttrRequest
 36  {
 37      return self.userSay;
 38  }
 39  
 40  - (void)setUp {
 41  
 42      self.ttrDidAppear = NO;
 43      self.isRateLimited = NO;
 44      self.userSay = YES;
 45  
 46      self.mockTTR = OCMClassMock([SecTapToRadar class]);
 47      OCMStub([self.mockTTR triggerTapToRadar:[OCMArg any]]).andCall(self, @selector(triggerTapToRadar:));
 48      OCMStub([self.mockTTR isRateLimited:[OCMArg any]]).andCall(self, @selector(isRateLimited:));
 49      OCMStub([self.mockTTR askUserIfTTR:[OCMArg any]]).andCall(self, @selector(askUserIfTTR:));
 50  }
 51  
 52  - (void)testSecTTRNormal {
 53  
 54      SecTapToRadar *ttr = [[SecTapToRadar alloc] initTapToRadar:@"alert" description:@"test" radar:@"1"];
 55  
 56      [ttr trigger];
 57      XCTAssertTrue(self.ttrDidAppear, "should have appeared");
 58  }
 59  
 60  - (void)testSecTTRRateLimit {
 61      SecTapToRadar *ttr = [[SecTapToRadar alloc] initTapToRadar:@"alert" description:@"test" radar:@"1"];
 62  
 63      self.isRateLimited = YES;
 64      [ttr trigger];
 65      XCTAssertFalse(self.ttrDidAppear, "should not have appered");
 66  }
 67  
 68  - (void)testSecTTRUserSupress {
 69  
 70      SecTapToRadar *ttr = [[SecTapToRadar alloc] initTapToRadar:@"alert" description:@"test" radar:@"1"];
 71  
 72      self.userSay = false;
 73      [ttr trigger];
 74      XCTAssertFalse(self.ttrDidAppear, "should not have appered");
 75  
 76      self.userSay = true;
 77      [ttr trigger];
 78      XCTAssertTrue(self.ttrDidAppear, "should have appeared");
 79  }
 80  
 81  - (void)testSecTTRRateLimiter {
 82  
 83      SecTapToRadar *ttr = [[SecTapToRadar alloc] initTapToRadar:@"alert" description:@"test" radar:@"1"];
 84      NSString *key = [SecTapToRadar keyname:ttr];
 85      NSUserDefaults *defaults = [[NSUserDefaults alloc] initWithSuiteName:@"com.apple.security"];
 86  
 87      [ttr clearRetryTimestamp];
 88  
 89      XCTAssertFalse([ttr isRateLimited], @"should not be rate-limited for first request");
 90      [ttr updateRetryTimestamp];
 91      XCTAssertTrue([ttr isRateLimited], @"should be rate-limited after first request");
 92  
 93      [ttr clearRetryTimestamp];
 94      XCTAssertFalse([ttr isRateLimited], @"should not be ratelitmied after clear");
 95  
 96      // check invalid settings
 97      [defaults setObject:@"invalid" forKey:key];
 98      XCTAssertNotNil([defaults objectForKey:key], "should have cleared setting");
 99      XCTAssertFalse([ttr isRateLimited], @"should not be rate-limited if invalid type");
100      XCTAssertNil([defaults objectForKey:key], "should have cleared setting");
101  
102  
103  }
104  
105  
106  @end
107  
108  #endif /* TARGET_OS_BRIDGE */