From 3ba4accc596979dd803ec4a2c85be46c4dc7bc84 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Halil=20=C4=B0brahim=20Kalkan?= <hi_kalkan@yahoo.com>
Date: Sun, 29 Mar 2020 13:12:14 +0300
Subject: [PATCH] Added ObjectExtensionManager_Tests

---
 .../Properties/AssemblyInfo.cs                |  3 +
 .../ObjectExtending/ObjectExtensionInfo.cs    |  9 +++
 .../ObjectExtending/ObjectExtensionManager.cs |  9 ++-
 .../ObjectExtensionManager_Tests.cs           | 63 +++++++++++++++++++
 4 files changed, 83 insertions(+), 1 deletion(-)
 create mode 100644 framework/src/Volo.Abp.ObjectExtending/Properties/AssemblyInfo.cs
 create mode 100644 framework/test/Volo.Abp.ObjectExtending.Tests/Volo/Abp/ObjectExtending/ObjectExtensionManager_Tests.cs

diff --git a/framework/src/Volo.Abp.ObjectExtending/Properties/AssemblyInfo.cs b/framework/src/Volo.Abp.ObjectExtending/Properties/AssemblyInfo.cs
new file mode 100644
index 0000000000..fe0b656805
--- /dev/null
+++ b/framework/src/Volo.Abp.ObjectExtending/Properties/AssemblyInfo.cs
@@ -0,0 +1,3 @@
+using System.Runtime.CompilerServices;
+
+[assembly: InternalsVisibleTo("Volo.Abp.ObjectExtending.Tests")]
diff --git a/framework/src/Volo.Abp.ObjectExtending/Volo/Abp/ObjectExtending/ObjectExtensionInfo.cs b/framework/src/Volo.Abp.ObjectExtending/Volo/Abp/ObjectExtending/ObjectExtensionInfo.cs
index 14d527991f..89ecbab574 100644
--- a/framework/src/Volo.Abp.ObjectExtending/Volo/Abp/ObjectExtending/ObjectExtensionInfo.cs
+++ b/framework/src/Volo.Abp.ObjectExtending/Volo/Abp/ObjectExtending/ObjectExtensionInfo.cs
@@ -65,5 +65,14 @@ namespace Volo.Abp.ObjectExtending
         {
             return Properties.Values.ToImmutableList();
         }
+
+        [CanBeNull]
+        public virtual ObjectExtensionPropertyInfo GetPropertyOrNull(
+            [NotNull] string propertyName)
+        {
+            Check.NotNullOrEmpty(propertyName, nameof(propertyName));
+
+            return Properties.GetOrDefault(propertyName);
+        }
     }
 }
\ No newline at end of file
diff --git a/framework/src/Volo.Abp.ObjectExtending/Volo/Abp/ObjectExtending/ObjectExtensionManager.cs b/framework/src/Volo.Abp.ObjectExtending/Volo/Abp/ObjectExtending/ObjectExtensionManager.cs
index cbc2aa3baa..9fd50da492 100644
--- a/framework/src/Volo.Abp.ObjectExtending/Volo/Abp/ObjectExtending/ObjectExtensionManager.cs
+++ b/framework/src/Volo.Abp.ObjectExtending/Volo/Abp/ObjectExtending/ObjectExtensionManager.cs
@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Collections.Immutable;
 using JetBrains.Annotations;
 using Volo.Abp.Data;
 
@@ -11,7 +12,7 @@ namespace Volo.Abp.ObjectExtending
 
         protected Dictionary<Type, ObjectExtensionInfo> ObjectsExtensions { get; }
 
-        protected ObjectExtensionManager()
+        protected internal ObjectExtensionManager()
         {
             ObjectsExtensions = new Dictionary<Type, ObjectExtensionInfo>();
         }
@@ -55,5 +56,11 @@ namespace Volo.Abp.ObjectExtending
 
             return ObjectsExtensions.GetOrDefault(type);
         }
+
+        [NotNull]
+        public virtual ImmutableList<ObjectExtensionInfo> GetExtendedObjects()
+        {
+            return ObjectsExtensions.Values.ToImmutableList();
+        }
     }
 }
\ No newline at end of file
diff --git a/framework/test/Volo.Abp.ObjectExtending.Tests/Volo/Abp/ObjectExtending/ObjectExtensionManager_Tests.cs b/framework/test/Volo.Abp.ObjectExtending.Tests/Volo/Abp/ObjectExtending/ObjectExtensionManager_Tests.cs
new file mode 100644
index 0000000000..857e2fd155
--- /dev/null
+++ b/framework/test/Volo.Abp.ObjectExtending.Tests/Volo/Abp/ObjectExtending/ObjectExtensionManager_Tests.cs
@@ -0,0 +1,63 @@
+using System.Linq;
+using Shouldly;
+using Xunit;
+
+namespace Volo.Abp.ObjectExtending
+{
+    public class ObjectExtensionManager_Tests
+    {
+        private readonly ObjectExtensionManager _objectExtensionManager;
+
+        public ObjectExtensionManager_Tests()
+        {
+            _objectExtensionManager = new ObjectExtensionManager();
+        }
+
+        [Fact]
+        public void Should_Not_Add_Same_Property_Multiple_Times()
+        {
+            _objectExtensionManager
+                .AddOrUpdateProperty<MyExtensibleObject, string>("TestProp")
+                .AddOrUpdateProperty<MyExtensibleObject, string>("TestProp");
+
+            var objectExtension = _objectExtensionManager.GetOrNull<MyExtensibleObject>();
+            objectExtension.ShouldNotBeNull();
+            
+            var properties = objectExtension.GetProperties();
+            properties.Count.ShouldBe(1);
+            properties.FirstOrDefault(p => p.Name == "TestProp").ShouldNotBeNull();
+        }
+
+        [Fact]
+        public void Should_Update_Property_Configuration()
+        {
+            _objectExtensionManager
+                .AddOrUpdateProperty<MyExtensibleObject, string>(
+                    "TestProp",
+                    options =>
+                    {
+                        options.Configuration["TestConfig1"] = "TestConfig1-Value";
+                    }
+                ).AddOrUpdateProperty<MyExtensibleObject, string>(
+                    "TestProp",
+                    options =>
+                    {
+                        options.Configuration["TestConfig2"] = "TestConfig2-Value";
+                    }
+                );
+
+            var objectExtension = _objectExtensionManager.GetOrNull<MyExtensibleObject>();
+            objectExtension.ShouldNotBeNull();
+
+            var property = objectExtension.GetPropertyOrNull("TestProp");
+            property.ShouldNotBeNull();
+            property.Configuration["TestConfig1"].ShouldBe("TestConfig1-Value");
+            property.Configuration["TestConfig2"].ShouldBe("TestConfig2-Value");
+        }
+
+        private class MyExtensibleObject : ExtensibleObject
+        {
+
+        }
+    }
+}