Files
app-check/AppCheckCore/Tests/Unit/Utils/GACFakeTimer.m
T

50 lines
1.4 KiB
Objective-C

/*
* Copyright 2021 Google LLC
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#import "AppCheckCore/Tests/Unit/Utils/GACFakeTimer.h"
@implementation GACFakeTimer
- (GACTimerProvider)fakeTimerProvider {
__weak __typeof__(self) weakSelf = self;
return ^id<GACAppCheckTimerProtocol> _Nullable(NSDate *fireDate, dispatch_queue_t queue,
dispatch_block_t handler) {
__typeof__(self) strongSelf = weakSelf;
if (!strongSelf) {
return nil;
}
@synchronized(strongSelf) {
strongSelf.handler = handler;
void (^createHandler)(NSDate *) = strongSelf.createHandler;
if (createHandler) {
createHandler(fireDate);
}
}
return strongSelf;
};
}
- (void)invalidate {
void (^invalidationHandler)(void) = self.invalidationHandler;
if (invalidationHandler) {
invalidationHandler();
}
}
@end