Fixed old test

BranchMetrics · Jun 28, 2024 · cb78ae8 · cb78ae8
1 parent 6545298
commit cb78ae8
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/Branch-TestBed/Branch-SDK-Tests/BranchClassTests.m b/Branch-TestBed/Branch-SDK-Tests/BranchClassTests.m
@@ -243,19 +243,20 @@ - (void)testSetDMAParamsForEEA {
 
 - (void)testSetConsumerProtectionAttributionLevel {
     // Set to Reduced and check
-    [Branch setConsumerProtectionAttributionLevel:BranchAttributionLevelReduced];
+    Branch *branch = [Branch getInstance];
+    [branch setConsumerProtectionAttributionLevel:BranchAttributionLevelReduced];
     XCTAssertEqual([BNCPreferenceHelper sharedInstance].attributionLevel, BranchAttributionLevelReduced);
 
     // Set to Minimal and check
-    [Branch setConsumerProtectionAttributionLevel:BranchAttributionLevelMinimal];
+    [branch setConsumerProtectionAttributionLevel:BranchAttributionLevelMinimal];
     XCTAssertEqual([BNCPreferenceHelper sharedInstance].attributionLevel, BranchAttributionLevelMinimal);
 
     // Set to None and check
-    [Branch setConsumerProtectionAttributionLevel:BranchAttributionLevelNone];
+    [branch setConsumerProtectionAttributionLevel:BranchAttributionLevelNone];
     XCTAssertEqual([BNCPreferenceHelper sharedInstance].attributionLevel, BranchAttributionLevelNone);
 
     // Set to Full and check
-    [Branch setConsumerProtectionAttributionLevel:BranchAttributionLevelFull];
+    [branch setConsumerProtectionAttributionLevel:BranchAttributionLevelFull];
     XCTAssertEqual([BNCPreferenceHelper sharedInstance].attributionLevel, BranchAttributionLevelFull);
 
 }