Skip to content

Commit

Permalink
tests for reqeust sender
Browse files Browse the repository at this point in the history
  • Loading branch information
@jakubch1
jakubch1 committed Jun 16, 2020
1 parent cbca05a commit 3176175
Show file tree
Hide file tree
Showing 2 changed files with 167 additions and 1 deletion.
1 change: 0 additions & 1 deletion src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/VsTestConsoleRequestSender.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -782,7 +782,6 @@ private async Task SendMessageAndListenAndReportFinalizationResultAsync(ICollect
// Also, connection termination might not kill the process which could result in files being locked by testhost.
}

// TODO: do we need events?
this.testPlatformEventSource.TranslationLayerMultiTestRunFinalizationStop();
}

Expand Down
167 changes: 167 additions & 0 deletions test/TranslationLayer.UnitTests/VsTestConsoleRequestSenderTests.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1899,6 +1899,173 @@ public async Task StartTestRunAsyncShouldLogErrorOnProcessExited()

#endregion

#region Finalization Tests

[TestMethod]
public async Task FinalizeTestsShouldCompleteWithZeroAttachments()
{
await this.InitializeCommunicationAsync();

var mockHandler = new Mock<IMultiTestRunFinalizationEventsHandler>();

var payload = new MultiTestRunFinalizationCompletePayload() { Attachments = new AttachmentSet[0] };
var finalizationComplete = new Message()
{
MessageType = MessageType.MultiTestRunFinalizationComplete,
Payload = JToken.FromObject(payload)
};
this.mockCommunicationManager.Setup(cm => cm.ReceiveMessageAsync(It.IsAny<CancellationToken>())).Returns(Task.FromResult(finalizationComplete));

await this.requestSenderAsync.FinalizeMultiTestRunAsync(new List<AttachmentSet> { new AttachmentSet(new Uri("http://www.bing.com"), "a") }, mockHandler.Object, CancellationToken.None);

mockCommunicationManager.Verify(c => c.SendMessage(MessageType.MultiTestRunFinalizationStart, It.IsAny<object>()));
mockCommunicationManager.Verify(c => c.SendMessage(MessageType.MultiTestRunFinalizationCancel), Times.Never);
mockHandler.Verify(mh => mh.HandleMultiTestRunFinalizationComplete(It.Is<ICollection<AttachmentSet>>(a => a.Count == 0)), Times.Once, "Discovery Complete must be called");
mockHandler.Verify(mh => mh.HandleLogMessage(It.IsAny<TestMessageLevel>(), It.IsAny<string>()), Times.Never, "TestMessage event must not be called");
}

[TestMethod]
public async Task FinalizeTestsShouldCompleteWithOneAttachment()
{
await this.InitializeCommunicationAsync();

var mockHandler = new Mock<IMultiTestRunFinalizationEventsHandler>();

var payload = new MultiTestRunFinalizationCompletePayload()
{
Attachments = new List<AttachmentSet> { new AttachmentSet(new Uri("http://www.bing.com"), "out") }
};
var finalizationComplete = new Message()
{
MessageType = MessageType.MultiTestRunFinalizationComplete,
Payload = JToken.FromObject(payload)
};
this.mockCommunicationManager.Setup(cm => cm.ReceiveMessageAsync(It.IsAny<CancellationToken>())).Returns(Task.FromResult(finalizationComplete));

await this.requestSenderAsync.FinalizeMultiTestRunAsync(new List<AttachmentSet> { new AttachmentSet(new Uri("http://www.bing.com"), "a") }, mockHandler.Object, CancellationToken.None);

mockCommunicationManager.Verify(c => c.SendMessage(MessageType.MultiTestRunFinalizationStart, It.IsAny<object>()));
mockCommunicationManager.Verify(c => c.SendMessage(MessageType.MultiTestRunFinalizationCancel), Times.Never);
mockHandler.Verify(mh => mh.HandleMultiTestRunFinalizationComplete(It.Is<ICollection<AttachmentSet>>(a => a.Count == 1)), Times.Once, "Discovery Complete must be called");
mockHandler.Verify(mh => mh.HandleLogMessage(It.IsAny<TestMessageLevel>(), It.IsAny<string>()), Times.Never, "TestMessage event must not be called");
}

[TestMethod]
public async Task FinalizeTestsShouldCompleteWithOneAttachmentAndTestMessage()
{
await this.InitializeCommunicationAsync();

var mockHandler = new Mock<IMultiTestRunFinalizationEventsHandler>();

var payload = new MultiTestRunFinalizationCompletePayload()
{
Attachments = new List<AttachmentSet> { new AttachmentSet(new Uri("http://www.bing.com"), "out") }
};
var finalizationComplete = new Message()
{
MessageType = MessageType.MultiTestRunFinalizationComplete,
Payload = JToken.FromObject(payload)
};

var mpayload = new TestMessagePayload() { MessageLevel = TestMessageLevel.Informational, Message = "Hello" };
var message = CreateMessage(MessageType.TestMessage, mpayload);

this.mockCommunicationManager.Setup(cm => cm.ReceiveMessageAsync(It.IsAny<CancellationToken>())).Returns(Task.FromResult(message));
mockHandler.Setup(mh => mh.HandleLogMessage(It.IsAny<TestMessageLevel>(), It.IsAny<string>())).Callback(
() => this.mockCommunicationManager.Setup(cm => cm.ReceiveMessageAsync(It.IsAny<CancellationToken>())).Returns(Task.FromResult(finalizationComplete)));

await this.requestSenderAsync.FinalizeMultiTestRunAsync(new List<AttachmentSet> { new AttachmentSet(new Uri("http://www.bing.com"), "a") }, mockHandler.Object, CancellationToken.None);

mockCommunicationManager.Verify(c => c.SendMessage(MessageType.MultiTestRunFinalizationStart, It.IsAny<object>()));
mockCommunicationManager.Verify(c => c.SendMessage(MessageType.MultiTestRunFinalizationCancel), Times.Never);
mockHandler.Verify(mh => mh.HandleMultiTestRunFinalizationComplete(It.Is<ICollection<AttachmentSet>>(a => a.Count == 1)), Times.Once, "Discovery Complete must be called");
mockHandler.Verify(mh => mh.HandleLogMessage(TestMessageLevel.Informational, "Hello"), Times.Once, "TestMessage event must be called");
}

