test: 增加健康检查单元测试
This commit is contained in:
parent
04b146369f
commit
f7c9a638ed
|
@ -3,6 +3,7 @@ using Microsoft.Extensions.Diagnostics.HealthChecks;
|
||||||
using System;
|
using System;
|
||||||
using System.Collections.Generic;
|
using System.Collections.Generic;
|
||||||
using System.Diagnostics;
|
using System.Diagnostics;
|
||||||
|
using System.Linq;
|
||||||
using System.Net.Http;
|
using System.Net.Http;
|
||||||
using System.Threading;
|
using System.Threading;
|
||||||
using System.Threading.Tasks;
|
using System.Threading.Tasks;
|
||||||
|
@ -23,7 +24,7 @@ namespace Bootstrap.Admin.HealthChecks
|
||||||
public GiteeHttpHealthCheck(IHttpClientFactory factory, IHttpContextAccessor accessor)
|
public GiteeHttpHealthCheck(IHttpClientFactory factory, IHttpContextAccessor accessor)
|
||||||
{
|
{
|
||||||
_client = factory.CreateClient();
|
_client = factory.CreateClient();
|
||||||
_client.BaseAddress = new Uri($"{accessor.HttpContext.Request.Scheme}://{accessor.HttpContext.Request.Host}/{accessor.HttpContext.Request.PathBase}");
|
_client.BaseAddress = new Uri($"{accessor.HttpContext.Request.Scheme}://{accessor.HttpContext.Request.Host}{accessor.HttpContext.Request.PathBase}");
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
|
@ -32,21 +33,30 @@ namespace Bootstrap.Admin.HealthChecks
|
||||||
/// <param name="context"></param>
|
/// <param name="context"></param>
|
||||||
/// <param name="cancellationToken"></param>
|
/// <param name="cancellationToken"></param>
|
||||||
/// <returns></returns>
|
/// <returns></returns>
|
||||||
public async Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context, CancellationToken cancellationToken = default)
|
public Task<HealthCheckResult> CheckHealthAsync(HealthCheckContext context, CancellationToken cancellationToken = default)
|
||||||
{
|
{
|
||||||
var urls = new string[] { "Issues", "Pulls", "Releases", "Builds" };
|
var urls = new string[] { "Issues", "Pulls", "Releases", "Builds" };
|
||||||
var data = new Dictionary<string, object>();
|
var data = new Dictionary<string, object>();
|
||||||
|
|
||||||
var sw = Stopwatch.StartNew();
|
urls.ToList().ForEach(url =>
|
||||||
foreach (var url in urls)
|
|
||||||
{
|
{
|
||||||
sw.Restart();
|
var sw = Stopwatch.StartNew();
|
||||||
await _client.GetStringAsync($"api/Gitee/{url}").ConfigureAwait(false);
|
try
|
||||||
sw.Stop();
|
{
|
||||||
data.Add(url, sw.Elapsed);
|
var task = _client.GetStringAsync($"/api/Gitee/{url}");
|
||||||
}
|
task.Wait(cancellationToken);
|
||||||
|
}
|
||||||
|
catch (Exception)
|
||||||
|
{
|
||||||
|
|
||||||
return HealthCheckResult.Healthy("Ok", data);
|
}
|
||||||
|
finally
|
||||||
|
{
|
||||||
|
sw.Stop();
|
||||||
|
data.Add(url, sw.Elapsed);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return Task.FromResult(HealthCheckResult.Healthy("Ok", data));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,23 @@
|
||||||
|
using Xunit;
|
||||||
|
|
||||||
|
namespace Bootstrap.Admin.Api
|
||||||
|
{
|
||||||
|
public class HealthCheckTest : ControllerTest
|
||||||
|
{
|
||||||
|
public HealthCheckTest(BAWebHost factory) : base(factory, "") { }
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public async void Get_Ok()
|
||||||
|
{
|
||||||
|
var cates = await Client.GetAsJsonAsync<object>("/Healths");
|
||||||
|
Assert.NotNull(cates);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public async void UI_Ok()
|
||||||
|
{
|
||||||
|
var cates = await Client.GetStringAsync("/Healths-ui");
|
||||||
|
Assert.Contains("健康检查", cates);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue