using Microsoft.EntityFrameworkCore;

    public class AppDbContext : DbContext
    {
        public AppDbContext(DbContextOptions<AppDbContext> options)
        : base(options)
        {
        }

        public DbSet<TestEntry> TestEntries { get; set; }

        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            var testEntryBuilder = modelBuilder.Entity<TestEntry>();
            testEntryBuilder.HasKey(e => new { e.PartitionKey, e.Id });
            testEntryBuilder.Property(e => e.ETag).IsETagConcurrency();
            testEntryBuilder.HasPartitionKey(e => e.PartitionKey);
        }
    }

    public class TestEntry
    {
        public static string CreatePartitionKey(string transactionId) => $"{transactionId}_test";
        public TestEntry() : base()
        {
            Id = Guid.NewGuid().ToString();
        }

        public string Id { get; set; }
        public string PartitionKey { get; set; }
        public string ETag { get; set; }
        public bool Confirmed { get; set; }
        public DateTime Updated { get; set; }
        public string TestValue { get; set; }
    }

var builder = WebApplication.CreateBuilder(args);
builder.Services.AddEndpointsApiExplorer();
builder.Services.AddSwaggerGen();
builder.Services.AddApplicationInsightsTelemetry();
builder.Services.AddDbContext<AppDbContext>(opt =>
{
    opt.UseCosmos(builder.Configuration.GetConnectionString("DefaultConnection"), "WebTests");
});

var app = builder.Build();

if (app.Environment.IsDevelopment())
{
    app.UseDeveloperExceptionPage();
}

app.UseSwagger();
app.MapGet("/", context =>
{
    context.Response.Redirect("./swagger/index.html", permanent: false);
    return Task.FromResult(0);
});

app.MapGet("/hello", () => "Hello World!");

app.MapGet("/test", async (HttpContext context, AppDbContext db) =>
{
    // Tested https://github.com/dotnet/efcore/issues/26627

    // New issue
    await db.Database.EnsureCreatedAsync();
    var dateVal = DateTime.UtcNow;
    var entry = new TestEntry()
    {
        PartitionKey = TestEntry.CreatePartitionKey("DateTest"),
        Id = "DateTest",
        Updated = dateVal,
        TestValue = dateVal.ToString()
    };
    db.Add(entry);
    await db.SaveChangesAsync();
    return JsonSerializer.Serialize(entry);
});

app.MapGet("/test-result", async (HttpContext context, AppDbContext db) =>
{
    var result = await db.TestEntries.FindAsync(TestEntry.CreatePartitionKey("DateTest"), "DateTest");
    return JsonSerializer.Serialize(result);
});

app.UseSwaggerUI();
await app.RunAsync();