forked from cedarbdd/cedar
/
NSInvocation+CDRXExample.m
40 lines (29 loc) · 1.16 KB
/
NSInvocation+CDRXExample.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
#import "NSInvocation+CDRXExample.h"
#import <objc/runtime.h>
const char *CDRXDispatcherKey;
const char *CDRXExamplesKey;
const char *CDRXSpecClassNameKey;
@implementation NSInvocation (CDRXExample)
- (CDRReportDispatcher *)cdr_dispatcher {
return objc_getAssociatedObject(self, &CDRXDispatcherKey);
}
- (void)cdr_setDispatcher:(CDRReportDispatcher *)dispatcher {
objc_setAssociatedObject(self, &CDRXDispatcherKey, dispatcher, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (NSArray *)cdr_examples {
return objc_getAssociatedObject(self, &CDRXExamplesKey);
}
- (void)cdr_setExamples:(NSArray *)examples {
objc_setAssociatedObject(self, &CDRXExamplesKey, examples, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (NSString *)cdr_specClassName {
return objc_getAssociatedObject(self, &CDRXSpecClassNameKey);
}
- (void)cdr_setSpecClassName:(NSString *)specClassName {
objc_setAssociatedObject(self, &CDRXSpecClassNameKey, specClassName, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
- (void)cdr_addSupplementaryExample:(CDRExample *)example {
NSArray *existingExamples = self.cdr_examples ?: @[];
self.cdr_examples = [existingExamples arrayByAddingObject:example];
}
@end