diff --git a/sdk/core/Azure.Core.TestFramework/src/MockCredential.cs b/sdk/core/Azure.Core.TestFramework/src/MockCredential.cs new file mode 100644 index 000000000000..9b42481b8d36 --- /dev/null +++ b/sdk/core/Azure.Core.TestFramework/src/MockCredential.cs @@ -0,0 +1,22 @@ +// Copyright (c) Microsoft Corporation. All rights reserved. +// Licensed under the MIT License. + +using System; +using System.Threading; +using System.Threading.Tasks; + +namespace Azure.Core.TestFramework +{ + public class MockCredential : TokenCredential + { + public override ValueTask GetTokenAsync(TokenRequestContext requestContext, CancellationToken cancellationToken) + { + return new ValueTask(GetToken(requestContext, cancellationToken)); + } + + public override AccessToken GetToken(TokenRequestContext requestContext, CancellationToken cancellationToken) + { + return new AccessToken("TEST TOKEN " + string.Join(" ", requestContext.Scopes), DateTimeOffset.MaxValue); + } + } +} diff --git a/sdk/core/Azure.Core.TestFramework/src/TestEnvironment.cs b/sdk/core/Azure.Core.TestFramework/src/TestEnvironment.cs index 948fbc9249c2..aaa52dfeef50 100644 --- a/sdk/core/Azure.Core.TestFramework/src/TestEnvironment.cs +++ b/sdk/core/Azure.Core.TestFramework/src/TestEnvironment.cs @@ -142,7 +142,7 @@ public TokenCredential Credential if (Mode == RecordedTestMode.Playback) { - _credential = new TestCredential(); + _credential = new MockCredential(); } else { @@ -283,18 +283,5 @@ private string GetRecordedValue(string name) return _recording.GetVariable(name, null); } - - private class TestCredential : TokenCredential - { - public override ValueTask GetTokenAsync(TokenRequestContext requestContext, CancellationToken cancellationToken) - { - return new ValueTask(GetToken(requestContext, cancellationToken)); - } - - public override AccessToken GetToken(TokenRequestContext requestContext, CancellationToken cancellationToken) - { - return new AccessToken("TEST TOKEN " + string.Join(" ", requestContext.Scopes), DateTimeOffset.MaxValue); - } - } } }