blob: 242a452d00aca301b8c4d1ee76bc21c94e508914 (
plain)
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
|
using Microsoft.AspNetCore.Mvc.Testing;
using System;
using System.Collections.Generic;
using System.Net.Http;
using System.Threading.Tasks;
using Timeline.Models.Http;
using Timeline.Tests.Helpers;
using Xunit;
namespace Timeline.Tests.IntegratedTests
{
public enum AuthType
{
None,
User,
Admin
}
public static class AuthTypeExtensions
{
public static MockUser GetMockUser(this AuthType authType)
{
return authType switch
{
AuthType.None => null,
AuthType.User => MockUser.User,
AuthType.Admin => MockUser.Admin,
_ => throw new InvalidOperationException("Unknown auth type.")
};
}
public static string GetUsername(this AuthType authType) => authType.GetMockUser().Username;
}
public abstract class IntegratedTestBase : IClassFixture<WebApplicationFactory<Startup>>, IDisposable
{
protected TestApplication TestApp { get; }
protected WebApplicationFactory<Startup> Factory => TestApp.Factory;
public IntegratedTestBase(WebApplicationFactory<Startup> factory)
{
TestApp = new TestApplication(factory);
}
protected virtual void OnDispose()
{
}
public void Dispose()
{
OnDispose();
TestApp.Dispose();
}
protected void CreateExtraMockUsers(int count)
{
TestApp.Database.CreateExtraMockUsers(count);
}
protected IReadOnlyList<MockUser> ExtraMockUsers => TestApp.Database.ExtraMockUsers;
public Task<HttpClient> CreateClientWithNoAuth()
{
return Task.FromResult(Factory.CreateDefaultClient());
}
public async Task<HttpClient> CreateClientWithCredential(string username, string password)
{
var client = Factory.CreateDefaultClient();
var response = await client.PostAsJsonAsync("/token/create",
new CreateTokenRequest { Username = username, Password = password });
var token = response.Should().HaveStatusCode(200)
.And.HaveJsonBody<CreateTokenResponse>().Which.Token;
client.DefaultRequestHeaders.Add("Authorization", "Bearer " + token);
return client;
}
public Task<HttpClient> CreateClientAs(MockUser user)
{
if (user == null)
return CreateClientWithNoAuth();
return CreateClientWithCredential(user.Username, user.Password);
}
public Task<HttpClient> CreateClientAs(AuthType authType) => CreateClientAs(authType.GetMockUser());
public Task<HttpClient> CreateClientAsUser() => CreateClientAs(MockUser.User);
public Task<HttpClient> CreateClientAsAdmin() => CreateClientAs(MockUser.Admin);
}
}
|
