WebApiClient/WebApiClient.Test/AuthTokens/AuthTokenFilterTest.cs at master · codesee/WebApiClient · GitHub

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
using System;
using System.Net.Http;
using System.Threading.Tasks;
using WebApiClient;
using WebApiClient.AuthTokens;
using WebApiClient.Contexts;
using Xunit;

namespace WebApiClient.Test.AuthTokens
{
    public class AuthTokenFilterTest
    {
        class TokenFilter : AuthTokenFilter
        {
            public bool IsRequestTokenResult { get; set; } = false;

            public bool IsRequestRefreshToken { get; set; } = false;

            public TokenResult TokenResult { get; set; }

            protected override Task<TokenResult> RequestTokenResultAsync()
            {
                this.IsRequestTokenResult = true;
                var token = new TokenResult
                {
                    IdToken = "0",
                    ExpiresIn = 1,
                    AccessToken = "AccessToken",
                    RefreshToken = "RefreshToken",
                    TokenType = "client_credentials"
                };
                return Task.FromResult(token);
            }

            protected override Task<TokenResult> RequestRefreshTokenAsync(string refresh_token)
            {
                this.IsRequestRefreshToken = true;
                var token = new TokenResult
                {
                    IdToken = "1",
                    ExpiresIn = 1,
                    AccessToken = "AccessToken",
                    RefreshToken = "RefreshToken",
                    TokenType = "client_credentials"
                };
                return Task.FromResult(token);
            }

            protected override void AccessTokenResult(ApiActionContext context, TokenResult tokenResult)
            {
                this.TokenResult = tokenResult;
            }
        }

        [Fact]
        public async Task TokenTestAsync()
        {
            var tokenFilter = new TokenFilter();
            IApiActionFilter filter = tokenFilter;

            await filter.OnBeginRequestAsync(null);
            Assert.True(tokenFilter.IsRequestTokenResult && tokenFilter.IsRequestRefreshToken == false && tokenFilter.TokenResult.IdToken == "0");
            await filter.OnEndRequestAsync(null);
            tokenFilter.IsRequestTokenResult = false;
            tokenFilter.IsRequestRefreshToken = false;

            await Task.Delay(500);
            await filter.OnBeginRequestAsync(null);
            Assert.True(tokenFilter.IsRequestTokenResult == false && tokenFilter.IsRequestRefreshToken == false && tokenFilter.TokenResult.IdToken == "0");
            await filter.OnEndRequestAsync(null);
            tokenFilter.IsRequestTokenResult = false;
            tokenFilter.IsRequestRefreshToken = false;


            await Task.Delay(1100);
            await filter.OnBeginRequestAsync(null);
            Assert.True(tokenFilter.IsRequestTokenResult == false && tokenFilter.IsRequestRefreshToken && tokenFilter.TokenResult.IdToken == "1");
            await filter.OnEndRequestAsync(null);
        }
    }
}