Files
firebase-ios-sdk/FirebaseDatabase/Tests/Unit/FPersistentConnectionTests.m
T

101 lines
3.3 KiB
Objective-C

/*
* Copyright 2026 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 <XCTest/XCTest.h>
#import "FirebaseDatabase/Sources/Api/FIRDatabaseConfig.h"
#import "FirebaseDatabase/Sources/Constants/FConstants.h"
#import "FirebaseDatabase/Sources/Core/FPersistentConnection.h"
#import "FirebaseDatabase/Sources/Core/FRepoInfo.h"
#import "FirebaseDatabase/Tests/Helpers/FTestHelpers.h"
@interface FPersistentConnectionTestDouble : FPersistentConnection
@property(nonatomic, strong) NSMutableArray<NSString *> *interruptedReasons;
@property(nonatomic, strong) NSMutableArray<NSString *> *resumedReasons;
@end
@implementation FPersistentConnectionTestDouble
- (void)interruptForReason:(NSString *)reason {
[self.interruptedReasons addObject:reason];
}
- (void)resumeForReason:(NSString *)reason {
[self.resumedReasons addObject:reason];
}
@end
@interface FPersistentConnection (Testing)
- (void)systemClockDidChange:(NSNotification *)notification;
@end
@interface FPersistentConnectionTests : XCTestCase
@property(nonatomic, strong) FPersistentConnectionTestDouble *connection;
@property(nonatomic, strong) dispatch_queue_t connectionQueue;
@end
@implementation FPersistentConnectionTests
- (void)setUp {
[super setUp];
self.connectionQueue = dispatch_queue_create(
"com.google.firebase.database.PersistentConnectionTests", DISPATCH_QUEUE_SERIAL);
FRepoInfo *repoInfo = [[FRepoInfo alloc] initWithHost:@"example.firebaseio.com"
isSecure:YES
withNamespace:@"example"];
self.connection =
[[FPersistentConnectionTestDouble alloc] initWithRepoInfo:repoInfo
dispatchQueue:self.connectionQueue
config:[FTestHelpers defaultConfig]];
self.connection.interruptedReasons = [NSMutableArray array];
self.connection.resumedReasons = [NSMutableArray array];
}
- (void)testSystemClockChangeRestartsConnection {
[self.connection systemClockDidChange:nil];
[self waitForConnectionQueue];
[self assertConnectionRestartedForSystemClockChange];
}
- (void)testSystemClockDidChangeNotificationRestartsConnection {
[[NSNotificationCenter defaultCenter] postNotificationName:NSSystemClockDidChangeNotification
object:nil];
[self waitForConnectionQueue];
[self assertConnectionRestartedForSystemClockChange];
}
- (void)waitForConnectionQueue {
dispatch_sync(self.connectionQueue, ^{
});
}
- (void)assertConnectionRestartedForSystemClockChange {
XCTAssertEqualObjects(self.connection.interruptedReasons,
(@[ kFInterruptReasonSystemClockChange ]));
XCTAssertEqualObjects(self.connection.resumedReasons, (@[ kFInterruptReasonSystemClockChange ]));
}
@end