[Storage] Refine track2 OAuth context logic (Azure#25954)

* refine track2 oauth context logic * add changelog --------- Co-authored-by: Jin Lei <54836179+msJinLei@users.noreply.github.com>
azure-sdk · Aug 29, 2024 · 0a0f8d7 · 0a0f8d7
1 parent 599b64c
commit 0a0f8d7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/Storage/Storage.Management/ChangeLog.md b/src/Storage/Storage.Management/ChangeLog.md
@@ -18,6 +18,7 @@
         - Additional information about change #1
 -->
 ## Upcoming Release
+* Updated Storage account cmdlet output context based on OAuth token 
 
 ## Version 7.3.0
 * Supported account tier Cold

diff --git a/src/Storage/Storage.Management/Common/LazyAzureStorageContext.cs b/src/Storage/Storage.Management/Common/LazyAzureStorageContext.cs
@@ -83,7 +83,11 @@ public override AzureSessionCredential Track2OauthToken {
             {
                 if (_track2OauthToken == null)
                 {
-                   _track2OauthToken = _track2OAuthTokenFactory();
+                    if (_track2OAuthTokenFactory == null)
+                    {
+                        return null;
+                    }
+                    _track2OauthToken = _track2OAuthTokenFactory();
                     return _track2OauthToken;
                 }
                 return _track2OauthToken;