API/tests/JSExportTests.mm

   1 /*
   2  * Copyright (C) 2014 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  23  * THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #import "JSExportTests.h"
  27
  28 #import <objc/runtime.h>
  29 #import <objc/objc.h>
  30
  31 #if JSC_OBJC_API_ENABLED
  32
  33 extern "C" void checkResult(NSString *description, bool passed);
  34
  35 @interface JSExportTests : NSObject
  36 + (void) exportInstanceMethodWithIdProtocolTest;
  37 + (void) exportInstanceMethodWithClassProtocolTest;
  38 + (void) exportDynamicallyGeneratedProtocolTest;
  39 @end
  40
  41 @protocol TruthTeller
  42 - (BOOL) returnTrue;
  43 @end
  44
  45 @interface TruthTeller : NSObject<TruthTeller>
  46 @end
  47
  48 @implementation TruthTeller
  49 - (BOOL) returnTrue
  50 {
  51     return true;
  52 }
  53 @end
  54
  55 @protocol ExportMethodWithIdProtocol <JSExport>
  56 - (void) methodWithIdProtocol:(id<TruthTeller>)object;
  57 @end
  58
  59 @interface ExportMethodWithIdProtocol : NSObject<ExportMethodWithIdProtocol>
  60 @end
  61
  62 @implementation ExportMethodWithIdProtocol
  63 - (void) methodWithIdProtocol:(id<TruthTeller>)object
  64 {
  65     checkResult(@"Exporting a method with id<protocol> in the type signature", [object returnTrue]);
  66 }
  67 @end
  68
  69 @protocol ExportMethodWithClassProtocol <JSExport>
  70 - (void) methodWithClassProtocol:(NSObject<TruthTeller> *)object;
  71 @end
  72
  73 @interface ExportMethodWithClassProtocol : NSObject<ExportMethodWithClassProtocol>
  74 @end
  75
  76 @implementation ExportMethodWithClassProtocol
  77 - (void) methodWithClassProtocol:(NSObject<TruthTeller> *)object
  78 {
  79     checkResult(@"Exporting a method with class<protocol> in the type signature", [object returnTrue]);
  80 }
  81 @end
  82
  83 @implementation JSExportTests
  84 + (void) exportInstanceMethodWithIdProtocolTest
  85 {
  86     JSContext *context = [[JSContext alloc] init];
  87     context[@"ExportMethodWithIdProtocol"] = [ExportMethodWithIdProtocol class];
  88     context[@"makeTestObject"] = ^{
  89         return [[ExportMethodWithIdProtocol alloc] init];
  90     };
  91     context[@"opaqueObject"] = [[TruthTeller alloc] init];
  92     [context evaluateScript:@"makeTestObject().methodWithIdProtocol(opaqueObject);"];
  93     checkResult(@"Successfully exported instance method", !context.exception);
  94 }
  95
  96 + (void) exportInstanceMethodWithClassProtocolTest
  97 {
  98     JSContext *context = [[JSContext alloc] init];
  99     context[@"ExportMethodWithClassProtocol"] = [ExportMethodWithClassProtocol class];
 100     context[@"makeTestObject"] = ^{
 101         return [[ExportMethodWithClassProtocol alloc] init];
 102     };
 103     context[@"opaqueObject"] = [[TruthTeller alloc] init];
 104     [context evaluateScript:@"makeTestObject().methodWithClassProtocol(opaqueObject);"];
 105     checkResult(@"Successfully exported instance method", !context.exception);
 106 }
 107
 108 + (void) exportDynamicallyGeneratedProtocolTest
 109 {
 110     JSContext *context = [[JSContext alloc] init];
 111     Protocol *dynProtocol = objc_allocateProtocol("NSStringJSExport");
 112     Protocol *jsExportProtocol = @protocol(JSExport);
 113     protocol_addProtocol(dynProtocol, jsExportProtocol);
 114     Method method = class_getInstanceMethod([NSString class], @selector(boolValue));
 115     protocol_addMethodDescription(dynProtocol, @selector(boolValue), method_getTypeEncoding(method), YES, YES);
 116     NSLog(@"type encoding = %s", method_getTypeEncoding(method));
 117     protocol_addMethodDescription(dynProtocol, @selector(boolValue), "B@:", YES, YES);
 118     objc_registerProtocol(dynProtocol);
 119     class_addProtocol([NSString class], dynProtocol);
 120
 121     context[@"NSString"] = [NSString class];
 122     context[@"myString"] = @"YES";
 123     JSValue *value = [context evaluateScript:@"myString.boolValue()"];
 124     checkResult(@"Dynamically generated JSExport-ed protocols are ignored", [value isUndefined] && !!context.exception);
 125 }
 126 @end
 127
 128 void runJSExportTests()
 129 {
 130     @autoreleasepool {
 131         [JSExportTests exportInstanceMethodWithIdProtocolTest];
 132         [JSExportTests exportInstanceMethodWithClassProtocolTest];
 133         [JSExportTests exportDynamicallyGeneratedProtocolTest];
 134     }
 135 }
 136
 137 #endif // JSC_OBJC_API_ENABLED