-
Notifications
You must be signed in to change notification settings - Fork 4.8k
/
AzureKeyCredentialPolicyTests.cs
49 lines (42 loc) · 1.71 KB
/
AzureKeyCredentialPolicyTests.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
using System.Threading;
using System.Threading.Tasks;
using Azure.Core.Pipeline;
using Azure.Core.TestFramework;
using NUnit.Framework;
using NUnit.Framework.Internal;
namespace Azure.Core.Tests
{
public class AzureKeyCredentialPolicyTests : PolicyTestBase
{
[Test]
public async Task SetsKey()
{
string keyValue = "test_key";
string header = "api_key";
var transport = new MockTransport(new MockResponse(200));
var keyPolicy = new AzureKeyCredentialPolicy(new AzureKeyCredential(keyValue), header);
await SendGetRequest(transport, keyPolicy);
Assert.True(transport.SingleRequest.TryGetHeader(header, out var key));
Assert.AreEqual(key, keyValue);
}
[Test]
public async Task VerifyRetry()
{
string keyValue = "test_key";
string header = "api_key";
var transport = new MockTransport(new MockResponse(200), new MockResponse(200));
var keyPolicy = new AzureKeyCredentialPolicy(new AzureKeyCredential(keyValue), header);
using (Request request = transport.CreateRequest())
{
request.Method = RequestMethod.Get;
var pipeline = new HttpPipeline(transport, new[] { keyPolicy });
await pipeline.SendRequestAsync(request, CancellationToken.None);
await pipeline.SendRequestAsync(request, CancellationToken.None);
}
Assert.True(transport.Requests[0].TryGetHeader(header, out var key));
Assert.AreEqual(key, keyValue);
}
}
}