public void SubscribingTopicAfterNoSubscriptionsWhenGCStateSetsStateToHasSubscription()
{
var dr = new DefaultDependencyResolver();
using (var bus = new TestMessageBus(dr))
{
var subscriber = new TestSubscriber(new[] { "key" });
int retries = 0;
// Make sure the topic is in the no subs state
bus.Subscribe(subscriber, null, (result, state) => TaskAsyncHelper.True, 10, null)
.Dispose();
bus.BeforeTopicMarked = (key, t) =>
{
if (retries == 0)
{
// Need to garbage collect twice to force the topic into the dead state
bus.GarbageCollectTopics();
Assert.Equal(TopicState.Dying, t.State);
}
retries++;
};
bus.AfterTopicMarked = (key, t, state) =>
{
if (retries == 1)
{
// Assert that we've revived the topic from dying since we've subscribed to the topic
Assert.Equal(TopicState.HasSubscriptions, state);
}
};
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);
Assert.Equal(1, retries);
}
}
public VsixRunSummary Run(VsixTestCase testCase, IMessageBus messageBus)
{
messageBus.QueueMessage(new DiagnosticMessage("Running {0}", testCase.DisplayName));
var aggregator = new ExceptionAggregator();
var runner = _collectionRunnerMap.GetOrAdd(testCase.TestMethod.TestClass.TestCollection, tc => new VsRemoteTestCollectionRunner(tc, _assemblyFixtureMappings, _collectionFixtureMappings));
if (SynchronizationContext.Current == null)
{
SynchronizationContext.SetSynchronizationContext(new SynchronizationContext());
}
try
{
using (var bus = new TestMessageBus(messageBus))
{
var result = runner.RunAsync(testCase, bus, aggregator)
.Result
.ToVsixRunSummary();
if (aggregator.HasExceptions && result != null)
{
result.Exception = aggregator.ToException();
}
return(result);
}
}
catch (AggregateException aex)
{
return(new VsixRunSummary
{
Failed = 1,
Exception = aex.Flatten().InnerException
});
}
}
public void GettingTopicAfterNoSubscriptionsWhenGCStateSetsStateToHasSubscriptions()
{
var dr = new DefaultDependencyResolver();
using (var bus = new TestMessageBus(dr))
{
var subscriber = new TestSubscriber(new[] { "key" });
int retries = 0;
// Make sure the topic is in the no subs state
bus.Subscribe(subscriber, null, (result, state) => TaskAsyncHelper.True, 10, null)
.Dispose();
bus.BeforeTopicMarked = (key, t) =>
{
if (retries == 0)
{
bus.GarbageCollectTopics();
}
retries++;
};
bus.AfterTopicMarked = (key, t, state) =>
{
if (retries == 1)
{
Assert.Equal(TopicState.Dead, state);
}
};
Topic topic = bus.GetTopic("key");
Assert.Equal(1, bus.Topics.Count);
Assert.True(bus.Topics.TryGetValue("key", out topic));
Assert.Equal(TopicState.HasSubscriptions, topic.State);
Assert.Equal(2, retries);
}
}
public void MultipleSubscribeTopicCallsToDeadTopicWork()
{
var dr = new DefaultDependencyResolver();
var configuration = dr.Resolve<IConfigurationManager>();
Topic topic;
configuration.DisconnectTimeout = TimeSpan.FromSeconds(6);
configuration.KeepAlive = null;
using (var bus = new TestMessageBus(dr))
{
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);
}
}
public void SubscribingTopicAfterNoSubscriptionsWhenGCStateSetsStateToHasSubscription()
{
var dr = new DefaultDependencyResolver();
using (var bus = new TestMessageBus(dr))
{
var subscriber = new TestSubscriber(new[] { "key" });
int retries = 0;
// Make sure the topic is in the no subs state
bus.Subscribe(subscriber, null, (result, state) => TaskAsyncHelper.True, 10, null)
.Dispose();
bus.BeforeTopicMarked = (key, t) =>
{
if (retries == 0)
{
// Need to garbage collect twice to force the topic into the dead state
bus.GarbageCollectTopics();
Assert.Equal(TopicState.Dying, t.State);
}
retries++;
};
bus.AfterTopicMarked = (key, t, state) =>
{
if (retries == 1)
{
// Assert that we've revived the topic from dying since we've subscribed to the topic
Assert.Equal(TopicState.HasSubscriptions, state);
}
};
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);
Assert.Equal(1, retries);
}
}
public void GettingTopicAfterNoSubscriptionsWhenGCStateSetsStateToHasSubscriptions()
{
var dr = new DefaultDependencyResolver();
using (var bus = new TestMessageBus(dr))
{
var subscriber = new TestSubscriber(new[] { "key" });
int retries = 0;
// Make sure the topic is in the no subs state
bus.Subscribe(subscriber, null, _ => TaskAsyncHelper.True, 10)
.Dispose();
bus.BeforeTopicMarked = (key, t) =>
{
if (retries == 0)
{
bus.GarbageCollectTopics();
}
retries++;
};
bus.AfterTopicMarked = (key, t, state) =>
{
if (retries == 1)
{
Assert.Equal(TopicState.Dead, state);
}
};
Topic topic = bus.GetTopic("key");
Assert.Equal(1, bus.Topics.Count);
Assert.True(bus.Topics.TryGetValue("key", out topic));
Assert.Equal(TopicState.HasSubscriptions, topic.State);
Assert.Equal(2, retries);
}
}
public void MultipleSubscribeTopicCallsToDeadTopicWork()
{
var dr = new DefaultDependencyResolver();
var configuration = dr.Resolve <IConfigurationManager>();
Topic topic;
configuration.DisconnectTimeout = TimeSpan.FromSeconds(6);
configuration.KeepAlive = null;
using (var bus = new TestMessageBus(dr))
{
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);
}
}
