Only generate specified document (#59097) (#59535)

Author: WretchedDade

src/Tools/GetDocumentInsider/src/Commands/GetDocumentCommandWorker.cs

@@ -254,17 +254,23 @@ private bool GetDocuments(IServiceProvider services)
             return false;
         }
 
-        // If an explicit document name is provided, then generate only that document.
+        // Get document names
         var documentNames = (IEnumerable<string>)InvokeMethod(getDocumentsMethod, service, _getDocumentsArguments);
         if (documentNames == null)
         {
             return false;
         }
 
-        if (!string.IsNullOrEmpty(_context.DocumentName) && !documentNames.Contains(_context.DocumentName))
+        // If an explicit document name is provided, then generate only that document.
+        if (!string.IsNullOrEmpty(_context.DocumentName))
         {
-            _reporter.WriteError(Resources.FormatDocumentNotFound(_context.DocumentName));
-            return false;
+            if (!documentNames.Contains(_context.DocumentName))
+            {
+                _reporter.WriteError(Resources.FormatDocumentNotFound(_context.DocumentName));
+                return false;
+            }
+
+            documentNames = [_context.DocumentName];
         }
 
         if (!string.IsNullOrWhiteSpace(_context.FileName) && !Regex.IsMatch(_context.FileName, "^([A-Za-z0-9-_]+)$"))

src/Tools/GetDocumentInsider/tests/GetDocumentTests.cs

@@ -113,7 +113,14 @@ public void GetDocument_WithDocumentName_Works()
         ], new GetDocumentCommand(_console), throwOnUnexpectedArg: false);
 
         // Assert
-        var document = new OpenApiStreamReader().Read(File.OpenRead(Path.Combine(outputPath.FullName, "Sample_internal.json")), out var diagnostic);
+        var expectedDocumentPath = Path.Combine(outputPath.FullName, "Sample_internal.json");
+
+        // There should only be one document when document name is specified
+        var documentNames = Directory.GetFiles(outputPath.FullName).Where(documentName => documentName.EndsWith(".json", StringComparison.Ordinal)).ToList();
+        Assert.Single(documentNames);
+        Assert.Contains(expectedDocumentPath, documentNames);
+
+        var document = new OpenApiStreamReader().Read(File.OpenRead(Path.Combine(outputPath.FullName, expectedDocumentPath)), out var diagnostic);
         Assert.Empty(diagnostic.Errors);
         Assert.Equal(OpenApiSpecVersion.OpenApi3_0, diagnostic.SpecificationVersion);
         // Document name in the title gives us a clue that the correct document was actually resolved