From 5342a574ceb99d78f0854810ca8f8c9e49dd44f2 Mon Sep 17 00:00:00 2001
From: maliming <malimings@gmail.com>
Date: Tue, 18 May 2021 09:12:35 +0800
Subject: [PATCH] Update Switch_Current_Checker_Test,

---
 ...ermissionsSimpleBatchStateChecker_Tests.cs | 25 ++++++++++++++++---
 1 file changed, 21 insertions(+), 4 deletions(-)

diff --git a/framework/test/Volo.Abp.Authorization.Tests/Volo/Abp/Authorization/RequirePermissionsSimpleBatchStateChecker_Tests.cs b/framework/test/Volo.Abp.Authorization.Tests/Volo/Abp/Authorization/RequirePermissionsSimpleBatchStateChecker_Tests.cs
index 29d542762c..a3b7a96eea 100644
--- a/framework/test/Volo.Abp.Authorization.Tests/Volo/Abp/Authorization/RequirePermissionsSimpleBatchStateChecker_Tests.cs
+++ b/framework/test/Volo.Abp.Authorization.Tests/Volo/Abp/Authorization/RequirePermissionsSimpleBatchStateChecker_Tests.cs
@@ -19,13 +19,20 @@ namespace Volo.Abp.Authorization
         [Fact]
         public void Switch_Current_Checker_Test()
         {
-            var checker = RequirePermissionsSimpleBatchStateChecker<MyStateEntity>.Current;
+            var checker = RequirePermissionsSimpleBatchStateChecker<MyStateEntity2>.Current;
+            checker.ShouldNotBeNull();
 
-            using (RequirePermissionsSimpleBatchStateChecker<MyStateEntity>.Use(new RequirePermissionsSimpleBatchStateChecker<MyStateEntity>()))
+            RequirePermissionsSimpleBatchStateChecker<MyStateEntity2> checker2 = null;
+
+            using (RequirePermissionsSimpleBatchStateChecker<MyStateEntity2>.Use(new RequirePermissionsSimpleBatchStateChecker<MyStateEntity2>()))
             {
-                RequirePermissionsSimpleBatchStateChecker<MyStateEntity>.Current.ShouldNotBeNull();
-                RequirePermissionsSimpleBatchStateChecker<MyStateEntity>.Current.ShouldNotBe(checker);
+                checker2 = RequirePermissionsSimpleBatchStateChecker<MyStateEntity2>.Current;
+                checker2.ShouldNotBeNull();
+                checker2.ShouldNotBe(checker);
             }
+
+            checker2.ShouldNotBeNull();
+            checker2.ShouldNotBe(checker);
         }
 
         [Fact]
@@ -58,5 +65,15 @@ namespace Volo.Abp.Authorization
                 StateCheckers = new List<ISimpleStateChecker<MyStateEntity>>();
             }
         }
+
+        class MyStateEntity2 : IHasSimpleStateCheckers<MyStateEntity2>
+        {
+            public List<ISimpleStateChecker<MyStateEntity2>> StateCheckers { get; }
+
+            public MyStateEntity2()
+            {
+                StateCheckers = new List<ISimpleStateChecker<MyStateEntity2>>();
+            }
+        }
     }
 }