DYN-1476: AllowRankReductionAttribute on Zero Touch seems to be broken (#9542)

scottmitchell · web-flow · commit 0dbc0cc413cf · 2019-03-07T13:59:23.000-05:00
* update

* Brought truncated search results and full query search out from behind debug menu. Deleted debug menu items.

* Added CheckForRankReductionAttribute() method to FFIMemberInfo

* Cleaned up CheckForRankReductionAttribute

* Added AllowRankReductionAttribute tests

* Comments. Change method name.

* Update tests, method name

* Corrected test names

* Fixed annoying diff
diff --git a/src/Engine/ProtoCore/FFI/CLRDLLModule.cs b/src/Engine/ProtoCore/FFI/CLRDLLModule.cs
@@ -393,7 +393,7 @@ private ClassDeclNode ParseSystemType(Type type, string alias)
             bool isDisposable = typeof(IDisposable).IsAssignableFrom(type);
             MethodInfo[] methods = type.GetMethods(flags);
             bool hasDisposeMethod = false;
-
+            
             foreach (var m in methods)
             {
                 if (SupressesImport(m, mGetterAttributes))
@@ -782,7 +782,10 @@ private void RegisterFunctionPointer(string functionName, MemberInfo method, Lis
             if (CoreUtils.IsDisposeMethod(functionName))
                 f = new DisposeFunctionPointer(Module, method, retype);
             else if (CoreUtils.IsGetter(functionName))
+            {
                 f = new GetterFunctionPointer(Module, functionName, method, retype);
+                (f as GetterFunctionPointer).ReflectionInfo.CheckForRankReductionAttribute(mGetterAttributes);
+            }
             else
                 f = new CLRFFIFunctionPointer(Module, functionName, method, argTypes, retype);
 
diff --git a/src/Engine/ProtoCore/FFI/CLRFFIFunctionPointer.cs b/src/Engine/ProtoCore/FFI/CLRFFIFunctionPointer.cs
@@ -64,6 +64,31 @@ protected FFIMemberInfo(MemberInfo info)
             Info = info;
         }
 
+        /// <summary>
+        /// This method is used to copy the AllowRankReductionAttribute 
+        /// from a Zero Touch property to the property's getter function.
+        /// </summary>
+        /// <param name="getterAttributes"></param>
+        internal void CheckForRankReductionAttribute(Dictionary<MethodInfo, Attribute[]> getterAttributes)
+        {
+            var info = Info as MethodInfo;
+            if (info != null)
+            {
+                Attribute[] attributes = null;
+                if (getterAttributes.TryGetValue(info, out attributes))
+                {
+                    foreach (var attr in attributes)
+                    {
+                        if (attr is AllowRankReductionAttribute)
+                        {
+                            mAllowRankReduction = true;
+                            mRankReducer = attr as AllowRankReductionAttribute;
+                        }
+                    }
+                }
+            }
+        }
+
         public object ReduceReturnedCollectionToSingleton(object collection)
         {
             if (null == mAllowRankReduction)
diff --git a/test/Engine/FFITarget/TestObjects.cs b/test/Engine/FFITarget/TestObjects.cs
@@ -457,4 +457,35 @@ public void Dispose()
         }
     }
 
+    public class TestRankReduce
+    {
+        private string RankReduceTestField;
+
+        [AllowRankReduction]
+        public List<string> RankReduceProperty
+        {
+            get { return new List<string> { RankReduceTestField }; }
+        }
+
+        public List<string> Property
+        {
+            get { return new List<string> { RankReduceTestField }; }
+        }
+
+        public TestRankReduce(string s)
+        {
+            RankReduceTestField = s;
+        }
+        
+        [AllowRankReduction]
+        public List<string> RankReduceMethod()
+        {
+            return new List<string> { RankReduceTestField };
+        }
+
+        public List<string> Method()
+        {
+            return new List<string> { RankReduceTestField };
+        }
+    }
 }
diff --git a/test/Engine/ProtoTest/FFITests/CSFFITest.cs b/test/Engine/ProtoTest/FFITests/CSFFITest.cs
@@ -1387,6 +1387,32 @@ public void InheritenceMethodInvokeWithoutImportDerivedClass()
             thisTest.Verify("value", new double[] { 1, 2, 3 });
             thisTest.Verify("newPoint", FFITarget.DummyPoint.ByCoordinates(2, 4, 6));
         }
+
+        [Test]
+        public void AllowRankReductionAttributeWorksForProperty()
+        {
+            string code =
+                @"import(""FFITarget.dll"");
+rankReduceTestObject = FFITarget.TestRankReduce(""test"");
+property = rankReduceTestObject.Property; 
+reducedProperty = rankReduceTestObject.RankReduceProperty; ";
+            thisTest.RunScriptSource(code);
+            thisTest.Verify("property", new List<string> { "test" });
+            thisTest.Verify("reducedProperty", "test");
+        }
+
+        [Test]
+        public void AllowRankReductionAttributeWorksForMethod()
+        {
+            string code =
+                @"import(""FFITarget.dll"");
+rankReduceTestObject = FFITarget.TestRankReduce(""test"");
+method = rankReduceTestObject.Method(); 
+reducedMethod = rankReduceTestObject.RankReduceMethod(); ";
+            thisTest.RunScriptSource(code);
+            thisTest.Verify("method", new List<string> { "test" });
+            thisTest.Verify("reducedMethod", "test");
+        }
     }
 }
 
