Update tests.

Author: antrix1989

IdentityCore/tests/MSIDAADV1WebviewFactoryTests.m

@@ -35,6 +35,8 @@
 #import "MSIDPkce.h"
 #import "MSIDInteractiveTokenRequestParameters.h"
 #import "MSIDAccountIdentifier.h"
+#import "MSIDFlightManager.h"
+#import "MSIDConstants.h"
 
 @interface MSIDAADV1WebviewFactoryTests : XCTestCase
 
@@ -77,6 +79,12 @@ - (void)testAuthorizationParametersFromParameters_withValidParams_shouldContainA
                                           }];
 
     [expectedQPs addEntriesFromDictionary:[MSIDDeviceId deviceId]];
+#if TARGET_OS_IPHONE
+    if ([MSIDFlightManager.sharedInstance boolForKey:MSID_FLIGHT_SUPPORT_DUNA_CBA])
+    {
+        expectedQPs[@"switch_browser"] = @"1";
+    }
+#endif
 
     XCTAssertTrue([expectedQPs compareAndPrintDiff:params]);
 }
@@ -110,6 +118,12 @@ - (void)testAuthorizationParametersFromConfiguration_withValidParamsWithScopes_s
                                           }];
 
     [expectedQPs addEntriesFromDictionary:[MSIDDeviceId deviceId]];
+#if TARGET_OS_IPHONE
+    if ([MSIDFlightManager.sharedInstance boolForKey:MSID_FLIGHT_SUPPORT_DUNA_CBA])
+    {
+        expectedQPs[@"switch_browser"] = @"1";
+    }
+#endif
 
     XCTAssertTrue([expectedQPs compareAndPrintDiff:params]);
 }

IdentityCore/tests/MSIDAADV2WebviewFactoryTests.m

@@ -35,6 +35,8 @@
 #import "MSIDPkce.h"
 #import "MSIDInteractiveTokenRequestParameters.h"
 #import "MSIDAccountIdentifier.h"
+#import "MSIDFlightManager.h"
+#import "MSIDConstants.h"
 
 @interface MSIDAADV2WebviewFactoryTests : XCTestCase
 
@@ -78,6 +80,12 @@ - (void)testAuthorizationParametersFromParameters_withValidParams_shouldContainA
                                           }];
 
     [expectedQPs addEntriesFromDictionary:[MSIDDeviceId deviceId]];
+#if TARGET_OS_IPHONE
+    if ([MSIDFlightManager.sharedInstance boolForKey:MSID_FLIGHT_SUPPORT_DUNA_CBA])
+    {
+        expectedQPs[@"switch_browser"] = @"1";
+    }
+#endif
 
     XCTAssertTrue([expectedQPs compareAndPrintDiff:params]);
 }

IdentityCore/tests/MSIDAADWebviewFactoryTests.m

@@ -46,6 +46,8 @@
 #import "MSIDPkce.h"
 #import "MSIDWebAADAuthCodeResponse.h"
 #import "MSIDBrokerConstants.h"
+#import "MSIDFlightManager.h"
+#import "MSIDConstants.h"
 
 @interface MSIDAADWebviewFactoryTests : XCTestCase
 
@@ -98,6 +100,12 @@ - (void)testAuthorizationParametersFromParameters_withValidParams_shouldContainA
                                           @"X-AnchorMailbox" : [MSIDTestRequireValueSentinel new],
                                           }];
     [expectedQPs addEntriesFromDictionary:[MSIDDeviceId deviceId]];
+#if TARGET_OS_IPHONE
+    if ([MSIDFlightManager.sharedInstance boolForKey:MSID_FLIGHT_SUPPORT_DUNA_CBA])
+    {
+        expectedQPs[@"switch_browser"] = @"1";
+    }
+#endif
 
     XCTAssertTrue([expectedQPs compareAndPrintDiff:params]);
 }
@@ -161,6 +169,13 @@ - (void)testAuthorizationParametersFromParameters_withInstanceAwareFlagSet_shoul
                                           @"code_challenge" : pkce.codeChallenge
                                           }];
     [expectedQPs addEntriesFromDictionary:[MSIDDeviceId deviceId]];
+#if TARGET_OS_IPHONE
+    if ([MSIDFlightManager.sharedInstance boolForKey:MSID_FLIGHT_SUPPORT_DUNA_CBA])
+    {
+        expectedQPs[@"switch_browser"] = @"1";
+    }
+#endif
+    
     XCTAssertTrue([expectedQPs compareAndPrintDiff:params]);
 }