test/future.m

   1 /*
   2 TEST_BUILD
   3     $C{COMPILE} $DIR/future0.m -o future0.dylib -dynamiclib
   4     $C{COMPILE} $DIR/future2.m -x none future0.dylib -o future2.dylib -dynamiclib
   5     $C{COMPILE} $DIR/future.m -x none future0.dylib -o future.out
   6 END
   7 */
   8
   9 #include "test.h"
  10
  11 #if __has_feature(objc_arc)
  12
  13 int main()
  14 {
  15     testwarn("rdar://10041403 future class API is not ARC-compatible");
  16     succeed(__FILE__);
  17 }
  18
  19
  20 #else
  21
  22 #include <objc/runtime.h>
  23 #include <malloc/malloc.h>
  24 #include <string.h>
  25 #include <dlfcn.h>
  26 #include "future.h"
  27
  28 @implementation Sub2
  29 +(int)method {
  30     return 2;
  31 }
  32 +(Class)classref {
  33     return [Sub2 class];
  34 }
  35 @end
  36
  37 @implementation SubSub2
  38 +(int)method {
  39     return 1 + [super method];
  40 }
  41 @end
  42
  43 int main()
  44 {
  45     Class oldTestRoot;
  46     Class oldSub1;
  47     Class newSub1;
  48
  49     // objc_getFutureClass with existing class
  50     oldTestRoot = objc_getFutureClass("TestRoot");
  51     testassert(oldTestRoot == [TestRoot class]);
  52     testassert(! _class_isFutureClass(oldTestRoot));
  53
  54     // objc_getFutureClass with missing class
  55     oldSub1 = objc_getFutureClass("Sub1");
  56     testassert(oldSub1);
  57     testassert(malloc_size(objc_unretainedPointer(oldSub1)) > 0);
  58     testassert(objc_getClass("Sub1") == Nil);
  59     testassert(_class_isFutureClass(oldSub1));
  60     testassert(0 == strcmp(class_getName(oldSub1), "Sub1"));
  61     testassert(object_getClass(oldSub1) == Nil);  // CF expects this
  62
  63     // objc_getFutureClass a second time
  64     testassert(oldSub1 == objc_getFutureClass("Sub1"));
  65
  66     // Load class Sub1
  67     dlopen("future2.dylib", 0);
  68
  69     // Verify use of future class
  70     newSub1 = objc_getClass("Sub1");
  71     testassert(oldSub1 == newSub1);
  72     testassert(newSub1 == [newSub1 classref]);
  73     testassert(newSub1 == class_getSuperclass(objc_getClass("SubSub1")));
  74     testassert(! _class_isFutureClass(newSub1));
  75
  76     testassert(1 == [oldSub1 method]);
  77     testassert(1 == [newSub1 method]);
  78
  79     succeed(__FILE__);
  80 }
  81
  82 #endif