mirror of https://github.com/dahall/Vanara.git
118 lines
4.6 KiB
C#
118 lines
4.6 KiB
C#
using Newtonsoft.Json;
|
|
using NUnit.Framework;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using Vanara.Extensions;
|
|
using Vanara.InteropServices;
|
|
|
|
namespace Vanara.PInvoke.Tests
|
|
{
|
|
public static class TestHelper
|
|
{
|
|
private const string testApp = @"C:\Users\dahall\Documents\Visual Studio 2017\Projects\TestSysConsumption\bin\Debug\netcoreapp3.0\TestSysConsumption.exe";
|
|
|
|
private static readonly Lazy<JsonSerializerSettings> jsonSet = new(() =>
|
|
new JsonSerializerSettings()
|
|
{
|
|
Converters = new JsonConverter[] { new Newtonsoft.Json.Converters.StringEnumConverter(), new SizeTConverter() },
|
|
ReferenceLoopHandling = ReferenceLoopHandling.Serialize
|
|
});
|
|
|
|
public static Process RunThrottleApp() => Process.Start(testApp);
|
|
|
|
public static void SetThrottle(string type, bool on)
|
|
{
|
|
using var evt = new EventWaitHandle(false, EventResetMode.AutoReset, (on ? "" : "End") + type);
|
|
evt.Set();
|
|
}
|
|
|
|
public static IList<string> GetNestedStructSizes(this Type type, params string[] filters) =>
|
|
type.GetNestedTypes(System.Reflection.BindingFlags.Public | System.Reflection.BindingFlags.NonPublic).GetStructSizes(false, filters);
|
|
|
|
public static IList<string> GetStructSizes(this Type[] types, bool fullName = false, params string[] filters)
|
|
{
|
|
System.Reflection.TypeAttributes attr = System.Reflection.TypeAttributes.SequentialLayout | System.Reflection.TypeAttributes.ExplicitLayout;
|
|
return types.Where(t => t.IsValueType && !t.IsEnum && !t.IsGenericType && (t.Attributes & attr) != 0 && ((filters?.Length ?? 0) == 0 || filters.Any(s => t.Name.Contains(s)))).
|
|
OrderBy(t => fullName ? t.FullName : t.Name).Select(t => $"{(fullName ? t.FullName : t.Name)} = {GetTypeSize(t)}").ToList();
|
|
|
|
static long GetTypeSize(Type t) { try { return (long)InteropExtensions.SizeOf(t); } catch { return -1; } }
|
|
}
|
|
|
|
public static void RunForEach<TEnum>(Type lib, string name, Func<TEnum, object[]> makeParam, Action<TEnum, object, object[]> action = null, Action<Exception> error = null) where TEnum : Enum =>
|
|
RunForEach(lib, name, makeParam, (e, ex) => error?.Invoke(ex), action);
|
|
|
|
public static void RunForEach<TEnum>(Type lib, string name, Func<TEnum, object[]> makeParam, Action<TEnum, Exception> error = null, Action<TEnum, object, object[]> action = null, CorrespondingAction? filter = null) where TEnum : Enum
|
|
{
|
|
System.Reflection.MethodInfo mi = lib.GetMethods(System.Reflection.BindingFlags.Public | System.Reflection.BindingFlags.Static).Where(m => m.IsGenericMethod && m.Name == name).First();
|
|
if (mi is null) throw new ArgumentException("Unable to find method.");
|
|
foreach (TEnum e in Enum.GetValues(typeof(TEnum)).Cast<TEnum>())
|
|
{
|
|
Type type = (filter.HasValue ? CorrespondingTypeAttribute.GetCorrespondingTypes(e, filter.Value) : CorrespondingTypeAttribute.GetCorrespondingTypes(e)).FirstOrDefault();
|
|
if (type is null)
|
|
{
|
|
TestContext.WriteLine($"No corresponding type found for {e}.");
|
|
continue;
|
|
}
|
|
System.Reflection.MethodInfo gmi = mi.MakeGenericMethod(type);
|
|
var param = makeParam(e);
|
|
try
|
|
{
|
|
var ret = gmi.Invoke(null, param);
|
|
action?.Invoke(e, ret, param);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
error?.Invoke(e, ex.InnerException);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static string GetStringVal(this object value)
|
|
{
|
|
switch (value)
|
|
{
|
|
case null:
|
|
return "(null)";
|
|
|
|
case System.Runtime.InteropServices.ComTypes.FILETIME ft:
|
|
value = ft.ToDateTime();
|
|
goto Simple;
|
|
|
|
case SYSTEMTIME st:
|
|
value = st.ToDateTime(DateTimeKind.Local);
|
|
goto Simple;
|
|
|
|
case DateTime:
|
|
case decimal:
|
|
case var v when v.GetType().IsPrimitive || v.GetType().IsEnum:
|
|
Simple:
|
|
return $"{value.GetType().Name} : [{value}]";
|
|
|
|
case string s:
|
|
return string.Concat("\"", s, "\"");
|
|
|
|
case byte[] bytes:
|
|
return string.Join(" ", Array.ConvertAll(bytes, b => $"{b:X2}"));
|
|
|
|
case SafeAllocatedMemoryHandleBase mem:
|
|
return mem.Dump;
|
|
|
|
default:
|
|
try { return JsonConvert.SerializeObject(value, Formatting.Indented, jsonSet.Value); }
|
|
catch (Exception e) { return e.ToString(); }
|
|
}
|
|
}
|
|
|
|
public static void WriteValues(this object value) => TestContext.WriteLine(GetStringVal(value));
|
|
|
|
private class SizeTConverter : JsonConverter<SizeT>
|
|
{
|
|
public override SizeT ReadJson(JsonReader reader, Type objectType, SizeT existingValue, bool hasExistingValue, JsonSerializer serializer) => reader.Value is ulong ul ? new SizeT(ul) : new SizeT(0);
|
|
|
|
public override void WriteJson(JsonWriter writer, SizeT value, JsonSerializer serializer) => writer.WriteValue(value.Value);
|
|
}
|
|
}
|
|
} |