88#import < XCTest/XCTest.h>
99#import " OCMock/OCMock.h"
1010#import " Instabug/Instabug.h"
11+ #import " Instabug/IBGSurvey.h"
1112#import " InstabugReactBridge.h"
1213#import < Instabug/IBGTypes.h>
1314#import " IBGConstants.h"
@@ -72,7 +73,7 @@ - (void)testInit {
7273 NSArray *invocationEvents = [NSArray arrayWithObjects: [NSNumber numberWithInteger: floatingButtonInvocationEvent], nil ];
7374 BOOL useNativeNetworkInterception = YES ;
7475 IBGSDKDebugLogsLevel sdkDebugLogsLevel = IBGSDKDebugLogsLevelDebug;
75-
76+
7677 OCMStub ([mock setCodePushVersion: codePushVersion]);
7778
7879 [self .instabugBridge init: appToken invocationEvents: invocationEvents debugLogsLevel: sdkDebugLogsLevel useNativeNetworkInterception: useNativeNetworkInterception codePushVersion: codePushVersion];
@@ -84,9 +85,9 @@ - (void)testInit {
8485- (void )testSetCodePushVersion {
8586 id mock = OCMClassMock ([Instabug class ]);
8687 NSString *codePushVersion = @" 123" ;
87-
88+
8889 [self .instabugBridge setCodePushVersion: codePushVersion];
89-
90+
9091 OCMVerify ([mock setCodePushVersion: codePushVersion]);
9192}
9293
@@ -332,12 +333,7 @@ - (void)testNetworkLogIOS {
332333 double duration = 150 ;
333334 NSString *gqlQueryName = nil ;
334335 NSString *serverErrorMessage = nil ;
335- NSDictionary * w3cExternalTraceAttributes = nil ;
336- NSNumber *isW3cCaughted = nil ;
337- NSNumber *partialID = nil ;
338- NSNumber *timestamp= nil ;
339- NSString *generatedW3CTraceparent= nil ;
340- NSString *caughtedW3CTraceparent= nil ;
336+
341337 [self .instabugBridge networkLogIOS: url
342338 method: method
343339 requestBody: requestBody
@@ -353,8 +349,7 @@ - (void)testNetworkLogIOS {
353349 startTime: startTime
354350 duration: duration
355351 gqlQueryName: gqlQueryName
356- w3cExternalTraceAttributes: nil
357- ];
352+ serverErrorMessage: serverErrorMessage];
358353
359354 OCMVerify ([mIBGNetworkLogger addNetworkLogWithUrl: url
360355 method: method
@@ -371,13 +366,7 @@ - (void)testNetworkLogIOS {
371366 startTime: startTime * 1000
372367 duration: duration * 1000
373368 gqlQueryName: gqlQueryName
374- isW3cCaughted: nil
375- partialID: nil
376- timestamp: nil
377- generatedW3CTraceparent: nil
378- caughtedW3CTraceparent: nil
379-
380- ]);
369+ serverErrorMessage: serverErrorMessage]);
381370}
382371
383372- (void )testSetFileAttachment {
@@ -509,102 +498,4 @@ - (void)testClearAllExperiments {
509498 OCMVerify ([mock clearAllExperiments ]);
510499}
511500
512-
513- - (void ) testIsW3ExternalTraceIDEnabled {
514- id mock = OCMClassMock ([IBGNetworkLogger class ]);
515- NSNumber *expectedValue = @(YES );
516-
517- OCMStub ([mock w3ExternalTraceIDEnabled ]).andReturn ([expectedValue boolValue ]);
518-
519- XCTestExpectation *expectation = [self expectationWithDescription: @" Call completion handler" ];
520- RCTPromiseResolveBlock resolve = ^(NSNumber *result) {
521- XCTAssertEqualObjects (result, expectedValue);
522- [expectation fulfill ];
523- };
524-
525- [self .instabugBridge isW3ExternalTraceIDEnabled: resolve :nil ];
526-
527- [self waitForExpectationsWithTimeout: 1.0 handler: nil ];
528-
529- OCMVerify ([mock w3ExternalTraceIDEnabled ]);
530- }
531-
532- - (void ) testIsW3ExternalGeneratedHeaderEnabled {
533- id mock = OCMClassMock ([IBGNetworkLogger class ]);
534- NSNumber *expectedValue = @(YES );
535-
536- OCMStub ([mock w3ExternalGeneratedHeaderEnabled ]).andReturn ([expectedValue boolValue ]);
537-
538- XCTestExpectation *expectation = [self expectationWithDescription: @" Call completion handler" ];
539- RCTPromiseResolveBlock resolve = ^(NSNumber *result) {
540- XCTAssertEqualObjects (result, expectedValue);
541- [expectation fulfill ];
542- };
543-
544- [self .instabugBridge isW3ExternalGeneratedHeaderEnabled: resolve :nil ];
545-
546- [self waitForExpectationsWithTimeout: 1.0 handler: nil ];
547-
548- OCMVerify ([mock w3ExternalGeneratedHeaderEnabled ]);
549- }
550-
551- - (void ) testIsW3CaughtHeaderEnabled {
552- id mock = OCMClassMock ([IBGNetworkLogger class ]);
553- NSNumber *expectedValue = @(YES );
554-
555- OCMStub ([mock w3CaughtHeaderEnabled ]).andReturn ([expectedValue boolValue ]);
556-
557- XCTestExpectation *expectation = [self expectationWithDescription: @" Call completion handler" ];
558- RCTPromiseResolveBlock resolve = ^(NSNumber *result) {
559- XCTAssertEqualObjects (result, expectedValue);
560- [expectation fulfill ];
561- };
562-
563- [self .instabugBridge isW3CaughtHeaderEnabled: resolve :nil ];
564-
565- [self waitForExpectationsWithTimeout: 1.0 handler: nil ];
566-
567- OCMVerify ([mock w3CaughtHeaderEnabled ]);
568- }
569-
570-
571- - (void )testAddFeatureFlags {
572- id mock = OCMClassMock ([Instabug class ]);
573- NSDictionary *featureFlagsMap = @{ @" key13" : @" value1" , @" key2" : @" value2" };
574-
575- OCMStub ([mock addFeatureFlags :[OCMArg any ]]);
576- [self .instabugBridge addFeatureFlags: featureFlagsMap];
577- OCMVerify ([mock addFeatureFlags: [OCMArg checkWithBlock: ^(id value) {
578- NSArray <IBGFeatureFlag *> *featureFlags = value;
579- NSString * firstFeatureFlagName = [featureFlags objectAtIndex: 0 ].name ;
580- NSString * firstFeatureFlagKey = [[featureFlagsMap allKeys ] objectAtIndex: 0 ] ;
581- if ([ firstFeatureFlagKey isEqualToString: firstFeatureFlagName]){
582- return YES ;
583- }
584- return NO ;
585- }]]);
586- }
587-
588- - (void )testRemoveFeatureFlags {
589- id mock = OCMClassMock ([Instabug class ]);
590- NSArray *featureFlags = @[@" exp1" , @" exp2" ];
591- [self .instabugBridge removeFeatureFlags: featureFlags];
592- OCMVerify ([mock removeFeatureFlags: [OCMArg checkWithBlock: ^(id value) {
593- NSArray <IBGFeatureFlag *> *featureFlagsObJ = value;
594- NSString * firstFeatureFlagName = [featureFlagsObJ objectAtIndex: 0 ].name ;
595- NSString * firstFeatureFlagKey = [featureFlags firstObject ] ;
596- if ([ firstFeatureFlagKey isEqualToString: firstFeatureFlagName]){
597- return YES ;
598- }
599- return NO ;
600- }]]);
601- }
602-
603- - (void )testRemoveAllFeatureFlags {
604- id mock = OCMClassMock ([Instabug class ]);
605- OCMStub ([mock removeAllFeatureFlags ]);
606- [self .instabugBridge removeAllFeatureFlags ];
607- OCMVerify ([mock removeAllFeatureFlags ]);
608- }
609-
610501@end
0 commit comments