public void SubscribingTopicAfterNoSubscriptionsStateSetsStateToHasSubscription() { var sp = ServiceProviderHelper.CreateServiceProvider(services => { services.Configure <MessageBusOptions>(options => { options.TopicTTL = TopicTtl(TimeSpan.FromSeconds(6)); }); }); using (var bus = (MessageBus)sp.GetRequiredService <IMessageBus>()) { var subscriber = new TestSubscriber(new[] { "key" }); // Make sure the topic is in the no subs state bus.Subscribe(subscriber, null, (result, state) => TaskAsyncHelper.True, 10, null) .Dispose(); Topic topic = bus.SubscribeTopic("key"); Assert.Equal(1, bus.Topics.Count); Assert.True(bus.Topics.TryGetValue("key", out topic)); Assert.Equal(TopicState.HasSubscriptions, topic.State); } }
public void SubscriptionWithMultipleExistingCursors() { var sp = ServiceProviderHelper.CreateServiceProvider(services => { var passThroughMinfier = new PassThroughStringMinifier(); services.AddSingleton <IStringMinifier>(passThroughMinfier); }); using (var bus = (MessageBus)sp.GetRequiredService <IMessageBus>()) { Func <ISubscriber> subscriberFactory = () => new TestSubscriber(new[] { "key", "key2" }); var cdKey = new CountDownRange <int>(Enumerable.Range(2, 4)); var cdKey2 = new CountDownRange <int>(new[] { 1, 2, 10 }); IDisposable subscription = null; string prefix = DefaultSubscription._defaultCursorPrefix; // Pretend like we had an initial subscription bus.Subscribe(subscriberFactory(), null, (result, state) => TaskAsyncHelper.True, 10, null) .Dispose(); // This simulates a reconnect bus.Publish("test", "key", "1").Wait(); bus.Publish("test", "key", "2").Wait(); bus.Publish("test", "key", "3").Wait(); bus.Publish("test", "key", "4").Wait(); bus.Publish("test", "key2", "1").Wait(); bus.Publish("test", "key2", "2").Wait(); try { subscription = bus.Subscribe(subscriberFactory(), prefix + "key,00000001|key2,00000000", (result, state) => { foreach (var m in result.GetMessages()) { int n = Int32.Parse(m.GetString()); if (m.Key == "key") { Assert.True(cdKey.Mark(n)); } else { Assert.True(cdKey2.Mark(n)); } } return(TaskAsyncHelper.True); }, 10, null); bus.Publish("test", "key", "5"); bus.Publish("test", "key2", "10"); Assert.True(cdKey.Wait(TimeSpan.FromSeconds(5))); Assert.True(cdKey2.Wait(TimeSpan.FromSeconds(5))); } finally { if (subscription != null) { subscription.Dispose(); } } } }
public void MultipleSubscribeTopicCallsToDeadTopicWork() { var sp = ServiceProviderHelper.CreateServiceProvider(services => { services.Configure <MessageBusOptions>(options => { options.TopicTTL = TopicTtl(TimeSpan.FromSeconds(6)); }); }); Topic topic; using (var bus = ActivatorUtilities.CreateInstance <TestMessageBus>(sp)) { var subscriber = new TestSubscriber(new[] { "key" }); int count = 0; // Make sure the topic is in the no subs state bus.Subscribe(subscriber, null, (result, state) => TaskAsyncHelper.True, 10, null) .Dispose(); bus.BeforeTopicCreated = (key) => { bus.Topics.TryGetValue(key, out topic); if (count == 1) { // Should have been removed by our double garbage collect in BeforeTopicMarked Assert.Null(topic); } if (count == 3) { // Ensure that we have a topic now created from the original thread Assert.NotNull(topic); } }; bus.BeforeTopicMarked = (key, t) => { count++; if (count == 1) { bus.GarbageCollectTopics(); bus.GarbageCollectTopics(); // We garbage collect twice to mark the current topic as dead (it will remove it from the topics list) Assert.Equal(t.State, TopicState.Dead); bus.SubscribeTopic("key"); // Topic should still be dead Assert.Equal(t.State, TopicState.Dead); Assert.Equal(count, 2); // Increment up to 3 so we don't execute same code path in after marked count++; } if (count == 2) { // We've just re-created the topic from the second bus.SubscribeTopic so we should have 0 subscriptions Assert.Equal(t.State, TopicState.NoSubscriptions); } if (count == 4) { // Ensure that we pulled the already created subscription (therefore it has subscriptions) Assert.Equal(t.State, TopicState.HasSubscriptions); } }; bus.AfterTopicMarked = (key, t, state) => { if (count == 2) { // After re-creating the topic from the second bus.SubscribeTopic we should then move the topic state // into the has subscriptions state Assert.Equal(state, TopicState.HasSubscriptions); } if (count == 3) { Assert.Equal(state, TopicState.Dead); } }; bus.SubscribeTopic("key"); Assert.Equal(1, bus.Topics.Count); Assert.True(bus.Topics.TryGetValue("key", out topic)); Assert.Equal(TopicState.HasSubscriptions, topic.State); } }