Fixed the waiter

src/EditorFeatures/CSharpTest/EventHookup/EventHookupTestState.cs

@@ -7,11 +7,15 @@
 using System.Threading;
 using System.Xml.Linq;
 using Microsoft.CodeAnalysis.Editor.CSharp.EventHookup;
+using Microsoft.CodeAnalysis.Editor.CSharp.UnitTests.EventHookup;
 using Microsoft.CodeAnalysis.Editor.Implementation.Commands;
 using Microsoft.CodeAnalysis.Editor.UnitTests;
 using Microsoft.CodeAnalysis.Editor.UnitTests.Extensions;
 using Microsoft.CodeAnalysis.Options;
 using Xunit;
+using System.ComponentModel.Composition;
+using System.ComponentModel.Composition.Primitives;
+using Microsoft.VisualStudio.Composition;
 
 namespace Microsoft.CodeAnalysis.Editor.CSharp.UnitTests.EventHookup
 {
@@ -21,7 +25,7 @@ internal sealed class EventHookupTestState : AbstractCommandHandlerTestState
         private Mutex _testSessionHookupMutex;
 
         public EventHookupTestState(XElement workspaceElement, IDictionary<OptionKey, object> options)
-            : base(workspaceElement, null, false)
+            : base(workspaceElement, GetExtraParts(), false)
         {
             CommandHandlerService t = (CommandHandlerService)Workspace.GetService<ICommandHandlerServiceFactory>().GetService(Workspace.Documents.Single().TextBuffer);
             var field = t.GetType().GetField("_commandHandlers", BindingFlags.NonPublic | BindingFlags.GetField | BindingFlags.Instance);
@@ -33,6 +37,11 @@ public EventHookupTestState(XElement workspaceElement, IDictionary<OptionKey, ob
             Workspace.ApplyOptions(options);
         }
 
+        private static ComposableCatalog GetExtraParts()
+        {
+            return MinimalTestExportProvider.CreateTypeCatalog(new[] { typeof(EventHookupWaiter) });
+        }
+
         public static EventHookupTestState CreateTestState(string markup, IDictionary<OptionKey, object> options = null)
         {
             var workspaceXml = string.Format(@"

src/EditorFeatures/Test/EditorServicesTest.csproj

@@ -179,7 +179,6 @@
     <Compile Include="EditorAdapter\SpanExtensionsTest.cs" />
     <Compile Include="EditorAdapter\TextSnapshotImplementationTest.cs" />
     <Compile Include="EditorAdapter\TextSpanExtensionsTest.cs" />
-    <Compile Include="EventHookup\EventHookupWaiter.cs" />
     <Compile Include="Extensions\CollectionExtensionsTest.cs" />
     <Compile Include="Extensions\EnumerableExtensionTests.cs" />
     <Compile Include="Extensions\IProjectionBufferFactoryServiceExtensionsTests.cs" />

src/EditorFeatures/TestUtilities/ServicesTestUtilities.csproj

@@ -185,6 +185,7 @@
     <Compile Include="EditAndContinue\SemanticEditDescription.cs" />
     <Compile Include="EditAndContinue\TestActiveStatementTrackingService.cs" />
     <Compile Include="EditorFactory.cs" />
+    <Compile Include="EventHookup\EventHookupWaiter.cs" />
     <Compile Include="ExceptionUtilities.cs" />
     <Compile Include="Extensions\WorkspaceExtensions.cs" />
     <Compile Include="Extensions\XElementExtensions.cs" />