From 6a2d8337f097cec27e81689073cb10627be671f8 Mon Sep 17 00:00:00 2001
From: maliming <6908465+maliming@users.noreply.github.com>
Date: Thu, 4 Jun 2020 10:09:26 +0800
Subject: [PATCH] Fix identity unit test.

Resolve #4208
---
 .../OrganizationUnitRepository_Tests.cs       | 20 +++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/modules/identity/test/Volo.Abp.Identity.TestBase/Volo/Abp/Identity/OrganizationUnitRepository_Tests.cs b/modules/identity/test/Volo.Abp.Identity.TestBase/Volo/Abp/Identity/OrganizationUnitRepository_Tests.cs
index bbb7d649c9..5ff0284194 100644
--- a/modules/identity/test/Volo.Abp.Identity.TestBase/Volo/Abp/Identity/OrganizationUnitRepository_Tests.cs
+++ b/modules/identity/test/Volo.Abp.Identity.TestBase/Volo/Abp/Identity/OrganizationUnitRepository_Tests.cs
@@ -167,13 +167,21 @@ namespace Volo.Abp.Identity
         [Fact]
         public async Task RemoveAllRolesOfOrganizationUnit()
         {
-            OrganizationUnit ou = await _organizationUnitRepository.GetAsync("OU111", true);
-            var rolesCount = await _organizationUnitRepository.GetRolesCountAsync(ou);
-            rolesCount.ShouldBeGreaterThan(1);
+            using (var uow = _unitOfWorkManager.Begin())
+            {
+                OrganizationUnit ou = await _organizationUnitRepository.GetAsync("OU111", true);
+                var rolesCount = await _organizationUnitRepository.GetRolesCountAsync(ou);
+                rolesCount.ShouldBeGreaterThan(1);
 
-            await _organizationUnitRepository.RemoveAllRolesAsync(ou);
-            var newCount = await _organizationUnitRepository.GetRolesCountAsync(ou);
-            newCount.ShouldBe(0);
+                await _organizationUnitRepository.RemoveAllRolesAsync(ou);
+
+                await uow.SaveChangesAsync();
+
+                var newCount = await _organizationUnitRepository.GetRolesCountAsync(ou);
+                newCount.ShouldBe(0);
+
+                await uow.CompleteAsync();
+            }
         }
     }
 }