From 05ccb4d8f1bbe3fb64e117136b4a89bcfb0b0b33 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Tue, 27 Oct 2020 19:21:35 +0800
Subject: Split front and back end.

---
 .../IntegratedTests/IntegratedTestBase.cs          | 164 +++++++++++++++++++++
 1 file changed, 164 insertions(+)
 create mode 100644 BackEnd/Timeline.Tests/IntegratedTests/IntegratedTestBase.cs

(limited to 'BackEnd/Timeline.Tests/IntegratedTests/IntegratedTestBase.cs')

diff --git a/BackEnd/Timeline.Tests/IntegratedTests/IntegratedTestBase.cs b/BackEnd/Timeline.Tests/IntegratedTests/IntegratedTestBase.cs
new file mode 100644
index 00000000..7cf27297
--- /dev/null
+++ b/BackEnd/Timeline.Tests/IntegratedTests/IntegratedTestBase.cs
@@ -0,0 +1,164 @@
+using FluentAssertions;
+using Microsoft.AspNetCore.TestHost;
+using Microsoft.Extensions.DependencyInjection;
+using System;
+using System.Collections.Generic;
+using System.Net.Http;
+using System.Text.Json;
+using System.Text.Json.Serialization;
+using System.Threading.Tasks;
+using Timeline.Models;
+using Timeline.Models.Converters;
+using Timeline.Models.Http;
+using Timeline.Services;
+using Timeline.Tests.Helpers;
+using Xunit;
+
+namespace Timeline.Tests.IntegratedTests
+{
+    public abstract class IntegratedTestBase : IAsyncLifetime
+    {
+        protected TestApplication TestApp { get; }
+
+        public IReadOnlyList<UserInfo> UserInfos { get; private set; }
+
+        private readonly int _userCount;
+
+        public IntegratedTestBase() : this(1)
+        {
+
+        }
+
+        public IntegratedTestBase(int userCount)
+        {
+            if (userCount < 0)
+                throw new ArgumentOutOfRangeException(nameof(userCount), userCount, "User count can't be negative.");
+
+            _userCount = userCount;
+
+            TestApp = new TestApplication();
+        }
+
+        protected virtual Task OnInitializeAsync()
+        {
+            return Task.CompletedTask;
+        }
+
+        protected virtual Task OnDisposeAsync()
+        {
+            return Task.CompletedTask;
+        }
+
+        protected virtual void OnDispose()
+        {
+
+        }
+
+        public async Task InitializeAsync()
+        {
+            await TestApp.InitializeAsync();
+
+            using (var scope = TestApp.Host.Services.CreateScope())
+            {
+                var users = new List<User>()
+                {
+                    new User
+                    {
+                        Username = "admin",
+                        Password = "adminpw",
+                        Administrator = true,
+                        Nickname = "administrator"
+                    }
+                };
+
+                for (int i = 1; i <= _userCount; i++)
+                {
+                    users.Add(new User
+                    {
+                        Username = $"user{i}",
+                        Password = $"user{i}pw",
+                        Administrator = false,
+                        Nickname = $"imuser{i}"
+                    });
+                }
+
+                var userInfoList = new List<UserInfo>();
+
+                var userService = scope.ServiceProvider.GetRequiredService<IUserService>();
+                foreach (var user in users)
+                {
+                    await userService.CreateUser(user);
+                }
+
+                using var client = await CreateDefaultClient();
+                var options = new JsonSerializerOptions
+                {
+                    PropertyNamingPolicy = JsonNamingPolicy.CamelCase
+                };
+                options.Converters.Add(new JsonStringEnumConverter());
+                options.Converters.Add(new JsonDateTimeConverter());
+                foreach (var user in users)
+                {
+                    var s = await client.GetStringAsync($"users/{user.Username}");
+                    userInfoList.Add(JsonSerializer.Deserialize<UserInfo>(s, options));
+                }
+
+                UserInfos = userInfoList;
+            }
+
+            await OnInitializeAsync();
+        }
+
+        public async Task DisposeAsync()
+        {
+            await OnDisposeAsync();
+            OnDispose();
+            await TestApp.DisposeAsync();
+        }
+
+        public Task<HttpClient> CreateDefaultClient(bool setApiBase = true)
+        {
+            var client = TestApp.Host.GetTestServer().CreateClient();
+            if (setApiBase)
+            {
+                client.BaseAddress = new Uri(client.BaseAddress, "api/");
+            }
+            return Task.FromResult(client);
+        }
+
+        public async Task<HttpClient> CreateClientWithCredential(string username, string password, bool setApiBase = true)
+        {
+            var client = TestApp.Host.GetTestServer().CreateClient();
+            if (setApiBase)
+            {
+                client.BaseAddress = new Uri(client.BaseAddress, "api/");
+            }
+            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(int userNumber, bool setApiBase = true)
+        {
+            if (userNumber < 0)
+                return CreateDefaultClient(setApiBase);
+            if (userNumber == 0)
+                return CreateClientWithCredential("admin", "adminpw", setApiBase);
+            else
+                return CreateClientWithCredential($"user{userNumber}", $"user{userNumber}pw", setApiBase);
+        }
+
+        public Task<HttpClient> CreateClientAsAdministrator(bool setApiBase = true)
+        {
+            return CreateClientAs(0, setApiBase);
+        }
+
+        public Task<HttpClient> CreateClientAsUser(bool setApiBase = true)
+        {
+            return CreateClientAs(1, setApiBase);
+        }
+    }
+}
-- 
cgit v1.2.3