9
9
#import " GREYIdlingResourcePrettyPrint.h"
10
10
@import ObjectiveC;
11
11
12
- static NSMapTable <NSString *, id >* __tarckedObjectsMapping;
12
+ static NSMapTable <GREYAppStateTrackerObject *, id >* __tarckedObjectsMapping;
13
13
14
14
@interface GREYAppStateTracker (PrettyPrint) @end
15
15
16
16
@implementation GREYAppStateTracker (PrettyPrint)
17
17
18
- - (NSString *)_pp__trackState : (GREYAppState)state forElement : (id )element ;
18
+ - (GREYAppStateTrackerObject *)_pp__trackState : (GREYAppState)state forObject : (id )element ;
19
19
{
20
- NSString * rv = [self _pp__trackState: state forElement : element];
20
+ GREYAppStateTrackerObject * rv = [self _pp__trackState: state forObject : element];
21
21
22
22
[__tarckedObjectsMapping setObject: element forKey: rv];
23
23
24
24
return rv;
25
25
}
26
26
27
- - (void )_pp__untrackState : (GREYAppState)state forElementWithID : ( NSString *) elementID ;
27
+ - (void )_pp__untrackState : (GREYAppState)state forObject : (GREYAppStateTrackerObject*) obj ;
28
28
{
29
- [self _pp__untrackState: state forElementWithID: elementID ];
29
+ [self _pp__untrackState: state forObject: obj ];
30
30
31
- [__tarckedObjectsMapping removeObjectForKey: elementID ];
31
+ [__tarckedObjectsMapping removeObjectForKey: obj ];
32
32
}
33
33
34
34
@@ -38,13 +38,13 @@ + (void)load
38
38
dispatch_once (&onceToken, ^{
39
39
__tarckedObjectsMapping = [NSMapTable mapTableWithKeyOptions: NSMapTableStrongMemory valueOptions: NSMapTableWeakMemory ];
40
40
41
- Method m1 = class_getInstanceMethod (self, @selector (trackState:forElement : ));
42
- Method m2 = class_getInstanceMethod (self, @selector (_pp__trackState:forElement : ));
41
+ Method m1 = class_getInstanceMethod (self, @selector (trackState:forObject : ));
42
+ Method m2 = class_getInstanceMethod (self, @selector (_pp__trackState:forObject : ));
43
43
44
44
method_exchangeImplementations (m1, m2);
45
45
46
- m1 = class_getInstanceMethod (self, @selector (untrackState:forElementWithID : ));
47
- m2 = class_getInstanceMethod (self, @selector (_pp__untrackState:forElementWithID : ));
46
+ m1 = class_getInstanceMethod (self, @selector (untrackState:forObject : ));
47
+ m2 = class_getInstanceMethod (self, @selector (_pp__untrackState:forObject : ));
48
48
49
49
method_exchangeImplementations (m1, m2);
50
50
});
0 commit comments