|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.CommandLine;
using Microsoft.CodeAnalysis.Test.Utilities;
using Xunit;
using Xunit.Abstractions;
namespace Microsoft.CodeAnalysis.CompilerServer.UnitTests
{
public sealed class BuildServerConnectionTests : IDisposable
{
internal TempRoot TempRoot { get; } = new TempRoot();
internal XunitCompilerServerLogger Logger { get; }
public BuildServerConnectionTests(ITestOutputHelper testOutputHelper)
{
Logger = new XunitCompilerServerLogger(testOutputHelper);
}
public void Dispose()
{
TempRoot.Dispose();
}
[Fact]
public async Task OnlyStartOneServer()
{
ServerData? serverData = null;
try
{
var pipeName = ServerUtil.GetPipeName();
var workingDirectory = TempRoot.CreateDirectory().Path;
for (var i = 0; i < 5; i++)
{
var response = await BuildServerConnection.RunServerBuildRequestAsync(
ProtocolUtil.CreateEmptyCSharp(workingDirectory),
pipeName,
timeoutOverride: Timeout.Infinite,
tryCreateServerFunc: (pipeName, logger) =>
{
Assert.Null(serverData);
serverData = ServerData.Create(logger, pipeName);
return true;
},
Logger,
cancellationToken: default);
Assert.True(response is CompletedBuildResponse);
}
}
finally
{
serverData?.Dispose();
}
}
[Fact]
public async Task UseExistingServer()
{
using var serverData = await ServerUtil.CreateServer(Logger);
var ran = false;
var workingDirectory = TempRoot.CreateDirectory().Path;
for (var i = 0; i < 5; i++)
{
var response = await BuildServerConnection.RunServerBuildRequestAsync(
ProtocolUtil.CreateEmptyCSharp(workingDirectory),
serverData.PipeName,
timeoutOverride: Timeout.Infinite,
tryCreateServerFunc: (_, _) =>
{
ran = true;
return false;
},
Logger,
cancellationToken: default);
Assert.True(response is CompletedBuildResponse);
}
Assert.False(ran);
}
/// <summary>
/// Simulate the case where the server process crashes or hangs on startup
/// and make sure the client properly fails
/// </summary>
[Fact]
public async Task SimulateServerCrashingOnStartup()
{
var pipeName = ServerUtil.GetPipeName();
var ran = false;
var response = await BuildServerConnection.RunServerBuildRequestAsync(
ProtocolUtil.CreateEmptyCSharp(TempRoot.CreateDirectory().Path),
pipeName,
timeoutOverride: (int)TimeSpan.FromSeconds(2).TotalMilliseconds,
tryCreateServerFunc: (_, _) =>
{
ran = true;
// Correct this is a lie. The server did not start. But it also does a nice
// job of simulating a hung or crashed server.
return true;
},
Logger,
cancellationToken: default);
Assert.True(response is CannotConnectResponse);
Assert.True(ran);
}
[Fact]
public async Task FailedServer()
{
var pipeName = ServerUtil.GetPipeName();
var workingDirectory = TempRoot.CreateDirectory().Path;
var count = 0;
for (var i = 0; i < 5; i++)
{
var response = await BuildServerConnection.RunServerBuildRequestAsync(
ProtocolUtil.CreateEmptyCSharp(workingDirectory),
pipeName,
timeoutOverride: Timeout.Infinite,
tryCreateServerFunc: (_, _) =>
{
count++;
return false;
},
Logger,
cancellationToken: default);
Assert.True(response is CannotConnectResponse);
}
Assert.Equal(5, count);
}
}
}
|