114 lines
4.0 KiB
C#
114 lines
4.0 KiB
C#
// Copyright (c) Argo Zhang (argo@163.com). All rights reserved.
|
|
// Licensed under the LGPL License, Version 3.0. See License.txt in the project root for license information.
|
|
// Website: https://admin.blazor.zone
|
|
|
|
using Bootstrap.DataAccess;
|
|
using Longbow.Data;
|
|
using Microsoft.AspNetCore.Hosting;
|
|
using Microsoft.Extensions.Configuration;
|
|
using PetaPoco;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
|
|
namespace Bootstrap.Admin
|
|
{
|
|
public static class TestHelper
|
|
{
|
|
/// <summary>
|
|
/// 获得当前工程解决方案目录
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public static string RetrieveSolutionPath()
|
|
{
|
|
var dirSeparator = Path.DirectorySeparatorChar;
|
|
var paths = AppContext.BaseDirectory.SpanSplit($"{dirSeparator}.vs{dirSeparator}");
|
|
return paths.Count > 1 ? paths[0] : Path.Combine(AppContext.BaseDirectory, $"..{dirSeparator}..{dirSeparator}..{dirSeparator}..{dirSeparator}");
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <param name="folder"></param>
|
|
/// <returns></returns>
|
|
public static string RetrievePath(string folder)
|
|
{
|
|
var soluFolder = RetrieveSolutionPath();
|
|
return Path.Combine(soluFolder, folder);
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <param name="callback"></param>
|
|
public static void RevokeMapper(Action callback)
|
|
{
|
|
var t = typeof(App);
|
|
var map = Mappers.GetMapper(t, null);
|
|
Mappers.Revoke(map);
|
|
|
|
var foo = new FooMapper();
|
|
Mappers.Register(t.Assembly, foo);
|
|
try { callback(); }
|
|
catch (Exception) { throw; }
|
|
finally
|
|
{
|
|
Mappers.Revoke(foo);
|
|
Mappers.Register(t.Assembly, map);
|
|
}
|
|
}
|
|
|
|
public static void RevokePocoMapper<T>(Action callback)
|
|
{
|
|
var foo = new FooMapper();
|
|
Mappers.Register(typeof(T), foo);
|
|
try { callback(); }
|
|
catch (Exception) { throw; }
|
|
finally
|
|
{
|
|
Mappers.Revoke(foo);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <param name="builder"></param>
|
|
/// <param name="providerName"></param>
|
|
public static void ConfigureDatabase(this IWebHostBuilder builder, DatabaseProviderType providerName = DatabaseProviderType.SQLite)
|
|
{
|
|
if (providerName == DatabaseProviderType.SqlServer)
|
|
{
|
|
builder.ConfigureAppConfiguration(app => app.AddInMemoryCollection(new KeyValuePair<string, string>[] {
|
|
new KeyValuePair<string, string>("DB:0:Enabled", "false"),
|
|
new KeyValuePair<string, string>("DB:1:Enabled", "true")
|
|
}));
|
|
}
|
|
|
|
if (providerName == DatabaseProviderType.MySql)
|
|
{
|
|
builder.ConfigureAppConfiguration(app => app.AddInMemoryCollection(new KeyValuePair<string, string>[] {
|
|
new KeyValuePair<string, string>("DB:0:Enabled", "false"),
|
|
new KeyValuePair<string, string>("DB:1:Enabled", "false"),
|
|
new KeyValuePair<string, string>("DB:2:Enabled", "true")
|
|
}));
|
|
}
|
|
|
|
if (providerName == DatabaseProviderType.Npgsql)
|
|
{
|
|
builder.ConfigureAppConfiguration(app => app.AddInMemoryCollection(new KeyValuePair<string, string>[] {
|
|
new KeyValuePair<string, string>("DB:0:Enabled", "false"),
|
|
new KeyValuePair<string, string>("DB:1:Enabled", "false"),
|
|
new KeyValuePair<string, string>("DB:2:Enabled", "false"),
|
|
new KeyValuePair<string, string>("DB:3:Enabled", "true")
|
|
}));
|
|
}
|
|
}
|
|
|
|
private class FooMapper : ConventionMapper
|
|
{
|
|
public override TableInfo GetTableInfo(Type pocoType) => throw new Exception();
|
|
}
|
|
}
|
|
}
|