From 2564e798eef3b95d0d05c75dbda543d57cc9bdc3 Mon Sep 17 00:00:00 2001
From: maliming <malimings@gmail.com>
Date: Wed, 28 Jun 2023 14:07:02 +0800
Subject: [PATCH] Fix `AddItem` extension method of `UnitOfWorkExtensions`.

---
 .../Volo.Abp.Uow/Volo/Abp/Uow/UnitOfWorkExtensions.cs    | 9 +--------
 .../Volo/Abp/Uow/UnitOfWorkExtensions_Tests.cs           | 5 +++++
 2 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/framework/src/Volo.Abp.Uow/Volo/Abp/Uow/UnitOfWorkExtensions.cs b/framework/src/Volo.Abp.Uow/Volo/Abp/Uow/UnitOfWorkExtensions.cs
index 1396cd6ee5..a732e5ad21 100644
--- a/framework/src/Volo.Abp.Uow/Volo/Abp/Uow/UnitOfWorkExtensions.cs
+++ b/framework/src/Volo.Abp.Uow/Volo/Abp/Uow/UnitOfWorkExtensions.cs
@@ -19,14 +19,7 @@ public static class UnitOfWorkExtensions
     {
         Check.NotNull(unitOfWork, nameof(unitOfWork));
 
-        if (!unitOfWork.Items.ContainsKey(key))
-        {
-            unitOfWork.Items[key] = value;
-        }
-        else
-        {
-            unitOfWork.Items.Add(key, value);
-        }
+        unitOfWork.Items[key] = value;
     }
 
     public static TValue GetItemOrDefault<TValue>([NotNull] this IUnitOfWork unitOfWork, string key)
diff --git a/framework/test/Volo.Abp.Uow.Tests/Volo/Abp/Uow/UnitOfWorkExtensions_Tests.cs b/framework/test/Volo.Abp.Uow.Tests/Volo/Abp/Uow/UnitOfWorkExtensions_Tests.cs
index 6c88f70d9d..57f9d0a641 100644
--- a/framework/test/Volo.Abp.Uow.Tests/Volo/Abp/Uow/UnitOfWorkExtensions_Tests.cs
+++ b/framework/test/Volo.Abp.Uow.Tests/Volo/Abp/Uow/UnitOfWorkExtensions_Tests.cs
@@ -21,6 +21,11 @@ public class UnitOfWorkExtensions_Tests : AbpIntegratedTest<AbpUnitOfWorkModule>
 
         uow.Items.ShouldContainKey("testKey");
         uow.Items.ContainsValue("testValue");
+
+        uow.AddItem("testKey", "testValue2");
+
+        uow.Items.ShouldContainKey("testKey");
+        uow.Items.ContainsValue("testValue2");
     }
 
     [Fact]