This commit is contained in:
melekhin
2026-06-09 16:45:22 +07:00
parent 39ee5bdddf
commit 6765aa23b1
22 changed files with 2493 additions and 1328 deletions

View File

@@ -1,6 +1,5 @@
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.Text;
using System.Collections.Generic;
using System.Collections.Immutable;
using System.Linq;
using System.Text;
@@ -10,9 +9,6 @@ namespace G;
[Generator]
public class TestsGenerator : IIncrementalGenerator
{
private const string UnitAttributeFullName = "QWERTYkez.Mensura.UnitGeneratorAttribute";
private const string ComplexAttributeFullName = "QWERTYkez.Mensura.ComplexUnitGeneratorAttribute";
public void Initialize(IncrementalGeneratorInitializationContext context)
{
var compilationProvider = context.CompilationProvider;
@@ -24,116 +20,575 @@ public class TestsGenerator : IIncrementalGenerator
var mensuraAssembly = compilation.References
.Select(r => compilation.GetAssemblyOrModuleSymbol(r) as IAssemblySymbol)
.FirstOrDefault(a => a?.Name == "QWERTYkez.Mensura" || a?.Name?.EndsWith(".Mensura") == true);
.FirstOrDefault(a => a?.Name == "QWERTYkez.Mensura");
if (mensuraAssembly == null)
return;
var targetTypes = new List<INamedTypeSymbol>();
CollectTypesWithAttribute(mensuraAssembly.GlobalNamespace, targetTypes, UnitAttributeFullName);
CollectTypesWithAttribute(mensuraAssembly.GlobalNamespace, targetTypes, ComplexAttributeFullName);
var unitTypes = CollectAllUnitTypes(mensuraAssembly.GlobalNamespace);
if (unitTypes.Count == 0) return;
if (targetTypes.Count == 0)
return;
// 1. Сериализация
GenerateSerializationTests(spc, unitTypes);
foreach (var type in targetTypes)
{
var unitProperties = GetStaticProperties(type);
if (unitProperties.Length == 0) continue;
var firstNonBase = unitProperties.FirstOrDefault(p => !p.Name.StartsWith("_"));
if (firstNonBase.Name == null) continue;
// 2. Операторы (с атрибутом)
var operators = CollectOperatorsFromUnitTypes(unitTypes);
if (operators.Length > 0)
GenerateOperatorTests(spc, operators);
// Генерируем два файла
GenerateSerializationTest(spc, type.Name, firstNonBase.Name, "SystemText", "System.Text.Json");
GenerateSerializationTest(spc, type.Name, firstNonBase.Name, "Newtonsoft", "Newtonsoft.Json");
}
// 3. Коллекционные методы
GenerateCollectionTests(spc, unitTypes);
});
}
private static void GenerateSerializationTest(SourceProductionContext spc, string typeName, string unitName, string framework, string namespaceName)
// ========== 1. СЕРИАЛИЗАЦИЯ ==========
private static void GenerateSerializationTests(SourceProductionContext spc, List<INamedTypeSymbol> types)
{
var sb = new StringBuilder();
sb.AppendLine("// <auto-generated/>");
sb.AppendLine("#pragma warning disable");
sb.AppendLine();
sb.AppendLine("using Xunit;");
sb.AppendLine($"using {namespaceName};");
sb.AppendLine("using System.Text.Json;");
sb.AppendLine("using Newtonsoft.Json;");
sb.AppendLine("using QWERTYkez.Mensura.Units;");
sb.AppendLine();
sb.AppendLine($"namespace QWERTYkez.Mensura.Tests");
sb.AppendLine("namespace QWERTYkez.Mensura.Tests");
sb.AppendLine("{");
sb.AppendLine($" public class {typeName}Serialization_{framework}Tests");
sb.AppendLine(" public class SerializationTests");
sb.AppendLine(" {");
sb.AppendLine($" private const double Tolerance = 1e-12;");
sb.AppendLine($" private readonly {typeName} _testValue = {typeName}.{unitName};");
sb.AppendLine(" private const double Tolerance = 1e-12;");
sb.AppendLine();
// Тест
sb.AppendLine(" [Fact]");
sb.AppendLine($" public void {framework}_SerializeDeserialize_ReturnsEqualValue()");
sb.AppendLine(" {");
if (framework == "SystemText")
foreach (var type in types)
{
sb.AppendLine(" var json = System.Text.Json.JsonSerializer.Serialize(_testValue);");
var baseProp = GetBaseUnitProperty(type);
if (baseProp == null) continue;
var typeName = type.Name;
var propName = baseProp.Name;
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_SystemTextJson_SerializeDeserialize()");
sb.AppendLine($" {{");
sb.AppendLine($" var original = {typeName}.{propName};");
sb.AppendLine($" var json = System.Text.Json.JsonSerializer.Serialize(original);");
sb.AppendLine($" var deserialized = System.Text.Json.JsonSerializer.Deserialize<{typeName}>(json);");
}
else
{
sb.AppendLine(" var json = Newtonsoft.Json.JsonConvert.SerializeObject(_testValue);");
sb.AppendLine($" Assert.Equal((double)original, (double)deserialized, Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_NewtonsoftJson_SerializeDeserialize()");
sb.AppendLine($" {{");
sb.AppendLine($" var original = {typeName}.{propName};");
sb.AppendLine($" var json = Newtonsoft.Json.JsonConvert.SerializeObject(original);");
sb.AppendLine($" var deserialized = Newtonsoft.Json.JsonConvert.DeserializeObject<{typeName}>(json);");
sb.AppendLine($" Assert.Equal((double)original, (double)deserialized, Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
}
sb.AppendLine(" Assert.Equal((double)_testValue, (double)deserialized, Tolerance);");
sb.AppendLine(" }");
sb.AppendLine();
sb.AppendLine(" }");
sb.AppendLine("}");
spc.AddSource($"Serialize.{framework}.{typeName}.g.cs", SourceText.From(sb.ToString(), Encoding.UTF8));
spc.AddSource("SerializationTests.g.cs", SourceText.From(sb.ToString(), Encoding.UTF8));
}
private static void CollectTypesWithAttribute(INamespaceSymbol ns, List<INamedTypeSymbol> results, string attributeFullName)
// ========== 2. ОПЕРАТОРЫ (только с атрибутом) ==========
private static ImmutableArray<OperatorInfo> CollectOperatorsFromUnitTypes(List<INamedTypeSymbol> unitTypes)
{
foreach (var type in ns.GetTypeMembers())
var result = new List<OperatorInfo>();
foreach (var containingType in unitTypes)
{
if (HasAttribute(type, attributeFullName))
results.Add(type);
}
foreach (var childNs in ns.GetNamespaceMembers())
{
CollectTypesWithAttribute(childNs, results, attributeFullName);
}
}
private static bool HasAttribute(INamedTypeSymbol type, string attributeFullName)
{
foreach (var attr in type.GetAttributes())
{
if (attr.AttributeClass?.ToString() == attributeFullName)
return true;
}
return false;
}
private static ImmutableArray<UnitProperty> GetStaticProperties(INamedTypeSymbol type)
{
var props = new List<UnitProperty>();
foreach (var member in type.GetMembers())
{
if (member is IPropertySymbol prop && prop.IsStatic && prop.DeclaredAccessibility == Accessibility.Public)
foreach (var member in containingType.GetMembers())
{
var propName = prop.Name;
if (propName == "EqualityContract" || propName.StartsWith("<"))
if (member is not IMethodSymbol method || method.MethodKind != MethodKind.UserDefinedOperator)
continue;
props.Add(new UnitProperty(propName, prop.Type.ToString()));
bool hasAttr = method.GetAttributes().Any(attr =>
attr.AttributeClass?.Name == "OperatorsGeneratorAttribute" ||
attr.AttributeClass?.ToString() == "QWERTYkez.Mensura.OperatorsGeneratorAttribute");
if (!hasAttr) continue;
string opSymbol = method.Name switch
{
"op_Multiply" => "*",
"op_Division" => "/",
_ => null
};
if (opSymbol == null) continue;
var pars = method.Parameters;
if (pars.Length != 2) continue;
var leftType = pars[0].Type as INamedTypeSymbol;
var rightType = pars[1].Type as INamedTypeSymbol;
var returnType = method.ReturnType as INamedTypeSymbol;
if (leftType == null || rightType == null || returnType == null) continue;
if (!unitTypes.Contains(leftType, SymbolEqualityComparer.Default) ||
!unitTypes.Contains(rightType, SymbolEqualityComparer.Default) ||
!unitTypes.Contains(returnType, SymbolEqualityComparer.Default))
continue;
string? coeffField = null;
foreach (var attr in method.GetAttributes())
{
if (attr.AttributeClass?.Name == "OperatorsGeneratorAttribute" ||
attr.AttributeClass?.ToString() == "QWERTYkez.Mensura.OperatorsGeneratorAttribute")
{
foreach (var arg in attr.ConstructorArguments)
{
if (arg.Value is string s)
{
coeffField = s;
break;
}
}
break;
}
}
result.Add(new OperatorInfo(opSymbol, leftType, rightType, returnType, coeffField, containingType));
}
}
return props.ToImmutableArray();
return result.ToImmutableArray();
}
private readonly struct UnitProperty
private static void GenerateOperatorTests(SourceProductionContext spc, ImmutableArray<OperatorInfo> operators)
{
public string Name { get; }
public string Type { get; }
public UnitProperty(string name, string type) => (Name, Type) = (name, type);
var sb = new StringBuilder();
sb.AppendLine("// <auto-generated/>");
sb.AppendLine("#pragma warning disable");
sb.AppendLine();
sb.AppendLine("using Xunit;");
sb.AppendLine("using QWERTYkez.Mensura.Units;");
sb.AppendLine("using System.Reflection;");
sb.AppendLine();
sb.AppendLine("namespace QWERTYkez.Mensura.Tests");
sb.AppendLine("{");
sb.AppendLine(" public class OperatorsTests");
sb.AppendLine(" {");
sb.AppendLine(" private const double Tolerance = 1e-12;");
sb.AppendLine();
foreach (var op in operators)
{
var leftBase = GetBaseUnitProperty(op.LeftType);
var rightBase = GetBaseUnitProperty(op.RightType);
if (leftBase == null || rightBase == null) continue;
string leftTypeName = op.LeftType.Name;
string rightTypeName = op.RightType.Name;
string returnTypeName = op.ReturnType.Name;
string leftProp = leftBase.Name;
string rightProp = rightBase.Name;
string testName = $"{op.OperatorSymbol}_{leftTypeName}_{rightTypeName}_Returns_{returnTypeName}"
.Replace("<", "_").Replace(">", "_").Replace(",", "_").Replace(" ", "")
.Replace("*", "Mul").Replace("/", "Div");
string expectedExpr;
if (op.CoefficientField != null)
{
expectedExpr = $@"(double)left {op.OperatorSymbol} (double)right * " +
$@"(double)typeof({op.ContainingType.Name}).GetField(""{op.CoefficientField}"", " +
$@"BindingFlags.NonPublic | BindingFlags.Static).GetValue(null)";
}
else
{
expectedExpr = $"(double)left {op.OperatorSymbol} (double)right";
}
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {testName}()");
sb.AppendLine($" {{");
sb.AppendLine($" var left = {leftTypeName}.{leftProp};");
sb.AppendLine($" var right = {rightTypeName}.{rightProp};");
sb.AppendLine($" var result = left {op.OperatorSymbol} right;");
sb.AppendLine($" var expected = {expectedExpr};");
sb.AppendLine($" Assert.Equal(expected, (double)result, Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
}
if (operators.Length == 0)
sb.AppendLine(" // No operators with [OperatorsGenerator] found");
sb.AppendLine(" }");
sb.AppendLine("}");
spc.AddSource("OperatorsTests.g.cs", SourceText.From(sb.ToString(), Encoding.UTF8));
}
// ========== 3. КОЛЛЕКЦИОННЫЕ МЕТОДЫ ==========
private static void GenerateCollectionTests(SourceProductionContext spc, List<INamedTypeSymbol> types)
{
var sb = new StringBuilder();
sb.AppendLine("// <auto-generated/>");
sb.AppendLine("#pragma warning disable");
sb.AppendLine();
sb.AppendLine("using Xunit;");
sb.AppendLine("using QWERTYkez.Mensura.Units;");
sb.AppendLine("using QWERTYkez.Mensura.Extensions;");
sb.AppendLine("using System.Collections.Generic;");
sb.AppendLine("using System.Linq;");
sb.AppendLine();
sb.AppendLine("namespace QWERTYkez.Mensura.Tests");
sb.AppendLine("{");
sb.AppendLine(" public class CollectionTests");
sb.AppendLine(" {");
sb.AppendLine(" private const double Tolerance = 1e-12;");
sb.AppendLine();
foreach (var type in types)
{
var baseProp = GetBaseUnitProperty(type);
if (baseProp == null) continue;
string typeName = type.Name;
string propName = baseProp.Name;
// Создаём массив из двух тестовых значений
sb.AppendLine($" private {typeName}[] GetTestArray() => new[] {{ {typeName}.{propName}, {typeName}.{propName} }};");
sb.AppendLine();
// ========== Multiply ==========
// scalar * array
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Multiply_ScalarByArray_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var arr = GetTestArray();");
sb.AppendLine($" var result = scalar.Multiply(arr);");
sb.AppendLine($" Assert.Equal(2, result.Length);");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// array * scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Multiply_ArrayByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var arr = GetTestArray();");
sb.AppendLine($" var result = arr.Multiply(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Length);");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// scalar * List
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Multiply_ScalarByList_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var list = GetTestArray().ToList();");
sb.AppendLine($" var result = scalar.Multiply(list);");
sb.AppendLine($" Assert.Equal(2, result.Count);");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// List * scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Multiply_ListByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var list = GetTestArray().ToList();");
sb.AppendLine($" var result = list.Multiply(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Count);");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// scalar * IEnumerable
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Multiply_ScalarByEnumerable_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var enumerable = GetTestArray().AsEnumerable();");
sb.AppendLine($" var result = scalar.Multiply(enumerable);");
sb.AppendLine($" Assert.Equal(2, result.Count());");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result.ElementAt(0), Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result.ElementAt(1), Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// IEnumerable * scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Multiply_EnumerableByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var enumerable = GetTestArray().AsEnumerable();");
sb.AppendLine($" var result = enumerable.Multiply(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Count());");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result.ElementAt(0), Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar * (double)scalar, (double)result.ElementAt(1), Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// ========== Divide ==========
// scalar / array
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Divide_ScalarByArray_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var arr = GetTestArray();");
sb.AppendLine($" var result = scalar.Divide(arr);");
sb.AppendLine($" Assert.Equal(2, result.Length);");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// array / scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Divide_ArrayByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var arr = GetTestArray();");
sb.AppendLine($" var result = arr.Divide(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Length);");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// scalar / List
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Divide_ScalarByList_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var list = GetTestArray().ToList();");
sb.AppendLine($" var result = scalar.Divide(list);");
sb.AppendLine($" Assert.Equal(2, result.Count);");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// List / scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Divide_ListByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var list = GetTestArray().ToList();");
sb.AppendLine($" var result = list.Divide(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Count);");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// scalar / IEnumerable
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Divide_ScalarByEnumerable_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var enumerable = GetTestArray().AsEnumerable();");
sb.AppendLine($" var result = scalar.Divide(enumerable);");
sb.AppendLine($" Assert.Equal(2, result.Count());");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result.ElementAt(0), Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result.ElementAt(1), Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// IEnumerable / scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Divide_EnumerableByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var enumerable = GetTestArray().AsEnumerable();");
sb.AppendLine($" var result = enumerable.Divide(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Count());");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result.ElementAt(0), Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar / (double)scalar, (double)result.ElementAt(1), Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// ========== Plus ==========
// scalar + array
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Plus_ScalarByArray_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var arr = GetTestArray();");
sb.AppendLine($" var result = scalar.Plus(arr);");
sb.AppendLine($" Assert.Equal(2, result.Length);");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// array + scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Plus_ArrayByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var arr = GetTestArray();");
sb.AppendLine($" var result = arr.Plus(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Length);");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// scalar + List
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Plus_ScalarByList_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var list = GetTestArray().ToList();");
sb.AppendLine($" var result = scalar.Plus(list);");
sb.AppendLine($" Assert.Equal(2, result.Count);");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// List + scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Plus_ListByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var list = GetTestArray().ToList();");
sb.AppendLine($" var result = list.Plus(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Count);");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// scalar + IEnumerable
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Plus_ScalarByEnumerable_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var enumerable = GetTestArray().AsEnumerable();");
sb.AppendLine($" var result = scalar.Plus(enumerable);");
sb.AppendLine($" Assert.Equal(2, result.Count());");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result.ElementAt(0), Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result.ElementAt(1), Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// IEnumerable + scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Plus_EnumerableByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var enumerable = GetTestArray().AsEnumerable();");
sb.AppendLine($" var result = enumerable.Plus(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Count());");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result.ElementAt(0), Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar + (double)scalar, (double)result.ElementAt(1), Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// ========== Minus ==========
// scalar - array
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Minus_ScalarByArray_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var arr = GetTestArray();");
sb.AppendLine($" var result = scalar.Minus(arr);");
sb.AppendLine($" Assert.Equal(2, result.Length);");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// array - scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Minus_ArrayByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var arr = GetTestArray();");
sb.AppendLine($" var result = arr.Minus(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Length);");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// scalar - List
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Minus_ScalarByList_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var list = GetTestArray().ToList();");
sb.AppendLine($" var result = scalar.Minus(list);");
sb.AppendLine($" Assert.Equal(2, result.Count);");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// List - scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Minus_ListByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var list = GetTestArray().ToList();");
sb.AppendLine($" var result = list.Minus(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Count);");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result[0], Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result[1], Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// scalar - IEnumerable
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Minus_ScalarByEnumerable_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var enumerable = GetTestArray().AsEnumerable();");
sb.AppendLine($" var result = scalar.Minus(enumerable);");
sb.AppendLine($" Assert.Equal(2, result.Count());");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result.ElementAt(0), Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result.ElementAt(1), Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
// IEnumerable - scalar
sb.AppendLine($" [Fact]");
sb.AppendLine($" public void {typeName}_Minus_EnumerableByScalar_Works()");
sb.AppendLine($" {{");
sb.AppendLine($" var scalar = {typeName}.{propName};");
sb.AppendLine($" var enumerable = GetTestArray().AsEnumerable();");
sb.AppendLine($" var result = enumerable.Minus(scalar);");
sb.AppendLine($" Assert.Equal(2, result.Count());");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result.ElementAt(0), Tolerance);");
sb.AppendLine($" Assert.Equal((double)scalar - (double)scalar, (double)result.ElementAt(1), Tolerance);");
sb.AppendLine($" }}");
sb.AppendLine();
}
sb.AppendLine(" }");
sb.AppendLine("}");
spc.AddSource("CollectionTests.g.cs", SourceText.From(sb.ToString(), Encoding.UTF8));
}
// ========== ВСПОМОГАТЕЛЬНЫЕ МЕТОДЫ ==========
private static List<INamedTypeSymbol> CollectAllUnitTypes(INamespaceSymbol root)
{
var result = new List<INamedTypeSymbol>();
CollectTypesRecursive(root, result, "QWERTYkez.Mensura.Units");
return result;
}
private static void CollectTypesRecursive(INamespaceSymbol ns, List<INamedTypeSymbol> result, string targetNamespacePrefix)
{
var nsFullName = ns.ToString();
if (nsFullName.StartsWith(targetNamespacePrefix))
{
foreach (var type in ns.GetTypeMembers())
result.Add(type);
}
foreach (var childNs in ns.GetNamespaceMembers())
CollectTypesRecursive(childNs, result, targetNamespacePrefix);
}
private static IPropertySymbol? GetBaseUnitProperty(ITypeSymbol type)
{
foreach (var member in type.GetMembers())
{
if (member is IPropertySymbol prop && prop.IsStatic && prop.Name.StartsWith("_") && SymbolEqualityComparer.Default.Equals(prop.Type, type))
return prop;
}
return null;
}
private readonly struct OperatorInfo(string symbol, ITypeSymbol left, ITypeSymbol right, ITypeSymbol ret, string? coeff, ITypeSymbol containing)
{
public string OperatorSymbol { get; } = symbol;
public ITypeSymbol LeftType { get; } = left;
public ITypeSymbol RightType { get; } = right;
public ITypeSymbol ReturnType { get; } = ret;
public string? CoefficientField { get; } = coeff;
public ITypeSymbol ContainingType { get; } = containing;
}
}