-
Notifications
You must be signed in to change notification settings - Fork 3.1k
/
NorthwindQueryCosmosFixture.cs
43 lines (32 loc) · 1.51 KB
/
NorthwindQueryCosmosFixture.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using Microsoft.EntityFrameworkCore.TestModels.Northwind;
namespace Microsoft.EntityFrameworkCore.Query;
public class NorthwindQueryCosmosFixture<TModelCustomizer> : NorthwindQueryFixtureBase<TModelCustomizer>
where TModelCustomizer : IModelCustomizer, new()
{
protected override ITestStoreFactory TestStoreFactory
=> CosmosNorthwindTestStoreFactory.Instance;
protected override bool UsePooling
=> false;
public TestSqlLoggerFactory TestSqlLoggerFactory
=> (TestSqlLoggerFactory)ServiceProvider.GetRequiredService<ILoggerFactory>();
protected override bool ShouldLogCategory(string logCategory)
=> logCategory == DbLoggerCategory.Query.Name;
protected override void OnModelCreating(ModelBuilder modelBuilder, DbContext context)
{
base.OnModelCreating(modelBuilder, context);
modelBuilder
.Entity<CustomerQuery>()
.HasDiscriminator<string>("Discriminator").HasValue("Customer");
modelBuilder
.Entity<OrderQuery>()
.HasDiscriminator<string>("Discriminator").HasValue("Order");
modelBuilder
.Entity<ProductQuery>()
.HasDiscriminator<string>("Discriminator").HasValue("Product");
modelBuilder
.Entity<CustomerQueryWithQueryFilter>()
.HasDiscriminator<string>("Discriminator").HasValue("Customer");
}
}