Add ResetCount method to ITestCounter.

abpframework · Mar 15, 2024 · 84c6999 · 84c6999
1 parent d3e81a5
commit 84c6999
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/framework/src/Volo.Abp.TestBase/Volo/Abp/Testing/Utils/ITestCounter.cs b/framework/src/Volo.Abp.TestBase/Volo/Abp/Testing/Utils/ITestCounter.cs
@@ -9,4 +9,6 @@ public interface ITestCounter
     int Increment(string name);
 
     int GetValue(string name);
+
+    void ResetCount(string name);
 }
diff --git a/framework/src/Volo.Abp.TestBase/Volo/Abp/Testing/Utils/TestCounter.cs b/framework/src/Volo.Abp.TestBase/Volo/Abp/Testing/Utils/TestCounter.cs
@@ -39,4 +39,12 @@ public int GetValue(string name)
             return _values.GetOrDefault(name);
         }
     }
+
+    public void ResetCount(string name)
+    {
+        lock (_values)
+        {
+            _values[name] = 0;
+        }
+    }
 }
diff --git a/...t/Volo.Abp.Identity.Domain.Tests/Volo/Abp/Identity/Distributed_User_Change_Event_Tests.cs b/...t/Volo.Abp.Identity.Domain.Tests/Volo/Abp/Identity/Distributed_User_Change_Event_Tests.cs
@@ -45,8 +45,11 @@ public void Should_Register_Handler()
     [Fact]
     public async Task Should_Trigger_Distributed_EntityUpdated_Event()
     {
+        _testCounter.ResetCount("EntityUpdatedEto<UserEto>");
         using (var uow = _unitOfWorkManager.Begin())
         {
+            _testCounter.GetValue("EntityUpdatedEto<UserEto>").ShouldBe(0);
+
             var user = await _userRepository.FindByNormalizedUserNameAsync(_lookupNormalizer.NormalizeName("john.nash"));
             await _userManager.SetEmailAsync(user, "john.nash_UPDATED@abp.io");