Files
firebase-ios-sdk/GoogleDataTransport/GDTCORTests/Integration/Helpers/GDTCORIntegrationTestPrioritizer.m
T
Doudou Nan d69c642277 Delete GDTCORStoredEvent, modify apis and tests (#4880)
* Delete GDTCORStoredEvent, modify apis and tests

* styled

* Delete GDTCORDataFuture

* header changes

* add writeToURL in COREvent

* Remove dataObjectTransportBytes property in COREvent

* Set fileURL in COREvent nullable

* Change method writeToURL private and modify comments

* Modify encoding keys in GDTCORStorage and GDTCORUploadPackage

* comments and headers changes

* testMigration passed without fileURL

* Regression test passed with retrieving fileURL

* styled and added fileURL NOTNIL assert in test

* Modify backwords compatibility of GDTCORStoredEvent decoding

* Comments changes

* Add GDTCOREvent translation mapping in UploadPackage
2020-02-18 16:09:07 -08:00

85 lines
2.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 "GDTCORTests/Integration/Helpers/GDTCORIntegrationTestPrioritizer.h"
#import <GoogleDataTransport/GDTCOREvent.h>
#import <GoogleDataTransport/GDTCORRegistrar.h>
#import "GDTCORTests/Integration/Helpers/GDTCORIntegrationTestUploadPackage.h"
@interface GDTCORIntegrationTestPrioritizer ()
/** Events that are only supposed to be uploaded whilst on wifi. */
@property(nonatomic) NSMutableSet<GDTCOREvent *> *wifiOnlyEvents;
/** Events that can be uploaded on any type of connection. */
@property(nonatomic) NSMutableSet<GDTCOREvent *> *nonWifiEvents;
/** The queue on which this prioritizer operates. */
@property(nonatomic) dispatch_queue_t queue;
@end
@implementation GDTCORIntegrationTestPrioritizer
- (instancetype)init {
self = [super init];
if (self) {
_queue =
dispatch_queue_create("com.google.GDTCORIntegrationTestPrioritizer", DISPATCH_QUEUE_SERIAL);
_wifiOnlyEvents = [[NSMutableSet alloc] init];
_nonWifiEvents = [[NSMutableSet alloc] init];
[[GDTCORRegistrar sharedInstance] registerPrioritizer:self target:kGDTCORIntegrationTestTarget];
}
return self;
}
- (void)prioritizeEvent:(GDTCOREvent *)event {
dispatch_async(_queue, ^{
if (event.qosTier == GDTCOREventQoSWifiOnly) {
[self.wifiOnlyEvents addObject:event];
} else {
[self.nonWifiEvents addObject:event];
}
});
}
- (GDTCORUploadPackage *)uploadPackageWithTarget:(GDTCORTarget)target
conditions:(GDTCORUploadConditions)conditions {
__block GDTCORIntegrationTestUploadPackage *uploadPackage =
[[GDTCORIntegrationTestUploadPackage alloc] initWithTarget:target];
dispatch_sync(_queue, ^{
if ((conditions & GDTCORUploadConditionWifiData) == GDTCORUploadConditionWifiData) {
uploadPackage.events = [self.wifiOnlyEvents setByAddingObjectsFromSet:self.nonWifiEvents];
} else {
uploadPackage.events = self.nonWifiEvents;
}
});
return uploadPackage;
}
- (void)packageDelivered:(GDTCORUploadPackage *)package successful:(BOOL)successful {
dispatch_async(_queue, ^{
for (GDTCOREvent *event in package.events) {
[self.wifiOnlyEvents removeObject:event];
[self.nonWifiEvents removeObject:event];
}
});
}
@end