From c8ef9fadca9eb1769b16959bb5868b1e3335a676 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?G=C3=A1bor=20Szab=C3=B3?= <szabo.gabor@podnet.hu>
Date: Fri, 9 Feb 2024 20:27:10 +0100
Subject: [PATCH] Store the provided values for custom Fakes to allow
 inspection and testing

---
 .../CodeAnalysis/Fakes/AdditionalText.cs               |  7 +++++--
 .../Fakes/AnalyzerConfigOptionsProvider.cs             | 10 ++++++++--
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/Analyzers.Testing/CodeAnalysis/Fakes/AdditionalText.cs b/src/Analyzers.Testing/CodeAnalysis/Fakes/AdditionalText.cs
index 02e4fbb..9fd43c8 100644
--- a/src/Analyzers.Testing/CodeAnalysis/Fakes/AdditionalText.cs
+++ b/src/Analyzers.Testing/CodeAnalysis/Fakes/AdditionalText.cs
@@ -12,9 +12,12 @@ public AdditionalText(string path, string text) : this(path, SourceText.From(tex
 
     /// <summary>The "path" for the text file. Doesn't necessarily point to a valid location.</summary>
     public override string Path => path;
-    
+
+    /// <summary>Retrieves the text stored with this instance.</summary>
+    public SourceText? Text => text;
+
     /// <summary>Return the wrapped text of the current fake.</summary>
     /// <param name="cancellationToken">The cancellation token. Unused.</param>
     /// <returns>The source text that was referenced during construction of this instance.</returns>
-    public override SourceText? GetText(CancellationToken cancellationToken = default) => text;
+    public override SourceText? GetText(CancellationToken cancellationToken = default) => Text;
 }
\ No newline at end of file
diff --git a/src/Analyzers.Testing/CodeAnalysis/Fakes/AnalyzerConfigOptionsProvider.cs b/src/Analyzers.Testing/CodeAnalysis/Fakes/AnalyzerConfigOptionsProvider.cs
index 96031c7..bcad028 100644
--- a/src/Analyzers.Testing/CodeAnalysis/Fakes/AnalyzerConfigOptionsProvider.cs
+++ b/src/Analyzers.Testing/CodeAnalysis/Fakes/AnalyzerConfigOptionsProvider.cs
@@ -14,12 +14,18 @@ public class AnalyzerConfigOptionsProvider(
     /// <summary>The global options, as they are available from the options provided during construction.</summary>
     public override AnalyzerConfigOptions GlobalOptions => globalOptions;
 
+    /// <summary>The lookup options for each additional syntax tree.</summary>
+    public Dictionary<SyntaxTree, AnalyzerConfigOptions> OptionsForSyntaxTrees { get; } = optionsForSyntaxTrees;
+
+    /// <summary>The lookup options for each additional text file.</summary>
+    public Dictionary<Microsoft.CodeAnalysis.AdditionalText, AnalyzerConfigOptions> OptionsForAdditionalTexts { get; } = optionsForAdditionalTexts;
+
     /// <summary>Gets the <see cref="GlobalOptions"/>, and any additional options specific to <paramref name="key"/>.</summary>
     /// <param name="key">The syntax tree to look up additional options for.</param>
     /// <returns>The <see cref="GlobalOptions"/>, and any additional options specific to <paramref name="key"/>.</returns>
     public override AnalyzerConfigOptions GetOptions(SyntaxTree key)
     {
-        if (optionsForSyntaxTrees.TryGetValue(key, out var value))
+        if (OptionsForSyntaxTrees.TryGetValue(key, out var value))
             return new(new Dictionary<string, string?>(GlobalOptions.Values.Concat(value.Values)));
         return GlobalOptions;
     }
@@ -27,7 +33,7 @@ public override AnalyzerConfigOptions GetOptions(SyntaxTree key)
     /// <inheritdoc cref="GetOptions(SyntaxTree)"/>
     public override AnalyzerConfigOptions GetOptions(Microsoft.CodeAnalysis.AdditionalText key)
     {
-        if (optionsForAdditionalTexts.TryGetValue(key, out var value))
+        if (OptionsForAdditionalTexts.TryGetValue(key, out var value))
             return new(new Dictionary<string, string?>(GlobalOptions.Values.Concat(value.Values)));
         return GlobalOptions;
     }