Hugging Face's logo

AbdulElahGwaith
/
google-api-dotnet-client

Model card Files
xet
Community
google-api-dotnet-client
File size: 8,069 Bytes
7b715bc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
 
/*
Copyright 2024 Google Inc

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

using Google.Apis.Auth.OAuth2;
using Google.Apis.Auth.Tests.OAuth2;
using Google.Apis.Http;
using System;
using System.Net;
using System.Net.Http;
using System.Threading.Tasks;
using Xunit;

namespace Google.Apis.Auth.Tests;

public class GoogleAuthConstsTests
{
    [Fact]
    public void OAuth2TokenEndpointRecommendedRetry()
    {
        var httpClient = new ConfigurableHttpClient(new ConfigurableMessageHandler(new DelegatedMessageHandler(request => Task.FromResult(new HttpResponseMessage()))));

        GoogleAuthConsts.OAuth2TokenEndpointRecommendedRetry.Initialize(httpClient);

        // MessageHandler no longer provides a supported way for clients to query the list of handlers,
        // but we rely on the obsolete property as an implementation detail here.
#pragma warning disable CS0618 // Type or member is obsolete
        var badResponseHandler = Assert.Single(httpClient.MessageHandler.UnsuccessfulResponseHandlers);
        var exceptionHandler = Assert.Single(httpClient.MessageHandler.ExceptionHandlers);
#pragma warning restore CS0618 // Type or member is obsolete

        var badResponseBackoffHandler = Assert.IsType<BackOffHandler>(badResponseHandler);
        var exceptionBackoffHandler = Assert.IsType<BackOffHandler>(exceptionHandler);
        Assert.Same(badResponseBackoffHandler, exceptionBackoffHandler);

        Assert.Equal(TimeSpan.MaxValue, badResponseBackoffHandler.MaxTimeSpan);

        // Check that it doesn't handle (some) exceptions.
        Assert.False(badResponseBackoffHandler.HandleExceptionFunc(new Exception()));
        Assert.False(badResponseBackoffHandler.HandleExceptionFunc(new HttpRequestException()));
        Assert.False(badResponseBackoffHandler.HandleExceptionFunc(new TaskCanceledException()));
        Assert.False(badResponseBackoffHandler.HandleExceptionFunc(new OperationCanceledException()));

        // Check that it handles the expected bad responses.
        Assert.True(badResponseBackoffHandler.HandleUnsuccessfulResponseFunc(new HttpResponseMessage(HttpStatusCode.RequestTimeout)));
        Assert.True(badResponseBackoffHandler.HandleUnsuccessfulResponseFunc(new HttpResponseMessage((HttpStatusCode) 429)));
        Assert.True(badResponseBackoffHandler.HandleUnsuccessfulResponseFunc(new HttpResponseMessage(HttpStatusCode.InternalServerError)));
        Assert.True(badResponseBackoffHandler.HandleUnsuccessfulResponseFunc(new HttpResponseMessage(HttpStatusCode.ServiceUnavailable)));
    }

    [Theory]
    [InlineData(ExponentialBackOffPolicy.None)]
    [InlineData(ExponentialBackOffPolicy.UnsuccessfulResponse503)]
    [InlineData(ExponentialBackOffPolicy.Exception)]
    [InlineData(ExponentialBackOffPolicy.UnsuccessfulResponse503 | ExponentialBackOffPolicy.Exception)]
    public void StripOAuth2TokenEndpointRecommendedPolicy_DoesNotStrip(ExponentialBackOffPolicy original)
    {
        var stripped = GoogleAuthConsts.StripOAuth2TokenEndpointRecommendedPolicy(original);
        Assert.Equal(original, stripped);
    }

    [Theory]
    [InlineData(ExponentialBackOffPolicy.RecommendedOrDefault, ExponentialBackOffPolicy.None)]
    [InlineData(ExponentialBackOffPolicy.RecommendedOrDefault | ExponentialBackOffPolicy.UnsuccessfulResponse503, ExponentialBackOffPolicy.None)]
    [InlineData(ExponentialBackOffPolicy.RecommendedOrDefault | ExponentialBackOffPolicy.Exception, ExponentialBackOffPolicy.Exception)]
    [InlineData(ExponentialBackOffPolicy.RecommendedOrDefault | ExponentialBackOffPolicy.UnsuccessfulResponse503 | ExponentialBackOffPolicy.Exception, ExponentialBackOffPolicy.Exception)]
    public void StripOAuth2TokenEndpointRecommendedPolicy_Strips(ExponentialBackOffPolicy original, ExponentialBackOffPolicy expected)
    {
        var stripped = GoogleAuthConsts.StripOAuth2TokenEndpointRecommendedPolicy(original);
        Assert.Equal(expected, stripped);
    }

    [Fact]
    public void IamSignBlobTokenEndpointRecommendedRetry()
    {
        var httpClient = new ConfigurableHttpClient(new ConfigurableMessageHandler(new DelegatedMessageHandler(request => Task.FromResult(new HttpResponseMessage()))));

        GoogleAuthConsts.IamSignBlobEndpointRecommendedRetry.Initialize(httpClient);

        // MessageHandler no longer provides a supported way for clients to query the list of handlers,
        // but we rely on the obsolete property as an implementation detail here.
#pragma warning disable CS0618 // Type or member is obsolete
        var badResponseHandler = Assert.Single(httpClient.MessageHandler.UnsuccessfulResponseHandlers);
        var exceptionHandler = Assert.Single(httpClient.MessageHandler.ExceptionHandlers);
#pragma warning restore CS0618 // Type or member is obsolete

        var badResponseBackoffHandler = Assert.IsType<BackOffHandler>(badResponseHandler);
        var exceptionBackoffHandler = Assert.IsType<BackOffHandler>(exceptionHandler);
        Assert.Same(badResponseBackoffHandler, exceptionBackoffHandler);

        Assert.Equal(TimeSpan.MaxValue, badResponseBackoffHandler.MaxTimeSpan);

        // Check that it doesn't handle (some) exceptions.
        Assert.False(badResponseBackoffHandler.HandleExceptionFunc(new Exception()));
        Assert.False(badResponseBackoffHandler.HandleExceptionFunc(new HttpRequestException()));
        Assert.False(badResponseBackoffHandler.HandleExceptionFunc(new TaskCanceledException()));
        Assert.False(badResponseBackoffHandler.HandleExceptionFunc(new OperationCanceledException()));

        // Check that it handles the expected bad responses.
        Assert.True(badResponseBackoffHandler.HandleUnsuccessfulResponseFunc(new HttpResponseMessage(HttpStatusCode.InternalServerError)));
        Assert.True(badResponseBackoffHandler.HandleUnsuccessfulResponseFunc(new HttpResponseMessage(HttpStatusCode.BadGateway)));
        Assert.True(badResponseBackoffHandler.HandleUnsuccessfulResponseFunc(new HttpResponseMessage(HttpStatusCode.ServiceUnavailable)));
        Assert.True(badResponseBackoffHandler.HandleUnsuccessfulResponseFunc(new HttpResponseMessage(HttpStatusCode.GatewayTimeout)));
    }

    [Theory]
    [InlineData(ExponentialBackOffPolicy.None)]
    [InlineData(ExponentialBackOffPolicy.UnsuccessfulResponse503)]
    [InlineData(ExponentialBackOffPolicy.Exception)]
    [InlineData(ExponentialBackOffPolicy.UnsuccessfulResponse503 | ExponentialBackOffPolicy.Exception)]
    public void StripIamSignBlobEndpointRecommendedPolicy_DoesNotStrip(ExponentialBackOffPolicy original)
    {
        var stripped = GoogleAuthConsts.StripIamSignBlobEndpointRecommendedPolicy(original);
        Assert.Equal(original, stripped);
    }

    [Theory]
    [InlineData(ExponentialBackOffPolicy.RecommendedOrDefault, ExponentialBackOffPolicy.None)]
    [InlineData(ExponentialBackOffPolicy.RecommendedOrDefault | ExponentialBackOffPolicy.UnsuccessfulResponse503, ExponentialBackOffPolicy.None)]
    [InlineData(ExponentialBackOffPolicy.RecommendedOrDefault | ExponentialBackOffPolicy.Exception, ExponentialBackOffPolicy.Exception)]
    [InlineData(ExponentialBackOffPolicy.RecommendedOrDefault | ExponentialBackOffPolicy.UnsuccessfulResponse503 | ExponentialBackOffPolicy.Exception, ExponentialBackOffPolicy.Exception)]
    public void StripIamSignBlobEndpointRecommendedPolicy_Strips(ExponentialBackOffPolicy original, ExponentialBackOffPolicy expected)
    {
        var stripped = GoogleAuthConsts.StripOAuth2TokenEndpointRecommendedPolicy(original);
        Assert.Equal(expected, stripped);
    }
}