public void GetEffectiveCoresCount_RetrunsExpectedResult()
{
TestEnvironment env = new TestEnvironment();
Assert.Equal(Environment.ProcessorCount, EnvironmentExtensions.GetEffectiveCoresCount(env));
env.Clear();
env.SetEnvironmentVariable(EnvironmentSettingNames.AzureWebsiteSku, ScriptConstants.DynamicSku);
Assert.Equal(1, EnvironmentExtensions.GetEffectiveCoresCount(env));
env.Clear();
env.SetEnvironmentVariable(EnvironmentSettingNames.AzureWebsiteSku, ScriptConstants.DynamicSku);
env.SetEnvironmentVariable(EnvironmentSettingNames.RoleInstanceId, "dw0SmallDedicatedWebWorkerRole_hr0HostRole-0-VM-1");
Assert.Equal(Environment.ProcessorCount, EnvironmentExtensions.GetEffectiveCoresCount(env));
}
public void IsWarmUpRequest_ReturnsExpectedValue()
{
var environment = new TestEnvironment();
var hostEnvironment = new ScriptWebHostEnvironment(environment);
var request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/warmup");
Assert.False(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment));
// Reset environment
environment.Clear();
hostEnvironment = new ScriptWebHostEnvironment(environment);
environment.SetEnvironmentVariable(EnvironmentSettingNames.AzureWebsitePlaceholderMode, "1");
Assert.False(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment));
environment.SetEnvironmentVariable(EnvironmentSettingNames.AzureWebsiteInstanceId, "12345");
Assert.True(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment));
request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/csharphttpwarmup");
Assert.True(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment));
request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/warmup");
request.Headers.Add(ScriptConstants.AntaresLogIdHeaderName, "xyz123");
Assert.False(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment));
request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/foo");
Assert.False(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment));
// Reset environment
environment.Clear();
hostEnvironment = new ScriptWebHostEnvironment(environment);
environment.SetEnvironmentVariable(EnvironmentSettingNames.AzureWebsitePlaceholderMode, "1");
Assert.False(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment));
request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/warmup");
environment.SetEnvironmentVariable(EnvironmentSettingNames.ContainerName, "TestContainer");
Assert.True(environment.IsLinuxConsumption());
Assert.True(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment));
}
public void Dispose()
{
_testEnvironment.Clear();
}
