mirror of
https://github.com/firebase/firebase-ios-sdk
synced 2026-11-09 22:36:01 +05:45
Ryan Wilson
19159a4b25
* Fix tvOS storage issues. On tvOS devices, the ApplicationSupport directory cannot be used. Unfortunately this doesn't come up in simulator testing, only on device. This hasn't been fully tested on device but given that any writes to the ApplicationSupport directory fail on tvOS, this change can't break much. Fixes #6612 @defagos can you please give this a test? * Fixed changed dir name for RC * Fixed test naming, too. * Fix the documentation comment for GoogleUtilities. * Updated changelog. * Review feedback
123 lines
4.8 KiB
Objective-C
123 lines
4.8 KiB
Objective-C
// Copyright 2019 Google
|
|
//
|
|
// 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 "FirebaseSegmentation/Sources/SEGDatabaseManager.h"
|
|
|
|
#import <OCMock/OCMock.h>
|
|
|
|
@interface SEGDatabaseManager (Test)
|
|
- (NSString *)pathForSegmentationDatabase;
|
|
@end
|
|
|
|
@interface SEGDatabaseManagerTests : XCTestCase {
|
|
long _expectationTimeout;
|
|
}
|
|
@property(nonatomic) id databaseManagerMock;
|
|
|
|
@end
|
|
|
|
@implementation SEGDatabaseManagerTests
|
|
|
|
- (void)setUp {
|
|
// Override the database path to create a test database.
|
|
self.databaseManagerMock = OCMClassMock([SEGDatabaseManager class]);
|
|
OCMStub([self.databaseManagerMock pathForSegmentationDatabase])
|
|
.andReturn([self pathForSegmentationTestDatabase]);
|
|
|
|
// Expectation timeout for each test.
|
|
_expectationTimeout = 2;
|
|
}
|
|
|
|
- (void)tearDown {
|
|
[self.databaseManagerMock stopMocking];
|
|
self.databaseManagerMock = nil;
|
|
}
|
|
|
|
- (void)testDatabaseCreateOrOpen {
|
|
XCTestExpectation *expectation = [self expectationWithDescription:@"testDatabaseLoad"];
|
|
// Initialize the database manager.
|
|
SEGDatabaseManager *databaseManager = [[SEGDatabaseManager alloc] init];
|
|
XCTAssertNotNil(databaseManager);
|
|
// Load all data from the database.
|
|
[databaseManager createOrOpenDatabaseWithCompletion:^(BOOL success, NSDictionary *result) {
|
|
XCTAssertTrue(success);
|
|
XCTAssertTrue(result.count == 0);
|
|
[expectation fulfill];
|
|
}];
|
|
[self waitForExpectationsWithTimeout:_expectationTimeout handler:nil];
|
|
}
|
|
|
|
- (void)testDatabaseInsertAndRead {
|
|
XCTestExpectation *expectation = [self expectationWithDescription:@"testDatabaseLoad"];
|
|
// Initialize the database manager.
|
|
SEGDatabaseManager *databaseManager = [[SEGDatabaseManager alloc] init];
|
|
XCTAssertNotNil(databaseManager);
|
|
// Load all data from the database.
|
|
[databaseManager createOrOpenDatabaseWithCompletion:^(BOOL success, NSDictionary *result) {
|
|
XCTAssertTrue(success);
|
|
XCTAssertTrue(result.count == 0, "Result was %@", result);
|
|
|
|
// Insert data.
|
|
[databaseManager
|
|
insertMainTableApplicationNamed:@"firebase_test_app"
|
|
customInstanceIdentifier:@"custom-123"
|
|
firebaseInstanceIdentifier:@"firebase-123"
|
|
associationStatus:kSEGAssociationStatusPending
|
|
completionHandler:^(BOOL success, NSDictionary *result) {
|
|
XCTAssertTrue(success);
|
|
XCTAssertNil(result);
|
|
|
|
// Read data.
|
|
[databaseManager loadMainTableWithCompletion:^(BOOL success,
|
|
NSDictionary *result) {
|
|
XCTAssertTrue(success);
|
|
XCTAssertEqual(result.count, 1);
|
|
NSDictionary *associations = [result objectForKey:@"firebase_test_app"];
|
|
XCTAssertNotNil(associations);
|
|
XCTAssertEqualObjects(
|
|
[associations objectForKey:kSEGCustomInstallationIdentifierKey],
|
|
@"custom-123");
|
|
XCTAssertEqualObjects(
|
|
[associations objectForKey:kSEGFirebaseInstallationIdentifierKey],
|
|
@"firebase-123");
|
|
XCTAssertEqualObjects(
|
|
[associations objectForKey:kSEGAssociationStatusKey],
|
|
kSEGAssociationStatusPending);
|
|
[expectation fulfill];
|
|
}];
|
|
}];
|
|
}];
|
|
[self waitForExpectationsWithTimeout:_expectationTimeout handler:nil];
|
|
}
|
|
|
|
#pragma mark Helpers
|
|
- (NSString *)pathForSegmentationTestDatabase {
|
|
#if TARGET_OS_TV
|
|
NSArray *dirPaths = NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES);
|
|
#else
|
|
NSArray *dirPaths =
|
|
NSSearchPathForDirectoriesInDomains(NSApplicationSupportDirectory, NSUserDomainMask, YES);
|
|
#endif
|
|
NSString *storageDir = dirPaths.firstObject;
|
|
NSString *databaseName =
|
|
[NSString stringWithFormat:@"FirebaseSegmentation-test-%d.sqlite3", (arc4random() % 100)];
|
|
NSArray *components = @[ storageDir, @"Google/FirebaseSegmentation", databaseName ];
|
|
NSString *dbPath = [NSString pathWithComponents:components];
|
|
NSLog(@"Created test database at: %@", dbPath);
|
|
return dbPath;
|
|
}
|
|
@end
|