Initialize CBDD solution and add a .NET-focused gitignore for generated artifacts.
This commit is contained in:
Joseph Doherty
425
tests/CBDD.Tests/MockEntities.cs
Executable file
425
tests/CBDD.Tests/MockEntities.cs
Executable file
@@ -0,0 +1,425 @@
|
||||
using ZB.MOM.WW.CBDD.Bson;
|
||||
using ZB.MOM.WW.CBDD.Core.Collections;
|
||||
using ZB.MOM.WW.CBDD.Core.Metadata;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.ComponentModel.DataAnnotations;
|
||||
using System.ComponentModel.DataAnnotations.Schema;
|
||||
|
||||
namespace ZB.MOM.WW.CBDD.Shared
|
||||
{
|
||||
// --- Basic Entities ---
|
||||
|
||||
public class User
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string Name { get; set; } = "";
|
||||
public int Age { get; set; }
|
||||
}
|
||||
|
||||
// --- Complex Entities (Nested) ---
|
||||
|
||||
public class ComplexUser
|
||||
{
|
||||
[BsonId]
|
||||
public ObjectId Id { get; set; }
|
||||
public string Name { get; set; } = "";
|
||||
|
||||
// Direct nested object
|
||||
public Address MainAddress { get; set; } = new();
|
||||
|
||||
// Collection of nested objects
|
||||
public List<Address> OtherAddresses { get; set; } = new();
|
||||
|
||||
// Primitive collection
|
||||
public List<string> Tags { get; set; } = new();
|
||||
|
||||
[BsonIgnore]
|
||||
public string Secret { get; set; } = "";
|
||||
}
|
||||
|
||||
public class Address
|
||||
{
|
||||
public string Street { get; set; } = "";
|
||||
public City City { get; set; } = new(); // Depth 2
|
||||
}
|
||||
|
||||
public class City
|
||||
{
|
||||
public string Name { get; set; } = "";
|
||||
public string ZipCode { get; set; } = "";
|
||||
}
|
||||
|
||||
// --- Primary Key Test Entities ---
|
||||
|
||||
public class IntEntity
|
||||
{
|
||||
public int Id { get; set; }
|
||||
public string? Name { get; set; }
|
||||
}
|
||||
|
||||
public class StringEntity
|
||||
{
|
||||
public required string Id { get; set; }
|
||||
public string? Value { get; set; }
|
||||
}
|
||||
|
||||
public class GuidEntity
|
||||
{
|
||||
public Guid Id { get; set; }
|
||||
public string? Name { get; set; }
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Entity with string key NOT named "Id" - tests custom key name support
|
||||
/// </summary>
|
||||
public class CustomKeyEntity
|
||||
{
|
||||
[System.ComponentModel.DataAnnotations.Key]
|
||||
public required string Code { get; set; }
|
||||
public string? Description { get; set; }
|
||||
}
|
||||
|
||||
// --- Multi-collection / Auto-init entities ---
|
||||
|
||||
public class AutoInitEntity
|
||||
{
|
||||
public int Id { get; set; }
|
||||
public string Name { get; set; } = string.Empty;
|
||||
}
|
||||
|
||||
public class Person
|
||||
{
|
||||
public int Id { get; set; }
|
||||
public string Name { get; set; } = "";
|
||||
public int Age { get; set; }
|
||||
}
|
||||
|
||||
public class Product
|
||||
{
|
||||
public int Id { get; set; }
|
||||
public string Title { get; set; } = "";
|
||||
public decimal Price { get; set; }
|
||||
}
|
||||
|
||||
public class AsyncDoc
|
||||
{
|
||||
public int Id { get; set; }
|
||||
public string Name { get; set; } = "";
|
||||
}
|
||||
|
||||
public class SchemaUser
|
||||
{
|
||||
public int Id { get; set; }
|
||||
public string Name { get; set; } = "";
|
||||
public Address Address { get; set; } = new();
|
||||
}
|
||||
|
||||
public class VectorEntity
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string Title { get; set; } = "";
|
||||
public float[] Embedding { get; set; } = Array.Empty<float>();
|
||||
}
|
||||
|
||||
public class GeoEntity
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string Name { get; set; } = "";
|
||||
public (double Latitude, double Longitude) Location { get; set; }
|
||||
}
|
||||
|
||||
public record OrderId(string Value)
|
||||
{
|
||||
public OrderId() : this(string.Empty) { }
|
||||
}
|
||||
|
||||
public class OrderIdConverter : ValueConverter<OrderId, string>
|
||||
{
|
||||
public override string ConvertToProvider(OrderId model) => model?.Value ?? string.Empty;
|
||||
public override OrderId ConvertFromProvider(string provider) => new OrderId(provider);
|
||||
}
|
||||
|
||||
public class Order
|
||||
{
|
||||
public OrderId Id { get; set; } = null!;
|
||||
public string CustomerName { get; set; } = "";
|
||||
}
|
||||
|
||||
public class TestDocument
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string Category { get; set; } = string.Empty;
|
||||
public int Amount { get; set; }
|
||||
public string Name { get; set; } = string.Empty;
|
||||
}
|
||||
|
||||
public class OrderDocument
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string ItemName { get; set; } = string.Empty;
|
||||
public int Quantity { get; set; }
|
||||
}
|
||||
|
||||
public class OrderItem
|
||||
{
|
||||
public string Name { get; set; } = string.Empty;
|
||||
public int Price { get; set; }
|
||||
}
|
||||
|
||||
public class ComplexDocument
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string Title { get; set; } = string.Empty;
|
||||
public Address ShippingAddress { get; set; } = new();
|
||||
public List<OrderItem> Items { get; set; } = new();
|
||||
}
|
||||
|
||||
[Table("custom_users", Schema = "test")]
|
||||
public class AnnotatedUser
|
||||
{
|
||||
[Key]
|
||||
public ObjectId Id { get; set; }
|
||||
|
||||
[Required]
|
||||
[Column("display_name")]
|
||||
[StringLength(50, MinimumLength = 3)]
|
||||
public string Name { get; set; } = "";
|
||||
|
||||
[Range(0, 150)]
|
||||
public int Age { get; set; }
|
||||
|
||||
[NotMapped]
|
||||
public string ComputedInfo => $"{Name} ({Age})";
|
||||
|
||||
[Column(TypeName = "geopoint")]
|
||||
public (double Lat, double Lon) Location { get; set; }
|
||||
}
|
||||
public class PersonV2
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string Name { get; set; } = string.Empty;
|
||||
public int Age { get; set; }
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Entity used to test DbContext inheritance
|
||||
/// </summary>
|
||||
public class ExtendedEntity
|
||||
{
|
||||
public int Id { get; set; }
|
||||
public string Description { get; set; } = string.Empty;
|
||||
public DateTime CreatedAt { get; set; }
|
||||
}
|
||||
|
||||
// ===== SOURCE GENERATOR FEATURE TESTS =====
|
||||
|
||||
/// <summary>
|
||||
/// Base entity with Id property - test inheritance
|
||||
/// </summary>
|
||||
public class BaseEntityWithId
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public DateTime CreatedAt { get; set; }
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Derived entity that inherits Id from base class
|
||||
/// </summary>
|
||||
public class DerivedEntity : BaseEntityWithId
|
||||
{
|
||||
public string Name { get; set; } = string.Empty;
|
||||
public string Description { get; set; } = string.Empty;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Entity with computed getter-only properties (should be excluded from serialization)
|
||||
/// </summary>
|
||||
public class EntityWithComputedProperties
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string FirstName { get; set; } = string.Empty;
|
||||
public string LastName { get; set; } = string.Empty;
|
||||
public int BirthYear { get; set; }
|
||||
|
||||
// Computed properties - should NOT be serialized
|
||||
public string FullName => $"{FirstName} {LastName}";
|
||||
public int Age => DateTime.Now.Year - BirthYear;
|
||||
public string DisplayInfo => $"{FullName} (Age: {Age})";
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Entity with advanced collection types (HashSet, ISet, LinkedList, etc.)
|
||||
/// </summary>
|
||||
public class EntityWithAdvancedCollections
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string Name { get; set; } = string.Empty;
|
||||
|
||||
// Various collection types that should all be recognized
|
||||
public HashSet<string> Tags { get; set; } = new();
|
||||
public ISet<int> Numbers { get; set; } = new HashSet<int>();
|
||||
public LinkedList<string> History { get; set; } = new();
|
||||
public Queue<string> PendingItems { get; set; } = new();
|
||||
public Stack<string> UndoStack { get; set; } = new();
|
||||
|
||||
// Nested objects in collections
|
||||
public HashSet<Address> Addresses { get; set; } = new();
|
||||
public ISet<City> FavoriteCities { get; set; } = new HashSet<City>();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Entity with private setters (requires reflection-based deserialization)
|
||||
/// </summary>
|
||||
public class EntityWithPrivateSetters
|
||||
{
|
||||
public ObjectId Id { get; private set; }
|
||||
public string Name { get; private set; } = string.Empty;
|
||||
public int Age { get; private set; }
|
||||
public DateTime CreatedAt { get; private set; }
|
||||
|
||||
// Factory method for creation
|
||||
public static EntityWithPrivateSetters Create(string name, int age)
|
||||
{
|
||||
return new EntityWithPrivateSetters
|
||||
{
|
||||
Id = ObjectId.NewObjectId(),
|
||||
Name = name,
|
||||
Age = age,
|
||||
CreatedAt = DateTime.UtcNow
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Entity with init-only setters (can use object initializer)
|
||||
/// </summary>
|
||||
public class EntityWithInitSetters
|
||||
{
|
||||
public ObjectId Id { get; init; }
|
||||
public required string Name { get; init; }
|
||||
public int Age { get; init; }
|
||||
public DateTime CreatedAt { get; init; }
|
||||
}
|
||||
|
||||
// ========================================
|
||||
// Circular Reference Test Entities
|
||||
// ========================================
|
||||
|
||||
/// <summary>
|
||||
/// Employee with self-referencing via ObjectIds (organizational hierarchy)
|
||||
/// Tests: self-reference using referencing (BEST PRACTICE)
|
||||
/// Recommended: Avoids embedding which can lead to large/circular documents
|
||||
/// </summary>
|
||||
public class Employee
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string Name { get; set; } = string.Empty;
|
||||
public string Department { get; set; } = string.Empty;
|
||||
public ObjectId? ManagerId { get; set; } // Reference to manager
|
||||
public List<ObjectId>? DirectReportIds { get; set; } // References to direct reports (best practice)
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Category with referenced products (N-N using ObjectId references)
|
||||
/// Tests: N-N relationships using referencing (BEST PRACTICE for document databases)
|
||||
/// Recommended: Avoids large documents, better for queries and updates
|
||||
/// </summary>
|
||||
public class CategoryRef
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string Name { get; set; } = string.Empty;
|
||||
public string Description { get; set; } = string.Empty;
|
||||
public List<ObjectId>? ProductIds { get; set; } // Only IDs - no embedding
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Product with referenced categories (N-N using ObjectId references)
|
||||
/// Tests: N-N relationships using referencing (BEST PRACTICE for document databases)
|
||||
/// Recommended: Avoids large documents, better for queries and updates
|
||||
/// </summary>
|
||||
public class ProductRef
|
||||
{
|
||||
public ObjectId Id { get; set; }
|
||||
public string Name { get; set; } = string.Empty;
|
||||
public decimal Price { get; set; }
|
||||
public List<ObjectId>? CategoryIds { get; set; } // Only IDs - no embedding
|
||||
}
|
||||
|
||||
// ========================================
|
||||
// Nullable String Key Test (UuidEntity scenario)
|
||||
// ========================================
|
||||
|
||||
/// <summary>
|
||||
/// Base entity class that simulates CleanCore's BaseEntity{TId, TEntity}
|
||||
/// This is the root of the hierarchy that causes the generator bug
|
||||
/// </summary>
|
||||
public abstract class MockBaseEntity<TId, TEntity>
|
||||
where TId : IEquatable<TId>
|
||||
where TEntity : class
|
||||
{
|
||||
[System.ComponentModel.DataAnnotations.Key]
|
||||
public virtual TId? Id { get; set; }
|
||||
|
||||
protected MockBaseEntity() { }
|
||||
|
||||
protected MockBaseEntity(TId? id)
|
||||
{
|
||||
Id = id;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Simulates CleanCore's UuidEntity{TEntity} which inherits from BaseEntity{string, TEntity}
|
||||
/// Tests the bug where generator incorrectly chooses ObjectIdMapperBase instead of StringMapperBase
|
||||
/// when the Id property is inherited and nullable
|
||||
/// </summary>
|
||||
public abstract class MockUuidEntity<TEntity> : MockBaseEntity<string, TEntity>
|
||||
where TEntity : class
|
||||
{
|
||||
protected MockUuidEntity() : base() { }
|
||||
|
||||
protected MockUuidEntity(string? id) : base(id) { }
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Concrete entity that inherits from MockUuidEntity, simulating Counter from CleanCore
|
||||
/// This is the actual entity that will be stored in the collection
|
||||
/// </summary>
|
||||
public class MockCounter : MockUuidEntity<MockCounter>
|
||||
{
|
||||
public MockCounter() : base() { }
|
||||
|
||||
public MockCounter(string? id) : base(id) { }
|
||||
|
||||
public string Name { get; set; } = string.Empty;
|
||||
public int Value { get; set; }
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Entity for testing temporal types: DateTimeOffset, TimeSpan, DateOnly, TimeOnly
|
||||
/// </summary>
|
||||
public class TemporalEntity
|
||||
{
|
||||
[Key]
|
||||
public ObjectId Id { get; set; }
|
||||
|
||||
public string Name { get; set; } = string.Empty;
|
||||
|
||||
// DateTime types
|
||||
public DateTime CreatedAt { get; set; }
|
||||
public DateTimeOffset UpdatedAt { get; set; }
|
||||
public DateTimeOffset? LastAccessedAt { get; set; }
|
||||
|
||||
// TimeSpan
|
||||
public TimeSpan Duration { get; set; }
|
||||
public TimeSpan? OptionalDuration { get; set; }
|
||||
|
||||
// DateOnly and TimeOnly (.NET 6+)
|
||||
public DateOnly BirthDate { get; set; }
|
||||
public DateOnly? Anniversary { get; set; }
|
||||
|
||||
public TimeOnly OpeningTime { get; set; }
|
||||
public TimeOnly? ClosingTime { get; set; }
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user