From b82ab38a362bf736a235450a187a6424adac44c6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micka=C3=ABl=20NIVET?= <mnivet@talentsoft.com>
Date: Mon, 18 Oct 2021 19:17:48 +0200
Subject: [PATCH] cherry pick 853eff3782d4d89427ef324c221820f6d42cba53 from
 PR785

Make possible to explicitly define the assembly that will be used to read metadata (assembly attributes) to produce the help.

https://github.com/commandlineparser/commandline/pull/785
---
 .../Infrastructure/ReflectionHelper.cs        | 20 +++++++++++--------
 src/CommandLine/Text/HelpText.cs              | 11 ++++++++++
 .../Unit/Text/HelpTextTests.cs                | 16 +++++++++++++++
 3 files changed, 39 insertions(+), 8 deletions(-)

diff --git a/src/CommandLine/Infrastructure/ReflectionHelper.cs b/src/CommandLine/Infrastructure/ReflectionHelper.cs
index 47fe70ea..11089d26 100644
--- a/src/CommandLine/Infrastructure/ReflectionHelper.cs
+++ b/src/CommandLine/Infrastructure/ReflectionHelper.cs
@@ -16,6 +16,14 @@ static class ReflectionHelper
         /// </summary>
         [ThreadStatic] private static IDictionary<Type, Attribute> _overrides;
 
+        private static Assembly _programAssembly;
+
+        public static Assembly ProgramAssembly
+        {
+            get => _programAssembly ?? GetExecutingOrEntryAssembly();
+            set => _programAssembly = value;
+        }
+
         /// <summary>
         /// Assembly attribute overrides for testing.
         /// </summary>
@@ -51,12 +59,10 @@ public static Maybe<TAttribute> GetAttribute<TAttribute>()
                         Maybe.Nothing<TAttribute>();
             }
 
-            var assembly = GetExecutingOrEntryAssembly();
-
 #if NET40
-            var attributes = assembly.GetCustomAttributes(typeof(TAttribute), false);
+            var attributes = ProgramAssembly.GetCustomAttributes(typeof(TAttribute), false);
 #else
-            var attributes = assembly.GetCustomAttributes<TAttribute>().ToArray();
+            var attributes = ProgramAssembly.GetCustomAttributes<TAttribute>().ToArray();
 #endif
 
             return attributes.Length > 0
@@ -66,14 +72,12 @@ public static Maybe<TAttribute> GetAttribute<TAttribute>()
 
         public static string GetAssemblyName()
         {
-            var assembly = GetExecutingOrEntryAssembly();
-            return assembly.GetName().Name;
+            return ProgramAssembly.GetName().Name;
         }
 
         public static string GetAssemblyVersion()
         {
-            var assembly = GetExecutingOrEntryAssembly();
-            return assembly.GetName().Version.ToStringInvariant();
+            return ProgramAssembly.GetName().Version.ToStringInvariant();
         }
 
         public static bool IsFSharpOptionType(Type type)
diff --git a/src/CommandLine/Text/HelpText.cs b/src/CommandLine/Text/HelpText.cs
index e9ce218d..a6a93690 100644
--- a/src/CommandLine/Text/HelpText.cs
+++ b/src/CommandLine/Text/HelpText.cs
@@ -305,6 +305,17 @@ public SentenceBuilder SentenceBuilder
             get { return sentenceBuilder; }
         }
 
+        /// <summary>
+        /// Gets or sets the <see cref="Assembly"/> that will be used to look for meta data (assembly attributes) to auto build the help.
+        /// By default the entry assembly is automatically selected, but this may not match what is really expected, in particular when
+        /// running unit tests.
+        /// </summary>
+        public static Assembly AutoBuildMetadataAssembly
+        {
+            get => ReflectionHelper.ProgramAssembly;
+            set => ReflectionHelper.ProgramAssembly = value;
+        }
+
         /// <summary>
         /// Creates a new instance of the <see cref="CommandLine.Text.HelpText"/> class using common defaults.
         /// </summary>
diff --git a/tests/CommandLine.Tests/Unit/Text/HelpTextTests.cs b/tests/CommandLine.Tests/Unit/Text/HelpTextTests.cs
index 9811f7be..948e5546 100644
--- a/tests/CommandLine.Tests/Unit/Text/HelpTextTests.cs
+++ b/tests/CommandLine.Tests/Unit/Text/HelpTextTests.cs
@@ -22,6 +22,7 @@ public class HelpTextTests : IDisposable
         public void Dispose()
         {
             ReflectionHelper.SetAttributeOverride(null);
+            HelpText.AutoBuildMetadataAssembly = null;
         }
 
         [Fact]
@@ -716,6 +717,21 @@ public void AutoBuild_with_assembly_company_attribute_only()
             actualResult.Copyright.Should().Be(string.Format("Copyright (C) {0} {1}", DateTime.Now.Year, expectedCompany));
         }
 
+        [Fact]
+        public void AutoBuild_with_AutoBuildMetadataAssembly_defined()
+        {
+            HelpText.AutoBuildMetadataAssembly = typeof(HelpText).Assembly;
+            string expectedHeading = "CommandLine 0.0.0";
+            string expectedCopyright = "Copyright (c) 2005 - 2020 Giacomo Stelluti Scala & Contributors";
+
+            ParserResult<Simple_Options> fakeResult = new NotParsed<Simple_Options>(
+                TypeInfo.Create(typeof(Simple_Options)), new Error[0]);
+            HelpText actualResult = HelpText.AutoBuild(fakeResult, ht => ht, ex => ex);
+
+            actualResult.Heading.Should().Be(expectedHeading);
+            actualResult.Copyright.Should().Be(expectedCopyright);
+        }
+
         [Fact]
         public void Add_line_with_two_empty_spaces_at_the_end()
         {