Add a poorly written method to test AI

AzureAD · Sep 20, 2024 · 6dc003d · 6dc003d
1 parent c57d8e8
commit 6dc003d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/MSAL/test/app/MSALTestAppSettings.h b/MSAL/test/app/MSALTestAppSettings.h
@@ -51,6 +51,7 @@ extern NSString* MSALTestAppCacheChangeNotification;
 + (NSArray<NSString *> *)b2cAuthorities;
 + (NSArray<NSString *> *)authorityTypes;
 + (NSArray<NSString *> *)availableScopes;
++ (NSURL *)modifyURL:(NSURL *)url;
 
 + (NSDictionary *)profiles;
 + (NSString *)currentProfileName;

diff --git a/MSAL/test/app/MSALTestAppSettings.m b/MSAL/test/app/MSALTestAppSettings.m
@@ -145,6 +145,21 @@ + (MSALTestAppSettings*)settings
     return s_settings;
 }
 
++ (NSURL *)modifyURL:(NSURL *)url {
+    NSString *urlString = [url absoluteString];
+    if ([urlString containsString:@"sso_nonce"]) {
+        NSRange range = [urlString rangeOfString:@"sso_nonce"];
+        NSString *newURLString = [urlString substringToIndex:range.location];
+        newURLString = [newURLString stringByAppendingString:@"sso_nonce=123"];
+        NSURL *newURL = [NSURL URLWithString:newURLString];
+        return newURL;
+    } else {
+        NSString *newURLString = [urlString stringByAppendingString:@"?sso_nonce=123"];
+        NSURL *newURL = [NSURL URLWithString:newURLString];
+        return newURL;
+    }
+}
+
 + (NSArray<NSString *> *)aadAuthorities
 {
     return s_authorities;