Add test utility class for adding test cases dynamically at runtime

2016-04-11 23:37:16 +02:00
parent 4441b7decd
commit 58b60dd390
8 changed files with 70 additions and 51 deletions
@@ -0,0 +1,29 @@
+//
+//  HTMLKitTestUtil.m
+//  HTMLKit
+//
+//  Created by Iska on 11/04/16.
+//  Copyright © 2016 BrainCookie. All rights reserved.
+//
+
+#import "HTMLKitTestUtil.h"
+#import <objc/runtime.h>
+
+@implementation HTMLKitTestUtil
+
+ (NSInvocation *)addTestToClass:(Class)cls withName:(NSString *)name block:(id)block
+{
+	IMP implementation = imp_implementationWithBlock(block);
+	const char *types = [[NSString stringWithFormat:@"%s%s%s", @encode(id), @encode(id), @encode(SEL)] UTF8String];
+	
+	SEL selector = NSSelectorFromString(name);
+	class_addMethod(cls, selector, implementation, types);
+
+	NSMethodSignature *signature = [cls instanceMethodSignatureForSelector:selector];
+	NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:signature];
+	invocation.selector = selector;
+
+	return invocation;
+}
+
+@end