Make HostedTest.UseMessageBus support N streams in FakeScaleoutBus

SignalR · May 30, 2013 · 516b151 · 516b151
1 parent 9ba1622
commit 516b151
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/Microsoft.AspNet.SignalR.Tests.Common/Infrastructure/HostedTest.cs b/tests/Microsoft.AspNet.SignalR.Tests.Common/Infrastructure/HostedTest.cs
@@ -31,7 +31,7 @@ protected ITestHost CreateHost(HostType hostType, TransportType transportType)
             return HostedTestFactory.CreateHost(hostType, transportType, detailedTestName);
         }
 
-        protected void UseMessageBus(MessageBusType type, IDependencyResolver resolver)
+        protected void UseMessageBus(MessageBusType type, IDependencyResolver resolver, int streams = 1)
         {
             IMessageBus bus = null;
 
@@ -40,10 +40,8 @@ protected void UseMessageBus(MessageBusType type, IDependencyResolver resolver)
                 case MessageBusType.Default:
                     break;
                 case MessageBusType.Fake:
-                    bus = new FakeScaleoutBus(resolver, streams: 1);
-                    break;
                 case MessageBusType.FakeMultiStream:
-                    bus  = new FakeScaleoutBus(resolver, streams: 3);
+                    bus = new FakeScaleoutBus(resolver, streams);
                     break;
                 case MessageBusType.SqlServer:
                     break;