Test update OCMock

third_party/ocmock/OCMock/NSValue+OCMAdditions.m

+/*
+ *  Copyright (c) 2014-2015 Erik Doernenburg and contributors
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License"); you may
+ *  not use these files except in compliance with the License. You may obtain
+ *  a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ *  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ *  License for the specific language governing permissions and limitations
+ *  under the License.
+ */
+
+#import "NSValue+OCMAdditions.h"
+
+@implementation NSValue (OCMAdditions)
+
+static CFNumberType OCMNumberTypeForObjCType(const char* objcType) {
+  switch (objcType[0]) {
+    case 'c':
+      return kCFNumberCharType;
+    case 'C':
+      return kCFNumberCharType;
+    case 'B':
+      return kCFNumberCharType;
+    case 's':
+      return kCFNumberShortType;
+    case 'S':
+      return kCFNumberShortType;
+    case 'i':
+      return kCFNumberIntType;
+    case 'I':
+      return kCFNumberIntType;
+    case 'l':
+      return kCFNumberLongType;
+    case 'L':
+      return kCFNumberLongType;
+    case 'q':
+      return kCFNumberLongLongType;
+    case 'Q':
+      return kCFNumberLongLongType;
+    case 'f':
+      return kCFNumberFloatType;
+    case 'd':
+      return kCFNumberDoubleType;
+    default:
+      return 0;
+  }
+}
+
+static NSNumber* OCMNumberForValue(NSValue* value) {
+#define CREATE_NUM(_type) \
+  ({                      \
+    _type _v;             \
+    [value getValue:&_v]; \
+    @(_v);                \
+  })
+  switch ([value objCType][0]) {
+    case 'c':
+      return CREATE_NUM(char);
+    case 'C':
+      return CREATE_NUM(unsigned char);
+    case 'B':
+      return CREATE_NUM(bool);
+    case 's':
+      return CREATE_NUM(short);
+    case 'S':
+      return CREATE_NUM(unsigned short);
+    case 'i':
+      return CREATE_NUM(int);
+    case 'I':
+      return CREATE_NUM(unsigned int);
+    case 'l':
+      return CREATE_NUM(long);
+    case 'L':
+      return CREATE_NUM(unsigned long);
+    case 'q':
+      return CREATE_NUM(long long);
+    case 'Q':
+      return CREATE_NUM(unsigned long long);
+    case 'f':
+      return CREATE_NUM(float);
+    case 'd':
+      return CREATE_NUM(double);
+    default:
+      return nil;
+  }
+}
+
+- (BOOL)getBytes:(void*)outputBuf objCType:(const char*)targetType {
+  /*
+   * See if they are similar number types, and if we can convert losslessly
+   * between them.
+   * For the most part, we set things up to use CFNumberGetValue, which returns
+   * false if
+   * conversion will be lossy.
+   */
+  CFNumberType inputType = OCMNumberTypeForObjCType([self objCType]);
+  CFNumberType outputType = OCMNumberTypeForObjCType(targetType);
+
+  if (inputType == 0 || outputType == 0)  // one or both are non-number types
+    return NO;
+
+  NSNumber* inputNumber = [self isKindOfClass:[NSNumber class]]
+                              ? (NSNumber*)self
+                              : OCMNumberForValue(self);
+
+  /*
+   * Due to some legacy, back-compatible requirements in CFNumber.c,
+   * CFNumberGetValue can return true for
+   * some conversions which should not be allowed (by reading source,
+   * conversions from integer types to
+   * 8-bit or 16-bit integer types).  So, check ourselves.
+   */
+  long long min;
+  long long max;
+  long long val = [inputNumber longLongValue];
+  switch (targetType[0]) {
+    case 'B':
+    case 'c':
+      min = CHAR_MIN;
+      max = CHAR_MAX;
+      break;
+    case 'C':
+      min = 0;
+      max = UCHAR_MAX;
+      break;
+    case 's':
+      min = SHRT_MIN;
+      max = SHRT_MAX;
+      break;
+    case 'S':
+      min = 0;
+      max = USHRT_MAX;
+      break;
+    default:
+      min = LLONG_MIN;
+      max = LLONG_MAX;
+      break;
+  }
+  if (val < min || val > max)
+    return NO;
+
+  /* Get the number, and return NO if the value was out of range or conversion
+   * was lossy */
+  return CFNumberGetValue((CFNumberRef)inputNumber, outputType, outputBuf);
+}
+
+@end