Files
firebase-ios-sdk/GoogleDataTransport/GDTCORTests/Unit/Helpers/GDTCOREventGenerator.m
T

62 lines
2.7 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 "GoogleDataTransport/GDTCORTests/Unit/Helpers/GDTCOREventGenerator.h"
#import "GoogleDataTransport/GDTCORLibrary/Public/GoogleDataTransport/GDTCORClock.h"
#import "GoogleDataTransport/GDTCORLibrary/Public/GoogleDataTransport/GDTCOREvent.h"
#import "GoogleDataTransport/GDTCORLibrary/Public/GoogleDataTransport/GDTCORTargets.h"
#import "GoogleDataTransport/GDTCORLibrary/Private/GDTCOREvent_Private.h"
#import "GoogleDataTransport/GDTCORTests/Unit/Helpers/GDTCORDataObjectTesterClasses.h"
@implementation GDTCOREventGenerator
+ (NSMutableSet<GDTCOREvent *> *)generate3Events {
static NSUInteger counter = 0;
int howManyToGenerate = 3;
NSMutableSet<GDTCOREvent *> *set = [[NSMutableSet alloc] initWithCapacity:howManyToGenerate];
for (int i = 0; i < howManyToGenerate; i++) {
GDTCOREvent *event = [[GDTCOREvent alloc] initWithMappingID:@"1337" target:kGDTCORTargetTest];
event.clockSnapshot = [GDTCORClock snapshot];
event.qosTier = GDTCOREventQosDefault;
event.dataObject = [[GDTCORDataObjectTesterSimple alloc] initWithString:@"testing!"];
NSString *filePath = [NSString stringWithFormat:@"test-%ld.txt", (unsigned long)counter];
[[NSFileManager defaultManager] createFileAtPath:filePath
contents:[NSData data]
attributes:nil];
[set addObject:event];
counter++;
}
return set;
}
+ (GDTCOREvent *)generateEventForTarget:(GDTCORTarget)target
qosTier:(nullable NSNumber *)qosTier
mappingID:(nullable NSString *)mappingID {
GDTCOREventQoS determinedQosTier =
qosTier == nil ? arc4random_uniform(GDTCOREventQoSWifiOnly) : qosTier.intValue;
NSString *determinedMappingID = mappingID == nil ? [NSDate date].description : mappingID;
GDTCOREvent *event = [[GDTCOREvent alloc] initWithMappingID:determinedMappingID target:target];
event.clockSnapshot = [GDTCORClock snapshot];
event.qosTier = determinedQosTier;
event.dataObject = [[GDTCORDataObjectTesterSimple alloc] initWithString:@"testing!"];
return event;
}
@end