136 lines
6.8 KiB
C#
Executable File
136 lines
6.8 KiB
C#
Executable File
using ZB.MOM.WW.CBDD.Bson;
|
|
using ZB.MOM.WW.CBDD.Core;
|
|
using ZB.MOM.WW.CBDD.Core.Collections;
|
|
using ZB.MOM.WW.CBDD.Core.Compression;
|
|
using ZB.MOM.WW.CBDD.Core.Indexing;
|
|
using ZB.MOM.WW.CBDD.Core.Metadata;
|
|
using ZB.MOM.WW.CBDD.Core.Storage;
|
|
|
|
namespace ZB.MOM.WW.CBDD.Shared;
|
|
|
|
/// <summary>
|
|
/// Test context with manual collection initialization
|
|
/// (Source Generator will automate this in the future)
|
|
/// </summary>
|
|
public partial class TestDbContext : DocumentDbContext
|
|
{
|
|
public DocumentCollection<ObjectId, AnnotatedUser> AnnotatedUsers { get; set; } = null!;
|
|
public DocumentCollection<OrderId, Order> Orders { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, TestDocument> TestDocuments { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, OrderDocument> OrderDocuments { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, ComplexDocument> ComplexDocuments { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, User> Users { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, ComplexUser> ComplexUsers { get; set; } = null!;
|
|
public DocumentCollection<int, AutoInitEntity> AutoInitEntities { get; set; } = null!;
|
|
public DocumentCollection<int, Person> People { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, PersonV2> PeopleV2 { get; set; } = null!;
|
|
public DocumentCollection<int, Product> Products { get; set; } = null!;
|
|
public DocumentCollection<int, IntEntity> IntEntities { get; set; } = null!;
|
|
public DocumentCollection<string, StringEntity> StringEntities { get; set; } = null!;
|
|
public DocumentCollection<Guid, GuidEntity> GuidEntities { get; set; } = null!;
|
|
public DocumentCollection<string, CustomKeyEntity> CustomKeyEntities { get; set; } = null!;
|
|
public DocumentCollection<int, AsyncDoc> AsyncDocs { get; set; } = null!;
|
|
public DocumentCollection<int, SchemaUser> SchemaUsers { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, VectorEntity> VectorItems { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, GeoEntity> GeoItems { get; set; } = null!;
|
|
|
|
// Source Generator Feature Tests
|
|
public DocumentCollection<ObjectId, DerivedEntity> DerivedEntities { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, EntityWithComputedProperties> ComputedPropertyEntities { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, EntityWithAdvancedCollections> AdvancedCollectionEntities { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, EntityWithPrivateSetters> PrivateSetterEntities { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, EntityWithInitSetters> InitSetterEntities { get; set; } = null!;
|
|
|
|
// Circular Reference Tests
|
|
public DocumentCollection<ObjectId, Employee> Employees { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, CategoryRef> CategoryRefs { get; set; } = null!;
|
|
public DocumentCollection<ObjectId, ProductRef> ProductRefs { get; set; } = null!;
|
|
|
|
// Nullable String Id Test (UuidEntity scenario with inheritance)
|
|
public DocumentCollection<string, MockCounter> MockCounters { get; set; } = null!;
|
|
|
|
// Temporal Types Test (DateTimeOffset, TimeSpan, DateOnly, TimeOnly)
|
|
public DocumentCollection<ObjectId, TemporalEntity> TemporalEntities { get; set; } = null!;
|
|
|
|
public TestDbContext(string databasePath)
|
|
: this(databasePath, PageFileConfig.Default, CompressionOptions.Default)
|
|
{
|
|
}
|
|
|
|
public TestDbContext(string databasePath, CompressionOptions compressionOptions)
|
|
: this(databasePath, PageFileConfig.Default, compressionOptions)
|
|
{
|
|
}
|
|
|
|
public TestDbContext(string databasePath, PageFileConfig pageFileConfig)
|
|
: this(databasePath, pageFileConfig, CompressionOptions.Default)
|
|
{
|
|
}
|
|
|
|
public TestDbContext(
|
|
string databasePath,
|
|
PageFileConfig pageFileConfig,
|
|
CompressionOptions? compressionOptions,
|
|
MaintenanceOptions? maintenanceOptions = null)
|
|
: base(databasePath, pageFileConfig, compressionOptions, maintenanceOptions)
|
|
{
|
|
}
|
|
|
|
protected override void OnModelCreating(ModelBuilder modelBuilder)
|
|
{
|
|
modelBuilder.Entity<AnnotatedUser>();
|
|
modelBuilder.Entity<User>().ToCollection("users");
|
|
modelBuilder.Entity<ComplexUser>().ToCollection("complex_users");
|
|
modelBuilder.Entity<AutoInitEntity>().ToCollection("auto_init_entities");
|
|
modelBuilder.Entity<Person>().ToCollection("people_collection").HasIndex(p => p.Age);
|
|
modelBuilder.Entity<PersonV2>().ToCollection("peoplev2_collection").HasIndex(p => p.Age);
|
|
modelBuilder.Entity<Product>().ToCollection("products_collection").HasIndex(p => p.Price);
|
|
modelBuilder.Entity<IntEntity>().HasKey(e => e.Id);
|
|
modelBuilder.Entity<StringEntity>().HasKey(e => e.Id);
|
|
modelBuilder.Entity<GuidEntity>().HasKey(e => e.Id);
|
|
modelBuilder.Entity<CustomKeyEntity>().HasKey(e => e.Code);
|
|
modelBuilder.Entity<AsyncDoc>().ToCollection("async_docs");
|
|
modelBuilder.Entity<SchemaUser>().ToCollection("schema_users").HasKey(e => e.Id);
|
|
modelBuilder.Entity<TestDocument>();
|
|
modelBuilder.Entity<OrderDocument>();
|
|
modelBuilder.Entity<ComplexDocument>();
|
|
|
|
modelBuilder.Entity<VectorEntity>()
|
|
.ToCollection("vector_items")
|
|
.HasVectorIndex(x => x.Embedding, dimensions: 3, metric: VectorMetric.L2, name: "idx_vector");
|
|
|
|
modelBuilder.Entity<GeoEntity>()
|
|
.ToCollection("geo_items")
|
|
.HasSpatialIndex(x => x.Location, name: "idx_spatial");
|
|
|
|
modelBuilder.Entity<Order>()
|
|
.HasKey(x => x.Id)
|
|
.HasConversion<OrderIdConverter>();
|
|
|
|
// Source Generator Feature Tests
|
|
modelBuilder.Entity<DerivedEntity>().ToCollection("derived_entities");
|
|
modelBuilder.Entity<EntityWithComputedProperties>().ToCollection("computed_property_entities");
|
|
modelBuilder.Entity<EntityWithAdvancedCollections>().ToCollection("advanced_collection_entities");
|
|
modelBuilder.Entity<EntityWithPrivateSetters>().ToCollection("private_setter_entities");
|
|
modelBuilder.Entity<EntityWithInitSetters>().ToCollection("init_setter_entities");
|
|
|
|
// Circular Reference Tests
|
|
modelBuilder.Entity<Employee>().ToCollection("employees");
|
|
modelBuilder.Entity<CategoryRef>().ToCollection("category_refs");
|
|
modelBuilder.Entity<ProductRef>().ToCollection("product_refs");
|
|
|
|
// Nullable String Id Test (UuidEntity scenario)
|
|
modelBuilder.Entity<MockCounter>().ToCollection("mock_counters").HasKey(e => e.Id);
|
|
|
|
// Temporal Types Test
|
|
modelBuilder.Entity<TemporalEntity>().ToCollection("temporal_entities").HasKey(e => e.Id);
|
|
}
|
|
|
|
public void ForceCheckpoint()
|
|
{
|
|
Engine.Checkpoint();
|
|
}
|
|
|
|
public StorageEngine Storage => Engine;
|
|
}
|