Add test

Author: davidwengier

src/Razor/test/Microsoft.AspNetCore.Razor.LanguageServer.Test/OpenDocumentGeneratorTest.cs

@@ -27,6 +27,37 @@ public class OpenDocumentGeneratorTest(ITestOutputHelper testOutput) : LanguageS
     private readonly HostProject _hostProject1 = new("c:/Test1/Test1.csproj", "c:/Test1/obj", RazorConfiguration.Default, "TestRootNamespace");
     private readonly HostProject _hostProject2 = new("c:/Test2/Test2.csproj", "c:/Test2/obj", RazorConfiguration.Default, "TestRootNamespace");
 
+    [Fact]
+    public async Task DocumentAdded_ProcessesOpenDocument()
+    {
+        // Arrange
+        var projectManager = CreateProjectSnapshotManager(new LspProjectEngineFactoryProvider(TestRazorLSPOptionsMonitor.Create()));
+        var listener = new TestDocumentProcessedListener();
+        using var generator = CreateOpenDocumentGenerator(projectManager, listener);
+
+        await projectManager.UpdateAsync(updater =>
+        {
+            updater.ProjectAdded(_hostProject1);
+            updater.ProjectAdded(_hostProject2);
+            updater.DocumentAdded(_hostProject1.Key, _documents[0], null!);
+            updater.DocumentOpened(_hostProject1.Key, _documents[0].FilePath, SourceText.From(string.Empty));
+        });
+
+        await listener.GetProcessedDocumentAsync(cancelAfter: TimeSpan.FromSeconds(10));
+        listener.Reset();
+
+        await projectManager.UpdateAsync(updater =>
+        {
+            updater.DocumentRemoved(_hostProject1.Key, _documents[0]);
+            updater.DocumentAdded(_hostProject2.Key, _documents[0], null!);
+        });
+
+        // Assert
+        var document = await listener.GetProcessedDocumentAsync(cancelAfter: TimeSpan.FromSeconds(10));
+        Assert.Equal(_hostProject2.Key, document.Project.Key);
+        Assert.Equal(_documents[0].FilePath, document.FilePath);
+    }
+
     [Fact]
     public async Task DocumentAdded_IgnoresClosedDocument()
     {
@@ -152,7 +183,7 @@ private OpenDocumentGenerator CreateOpenDocumentGenerator(
 
     private class TestDocumentProcessedListener : IDocumentProcessedListener
     {
-        private readonly TaskCompletionSource<IDocumentSnapshot> _tcs;
+        private TaskCompletionSource<IDocumentSnapshot> _tcs;
 
         public TestDocumentProcessedListener()
         {
@@ -178,5 +209,10 @@ public void DocumentProcessed(RazorCodeDocument codeDocument, IDocumentSnapshot
         {
             _tcs.SetResult(document);
         }
+
+        internal void Reset()
+        {
+            _tcs = new TaskCompletionSource<IDocumentSnapshot>();
+        }
     }
 }