public virtual async Task DefaultEnumTypesProviderShouldReturnEnumTypes()
{
const string sourceCodes = @"
public interface IDto {
}
public class DtoController<TDto>
where TDto : IDto
{
}
public class DtoWithEnum : IDto
{
public virtual int Id { get; set; }
public virtual TestGender? Gender { get; set; }
public virtual string Test { get; set; }
}
public enum TestGender
{
Man = 3, Woman = 12, Other
}
public enum TestGender2
{
Man = 3, Woman = 12, Other
}
public class DtoWithEnumController : DtoController<DtoWithEnum>
{
public class GetDtoWithEnumsByGenderParameters
{
public virtual TestGender gender { get; set; }
}
[ActionAttribute]
public virtual int GetDtoWithEnumsByGender(GetDtoWithEnumsByGenderParameters parameters)
{
return 1;
}
public class GetDtoWithEnumsByGender2Parameters
{
public virtual TestGender2 gender { get; set; }
}
[ActionAttribute]
public virtual int GetDtoWithEnumsByGender2(GetDtoWithEnumsByGender2Parameters parameters)
{
return 1;
}
}
";
IProjectDtoControllersProvider projectDtoControllersProvider = new DefaultProjectDtoControllersProvider();
IProjectEnumTypesProvider projectEnumTypesProvider = new DefaultProjectEnumTypesProvider(projectDtoControllersProvider, new DefaultProjectDtosProvider(projectDtoControllersProvider));
Project proj = CreateProjectFromSourceCodes(sourceCodes);
IList <EnumType> result = await projectEnumTypesProvider.GetProjectEnumTypes(proj, new[] { proj });
Assert.IsTrue(result.Select(enumType => enumType.EnumTypeSymbol.Name).SequenceEqual(new[] { "TestGender", "TestGender2" }));
Assert.IsTrue(result.First().Members.SequenceEqual(new[]
{
new EnumMember {
Name = "Man", Value = 3, Index = 0
},
new EnumMember {
Name = "Woman", Value = 12, Index = 1
},
new EnumMember {
Name = "Other", Value = 13, Index = 2
}
}, new EnumMemberEqualityComparer()));
}
public virtual async Task NestedObjectsTest()
{
string sourceProjectCodes = @"
public interface IDto {
}
public class DtoController<TDto>
where TDto : IDto
{
}
public class TestComplexController : DtoController<TestComplexDto>
{
[FunctionAttribute]
public System.Threading.Tasks.Task<ComplexObj3[]> GetComplexObjects()
{
return null;
}
public class SomeActionArgs
{
public ComplexObj5 Test5 { get; set; }
public string Test { get; set; }
}
[ActionAttribute]
public void SomeAction(SomeActionArgs args)
{
}
}
public class TestComplexDto : IDto
{
public virtual int EntityId { get; set; }
public virtual ComplexObj ComplexObj { get; set; }
public virtual ComplexObj2 ComplexObj2 { get; set; }
}
[System.ComponentModel.DataAnnotations.Schema.ComplexType]
public class ComplexObj
{
public virtual string Name { get; set; }
}
[System.ComponentModel.DataAnnotations.Schema.ComplexType]
public class ComplexObj3
{
public virtual string Name { get; set; }
public virtual ComplexObj4 Obj4 { get; set; }
}
[System.ComponentModel.DataAnnotations.Schema.ComplexType]
public class ComplexObj4
{
public virtual string Name { get; set; }
public virtual Test Test { get; set; }
}
[System.ComponentModel.DataAnnotations.Schema.ComplexType]
public class ComplexObj5
{
public virtual string Name { get; set; }
public virtual Test2 Test2 { get; set; }
}
public enum Test
{
}
public enum Test2
{
}
";
string otherProjectCodes = @"
namespace System.ComponentModel.DataAnnotations.Schema
{
public class ComplexTypeAttribute : Attribute
{
}
}
[System.ComponentModel.DataAnnotations.Schema.ComplexType]
public class ComplexObj2
{
public virtual string Name { get; set; }
public virtual Test Test { get; set; }
}
public enum Test3
{
}
";
Project otherProject = CreateProjectFromSourceCodes(otherProjectCodes);
Project sourceProject = CreateProjectFromSourceCodesWithExistingSolution(otherProject.Solution, sourceProjectCodes);
sourceProject = sourceProject.AddProjectReference(new ProjectReference(otherProject.Id));
DefaultProjectDtoControllersProvider dtoControllersProvider = new DefaultProjectDtoControllersProvider();
DefaultProjectDtosProvider projectDtosProvider = new DefaultProjectDtosProvider(dtoControllersProvider);
DefaultProjectEnumTypesProvider enumTypesProvider = new DefaultProjectEnumTypesProvider(dtoControllersProvider, projectDtosProvider);
Dto[] dtos = (await projectDtosProvider.GetProjectDtos(sourceProject)).ToArray();
EnumType[] enums = (await enumTypesProvider.GetProjectEnumTypes(sourceProject)).ToArray();
Assert.IsTrue(dtos.Select(d => d.DtoSymbol.Name).SequenceEqual(new[] { "ComplexObj5", "ComplexObj4", "ComplexObj3", "ComplexObj2", "ComplexObj", "TestComplexDto" }));
Assert.IsTrue(enums.Select(d => d.EnumTypeSymbol.Name).SequenceEqual(new[] { "Test2", "Test" }));
}
