/// <inheritdoc/>
public void HandleDiscoveryComplete(DiscoveryCompleteEventArgs discoveryCompleteEventArgs, IEnumerable <TestCase> lastChunk)
{
var totalTests = discoveryCompleteEventArgs.TotalCount;
var isAborted = discoveryCompleteEventArgs.IsAborted;
// we get discovery complete events from each host process
// so we cannot "complete" the actual operation until all sources are consumed
// We should not block last chunk results while we aggregate overall discovery data
if (lastChunk != null)
{
ConvertToRawMessageAndSend(MessageType.TestCasesFound, lastChunk);
this.HandleDiscoveredTests(lastChunk);
}
// Aggregate for final discovery complete
discoveryDataAggregator.Aggregate(totalTests, isAborted);
// Aggregate Discovery Data Metrics
discoveryDataAggregator.AggregateDiscoveryDataMetrics(discoveryCompleteEventArgs.Metrics);
// Do not send TestDiscoveryComplete to actual test discovery handler
// We need to see if there are still sources left - let the parallel manager decide
var parallelDiscoveryComplete = this.parallelProxyDiscoveryManager.HandlePartialDiscoveryComplete(
this.proxyDiscoveryManager,
totalTests,
null, // lastChunk should be null as we already sent this data above
isAborted);
if (parallelDiscoveryComplete)
{
// In case of sequential discovery - RawMessage would have contained a 'DiscoveryCompletePayload' object
// To send a raw message - we need to create raw message from an aggregated payload object
var testDiscoveryCompletePayload = new DiscoveryCompletePayload()
{
TotalTests = discoveryDataAggregator.TotalTests,
IsAborted = discoveryDataAggregator.IsAborted,
LastDiscoveredTests = null
};
// Collecting Final Discovery State
this.requestData.MetricsCollection.Add(TelemetryDataConstants.DiscoveryState, isAborted ? "Aborted" : "Completed");
// Collect Aggregated Metrics Data
var aggregatedDiscoveryDataMetrics = discoveryDataAggregator.GetAggregatedDiscoveryDataMetrics();
testDiscoveryCompletePayload.Metrics = aggregatedDiscoveryDataMetrics;
// we have to send raw messages as we block the discovery complete actual raw messages
this.ConvertToRawMessageAndSend(MessageType.DiscoveryComplete, testDiscoveryCompletePayload);
var finalDiscoveryCompleteEventArgs = new DiscoveryCompleteEventArgs(this.discoveryDataAggregator.TotalTests,
this.discoveryDataAggregator.IsAborted);
finalDiscoveryCompleteEventArgs.Metrics = aggregatedDiscoveryDataMetrics;
// send actual test discovery complete to clients
this.actualDiscoveryEventsHandler.HandleDiscoveryComplete(finalDiscoveryCompleteEventArgs, null);
}
}
/// <inheritdoc/>
public void HandleDiscoveryComplete(long totalTests, IEnumerable <TestCase> lastChunk, bool isAborted)
{
// we get discovery complete events from each host process
// so we cannot "complete" the actual operation until all sources are consumed
// We should not block last chunk results while we aggregate overall discovery data
if (lastChunk != null)
{
ConvertToRawMessageAndSend(MessageType.TestCasesFound, lastChunk);
this.HandleDiscoveredTests(lastChunk);
}
// Aggregate for final discoverycomplete
discoveryDataAggregator.Aggregate(totalTests, isAborted);
// Do not send TestDiscoveryComplete to actual test discovery handler
// We need to see if there are still sources left - let the parallel manager decide
var parallelDiscoveryComplete = this.parallelProxyDiscoveryManager.HandlePartialDiscoveryComplete(
this.proxyDiscoveryManager,
totalTests,
null, // lastChunk should be null as we already sent this data above
isAborted);
if (parallelDiscoveryComplete)
{
// In case of sequential discovery - RawMessage would have contained a 'DiscoveryCompletePayload' object
// To send a raw message - we need to create raw message from an aggregated payload object
var testDiscoveryCompletePayload = new DiscoveryCompletePayload()
{
TotalTests = discoveryDataAggregator.TotalTests,
IsAborted = discoveryDataAggregator.IsAborted,
LastDiscoveredTests = null
};
// we have to send raw messages as we block the discoverycomplete actual raw messages
this.ConvertToRawMessageAndSend(MessageType.DiscoveryComplete, testDiscoveryCompletePayload);
// send actual test discoverycomplete to clients
this.actualDiscoveryEventsHandler.HandleDiscoveryComplete(discoveryDataAggregator.TotalTests, null, discoveryDataAggregator.IsAborted);
}
}