keychain/trust/TrustedPeersTests/TPPolicyDocumentTests.m

   1 /*
   2  * Copyright (c) 2017 Apple Inc. All Rights Reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. Please obtain a copy of the License at
  10  * http://www.opensource.apple.com/apsl/ and read it before using this
  11  * file.
  12  *
  13  * The Original Code and all software distributed under the License are
  14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  18  * Please see the License for the specific language governing rights and
  19  * limitations under the License.
  20  *
  21  * @APPLE_LICENSE_HEADER_END@
  22  */
  23
  24 #import <XCTest/XCTest.h>
  25 #import <TrustedPeers/TrustedPeers.h>
  26
  27 @interface TPPolicyDocumentTests : XCTestCase
  28
  29 @end
  30
  31 @implementation TPPolicyDocumentTests
  32
  33 - (void)testRoundTrip
  34 {
  35     TPPolicyDocument *doc1
  36     = [TPPolicyDocument policyDocWithVersion:1
  37                              modelToCategory:@[
  38                                                @{ @"prefix": @"iPhone",  @"category": @"full" },
  39                                                @{ @"prefix": @"iPad",    @"category": @"full" },
  40                                                @{ @"prefix": @"Mac",     @"category": @"full" },
  41                                                @{ @"prefix": @"iMac",    @"category": @"full" },
  42                                                @{ @"prefix": @"AppleTV", @"category": @"tv" },
  43                                                @{ @"prefix": @"Watch",   @"category": @"watch" },
  44                                                ]
  45                             categoriesByView:@{
  46                                                @"WiFi":              @[ @"full", @"tv", @"watch" ],
  47                                                @"SafariCreditCards": @[ @"full" ],
  48                                                @"PCSEscrow":         @[ @"full" ]
  49                                                }
  50                        introducersByCategory:@{
  51                                                @"full":  @[ @"full" ],
  52                                                @"tv":    @[ @"full", @"tv" ],
  53                                                @"watch": @[ @"full", @"watch" ]
  54                                                }
  55                                   redactions:@{
  56                                                @"foo": [@"bar" dataUsingEncoding:NSUTF8StringEncoding]
  57                                                }
  58                                     hashAlgo:kTPHashAlgoSHA256];
  59
  60
  61     TPPolicyDocument *doc2 = [TPPolicyDocument policyDocWithHash:doc1.policyHash pList:doc1.pList];
  62     XCTAssert([doc1 isEqualToPolicyDocument:doc2]);
  63
  64     TPPolicyDocument *doc3 = [TPPolicyDocument policyDocWithHash:@"SHA256:foo" pList:doc1.pList];
  65     XCTAssertNil(doc3);
  66 }
  67
  68 @end