[TestMethod]
public async Task FinalizeTestsShouldSendCancelMessageIfCancellationTokenCancelled()
{
await this.InitializeCommunicationAsync();

var cts = new CancellationTokenSource();

var mockHandler = new Mock<IMultiTestRunFinalizationEventsHandler>();

var payload = new MultiTestRunFinalizationCompletePayload()
{
Attachments = new List<AttachmentSet> { new AttachmentSet(new Uri("http://www.bing.com"), "out") }
};
var finalizationComplete = new Message()
{
MessageType = MessageType.MultiTestRunFinalizationComplete,
Payload = JToken.FromObject(payload)
};

var mpayload = new TestMessagePayload() { MessageLevel = TestMessageLevel.Informational, Message = "Hello" };
var message = CreateMessage(MessageType.TestMessage, mpayload);

this.mockCommunicationManager.Setup(cm => cm.ReceiveMessageAsync(It.IsAny<CancellationToken>())).Returns(Task.FromResult(message));
mockHandler.Setup(mh => mh.HandleLogMessage(It.IsAny<TestMessageLevel>(), It.IsAny<string>())).Callback(() =>
{
cts.Cancel();
this.mockCommunicationManager.Setup(cm => cm.ReceiveMessageAsync(It.IsAny<CancellationToken>())).Returns(Task.FromResult(finalizationComplete));
});

await this.requestSenderAsync.FinalizeMultiTestRunAsync(new List<AttachmentSet> { new AttachmentSet(new Uri("http://www.bing.com"), "a") }, mockHandler.Object, cts.Token);

mockCommunicationManager.Verify(c => c.SendMessage(MessageType.MultiTestRunFinalizationStart, It.IsAny<object>()));
mockCommunicationManager.Verify(c => c.SendMessage(MessageType.MultiTestRunFinalizationCancel));
mockHandler.Verify(mh => mh.HandleMultiTestRunFinalizationComplete(It.Is<ICollection<AttachmentSet>>(a => a.Count == 1)), Times.Once, "Discovery Complete must be called");
mockHandler.Verify(mh => mh.HandleLogMessage(TestMessageLevel.Informational, "Hello"), Times.Once, "TestMessage event must be called");
}

[TestMethod]
public async Task FinalizeTestsShouldSendCancelMessageIfCancellationTokenCancelledAtTheBeginning()
{
await this.InitializeCommunicationAsync();

var cts = new CancellationTokenSource();
cts.Cancel();

var mockHandler = new Mock<IMultiTestRunFinalizationEventsHandler>();

var payload = new MultiTestRunFinalizationCompletePayload()
{
Attachments = new List<AttachmentSet> { new AttachmentSet(new Uri("http://www.bing.com"), "out") }
};
var finalizationComplete = new Message()
{
MessageType = MessageType.MultiTestRunFinalizationComplete,
Payload = JToken.FromObject(payload)
};

this.mockCommunicationManager.Setup(cm => cm.ReceiveMessageAsync(It.IsAny<CancellationToken>())).Returns(Task.FromResult(finalizationComplete));

await this.requestSenderAsync.FinalizeMultiTestRunAsync(new List<AttachmentSet> { new AttachmentSet(new Uri("http://www.bing.com"), "a") }, mockHandler.Object, cts.Token);

mockCommunicationManager.Verify(c => c.SendMessage(MessageType.MultiTestRunFinalizationStart, It.IsAny<object>()));
mockCommunicationManager.Verify(c => c.SendMessage(MessageType.MultiTestRunFinalizationCancel));
mockHandler.Verify(mh => mh.HandleMultiTestRunFinalizationComplete(It.Is<ICollection<AttachmentSet>>(a => a.Count == 1)), Times.Once, "Discovery Complete must be called");
mockHandler.Verify(mh => mh.HandleLogMessage(TestMessageLevel.Informational, "Hello"), Times.Never, "TestMessage event must be called");
}

[TestMethod]
public async Task FinalizeTestsShouldAbortOnExceptionInSendMessage()
{
var mockHandler = new Mock<IMultiTestRunFinalizationEventsHandler>();
this.mockCommunicationManager.Setup(cm => cm.SendMessage(MessageType.MultiTestRunFinalizationStart, It.IsAny<object>())).Throws(new IOException());

await this.requestSenderAsync.FinalizeMultiTestRunAsync(new List<AttachmentSet> { new AttachmentSet(new Uri("http://www.bing.com"), "out") }, mockHandler.Object, CancellationToken.None);

mockHandler.Verify(mh => mh.HandleMultiTestRunFinalizationComplete(null), Times.Once, "Discovery Complete must be called");
mockHandler.Verify(mh => mh.HandleLogMessage(TestMessageLevel.Error, It.IsAny<string>()), Times.Once, "TestMessage event must be called");
this.mockCommunicationManager.Verify(cm => cm.StopServer(), Times.Never);
}

#endregion



#region Private Methods

/// <summary>
Expand Down

0 comments on commit 3176175

Please sign in to comment.