1  //
  2  //  KCDerTest.m
  3  //  Security
  4  //
  5  //
  6  
  7  #import <XCTest/XCTest.h>
  8  
  9  #import <Foundation/Foundation.h>
 10  #import <KeychainCircle/KCDer.h>
 11  
 12  @interface KCDerTest : XCTestCase
 13  
 14  @end
 15  
 16  @implementation KCDerTest
 17  
 18  - (void)setUp {
 19      [super setUp];
 20      // Put setup code here. This method is called before the invocation of each test method in the class.
 21  }
 22  
 23  - (void)tearDown {
 24      // Put teardown code here. This method is called after the invocation of each test method in the class.
 25      [super tearDown];
 26  }
 27  
 28  - (void) roundTripData: (NSData*) data {
 29      NSError* error = nil;
 30      size_t size = kcder_sizeof_data(data, &error);
 31  
 32      XCTAssert(size != 0, @"Bad size: %@", data);
 33  
 34      if (size == 0)
 35          return;
 36  
 37      NSMutableData *buffer = [NSMutableData dataWithLength:size];
 38      error = nil;
 39      uint8_t* beginning = kcder_encode_data(data, &error, [buffer mutableBytes], [buffer mutableBytes] + size);
 40  
 41      XCTAssert(beginning != NULL, "Error encoding: %@", error);
 42  
 43      if (beginning == NULL)
 44          return;
 45  
 46      XCTAssertEqual(beginning, [buffer mutableBytes], @"Size != buffer use");
 47  
 48      NSData* recovered = nil;
 49  
 50      error = nil;
 51      const uint8_t* end = kcder_decode_data(&recovered, &error, [buffer mutableBytes], [buffer mutableBytes] + size);
 52  
 53      XCTAssert(end != NULL, "Error decoding: %@", error);
 54  
 55      if (end == NULL)
 56          return;
 57  
 58      XCTAssertEqual(end, [buffer mutableBytes] + size, @"readback didn't use all the buffer");
 59  
 60      XCTAssertEqualObjects(data, recovered, @"Didn't get equal object");
 61  
 62  }
 63  
 64  - (void)testData {
 65      [self roundTripData: [NSData data]];
 66  
 67      uint8_t bytes[] = { 1, 2, 3, 0xFF, 4, 0x0, 0xA };
 68      [self roundTripData: [NSData dataWithBytes:bytes length:sizeof(bytes)]];
 69  }
 70  
 71  - (void) roundTripString: (NSString*) string {
 72      NSError* error = nil;
 73  
 74      size_t size = kcder_sizeof_string(string, &error);
 75  
 76      XCTAssert(size != 0, @"Bad size: %@", string);
 77  
 78      if (size == 0)
 79          return;
 80  
 81      NSMutableData *buffer = [NSMutableData dataWithLength:size];
 82      error = nil;
 83      uint8_t* beginning = kcder_encode_string(string, &error, [buffer mutableBytes], [buffer mutableBytes] + size);
 84  
 85      XCTAssert(beginning != NULL, "Error encoding: %@", error);
 86  
 87      if (beginning == NULL)
 88          return;
 89  
 90      XCTAssertEqual(beginning, [buffer mutableBytes], @"Size != buffer use");
 91  
 92      NSString* recovered = nil;
 93  
 94      error = nil;
 95      const uint8_t* end = kcder_decode_string(&recovered, &error, [buffer mutableBytes], [buffer mutableBytes] + size);
 96  
 97      XCTAssert(end != NULL, "Error decoding: %@", error);
 98  
 99      if (end == NULL)
100          return;
101  
102      XCTAssertEqual(end, [buffer mutableBytes] + size, @"readback didn't use all the buffer");
103  
104      XCTAssertEqualObjects(string, recovered, @"Didn't get equal object");
105      
106  }
107  
108  - (void)testString {
109      [self roundTripString: [NSString stringWithCString:"Test" encoding:NSUTF8StringEncoding]];
110      [self roundTripString: [NSString stringWithCString:"ü😍🐸✝️₧➜" encoding:NSUTF8StringEncoding]];
111  }
112  
113  
114  @end