From 31e519087efa4f2c64dd48778a9b69711f7133f0 Mon Sep 17 00:00:00 2001
From: LiangShiWei <shiwei.liang@volosoft.com>
Date: Wed, 28 Sep 2022 11:38:53 +0800
Subject: [PATCH] Fix unit tests

---
 .../Volo/Abp/MemoryDb/DomainEvents/EntityCache_Tests.cs    | 7 +++++++
 .../Volo/Abp/TestApp/MemoryDb/TestAppMemoryDbContext.cs    | 6 ++++--
 .../Volo/Abp/MongoDB/AbpMongoDbTestModule.cs               | 2 +-
 .../Volo/Abp/TestApp/Testing/EntityCache_Tests.cs          | 3 ++-
 4 files changed, 14 insertions(+), 4 deletions(-)
 create mode 100644 framework/test/Volo.Abp.MemoryDb.Tests/Volo/Abp/MemoryDb/DomainEvents/EntityCache_Tests.cs

diff --git a/framework/test/Volo.Abp.MemoryDb.Tests/Volo/Abp/MemoryDb/DomainEvents/EntityCache_Tests.cs b/framework/test/Volo.Abp.MemoryDb.Tests/Volo/Abp/MemoryDb/DomainEvents/EntityCache_Tests.cs
new file mode 100644
index 0000000000..6718fabc4e
--- /dev/null
+++ b/framework/test/Volo.Abp.MemoryDb.Tests/Volo/Abp/MemoryDb/DomainEvents/EntityCache_Tests.cs
@@ -0,0 +1,7 @@
+using Volo.Abp.TestApp.Testing;
+
+namespace Volo.Abp.MemoryDb.DomainEvents;
+
+public class EntityCache_Tests : EntityCache_Tests<AbpMemoryDbTestModule>
+{
+}
\ No newline at end of file
diff --git a/framework/test/Volo.Abp.MemoryDb.Tests/Volo/Abp/TestApp/MemoryDb/TestAppMemoryDbContext.cs b/framework/test/Volo.Abp.MemoryDb.Tests/Volo/Abp/TestApp/MemoryDb/TestAppMemoryDbContext.cs
index 63eddc6f94..4336889b90 100644
--- a/framework/test/Volo.Abp.MemoryDb.Tests/Volo/Abp/TestApp/MemoryDb/TestAppMemoryDbContext.cs
+++ b/framework/test/Volo.Abp.MemoryDb.Tests/Volo/Abp/TestApp/MemoryDb/TestAppMemoryDbContext.cs
@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using Volo.Abp.MemoryDb;
 using Volo.Abp.TestApp.Domain;
+using Volo.Abp.TestApp.Testing;
 
 namespace Volo.Abp.TestApp.MemoryDb;
 
@@ -9,8 +10,9 @@ public class TestAppMemoryDbContext : MemoryDbContext
 {
     private static readonly Type[] EntityTypeList = {
             typeof(Person),
-            typeof(EntityWithIntPk)
-        };
+            typeof(EntityWithIntPk),
+            typeof(Product)
+    };
 
     public override IReadOnlyList<Type> GetEntityTypes()
     {
diff --git a/framework/test/Volo.Abp.MongoDB.Tests/Volo/Abp/MongoDB/AbpMongoDbTestModule.cs b/framework/test/Volo.Abp.MongoDB.Tests/Volo/Abp/MongoDB/AbpMongoDbTestModule.cs
index 0ad7554ea1..63b63f0873 100644
--- a/framework/test/Volo.Abp.MongoDB.Tests/Volo/Abp/MongoDB/AbpMongoDbTestModule.cs
+++ b/framework/test/Volo.Abp.MongoDB.Tests/Volo/Abp/MongoDB/AbpMongoDbTestModule.cs
@@ -30,7 +30,7 @@ public class AbpMongoDbTestModule : AbpModule
 
         context.Services.AddMongoDbContext<TestAppMongoDbContext>(options =>
         {
-            options.AddDefaultRepositories<ITestAppMongoDbContext>(true);
+            options.AddDefaultRepositories<ITestAppMongoDbContext>();
             options.AddRepository<City, CityRepository>();
 
             options.ReplaceDbContext<IThirdDbContext>();
diff --git a/framework/test/Volo.Abp.TestApp/Volo/Abp/TestApp/Testing/EntityCache_Tests.cs b/framework/test/Volo.Abp.TestApp/Volo/Abp/TestApp/Testing/EntityCache_Tests.cs
index 4d1f810705..7291fcf585 100644
--- a/framework/test/Volo.Abp.TestApp/Volo/Abp/TestApp/Testing/EntityCache_Tests.cs
+++ b/framework/test/Volo.Abp.TestApp/Volo/Abp/TestApp/Testing/EntityCache_Tests.cs
@@ -3,6 +3,7 @@ using System.Text.Json.Serialization;
 using System.Threading.Tasks;
 using Shouldly;
 using Volo.Abp.Domain.Entities;
+using Volo.Abp.Domain.Entities.Auditing;
 using Volo.Abp.Domain.Entities.Caching;
 using Volo.Abp.Domain.Repositories;
 using Volo.Abp.Modularity;
@@ -91,7 +92,7 @@ public abstract class EntityCache_Tests<TStartupModule> : TestAppTestBase<TStart
 }
 
 [Serializable]
-public class Product : Entity<Guid>
+public class Product : FullAuditedAggregateRoot<Guid>
 {
     public Product(Guid id, string name, decimal price)
        : base(id)